diff --git a/main.go b/main.go index fc23617..d43bf72 100644 --- a/main.go +++ b/main.go @@ -11,6 +11,7 @@ import ( "git.lxtend.com/qqbot/config" "git.lxtend.com/qqbot/service/exec" "git.lxtend.com/qqbot/sqlite3" + "git.lxtend.com/qqbot/webserver" _ "git.lxtend.com/qqbot/works" wsclient "git.lxtend.com/qqbot/ws_client" "github.com/gin-gonic/gin" @@ -65,7 +66,7 @@ func main() { time.Sleep(reconnectDelay) } }() - startRouter() + webserver.StartRouter() for { time.Sleep(1000 * time.Second) } diff --git a/router.go b/webserver/router.go similarity index 96% rename from router.go rename to webserver/router.go index 513e73f..22530ca 100644 --- a/router.go +++ b/webserver/router.go @@ -1,4 +1,4 @@ -package main +package webserver import ( "time" @@ -10,7 +10,7 @@ import ( "github.com/gin-gonic/gin" ) -func startRouter() { +func StartRouter() { ginServer := gin.New() ginServer.Use(cors.New(cors.Config{ AllowOrigins: []string{"https://www.lxtend.com", "https://git.lxtend.com"},