🆕初始化脚本以及一些工具类

This commit is contained in:
coward
2024-03-06 16:24:49 +08:00
parent a9333df157
commit 7fd186d298
15 changed files with 500 additions and 11 deletions

View File

@@ -2,6 +2,8 @@ package initialize
import (
"fmt"
"github.com/cowardmrx/go_aliyun_oss"
"github.com/go-resty/resty/v2"
"github.com/redis/go-redis/v9"
"golang.zx2c4.com/wireguard/wgctrl"
"gopkg.in/yaml.v3"
@@ -20,10 +22,12 @@ import (
// Init
// @description: 初始化
func Init() {
initConfig() // 读取配置文件
initWireguard() // 初始化wireguard客户端
initDatabase() // 初始化数据库
initRedis() // 初始化redis
initConfig() // 读取配置文件
initWireguard() // 初始化wireguard客户端
initDatabase() // 初始化数据库
initRedis() // 初始化redis
initHttpClient() // 初始化http客户端
initOSS() // 初始化oss客户端链接
}
// initConfig
@@ -98,3 +102,27 @@ func initRedis() {
client.Redis = c
}
// initHttpClient
// @description: 初始化http客户端
func initHttpClient() {
client.HttpClient = resty.New()
}
// initOSS
// @description: 初始化oss客户端
func initOSS() {
if config.Config.File.Type != "oss" {
return
}
ossConfig := &go_aliyun_oss.AliOssConfig{
EndPoint: config.Config.File.Endpoint,
AccessKeyId: config.Config.File.AccessId,
AccessKeySecret: config.Config.File.AccessSecret,
BucketName: config.Config.File.BucketName,
OriginalFileName: false,
}
ossClient := ossConfig.CreateOssConnect()
client.OSS = ossClient
}