init repo

cmd/db.go

@@ -0,0 +1,50 @@
+package main
+
+import (
+	"context"
+
+	"github.com/nose7en/ToyBoomServer/common"
+	"github.com/nose7en/ToyBoomServer/config"
+	"github.com/nose7en/ToyBoomServer/models"
+	"github.com/nose7en/ToyBoomServer/storage"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func initDatabase() {
+	c := context.Background()
+
+	common.Logger(c).Infof("start to init database, type: %s", config.GetSettings().DB.Type)
+	storage.MustInitDBManager(nil, config.GetSettings().DB.Type)
+
+	switch config.GetSettings().DB.Type {
+	case "sqlite":
+		if sqlitedb, err := gorm.Open(sqlite.Open(config.GetSettings().DB.DSN), &gorm.Config{}); err != nil {
+			common.Logger(c).Panic(err)
+		} else {
+			storage.GetDBManager().SetDB("sqlite", storage.DefaultDBName, sqlitedb)
+			common.Logger(c).Infof("init database success, data location: [%s]", config.GetSettings().DB.DSN)
+		}
+	case "mysql":
+		if mysqlDB, err := gorm.Open(mysql.Open(config.GetSettings().DB.DSN), &gorm.Config{}); err != nil {
+			common.Logger(c).Panic(err)
+		} else {
+			storage.GetDBManager().SetDB("mysql", storage.DefaultDBName, mysqlDB)
+			common.Logger(c).Infof("init database success, data type: [%s]", "mysql")
+		}
+	case "postgres":
+		if postgresDB, err := gorm.Open(postgres.Open(config.GetSettings().DB.DSN), &gorm.Config{}); err != nil {
+			common.Logger(c).Panic(err)
+		} else {
+			storage.GetDBManager().SetDB("postgres", storage.DefaultDBName, postgresDB)
+			common.Logger(c).Infof("init database success, data type: [%s]", "postgres")
+		}
+	default:
+		common.Logger(c).Panicf("currently unsupported database type: %s", config.GetSettings().DB.Type)
+	}
+
+	storage.GetDBManager().Init(models.Models()...)
+}

cmd/main.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	"context"
+
+	"github.com/nose7en/ToyBoomServer/biz"
+	"github.com/nose7en/ToyBoomServer/common"
+	"github.com/nose7en/ToyBoomServer/config"
+	"github.com/nose7en/ToyBoomServer/middleware"
+	"github.com/nose7en/ToyBoomServer/rpc"
+	"github.com/nose7en/ToyBoomServer/services/api"
+
+	"github.com/sourcegraph/conc"
+)
+
+// go:embed all:out/*
+// var fs embed.FS
+
+func main() {
+	c := context.Background()
+
+	config.InitSettings()
+	rpc.InitManager()
+	middleware.Init()
+	initDatabase()
+
+	// ------------ biz init ----------------
+	// --------------------------------------
+
+	r := biz.Router()
+	// biz.HandleStaticFile(r, fs, ar)
+
+	apiService := api.NewAPIHandler(config.GetSettings().APIListenAddr)
+	apiService.Init(r)
+	// watherService := watcher.NewClient(func() error {
+	// 	return nil
+	// })
+
+	common.Logger(c).Infof("------------------------------------------------")
+	common.Logger(c).Infof("init services done, start to run all services...")
+	common.Logger(c).Infof("------------------------------------------------")
+
+	var wg conc.WaitGroup
+	wg.Go(apiService.Run)
+	// wg.Go(watherService.Run)
+	wg.Wait()
+}