From 15dc6df83a839f5913997e4e2a252a0714dc2978 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 13 六月 2024 11:39:33 +0800
Subject: [PATCH] osd 添加上下行
---
config/config.go | 13 ++++++++++++-
1 files changed, 12 insertions(+), 1 deletions(-)
diff --git a/config/config.go b/config/config.go
index 9e6034b..cebb37e 100644
--- a/config/config.go
+++ b/config/config.go
@@ -47,8 +47,12 @@
ReportServer string `mapstructure:"report-server"`
ReportInterval int `mapstructure:"report-interval"`
RetryInterval int `mapstructure:"retry-interval"`
- CutFaceImage bool `mapstructure:"cut-face-image"`
Enable bool `mapstructure:"enable"`
+}
+
+type image struct {
+ CutFaceImage bool `mapstructure:"cut-face-image"`
+ Enlarge int `mapstructure:"enlarge"`
}
// 姊帶璁惧
@@ -69,6 +73,7 @@
var RateLimitConf = &rateLimitConfig{}
var ClientConf = &client{}
var NVCSConf = &nvcs{}
+var ImageConf = &image{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -77,6 +82,7 @@
v := viper.New()
v.SetConfigType("yaml")
v.SetConfigName("gat1400")
+ v.AddConfigPath("./")
v.AddConfigPath("./config/")
v.AddConfigPath("../config/")
err = v.ReadInConfig()
@@ -98,6 +104,7 @@
v.UnmarshalKey("rate-limit", RateLimitConf)
v.UnmarshalKey("client", ClientConf)
v.UnmarshalKey("nvcs", NVCSConf)
+ v.UnmarshalKey("image", ImageConf)
if LogConf.Level == "" {
LogConf.Level = "info"
@@ -119,5 +126,9 @@
ClientConf.Proto = "http"
}
+ if ImageConf.Enlarge == 0 {
+ ImageConf.Enlarge = 100
+ }
+
logger.SetLogLevel(LogConf.Level)
}
--
Gitblit v1.8.0