fix: only init on standard start

api/pkg/menu/main.go

@@ -3,6 +3,7 @@ package menu
 import (
 	"github.com/OhYee/blotter/mongo"
 	"github.com/OhYee/blotter/output"
+	"github.com/OhYee/blotter/utils/initial"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
@@ -10,9 +11,11 @@ import (
 const DatabaseName = "pages"
 
 func init() {
-	if exists, _ := mongo.CollectionExists("blotter", DatabaseName); exists == false {
-		initMenus()
-	}
+	initial.Register(func() {
+		if exists, _ := mongo.CollectionExists("blotter", DatabaseName); exists == false {
+			initMenus()
+		}
+	})
 }
 
 func initMenus() {

api/pkg/post/jieba.go

@@ -4,6 +4,7 @@ import (
 	"sync"
 
 	"github.com/OhYee/blotter/output"
+	"github.com/OhYee/blotter/utils/initial"
 	"github.com/yanyiwu/gojieba"
 )
 
@@ -31,5 +32,5 @@ func getJieba() *gojieba.Jieba {
 }
 
 func init() {
-	go initJieba()
+	initial.Register(initJieba)
 }

api/pkg/post/update.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/OhYee/blotter/mongo"
+	"github.com/OhYee/blotter/utils/initial"
 	"go.mongodb.org/mongo-driver/bson"
 )
 
@@ -35,5 +36,5 @@ func updatePublishedToStatus() {
 }
 
 func init() {
-	go updatePublishedToStatus()
+	initial.Register(updatePublishedToStatus)
 }

main.go

@@ -11,6 +11,7 @@ import (
 	"github.com/OhYee/blotter/http"
 	"github.com/OhYee/blotter/output"
 	"github.com/OhYee/blotter/register"
+	"github.com/OhYee/blotter/utils/initial"
 )
 
 var (
@@ -55,13 +56,14 @@ func main() {
 		}
 		f()
 		os.Exit(0)
+		return
 	}
 
-
 	cron.Start()
 	defer cron.Stop()
 
 	api.Register()
+	initial.Run()
 	// queue.Register().Register("extensions/queue")
 	// register.DebugApiMap()
 	output.Log("Server will start at http://%s", addr)

mongo/conn.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"sync"
 
+	"github.com/OhYee/blotter/utils/initial"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
@@ -32,5 +33,5 @@ func getClientOptions() *options.ClientOptions {
 }
 
 func init() {
-	go initClientOptions()
+	initial.Register(initClientOptions)
 }

utils/initial/initial.go

@@ -0,0 +1,15 @@
+package initial
+
+var (
+	initialFunction = make([]func(), 0)
+)
+
+func Register(f func()) {
+	initialFunction = append(initialFunction, f)
+}
+
+func Run() {
+	for _, f := range initialFunction {
+		go f()
+	}
+}