torchapi-hive-system/torchapi-hive-system.go at master · fankserver/torchapi-hive-system · GitHub

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
package main

import (
	"context"
	"flag"
	"fmt"
	"net/http"
	_ "net/http/pprof"
	"os"
	"os/signal"
	"time"

	"github.com/fankserver/torchapi-hive-system/src/hive"
	"github.com/fankserver/torchapi-hive-system/src/notification"
	"github.com/globalsign/mgo/bson"
	"github.com/gorilla/mux"
	"github.com/sirupsen/logrus"
)

var (
	dbConnection = flag.String("dbconn", "mongodb://localhost", "mongodb connection string")
)

func main() {
	flag.Parse()

	go func() {
		logrus.Println(http.ListenAndServe(":6060", nil))
	}()

	system, err := hive.NewSystem(*dbConnection)
	if err != nil {
		logrus.Fatalln(err.Error())
	}

	hub := notification.NewHub()
	hub.RegisterEventHandler(system.ProcessSectorEvent)
	go hub.Run()

	// subscribe to SIGINT signals
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, os.Interrupt)

	router := mux.NewRouter()
	router.HandleFunc("/", func(writer http.ResponseWriter, _ *http.Request) {
		fmt.Fprint(writer, "TorchAPI Hive System")
	}).Methods(http.MethodGet)
	router.HandleFunc("/ws/hive/{hive_id:[a-z0-9]+}/sector/{sector_id:[a-z0-9]+}", func(w http.ResponseWriter, r *http.Request) {
		vars := mux.Vars(r)

		hiveID := bson.ObjectIdHex(vars["hive_id"])
		sectorID := bson.ObjectIdHex(vars["sector_id"])

		valid, err := system.IsSectorValid(hiveID, sectorID)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}

		if !valid {
			http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
			return
		}

		notification.ServeWs(hub, w, r, hiveID.Hex(), sectorID.Hex())
	})
	router.HandleFunc("/api/hive", system.GetHives).Methods(http.MethodGet)
	router.HandleFunc("/api/hive", system.CreateHive).Methods(http.MethodPost)
	router.HandleFunc("/api/hive/{hive_id:[a-z0-9]+}/faction", system.GetFactions).Methods(http.MethodGet)
	router.HandleFunc("/api/hive/{hive_id:[a-z0-9]+}/faction", system.DeleteFactions).Methods(http.MethodDelete)
	router.HandleFunc("/api/hive/{hive_id:[a-z0-9]+}/sector", system.GetSectors).Methods(http.MethodGet)
	router.HandleFunc("/api/hive/{hive_id:[a-z0-9]+}/sector", system.CreateSector).Methods(http.MethodPost)
	router.HandleFunc("/api/hive/{hive_id:[a-z0-9]+}/sector/{sector_id:[a-z0-9]+}", system.DeleteSector).Methods(http.MethodDelete)

	srv := &http.Server{
		Addr:    ":8080",
		Handler: router,
	}
	go func() {
		logrus.Info("server started")
		if err := srv.ListenAndServe(); err != http.ErrServerClosed {
			logrus.Fatalf("listen: %s\n", err)
		}
	}()

	<-quit
	logrus.Println("shutting down server...")
	ctxTimeout, cancel := context.WithTimeout(context.Background(), 60*time.Second)
	defer cancel()
	if err := srv.Shutdown(ctxTimeout); err != nil {
		logrus.Fatalf("could not shutdown: %v", err)
	}
	logrus.Println("server gracefully stopped")
}