From 3737ab3dd0cc753be986638316c96cb3114601e4 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期日, 29 九月 2024 16:20:46 +0800
Subject: [PATCH] fix db column
---
config/config.go | 27 ++++++++++++++++++++++-----
1 files changed, 22 insertions(+), 5 deletions(-)
diff --git a/config/config.go b/config/config.go
index ed8e823..61d850f 100644
--- a/config/config.go
+++ b/config/config.go
@@ -15,10 +15,12 @@
}
type database struct {
- Driver string `mapstructure: "driver"`
- Host string `mapstructure: "host"`
- Port string `mapstructure: "port"`
- Name string `mapstructure: "name"`
+ Driver string `mapstructure: "driver"`
+ Host string `mapstructure: "host"`
+ Port string `mapstructure: "port"`
+ Name string `mapstructure: "name"`
+ Username string `mapstructure: "username"`
+ Password string `mapstructure: "password"`
}
type elastic struct {
@@ -30,6 +32,7 @@
TopHitsSize int `mapstructure: "topHitsSize"`
CameraSize int `mapstructure: "cameraSize"`
TimeInterval int `mapstructure: "timeInterval"`
+ BatchSize int `mapstructure: "batchSize"`
}
type app struct {
@@ -38,10 +41,21 @@
LogLevel string `mapstructure: "logLevel"`
}
+type api struct {
+ Host string `mapstructure: "host"`
+ Port string `mapstructure: "port"`
+ CapAddrDaysThreshold int `mapstructure:"capAddrDaysThreshold"`
+ TimeThreshold int `mapstructure:"timeThreshold"`
+ CsTimes int `mapstructure:"csTimes"`
+ CsHours int `mapstructure:"csHours"`
+ AInterval int `mapstructure:"aInterval"`
+}
+
var LogConf = &LogConfig{}
var DataBase = &database{}
var Elastic = &elastic{}
var App = &app{}
+var Api = &api{}
var LogBasePath string
var LogLevel int
@@ -49,7 +63,9 @@
var err error
viper.SetConfigType("yaml")
viper.SetConfigName(env)
- viper.AddConfigPath("config")
+ viper.AddConfigPath("./")
+ viper.AddConfigPath("./config")
+ viper.AddConfigPath("../config")
err = viper.ReadInConfig()
if err != nil {
log.Fatal("error on parsing configuration file", err)
@@ -57,6 +73,7 @@
viper.UnmarshalKey("elastic", Elastic)
viper.UnmarshalKey("database", DataBase)
viper.UnmarshalKey("app", App)
+ viper.UnmarshalKey("api", Api)
viper.UnmarshalKey("log", LogConf)
logger.SetLevel(LogConf.Level)
if viper.GetString("LogBasePath") != "" {
--
Gitblit v1.8.0