This commit is contained in:
2025-12-04 13:42:32 +08:00
parent aaf17321ff
commit 81639a81e6
3 changed files with 126 additions and 127 deletions

View File

@@ -14,93 +14,78 @@ func SetGlobalLogger(logger *zap.Logger) {
globalLogger = logger
}
// GetResourcesPDFDir 获取resources/pdf目录路径
// GetResourcesPDFDir 获取resources/pdf目录路径(绝对路径)
// resources目录和可执行文件同级例如
// /app/tyapi-server (可执行文件)
// /app/resources/pdf (资源文件)
// 生产环境:/app/tyapi-server (可执行文件) 和 /app/resources/pdf (资源文件)
// 开发环境:工作目录下的 resources/pdf 或 tyapi-server-gin/resources/pdf
func GetResourcesPDFDir() string {
// 从可执行文件所在目录查找resources和可执行文件同级
// 候选路径列表(按优先级排序
var candidatePaths []string
// 优先级1: 从可执行文件所在目录查找(生产环境和开发环境都适用)
if execPath, err := os.Executable(); err == nil {
execDir := filepath.Dir(execPath)
// 处理符号链接
if realPath, err := filepath.EvalSymlinks(execPath); err == nil {
execDir = filepath.Dir(realPath)
}
// resources目录和可执行文件同级
resourcesPath := filepath.Join(execDir, "resources", "pdf")
absPath, err := filepath.Abs(resourcesPath)
if err == nil {
if globalLogger != nil {
globalLogger.Debug("尝试从可执行文件目录查找资源",
zap.String("exec_path", execPath),
zap.String("exec_dir", execDir),
zap.String("resources_path", absPath),
)
}
if _, err := os.Stat(absPath); err == nil {
if globalLogger != nil {
globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
}
return absPath
} else if globalLogger != nil {
globalLogger.Debug("资源目录不存在", zap.String("path", absPath), zap.Error(err))
}
}
candidatePaths = append(candidatePaths, filepath.Join(execDir, "resources", "pdf"))
}
// 后备方案:从工作目录查找(开发环境)
// 优先级2: 从工作目录查找(开发环境)
if workDir, err := os.Getwd(); err == nil {
resourcesPath := filepath.Join(workDir, "resources", "pdf")
absPath, err := filepath.Abs(resourcesPath)
if err == nil {
if globalLogger != nil {
globalLogger.Debug("尝试从工作目录查找资源",
zap.String("work_dir", workDir),
zap.String("resources_path", absPath),
)
}
if _, err := os.Stat(absPath); err == nil {
if globalLogger != nil {
globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
}
return absPath
}
}
// 开发环境可能在工作目录的子目录
resourcesPath = filepath.Join(workDir, "tyapi-server-gin", "resources", "pdf")
absPath, err = filepath.Abs(resourcesPath)
if err == nil {
if globalLogger != nil {
globalLogger.Debug("尝试从工作目录子目录查找资源",
zap.String("resources_path", absPath),
)
}
if _, err := os.Stat(absPath); err == nil {
if globalLogger != nil {
globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
}
return absPath
}
}
candidatePaths = append(candidatePaths,
filepath.Join(workDir, "resources", "pdf"),
filepath.Join(workDir, "tyapi-server-gin", "resources", "pdf"),
)
}
// 最后的后备:返回绝对路径(即使目录不存在)
if workDir, err := os.Getwd(); err == nil {
resourcesPath := filepath.Join(workDir, "resources", "pdf")
if absPath, err := filepath.Abs(resourcesPath); err == nil {
// 尝试每个候选路径
for _, candidatePath := range candidatePaths {
absPath, err := filepath.Abs(candidatePath)
if err != nil {
continue
}
if globalLogger != nil {
globalLogger.Debug("尝试查找resources/pdf目录", zap.String("path", absPath))
}
// 检查目录是否存在
if info, err := os.Stat(absPath); err == nil && info.IsDir() {
if globalLogger != nil {
globalLogger.Warn("找到resources/pdf目录,返回后备路径", zap.String("path", absPath))
globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
}
return absPath
}
}
// 最后的最后:返回相对路径(不推荐
if globalLogger != nil {
globalLogger.Error("无法确定resources/pdf目录路径")
// 所有候选路径都不存在,返回第一个候选路径的绝对路径(作为后备
// 这样至少保证返回的是绝对路径,即使目录不存在
if len(candidatePaths) > 0 {
if absPath, err := filepath.Abs(candidatePaths[0]); err == nil {
if globalLogger != nil {
globalLogger.Warn("未找到resources/pdf目录返回后备绝对路径", zap.String("path", absPath))
}
return absPath
}
}
return filepath.Join("resources", "pdf")
// 最后的最后:从工作目录构建绝对路径
if workDir, err := os.Getwd(); err == nil {
absPath, err := filepath.Abs(filepath.Join(workDir, "resources", "pdf"))
if err == nil {
if globalLogger != nil {
globalLogger.Error("无法确定resources/pdf目录返回工作目录下的绝对路径", zap.String("path", absPath))
}
return absPath
}
}
// 完全无法确定路径
if globalLogger != nil {
globalLogger.Error("完全无法确定resources/pdf目录路径")
}
return ""
}