server.go


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

package main

import (
	"flag"
	"fmt"
	"log"

	"net/http"

	"github.com/gorilla/mux"
)

var (
	categories []string
	port       int
)

func main() {
	// Parse the command line flags
	initFlags()

	// Define a new gorilla/mux router and define some endpoints
	r := mux.NewRouter()

	r.HandleFunc("/", indexHandler)
	r.HandleFunc("/dumpall", dumpallHandler)
	r.HandleFunc("/{station}/{name}", getHandler)

	// Start the http server using the port provided using command line arguments
	// The Default port is 8080
	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", port), r))
}

func initFlags() {
	flag.IntVar(&port, "p", 8080, "the port the server should listen on")
	flag.Parse()
}