cutego/pkg/config/index.go

package config

import (
	config "cutego/pkg/config/models"
	"cutego/pkg/logging"
	"encoding/json"
	"fmt"
	"gopkg.in/yaml.v2"
	"io/ioutil"
	"os"
	"runtime"
	"strconv"
)

var (
	// AppCoreConfig 核心配置
	AppCoreConfig *config.ApplicationCoreStruct
	// AppEnvConfig 环境配置
	AppEnvConfig *config.ApplicationEnvStruct
)

// GetRootPath 获取项目根路径
func GetRootPath() string {
	rootPath, _ := os.Getwd()
	return rootPath
}

// GetPathSeparator 获取路径分隔符
func GetPathSeparator() string {
	return string(os.PathSeparator)
}

// PathExists 判断文件或文件夹是否存在
// 如果返回的错误为nil,说明文件或文件夹存在
// 如果返回的错误类型使用os.IsNotExist()判断为true,说明文件或文件夹不存在
// 如果返回的错误为其它类型,则不确定是否在存在
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
}

// LoadYamlFile 加载yaml文件
func LoadYamlFile(filename string, v interface{}) {
	data, err := ioutil.ReadFile(filename)
	if err != nil {
		panic(err.Error())
	}
	err = yaml.Unmarshal(data, v)
	if err != nil {
		panic(err.Error())
	}
}

// BaseConfigDirPath 配置文件所在路径
const BaseConfigDirPath = "resources"

func readAppYmlFile(resourcePath string) {
	// 读取主配置文件
	applicationCoreFileName := BaseConfigDirPath + "/application.yml"
	applicationCoreFilePath := resourcePath + GetPathSeparator() + applicationCoreFileName
	exists, _ := PathExists(applicationCoreFilePath)
	if !exists {
		panic(applicationCoreFileName + "配置文件不存在!")
	}
	AppCoreConfig = &config.ApplicationCoreStruct{}
	// 由于要改变appConfig内部的值, 所以这里要取址
	LoadYamlFile(applicationCoreFilePath, AppCoreConfig)

	// 读取环境文件
	applicationEnvFileName := fmt.Sprintf(BaseConfigDirPath+"/application-%s.yml", AppCoreConfig.CuteGoConfig.Active)
	applicationEnvFilePath := resourcePath + GetPathSeparator() + applicationEnvFileName
	exists, _ = PathExists(applicationEnvFilePath)
	if !exists {
		panic(applicationEnvFileName + "配置文件不存在!")
	}
	AppEnvConfig = &config.ApplicationEnvStruct{}
	// 由于要改变appConfig内部的值, 所以这里要取址
	LoadYamlFile(applicationEnvFilePath, AppEnvConfig)
}

// IntToString int转string
func IntToString(n int) string {
	return strconv.Itoa(n)
}

// StringToInt string转int
func StringToInt(s string) int {
	i, err := strconv.Atoi(s)
	if err != nil {
		panic(err)
	}
	return i
}

// StringToInt64 string转int64
func StringToInt64(s string) int64 {
	i, err := strconv.ParseInt(s, 10, 64)
	if err != nil {
		panic(err)
	}
	return i
}

// Int64ToString int64转string
func Int64ToString(n int64) string {
	return strconv.FormatInt(n, 10)
}

func mapToBytes(data map[string]interface{}) []byte {
	bytes, _ := json.Marshal(data)
	return bytes
}

// MapToStruct map转struct
func MapToStruct(data map[string]interface{}, v interface{}) {
	_ = json.Unmarshal(mapToBytes(data), v)
}

// GetDirPath 获取目录路径
func GetDirPath(resType string) string {
	sysType := runtime.GOOS
	switch sysType {
	case "linux":
		if resType == "logging" {
			return AppCoreConfig.CuteGoConfig.File.Linux.Logs
		} else if resType == "avatar" {
			return AppCoreConfig.CuteGoConfig.File.Linux.Avatar
		} else if resType == "file" {
			return AppCoreConfig.CuteGoConfig.File.Linux.Path
		}
		break
	case "windows":
		if resType == "logging" {
			return AppCoreConfig.CuteGoConfig.File.Windows.Logs
		} else if resType == "avatar" {
			return AppCoreConfig.CuteGoConfig.File.Windows.Avatar
		} else if resType == "file" {
			return AppCoreConfig.CuteGoConfig.File.Windows.Path
		}
		break
	case "mac":
		if resType == "logging" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Logs
		} else if resType == "avatar" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Avatar
		} else if resType == "file" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Path
		}
		break
	case "darwin":
		if resType == "logging" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Logs
		} else if resType == "avatar" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Avatar
		} else if resType == "file" {
			return AppCoreConfig.CuteGoConfig.File.Mac.Path
		}
	}
	return AppCoreConfig.CuteGoConfig.File.Linux.Logs
}

// CreateAllDir 递归创建文件夹
func CreateAllDir(filePath string) error {
	if !IsFileOrDirExist(filePath) {
		err := os.MkdirAll(filePath, os.ModePerm)
		if err != nil {
			fmt.Println("创建文件夹失败, error info:", err)
			return err
		}
		return err
	}
	return nil
}

// IsFileOrDirExist 判断所给路径文件/文件夹是否存在(返回true是存在)
func IsFileOrDirExist(path string) bool {
	// os.Stat获取文件信息
	_, err := os.Stat(path)
	if err != nil {
		if os.IsExist(err) {
			return true
		}
		return false
	}
	return true
}

func init() {
	// 资源文件所在的路径
	resourcePath := GetRootPath()
	logging.InfoLog("application init start...")
	readAppYmlFile(resourcePath)
	logging.InfoLog("application init start...")
}