Browse Source

修改bug

yjp 10 months ago
parent
commit
e4a0cd589d
2 changed files with 6 additions and 38 deletions
  1. 5 38
      baize.go
  2. 1 0
      framework/core/application/config.go

+ 5 - 38
baize.go

@@ -4,8 +4,6 @@ import (
 	"git.sxidc.com/go-framework/baize/framework/core/api"
 	"git.sxidc.com/go-framework/baize/framework/core/application"
 	"git.sxidc.com/go-framework/baize/framework/core/infrastructure"
-	"git.sxidc.com/go-framework/baize/framework/core/infrastructure/database/data_service"
-	"git.sxidc.com/go-framework/baize/framework/core/infrastructure/database/operations"
 )
 
 func NewApplication(conf application.Config) *application.App {
@@ -14,43 +12,12 @@ func NewApplication(conf application.Config) *application.App {
 	apiInstance := api.New(api.WithUrlPrefix(apiConfig.UrlPrefix), api.WithPort(apiConfig.Port))
 
 	// 创建基础设施
-	var infrastructureInstance *infrastructure.Infrastructure
+	infrastructureConfig := new(infrastructure.Config)
+	infrastructureConfig.DatabaseConfig.Operations = conf.InfrastructureConfig.Database.Operations
+	infrastructureConfig.DatabaseConfig.DataService = conf.InfrastructureConfig.Database.DataService
+	infrastructureConfig.CacheConfig = conf.InfrastructureConfig.Cache
 
-	infrastructureConfig := infrastructure.Config{
-		DatabaseConfig: infrastructure.DatabaseConfig{},
-	}
-
-	if conf.InfrastructureConfig.Database.Operations != nil {
-		operationsConfig := conf.InfrastructureConfig.Database.Operations
-		if operationsConfig != nil {
-			infrastructureConfig.DatabaseConfig.Operations = &operations.Config{
-				UserName:           operationsConfig.UserName,
-				Password:           operationsConfig.Password,
-				Address:            operationsConfig.Address,
-				Port:               operationsConfig.Port,
-				Database:           operationsConfig.Database,
-				MaxConnections:     operationsConfig.MaxConnections,
-				MaxIdleConnections: operationsConfig.MaxIdleConnections,
-			}
-		}
-	}
-
-	if conf.InfrastructureConfig.Database.DataService != nil {
-		dataServiceConfig := conf.InfrastructureConfig.Database.DataService
-		if dataServiceConfig != nil {
-			infrastructureConfig.DatabaseConfig.DataService = &data_service.Config{
-				Token:      dataServiceConfig.Token,
-				Address:    dataServiceConfig.Address,
-				HttpPort:   dataServiceConfig.HttpPort,
-				GrpcPort:   dataServiceConfig.GrpcPort,
-				Namespace:  dataServiceConfig.Namespace,
-				DataSource: dataServiceConfig.DataSource,
-				TimeoutSec: dataServiceConfig.TimeoutSec,
-			}
-		}
-	}
-
-	infrastructureInstance = infrastructure.NewInfrastructure(infrastructureConfig)
+	infrastructureInstance := infrastructure.NewInfrastructure(*infrastructureConfig)
 
 	return application.New(apiInstance, infrastructureInstance)
 }

+ 1 - 0
framework/core/application/config.go

@@ -21,6 +21,7 @@ type ApiConfig struct {
 
 type InfrastructureConfig struct {
 	Database infrastructure.DatabaseConfig `json:"database" yaml:"database"`
+	Cache    infrastructure.CacheConfig    `json:"cache" yaml:"cache"`
 }
 
 func LoadFromJsonFile(jsonFilePath string) (Config, error) {