73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"message-nest/pkg/table"
|
|
"message-nest/service/cron_service"
|
|
"message-nest/service/env_service"
|
|
"message-nest/service/send_message_service"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"message-nest/migrate"
|
|
"message-nest/models"
|
|
"message-nest/pkg/logging"
|
|
"message-nest/pkg/setting"
|
|
"message-nest/routers"
|
|
)
|
|
|
|
var (
|
|
//go:embed web/dist/*
|
|
f embed.FS
|
|
|
|
wg sync.WaitGroup
|
|
)
|
|
|
|
func init() {
|
|
logging.Setup()
|
|
migrate.Setup()
|
|
setting.Setup()
|
|
models.Setup()
|
|
table.Setup()
|
|
env_service.Setup()
|
|
cron_service.Setup()
|
|
}
|
|
|
|
func GinServerUp() {
|
|
gin.SetMode(setting.ServerSetting.RunMode)
|
|
|
|
routersInit := routers.InitRouter(f)
|
|
readTimeout := setting.ServerSetting.ReadTimeout
|
|
writeTimeout := setting.ServerSetting.WriteTimeout
|
|
endPoint := fmt.Sprintf(":%d", setting.ServerSetting.HttpPort)
|
|
maxHeaderBytes := 1 << 20
|
|
|
|
server := &http.Server{
|
|
Addr: endPoint,
|
|
Handler: routersInit,
|
|
ReadTimeout: readTimeout,
|
|
WriteTimeout: writeTimeout,
|
|
MaxHeaderBytes: maxHeaderBytes,
|
|
}
|
|
|
|
logging.Logger.Info("start http server listening http://0.0.0.0", endPoint)
|
|
|
|
err := server.ListenAndServe()
|
|
if err != nil {
|
|
logging.Logger.Error("Server err: ", err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
wg.Add(1)
|
|
|
|
go GinServerUp()
|
|
go send_message_service.MessageConsumer(&wg)
|
|
|
|
wg.Wait()
|
|
fmt.Println("Server exit...")
|
|
}
|