geo-smart-system/main.go

package main

import (
	"fmt"
	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	"github.com/go-redis/redis"
	socketio "github.com/googollee/go-socket.io"
	"github.com/supanadit/geosmartsystem/model"
	"log"
)

func main() {
	// Tile38
	client := redis.NewClient(&redis.Options{
		Addr: "localhost:9851",
	})

	router := gin.Default()
	router.Use(cors.New(cors.Config{
		AllowOrigins:     []string{"http://localhost:4200"},
		AllowMethods:     []string{"PUT", "PATCH", "POST", "GET"},
		AllowHeaders:     []string{"Origin"},
		ExposeHeaders:    []string{"Content-Length"},
		AllowWebSockets:  true,
		AllowCredentials: true,
		AllowWildcard:    true,
	}))
	router.GET("/point", func(c *gin.Context) {
		data, _ := model.FromScan(client, "sales")
		c.JSON(200, data)
	})
	// Socket.IO Start
	server, err := socketio.NewServer(nil)
	if err != nil {
		log.Fatal(err)
	}
	server.OnConnect("/", func(s socketio.Conn) error {
		s.SetContext("")
		fmt.Println("Connected:", s.ID())
		return nil
	})
	server.OnEvent("/", "message", func(s socketio.Conn, msg string) {
		s.Emit("message", "have "+msg)
	})
	server.OnEvent("/", "bye", func(s socketio.Conn) string {
		last := s.Context().(string)
		s.Emit("bye", last)
		_ = s.Close()
		return last
	})
	server.OnError("/", func(e error) {
		fmt.Println("Meet Error:", e)
	})
	server.OnDisconnect("/", func(s socketio.Conn, msg string) {
		fmt.Println("Closed", msg)
	})
	router.GET("/socket.io/", gin.WrapH(server))
	router.POST("/socket.io/", gin.WrapH(server))
	router.Handle("WS", "/socket.io/", WebSocketIO(server))
	router.Handle("WSS", "/socket.io/", WebSocketIO(server))
	router.GET("/ws", func(c *gin.Context) {
		server.ServeHTTP(c.Writer, c.Request)
	})
	go server.Serve()
	defer server.Close()
	// End Socket.IO
	_ = router.Run()
}

func WebSocketIO(server *socketio.Server) gin.HandlerFunc {
	return func(c *gin.Context) {
		server.ServeHTTP(c.Writer, c.Request)
	}
}