1. 小程序端
2. 视频优化
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
@@ -17,37 +18,102 @@ var DB *gorm.DB
|
||||
// JWTSecret is the secret key used for signing JWT tokens
|
||||
var JWTSecret = "your-secret-key" // Default value, should be set via JWT_SECRET environment variable in production
|
||||
|
||||
// LoadEnvFile 从可执行文件同级目录加载 .env 文件
|
||||
func LoadEnvFile() {
|
||||
// 获取可执行文件路径
|
||||
execPath, err := os.Executable()
|
||||
// utf8BOM 用于剥离 Windows 编辑器常写入的 UTF-8 BOM,否则 godotenv 会解析失败
|
||||
var utf8BOM = []byte{0xEF, 0xBB, 0xBF}
|
||||
|
||||
// tryLoadEnv 尝试加载指定路径的 .env;成功则记录日志并返回 true。
|
||||
// 会先去掉 UTF-8 BOM,再按 godotenv 默认语义:不覆盖已存在的 OS 环境变量。
|
||||
func tryLoadEnv(path string) bool {
|
||||
if path == "" {
|
||||
return false
|
||||
}
|
||||
raw, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
log.Printf("Warning: Failed to get executable path: %v. Using current working directory.", err)
|
||||
// 如果获取可执行文件路径失败,使用当前工作目录
|
||||
if err := godotenv.Load(".env"); err != nil {
|
||||
log.Printf("Info: .env file not found in current directory: %v", err)
|
||||
} else {
|
||||
log.Println("Loaded .env file from current directory")
|
||||
return false
|
||||
}
|
||||
raw = bytes.TrimPrefix(raw, utf8BOM)
|
||||
envMap, err := godotenv.Parse(bytes.NewReader(raw))
|
||||
if err != nil {
|
||||
log.Printf("Warning: found .env at %s but failed to parse: %v", path, err)
|
||||
return false
|
||||
}
|
||||
applied := 0
|
||||
for k, v := range envMap {
|
||||
if os.Getenv(k) == "" {
|
||||
_ = os.Setenv(k, v)
|
||||
applied++
|
||||
}
|
||||
return
|
||||
}
|
||||
log.Printf("Loaded .env file from %s (%d keys applied)", path, applied)
|
||||
return true
|
||||
}
|
||||
|
||||
// collectEnvCandidates 收集候选 .env 路径:可执行文件旁、cwd、cwd/server、向上 3 级
|
||||
func collectEnvCandidates() []string {
|
||||
seen := map[string]struct{}{}
|
||||
var list []string
|
||||
add := func(p string) {
|
||||
abs, err := filepath.Abs(p)
|
||||
if err != nil {
|
||||
abs = p
|
||||
}
|
||||
if _, ok := seen[abs]; ok {
|
||||
return
|
||||
}
|
||||
seen[abs] = struct{}{}
|
||||
list = append(list, abs)
|
||||
}
|
||||
|
||||
// 获取可执行文件所在目录
|
||||
execDir := filepath.Dir(execPath)
|
||||
envPath := filepath.Join(execDir, ".env")
|
||||
|
||||
// 检查 .env 文件是否存在
|
||||
if _, err := os.Stat(envPath); os.IsNotExist(err) {
|
||||
log.Printf("Info: .env file not found at %s, using environment variables or defaults", envPath)
|
||||
return
|
||||
}
|
||||
|
||||
// 加载 .env 文件
|
||||
if err := godotenv.Load(envPath); err != nil {
|
||||
log.Printf("Warning: Failed to load .env file from %s: %v", envPath, err)
|
||||
if execPath, err := os.Executable(); err == nil {
|
||||
add(filepath.Join(filepath.Dir(execPath), ".env"))
|
||||
} else {
|
||||
log.Printf("Loaded .env file from %s", envPath)
|
||||
log.Printf("Warning: Failed to get executable path: %v", err)
|
||||
}
|
||||
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
log.Printf("Warning: Failed to get working directory: %v", err)
|
||||
cwd = "."
|
||||
}
|
||||
add(filepath.Join(cwd, ".env"))
|
||||
add(filepath.Join(cwd, "server", ".env")) // 从仓库根启动时
|
||||
|
||||
// 向上最多 3 级父目录查找 .env 与 server/.env
|
||||
dir := cwd
|
||||
for i := 0; i < 3; i++ {
|
||||
parent := filepath.Dir(dir)
|
||||
if parent == dir {
|
||||
break
|
||||
}
|
||||
dir = parent
|
||||
add(filepath.Join(dir, ".env"))
|
||||
add(filepath.Join(dir, "server", ".env"))
|
||||
}
|
||||
return list
|
||||
}
|
||||
|
||||
// LoadEnvFile 依次尝试多个路径加载 .env(后加载补缺,不覆盖已有 OS 环境变量)
|
||||
func LoadEnvFile() {
|
||||
loadedPaths := []string{}
|
||||
for _, p := range collectEnvCandidates() {
|
||||
if tryLoadEnv(p) {
|
||||
loadedPaths = append(loadedPaths, p)
|
||||
}
|
||||
}
|
||||
|
||||
if len(loadedPaths) == 0 {
|
||||
log.Println("Info: no .env loaded; relying on process environment variables")
|
||||
} else {
|
||||
log.Printf("Env load summary: %d file(s) loaded", len(loadedPaths))
|
||||
}
|
||||
|
||||
appID := os.Getenv("WECHAT_APP_ID")
|
||||
secret := os.Getenv("WECHAT_APP_SECRET")
|
||||
log.Printf(
|
||||
"WeChat env: APP_ID_set=%v SECRET_set=%v",
|
||||
appID != "",
|
||||
secret != "",
|
||||
)
|
||||
}
|
||||
|
||||
func InitDB() {
|
||||
|
||||
Reference in New Issue
Block a user