main.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"

	"github.com/915604903T/ModelController/handlers"

	"github.com/NVIDIA/go-nvml/pkg/nvml"

	"github.com/gorilla/mux"
)

const NameExpression = "-a-zA-Z_0-9."

func main() {
	router := mux.NewRouter()
	router.HandleFunc("/render/scene/{name}", handlers.MakeReceiveFileHandler())
	router.HandleFunc("/relocalise/info", handlers.MakeReceiveRelocInfoHandler())
	router.HandleFunc("/relocalise/scene/{name}", handlers.MakeSendSceneModelHandler())

	// start server but not block
	go handlers.DealSignal()

	// start send resource info to center server
	go handlers.SendResourceInfo()

	// close nvml and two channel
	defer func() {
		ret := nvml.Shutdown()
		if ret != nvml.SUCCESS {
			log.Fatalf("Unable to shutdown NVML: %v", nvml.ErrorString(ret))
		}
	}()
	defer close(handlers.RenderFinish)
	defer close(handlers.RelocaliseFinish)

	// regard tcp port as env viarable
	tcpPort := os.Getenv("PORT")
	// tcpPort, _ := strconv.Atoi(os.Getenv("PORT"))
	handlers.HostAddr = "http://127.0.0.1" + ":" + tcpPort
	s := &http.Server{
		Addr:    fmt.Sprintf(":%s", tcpPort),
		Handler: router,
	}
	log.Println("listen on port: ", tcpPort)
	log.Fatal(s.ListenAndServe())
}