54 lines
1.8 KiB
Go
54 lines
1.8 KiB
Go
|
package pdf
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// GetResourcesPDFDir 获取resources/pdf目录路径
|
|||
|
|
// 支持开发和生产环境：
|
|||
|
|
// - 开发环境：从工作目录或可执行文件目录查找
|
|||
|
|
// - 生产环境：从可执行文件目录查找（Docker容器中工作目录为/app）
|
|||
|
|
func GetResourcesPDFDir() string {
|
|||
|
|
// 优先级1: 从工作目录查找（适用于开发环境和生产环境）
|
|||
|
|
if workDir, err := os.Getwd(); err == nil {
|
|||
|
|
// 检查当前工作目录下的resources/pdf
|
|||
|
|
resourcesPath := filepath.Join(workDir, "resources", "pdf")
|
|||
|
|
if _, err := os.Stat(resourcesPath); err == nil {
|
|||
|
|
return resourcesPath
|
|||
|
|
}
|
|||
|
|
// 检查tyapi-server-gin子目录（开发环境）
|
|||
|
|
resourcesPath = filepath.Join(workDir, "tyapi-server-gin", "resources", "pdf")
|
|||
|
|
if _, err := os.Stat(resourcesPath); err == nil {
|
|||
|
|
return resourcesPath
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 优先级2: 从可执行文件所在目录查找（适用于生产环境Docker容器）
|
|||
|
|
// 在生产环境中，可执行文件在/app/tyapi-server，资源在/app/resources
|
|||
|
|
if execPath, err := os.Executable(); err == nil {
|
|||
|
|
execDir := filepath.Dir(execPath)
|
|||
|
|
// 处理符号链接
|
|||
|
|
if realPath, err := filepath.EvalSymlinks(execPath); err == nil {
|
|||
|
|
execDir = filepath.Dir(realPath)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查可执行文件同目录下的resources/pdf
|
|||
|
|
resourcesPath := filepath.Join(execDir, "resources", "pdf")
|
|||
|
|
if _, err := os.Stat(resourcesPath); err == nil {
|
|||
|
|
return resourcesPath
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查可执行文件父目录下的resources/pdf
|
|||
|
|
// 适用于生产环境：/app/tyapi-server -> /app/resources
|
|||
|
|
resourcesPath = filepath.Join(filepath.Dir(execDir), "resources", "pdf")
|
|||
|
|
if _, err := os.Stat(resourcesPath); err == nil {
|
|||
|
|
return resourcesPath
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 优先级3: 返回相对路径作为后备（相对于当前工作目录）
|
|||
|
|
return filepath.Join("resources", "pdf")
|
|||
|
|
}
|
|||
|
|
|