DKube v1.0

main.go

@@ -3,16 +3,25 @@ package main
 import (
 	"dkube/config"
 	"dkube/controller"
+	"dkube/db"
+	"dkube/middle"
 	"dkube/service"
 	"github.com/gin-gonic/gin"
+	"net/http"
 )
 
 func main() {
-	//初始化k8s client
+	db.Init()
 	service.K8s.Init()
-	//初始化Gin对象
 	r := gin.Default()
+	r.Use(middle.Cors())
 	controller.Router.InitApiRouter(r)
-	//启动gin程序
+
+	go func() {
+		http.HandleFunc("/ws", service.Terminal.WsHandler)
+		http.ListenAndServe(":8081", nil)
+	}()
+
 	r.Run(config.ListenAddr)
+	db.Close()
 }