45 lines
872 B
Go
45 lines
872 B
Go
package utils
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"gopkg.in/yaml.v2"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// determine whether the file exists
|
|
func Exists(name string) bool {
|
|
if _, err := os.Stat(name); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// 嵌套创建文件
|
|
func CreatNestedFile(path string) (*os.File, error) {
|
|
basePath := filepath.Dir(path)
|
|
if !Exists(basePath) {
|
|
err := os.MkdirAll(basePath, 0700)
|
|
if err != nil {
|
|
log.Errorf("无法创建目录，%s", err)
|
|
return nil, err
|
|
}
|
|
}
|
|
return os.Create(path)
|
|
}
|
|
|
|
// write struct to yaml file
|
|
func WriteToYml(src string, conf interface{}) {
|
|
data, err := yaml.Marshal(conf)
|
|
if err != nil {
|
|
log.Errorf("Conf转[]byte失败:%s", err.Error())
|
|
}
|
|
err = ioutil.WriteFile(src, data, 0777)
|
|
if err != nil {
|
|
log.Errorf("写yml文件失败", err.Error())
|
|
}
|
|
}
|