single intilialize DataService/TweetSearchService/ObjectStorageService

internal/dao/dao.go

@@ -1,6 +1,8 @@
 package dao
 
 import (
+	"sync"
+
 	"github.com/rocboss/paopao-ce/internal/conf"
 	"github.com/rocboss/paopao-ce/internal/core"
 	"github.com/rocboss/paopao-ce/internal/dao/jinzhu"
@@ -11,23 +13,34 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
-func NewDataService() (s core.DataService) {
+var (
+	ts  core.TweetSearchService
+	ds  core.DataService
+	oss core.ObjectStorageService
+
+	onceTs, onceDs, onceOss sync.Once
+)
+
+func DataService() core.DataService {
+	onceDs.Do(func() {
 		var v core.VersionInfo
 		if conf.CfgIf("Gorm") {
-		s, v = jinzhu.NewDataService()
+			ds, v = jinzhu.NewDataService()
 		} else if conf.CfgIf("Sqlx") && conf.CfgIf("MySQL") {
-		s, v = sakila.NewDataService()
+			ds, v = sakila.NewDataService()
 		} else if conf.CfgIf("Sqlx") && (conf.CfgIf("Postgres") || conf.CfgIf("PostgreSQL")) {
-		s, v = slonik.NewDataService()
+			ds, v = slonik.NewDataService()
 		} else {
 			// default use gorm as orm for sql database
-		s, v = jinzhu.NewDataService()
+			ds, v = jinzhu.NewDataService()
 		}
 		logrus.Infof("use %s as data service with version %s", v.Name(), v.Version())
-	return
+	})
+	return ds
 }
 
-func NewObjectStorageService() (oss core.ObjectStorageService) {
+func ObjectStorageService() core.ObjectStorageService {
+	onceOss.Do(func() {
 		var v core.VersionInfo
 		if conf.CfgIf("AliOSS") {
 			oss, v = storage.NewAliossService()
@@ -46,14 +59,13 @@ func NewObjectStorageService() (oss core.ObjectStorageService) {
 			return
 		}
 		logrus.Infof("use %s as object storage by version %s", v.Name(), v.Version())
-	return
+	})
+	return oss
 }
 
-func NewTweetSearchService() core.TweetSearchService {
+func TweetSearchService() core.TweetSearchService {
-	var (
+	onceTs.Do(func() {
-		ts core.TweetSearchService
+		var v core.VersionInfo
-		v  core.VersionInfo
-	)
 		ams := newAuthorizationManageService()
 		if conf.CfgIf("Zinc") {
 			ts, v = search.NewZincTweetSearchService(ams)
@@ -65,7 +77,9 @@ func NewTweetSearchService() core.TweetSearchService {
 		}
 		logrus.Infof("use %s as tweet search serice by version %s", v.Name(), v.Version())
 
-	return search.NewBridgeTweetSearchService(ts)
+		ts = search.NewBridgeTweetSearchService(ts)
+	})
+	return ts
 }
 
 func newAuthorizationManageService() (s core.AuthorizationManageService) {

internal/routers/api/api.go

@@ -10,5 +10,5 @@ var (
 )
 
 func Initialize() {
-	objectStorage = dao.NewObjectStorageService()
+	objectStorage = dao.ObjectStorageService()
 }

internal/service/service.go

@@ -14,8 +14,8 @@ var (
 )
 
 func Initialize() {
-	ds = dao.NewDataService()
+	ds = dao.DataService()
-	ts = dao.NewTweetSearchService()
+	ts = dao.TweetSearchService()
-	oss = dao.NewObjectStorageService()
+	oss = dao.ObjectStorageService()
 	DisablePhoneVerify = !conf.CfgIf("Sms")
 }