Compare commits

..
3 Commits
+12 -3
View File
@@ -40,6 +40,11 @@ func loadKamojis(path string) Kamojis {
return kamojis return kamojis
} }
func randNum(i int) int {
rand.Seed(time.Now().UnixNano())
return rand.Intn(i)
}
func main() { func main() {
port := flag.String("port", "80", "http listening port") port := flag.String("port", "80", "http listening port")
kamojisPath := flag.String("kamojis", "kamojis.txt", "path to file with kamojis") kamojisPath := flag.String("kamojis", "kamojis.txt", "path to file with kamojis")
@@ -53,19 +58,23 @@ func main() {
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
allk := loadKamojis(*kamojisPath) allk := loadKamojis(*kamojisPath)
timestamp := time.Now().Unix() timestamp := time.Now().Unix()
randomNumber := rand.Intn(len(allk.Kamojis)) randomNumber := randNum(len(allk.Kamojis))
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
if time.Now().Unix()-timestamp > 60 { if time.Now().Unix()-timestamp > 60 {
randomNumber = rand.Intn(len(allk.Kamojis)) randomNumber = randNum(len(allk.Kamojis))
timestamp = time.Now().Unix() timestamp = time.Now().Unix()
log.Println("rotating kamoji.") log.Println("rotating kamoji.")
} }
log.Println("served kamoji to " + r.RemoteAddr + ".") log.Println("serving kamoji to " + r.Header.Get("x-forwarded-for") + ".")
k := allk.Kamojis[randomNumber] k := allk.Kamojis[randomNumber]
tmpl.Execute(w, k) tmpl.Execute(w, k)
}) })
http.ListenAndServe(":"+*port, nil) http.ListenAndServe(":"+*port, nil)
log.Println("starting webserver on port " + *port + ". press ctrl-c to exit.") log.Println("starting webserver on port " + *port + ". press ctrl-c to exit.")
} }