conf/config.go
@@ -47,6 +47,11 @@
   fileTemplateConf struct {
      InputSelfmadeAddr string //入库自制单模版地址
   }
   dingTalkConf struct {
      AlarmKey string
      AlarmUrl string
   }
)
var (
@@ -56,6 +61,7 @@
   LocalConf      = &localConf{}
   Viper          *viper.Viper
   GrpcServerConf = &grpcServerConf{}
   DingTalkConf   = &dingTalkConf{}
)
func Init() error {
@@ -76,6 +82,7 @@
   apsAddr := os.Getenv("APS_GRPC")
   crmAddr := os.Getenv("CRM_GRPC")
   srmAddr := os.Getenv("SRM_GRPC")
   companyName := os.Getenv("COMPANY_NAME")
   if len(GrpcPort) > 0 {
      WebConf.GrpcPort = GrpcPort
   }
@@ -93,6 +100,9 @@
   }
   if len(srmAddr) > 0 {
      GrpcServerConf.SrmAddr = srmAddr
   }
   if len(companyName) > 0 {
      WebConf.CompanyName = companyName
   }
   DBHost := os.Getenv("DB_HOST")
@@ -117,6 +127,7 @@
   _ = v.UnmarshalKey("db", DbConf)
   _ = v.UnmarshalKey("local", LocalConf)
   _ = v.UnmarshalKey("grpcServer", GrpcServerConf)
   _ = v.UnmarshalKey("dingTalk", DingTalkConf)
   showConfig()
}
@@ -127,5 +138,6 @@
   log.Printf("   DbConf:                 %+v", DbConf)
   log.Printf("   LocalConf:               %+v", LocalConf)
   log.Printf("   GrpcServerConf:               %+v", GrpcServerConf)
   log.Printf("   DingTalkConf:               %+v", DingTalkConf)
   log.Println("......................................................")
}