From b0f374409775bd30ed8f0dc8d47d015d3edc0abb Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期二, 03 九月 2024 19:05:52 +0800
Subject: [PATCH] 1、修改查询策略 2、修改地址统计策略

---
 config/config.go |   23 ++++++++++++++++-------
 1 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/config/config.go b/config/config.go
index f6ab35f..f29e1cd 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 {
@@ -39,8 +42,12 @@
 }
 
 type api struct {
-	Host string `mapstructure: "host"`
-	Port string `mapstructure: "port"`
+	Host          string `mapstructure: "host"`
+	Port          string `mapstructure: "port"`
+	TimeThreshold int    `mapstructure:"timeThreshold"`
+	CsTimes       int    `mapstructure:"csTimes"`
+	CsHours       int    `mapstructure:"csHours"`
+	AInterval     int    `mapstructure:"aInterval"`
 }
 
 var LogConf = &LogConfig{}
@@ -55,7 +62,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)

--
Gitblit v1.8.0