-
Notifications
You must be signed in to change notification settings - Fork 23
Expand file tree
/
Copy pathmain.go
More file actions
99 lines (86 loc) · 2.36 KB
/
Copy pathmain.go
File metadata and controls
99 lines (86 loc) · 2.36 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package main
import (
_ "embed"
"encoding/json"
"errors"
"fmt"
"math/rand"
"net/http"
"os"
"github.com/gorilla/mux"
"github.com/rs/cors"
)
//go:embed greetings.json
var greetingsJson []byte
type Greeting struct {
Language string `json:"language"`
Greeting string `json:"greeting"`
}
func main() {
var greetings []*Greeting
err := json.Unmarshal(greetingsJson, &greetings)
if err != nil {
fmt.Printf("error loading greetings: %s\n", err)
os.Exit(1)
}
router := mux.NewRouter()
router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
fmt.Printf("got / request from %s\n", r.RemoteAddr)
w.Header().Set("Content-Type", "application/json")
greeting, err := SelectGreeting(greetings, "random")
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
}
_, err = w.Write([]byte(FormatResponse(greeting)))
if err != nil {
panic(err)
}
}).Methods("GET")
router.HandleFunc("/{language}", func(w http.ResponseWriter, r *http.Request) {
language := mux.Vars(r)["language"]
fmt.Printf("got /{language} request from %s\n", r.RemoteAddr)
w.Header().Set("Content-Type", "application/json")
greeting, err := SelectGreeting(greetings, language)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
}
_, err = w.Write([]byte(FormatResponse(greeting)))
if err != nil {
panic(err)
}
}).Methods("GET")
c := cors.New(cors.Options{
AllowedOrigins: []string{
"http://greetings.kylepenfound.com",
"https://dagger-demo.netlify.app",
"http://localhost:8081",
},
})
handler := c.Handler(router)
err = http.ListenAndServe(":8080", handler)
if errors.Is(err, http.ErrServerClosed) {
fmt.Printf("server closed\n")
} else if err != nil {
fmt.Printf("error starting server: %s\n", err)
os.Exit(1)
}
}
func FormatResponse(greeting *Greeting) string {
return fmt.Sprintf("{\"greeting\":\"%s\"}", greeting.Greeting)
}
func SelectGreeting(greetings []*Greeting, language string) (*Greeting, error) {
if len(greetings) == 0 {
return nil, errors.New("no greetings available")
}
if language == "random" {
// Get random item from greetings slice
randomIndex := rand.Intn(len(greetings))
return greetings[randomIndex], nil
}
for _, greeting := range greetings {
if greeting.Language == language {
return greeting, nil
}
}
return nil, fmt.Errorf("no greeting found for language '%s'", language)
}