package api
|
|
import (
|
"basic.com/valib/logger.git"
|
"bee-config/controller"
|
"bee-config/routers"
|
"github.com/gin-gonic/gin"
|
"net/http"
|
"sync"
|
)
|
|
var (
|
routines []routers.IRouter = []routers.IRouter{}
|
)
|
|
func init() {
|
routines = append(routines, controller.AddNode, controller.QueryNode)
|
}
|
|
func InitRouter(wg *sync.WaitGroup) *http.Server {
|
r := gin.Default()
|
|
for _, v := range routines {
|
methods := v.GetMethods()
|
if len(methods) > 0 && len(methods) > 0 {
|
for _, m := range methods {
|
r.Handle(m, v.GetPath(), v.GetHandles()...)
|
}
|
}
|
}
|
|
server := &http.Server{
|
Addr: ":8080",
|
Handler: r,
|
}
|
|
wg.Add(1)
|
go func() {
|
if err := server.ListenAndServe(); nil != err {
|
if err != http.ErrServerClosed {
|
logger.Fatal("InitRouter failed with error:", err)
|
} else {
|
logger.Info("InitRouter exited by user")
|
}
|
}
|
|
wg.Done()
|
}()
|
|
return server
|
}
|
