217 lines
7.7 KiB
Go
217 lines
7.7 KiB
Go
package srvconfig
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net"
|
||
"net/http"
|
||
"os"
|
||
"regexp"
|
||
"strings"
|
||
"xiawan/wx/clientsdk/baseinfo"
|
||
"xiawan/wx/clientsdk/baseutils"
|
||
"xiawan/wx/clientsdk/mmtls"
|
||
|
||
"github.com/gogf/gf/database/gredis"
|
||
)
|
||
|
||
// GlobalSetting 全局设置
|
||
var GlobalSetting Setting
|
||
|
||
// TaskExecWaitTimes 任务执行间隔时间 500毫秒
|
||
var TaskExecWaitTimes = uint32(500)
|
||
|
||
// ProxyConfig 代理配置
|
||
type ProxyConfig struct {
|
||
// 代理失效后是否允许直连（false=不允许，连接失败；true=允许，降级为直连）
|
||
AllowDirectOnProxyFail bool `json:"allowDirectOnProxyFail"`
|
||
// 长连接超时时间（秒）
|
||
LongConnTimeout int `json:"longConnTimeout"`
|
||
LongConnReadTimeout int `json:"longConnReadTimeout"`
|
||
// 长连接建立时的重试次数
|
||
LongConnRetryTimes int `json:"longConnRetryTimes"`
|
||
// 长连接建立时的重试间隔（毫秒）
|
||
LongConnRetryInterval int `json:"longConnRetryInterval"`
|
||
// 短连接超时时间（秒）
|
||
ShortConnTimeout int `json:"shortConnTimeout"`
|
||
// 长连接断开后最大重试恢复次数
|
||
MaxLongRetryTimes int `json:"maxLongRetryTimes"`
|
||
// 长连接断开后重试间隔（秒）
|
||
LongRetryInterval int `json:"longRetryInterval"`
|
||
}
|
||
|
||
// Setting 设置
|
||
type Setting struct {
|
||
Debug bool `json:"debug"`
|
||
Host string `json:"host"`
|
||
Port string `json:"port"`
|
||
ApiVersion string `json:"apiVersion"`
|
||
OpenClearMsg bool `json:"openClearMsg"`
|
||
SycnTargetIpUrl string `json:"sycnTargetIpUrl"` // 同步外网IP地址的url
|
||
|
||
WorkerPoolSize uint32 `json:"workerpoolsize"`
|
||
MaxWorkerTaskLen uint32 `json:"maxworkertasklen"`
|
||
WebDomain string `json:"webdomain"`
|
||
WebDomainNum string `json:"webdomainnum"`
|
||
SyncOnlineCount string `json:"syncOnlineCount"`
|
||
WebTaskName string `json:"webtaskname"`
|
||
WebTaskAppNumber string `json:"webtaskappnumber"`
|
||
//Redis
|
||
RedisConfig gredis.Config `json:"redisConfig"`
|
||
//消息同步 是否按微信id 发布消息
|
||
NewsSynWxId bool `json:"newsSynWxId"`
|
||
Dt bool `json:"dt"`
|
||
MysqlConnectStr string `json:"mySqlConnectStr"`
|
||
InitializeMySQLTables bool `json:"initializeMySQLTables"`
|
||
UserInputKey string `json:"userInputKey"`
|
||
Version string `json:"version"`
|
||
HttpSyncMsg bool `json:"httpSyncMsg"`
|
||
IsInitOld bool `json:"isInitOld"`
|
||
WsSyncMsg bool `json:"wsSyncMsg"`
|
||
SyncHisyory bool `json:"syncHisyory"`
|
||
LoginBeforeRequestUrl string `json:"loginBeforeRequestUrl"` // 登录前会请求该 url（二维码和唤醒）
|
||
// 当前服务器外网IP地址
|
||
TargetIp string `json:"targetIp"`
|
||
// 当前 HTTP-Api 服务的本地地址 基址
|
||
LocalApiUrl string
|
||
// 推广关注自己的微信公众号(新用户首次登录时关注)
|
||
GhWxid string `json:"ghWxid"`
|
||
// 管理接口授权 ADMIN_KEY
|
||
AdminKey string `json:"adminKey"`
|
||
// 是否开启 mac2ipad
|
||
Mac2Ipad bool `json:"mac2ipad"`
|
||
// 是否开启 car2ipad
|
||
Car2Ipad bool `json:"car2ipad"`
|
||
// 代理配置
|
||
ProxyConfig ProxyConfig `json:"proxyConfig"`
|
||
}
|
||
|
||
// getExternal 请求获取外网ip
|
||
func getExternal() string {
|
||
resp, err := http.Get(GlobalSetting.SycnTargetIpUrl)
|
||
if err != nil {
|
||
fmt.Printf("获取 ip 失败 -> 退出程序!---->")
|
||
os.Exit(1)
|
||
// return fmt.Sprintf("%s:%s", GlobalSetting.Host, GlobalSetting.Port)
|
||
}
|
||
defer resp.Body.Close()
|
||
content, _ := ioutil.ReadAll(resp.Body)
|
||
if content == nil {
|
||
fmt.Printf("获取 ip 失败 -> 退出程序!---->")
|
||
os.Exit(1)
|
||
// return fmt.Sprintf("%s:%s", GlobalSetting.Host, GlobalSetting.Port)
|
||
}
|
||
// 校验一下是不是 ip
|
||
if net.ParseIP(string(content)) == nil {
|
||
fmt.Printf("获取 ip 失败 -> 退出程序!---->")
|
||
os.Exit(1)
|
||
}
|
||
fmt.Printf("获取 ip 成功 -> %s ----> %s\n", content, GlobalSetting.SycnTargetIpUrl)
|
||
|
||
return fmt.Sprintf("%s:%s", content, GlobalSetting.Port)
|
||
}
|
||
|
||
func isValidWeChatID(wxid string) bool {
|
||
if strings.TrimSpace(wxid) == "" {
|
||
GlobalSetting.GhWxid = ""
|
||
return true
|
||
}
|
||
regex := `^[a-zA-Z][a-zA-Z0-9_-]{5,19}$`
|
||
r := regexp.MustCompile(regex)
|
||
return r.MatchString(wxid)
|
||
}
|
||
|
||
func ConfigSetUp() {
|
||
tmpHomeDir, err := os.Getwd()
|
||
if err != nil {
|
||
baseutils.PrintLog(err.Error())
|
||
return
|
||
}
|
||
baseinfo.HomeDIR = tmpHomeDir
|
||
|
||
// 读取配置文件
|
||
configData, err := baseutils.ReadFile(baseinfo.HomeDIR + "/assets/setting.json")
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = json.Unmarshal(configData, &GlobalSetting)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return
|
||
}
|
||
|
||
if len(strings.TrimSpace(GlobalSetting.AdminKey)) == 0 {
|
||
// 默认授权admin_key
|
||
GlobalSetting.AdminKey = "df704bd7012548049e68db25aff0507f"
|
||
}
|
||
fmt.Println("======== ADMIN_KEY ===", GlobalSetting.AdminKey, "========")
|
||
|
||
if !isValidWeChatID(GlobalSetting.GhWxid) {
|
||
fmt.Println("引流公众号(assets/setting.json ghWxid)不合法:", GlobalSetting.GhWxid, ";已重设为空")
|
||
GlobalSetting.GhWxid = ""
|
||
}
|
||
|
||
// 确保API版本格式正确，以斜杠开头
|
||
if GlobalSetting.ApiVersion != "" && !strings.HasPrefix(GlobalSetting.ApiVersion, "/") {
|
||
GlobalSetting.ApiVersion = "/" + GlobalSetting.ApiVersion
|
||
}
|
||
if GlobalSetting.ApiVersion == "/" {
|
||
GlobalSetting.ApiVersion = ""
|
||
}
|
||
// 如果ApiVersion不为空且不以斜杠结尾，则添加斜杠
|
||
if GlobalSetting.ApiVersion != "" && !strings.HasSuffix(GlobalSetting.ApiVersion, "/") {
|
||
GlobalSetting.ApiVersion += "/"
|
||
}
|
||
GlobalSetting.LocalApiUrl = fmt.Sprintf("http://127.0.0.1:%s%s", GlobalSetting.Port, GlobalSetting.ApiVersion)
|
||
|
||
// 获取当前服务器外网IP地址
|
||
// 如果没有指定 TargetIp, 同时 SycnTargetIpUrl 不为空, 则请求 SycnTargetIpUrl 获取外网 IP
|
||
if GlobalSetting.TargetIp == "" && GlobalSetting.SycnTargetIpUrl != "" {
|
||
GlobalSetting.TargetIp = getExternal()
|
||
}
|
||
|
||
// 设置代理配置默认值
|
||
if GlobalSetting.ProxyConfig.LongConnTimeout <= 0 {
|
||
GlobalSetting.ProxyConfig.LongConnTimeout = 15
|
||
}
|
||
if GlobalSetting.ProxyConfig.LongConnReadTimeout <= 0 {
|
||
readTimeout := GlobalSetting.ProxyConfig.LongConnTimeout
|
||
if readTimeout < 210 {
|
||
readTimeout = 210
|
||
}
|
||
GlobalSetting.ProxyConfig.LongConnReadTimeout = readTimeout
|
||
}
|
||
if GlobalSetting.ProxyConfig.LongConnRetryTimes <= 0 {
|
||
GlobalSetting.ProxyConfig.LongConnRetryTimes = 30
|
||
}
|
||
if GlobalSetting.ProxyConfig.LongConnRetryInterval <= 0 {
|
||
GlobalSetting.ProxyConfig.LongConnRetryInterval = 500
|
||
}
|
||
if GlobalSetting.ProxyConfig.ShortConnTimeout <= 0 {
|
||
GlobalSetting.ProxyConfig.ShortConnTimeout = 15
|
||
}
|
||
if GlobalSetting.ProxyConfig.MaxLongRetryTimes <= 0 {
|
||
GlobalSetting.ProxyConfig.MaxLongRetryTimes = 10
|
||
}
|
||
if GlobalSetting.ProxyConfig.LongRetryInterval <= 0 {
|
||
GlobalSetting.ProxyConfig.LongRetryInterval = 60
|
||
}
|
||
|
||
// 同步代理配置到 mmtls 全局配置
|
||
mmtls.GlobalProxyConfig.LongConnTimeout = GlobalSetting.ProxyConfig.LongConnTimeout
|
||
mmtls.GlobalProxyConfig.LongConnReadTimeout = GlobalSetting.ProxyConfig.LongConnReadTimeout
|
||
mmtls.GlobalProxyConfig.LongConnRetryTimes = GlobalSetting.ProxyConfig.LongConnRetryTimes
|
||
mmtls.GlobalProxyConfig.LongConnRetryInterval = GlobalSetting.ProxyConfig.LongConnRetryInterval
|
||
mmtls.GlobalProxyConfig.ShortConnTimeout = GlobalSetting.ProxyConfig.ShortConnTimeout
|
||
mmtls.GlobalProxyConfig.AllowDirectOnProxyFail = GlobalSetting.ProxyConfig.AllowDirectOnProxyFail
|
||
|
||
fmt.Printf("======== 代理配置已同步 ========\n")
|
||
fmt.Printf("代理失效后允许直连: %v\n", GlobalSetting.ProxyConfig.AllowDirectOnProxyFail)
|
||
fmt.Printf("长连接超时: %d秒\n", GlobalSetting.ProxyConfig.LongConnTimeout)
|
||
fmt.Printf("长连接重试次数: %d次\n", GlobalSetting.ProxyConfig.LongConnRetryTimes)
|
||
fmt.Printf("长连接重试间隔: %dms\n", GlobalSetting.ProxyConfig.LongConnRetryInterval)
|
||
fmt.Printf("短连接超时: %d秒\n", GlobalSetting.ProxyConfig.ShortConnTimeout)
|
||
fmt.Printf("========================\n")
|
||
}
|