Initial commit

utils/check.go

@@ -0,0 +1,16 @@
+package utils
+
+import (
+	log "github.com/sirupsen/logrus"
+	"net/url"
+)
+
+func GetCode(rawUrl string) string {
+	u,err:=url.Parse(rawUrl)
+	if err!=nil {
+		log.Errorf("解析url出错:%s",err.Error())
+		return ""
+	}
+	code:=u.Query().Get("code")
+	return code
+}

utils/file.go

@@ -0,0 +1,41 @@
+package utils
+
+import (
+	log "github.com/sirupsen/logrus"
+	"gopkg.in/yaml.v2"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+)
+
+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)
+}
+
+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())
+	}
+}