From 6597287fc412ce914aba41fe571ed05817cf371c Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期一, 27 五月 2024 11:34:58 +0800
Subject: [PATCH] fix script
---
config/config.go | 20 ++++++++++++++++----
1 files changed, 16 insertions(+), 4 deletions(-)
diff --git a/config/config.go b/config/config.go
index ed8e823..721adbe 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 {
@@ -38,10 +40,19 @@
LogLevel string `mapstructure: "logLevel"`
}
+type api struct {
+ Host string `mapstructure: "host"`
+ Port string `mapstructure: "port"`
+ 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
@@ -57,6 +68,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