1. 小程序端

2. 视频优化
This commit is contained in:
李琦
2026-07-30 14:11:26 +08:00
parent 66be7a3ba0
commit 4f4153ee0e
32 changed files with 638 additions and 7720 deletions

View File

@@ -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() {