187 lines
4.0 KiB
Go
187 lines
4.0 KiB
Go
package rest
|
||
|
||
import (
|
||
"fmt"
|
||
"go/build"
|
||
"io/ioutil"
|
||
"os"
|
||
"os/user"
|
||
"path/filepath"
|
||
"strings"
|
||
)
|
||
|
||
//判断文件或文件夹是否已经存在
|
||
func PathExists(path string) (bool, error) {
|
||
_, err := os.Stat(path)
|
||
if err == nil {
|
||
return true, nil
|
||
}
|
||
if os.IsNotExist(err) {
|
||
return false, nil
|
||
}
|
||
return false, err
|
||
}
|
||
|
||
//获取GOPATH路径
|
||
func GetGoPath() string {
|
||
|
||
return build.Default.GOPATH
|
||
|
||
}
|
||
|
||
//获取该应用可执行文件的位置。
|
||
//例如：C:\Users\lishuang\AppData\Local\Temp
|
||
func GetHomePath() string {
|
||
ex, err := os.Executable()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
exPath := filepath.Dir(ex)
|
||
|
||
//如果exPath中包含了 /private/var/folders 我们认为是在Mac的开发环境中
|
||
macDev := strings.HasPrefix(exPath, "/private/var/folders")
|
||
if macDev {
|
||
exPath = GetGoPath() + "/src/tank/tmp"
|
||
}
|
||
|
||
//如果exPath中包含了 \\AppData\\Local\\Temp 我们认为是在Win的开发环境中
|
||
systemUser, err := user.Current()
|
||
winDev := strings.HasPrefix(exPath, systemUser.HomeDir+"\\AppData\\Local\\Temp")
|
||
if winDev {
|
||
exPath = GetGoPath() + "/src/tank/tmp"
|
||
}
|
||
|
||
return exPath
|
||
}
|
||
|
||
//获取前端静态资源的位置。如果你在开发模式下，可以将这里直接返回tank/build下面的html路径。
|
||
//例如：C:/Users/lishuang/AppData/Local/Temp/html
|
||
func GetHtmlPath() string {
|
||
|
||
homePath := GetHomePath()
|
||
filePath := homePath + "/html"
|
||
exists, err := PathExists(filePath)
|
||
if err != nil {
|
||
panic("判断上传文件是否存在时出错！")
|
||
}
|
||
if !exists {
|
||
err = os.MkdirAll(filePath, 0777)
|
||
if err != nil {
|
||
panic("创建上传文件夹时出错！")
|
||
}
|
||
}
|
||
|
||
return filePath
|
||
}
|
||
|
||
//如果文件夹存在就不管，不存在就创建。 例如：/var/www/matter
|
||
func MakeDirAll(dirPath string) string {
|
||
|
||
exists, err := PathExists(dirPath)
|
||
if err != nil {
|
||
panic("判断文件是否存在时出错！")
|
||
}
|
||
if !exists {
|
||
//TODO:文件权限需要进一步考虑
|
||
err = os.MkdirAll(dirPath, 0777)
|
||
if err != nil {
|
||
panic("创建文件夹时出错！")
|
||
}
|
||
}
|
||
|
||
return dirPath
|
||
}
|
||
|
||
//尝试删除空文件夹
|
||
func DeleteEmptyDir(dirPath string) {
|
||
dir, err := ioutil.ReadDir(dirPath)
|
||
if err != nil {
|
||
LOGGER.Error("尝试读取目录%s时出错 %s", dirPath, err.Error())
|
||
panic("尝试读取目录时出错 " + err.Error())
|
||
}
|
||
|
||
if len(dir) == 0 {
|
||
//空文件夹
|
||
err = os.Remove(dirPath)
|
||
if err != nil {
|
||
LOGGER.Error("删除磁盘上的文件夹%s出错 %s", dirPath, err.Error())
|
||
}
|
||
} else {
|
||
LOGGER.Info("文件夹不为空，%v", len(dir))
|
||
}
|
||
}
|
||
|
||
//移除某个文件夹。 例如：/var/www/matter => /var/www
|
||
func RemoveDirectory(dirPath string) string {
|
||
|
||
exists, err := PathExists(dirPath)
|
||
if err != nil {
|
||
panic("判断文件是否存在时出错！")
|
||
}
|
||
if exists {
|
||
|
||
err = os.Remove(dirPath)
|
||
if err != nil {
|
||
panic("删除文件夹时出错！")
|
||
}
|
||
}
|
||
|
||
return dirPath
|
||
}
|
||
|
||
//获取配置文件存放的位置
|
||
//例如：C:\Users\lishuang\AppData\Local\Temp/conf
|
||
func GetConfPath() string {
|
||
|
||
homePath := GetHomePath()
|
||
filePath := homePath + "/conf"
|
||
exists, err := PathExists(filePath)
|
||
if err != nil {
|
||
panic("判断日志文件夹是否存在时出错！")
|
||
}
|
||
if !exists {
|
||
err = os.MkdirAll(filePath, 0777)
|
||
if err != nil {
|
||
panic("创建日志文件夹时出错！")
|
||
}
|
||
}
|
||
|
||
return filePath
|
||
}
|
||
|
||
//获取日志的路径
|
||
//例如：默认存放于 home/log
|
||
func GetLogPath() string {
|
||
|
||
homePath := GetHomePath()
|
||
filePath := homePath + "/log"
|
||
exists, err := PathExists(filePath)
|
||
if err != nil {
|
||
panic("判断日志文件夹是否存在时出错！")
|
||
}
|
||
if !exists {
|
||
err = os.MkdirAll(filePath, 0777)
|
||
if err != nil {
|
||
panic("创建日志文件夹时出错！")
|
||
}
|
||
}
|
||
|
||
return filePath
|
||
}
|
||
|
||
//获取到用户文件的根目录。
|
||
func GetUserFileRootDir(username string) (rootDirPath string) {
|
||
|
||
rootDirPath = fmt.Sprintf("%s/%s/%s", CONFIG.MatterPath, username, MATTER_ROOT)
|
||
|
||
return rootDirPath
|
||
}
|
||
|
||
//获取到用户缓存的根目录。
|
||
func GetUserCacheRootDir(username string) (rootDirPath string) {
|
||
|
||
rootDirPath = fmt.Sprintf("%s/%s/%s", CONFIG.MatterPath, username, MATTER_CACHE)
|
||
|
||
return rootDirPath
|
||
}
|