fix : 解决了config中Init未更新全局变量错误

xh-polaris · Aug 10, 2024 · fcc7f4a · fcc7f4a
1 parent 49ce896
commit fcc7f4a
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 7 deletions.
diff --git a/common/config/config.go b/common/config/config.go
@@ -1,6 +1,7 @@
 package config
 
 import (
+	"fmt"
 	"github.com/zeromicro/go-zero/core/stores/cache"
 	"os"
 
@@ -13,29 +14,32 @@ var config *Config
 type Config struct {
 	service.ServiceConf
 	ListenOn string
-	Cache    *cache.CacheConf
+	Cache    cache.CacheConf
 	Mongo    struct {
 		URL string
 		DB  string
 	}
 }
 
 func Init() {
-	c := new(Config)
+	config = new(Config)
 	path := os.Getenv("CONFIG_PATH")
 	if path == "" {
 		path = "etc/config.yaml"
 	}
-	err := conf.Load(path, c)
+	err := conf.Load(path, config)
 	if err != nil {
 		panic(err)
 	}
-	err = c.SetUp()
+	err = config.SetUp()
 	if err != nil {
 		panic(err)
 	}
 }
 
 func Get() *Config {
+	if config == nil {
+		fmt.Println("config is nil")
+	}
 	return config
 }
diff --git a/infra/mapper/follow/mongo.go b/infra/mapper/follow/mongo.go
@@ -35,7 +35,7 @@ type MongoMapper struct {
 
 func NewMongoMapper() IMongoMapper {
 	aConfig := config.Get()
-	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, *aConfig.Cache)
+	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, aConfig.Cache)
 	return &MongoMapper{
 		conn: conn,
 	}

diff --git a/infra/mapper/like/mongo.go b/infra/mapper/like/mongo.go
@@ -35,7 +35,7 @@ type MongoMapper struct {
 
 func NewMongoMapper() IMongoMapper {
 	aConfig := config.Get()
-	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, *aConfig.Cache)
+	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, aConfig.Cache)
 	return &MongoMapper{
 		conn: conn,
 	}

diff --git a/infra/mapper/share/mongo.go b/infra/mapper/share/mongo.go
@@ -34,7 +34,7 @@ type MongoMapper struct {
 
 func NewMongoMapper() IMongoMapper {
 	aConfig := config.Get()
-	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, *aConfig.Cache)
+	conn := monc.MustNewModel(aConfig.Mongo.URL, aConfig.Mongo.DB, CollectionName, aConfig.Cache)
 	return &MongoMapper{
 		conn: conn,
 	}