add utils

yifanshaoye · Jun 21, 2022 · cf37b5e · cf37b5e
1 parent 72f5709
commit cf37b5e
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/utils/common.go b/utils/common.go
@@ -0,0 +1,33 @@
+package utils
+
+import (
+	"os"
+	"os/user"
+	"path/filepath"
+)
+
+var dirPath string
+
+func getHomeDir() string {
+	path, err := os.UserHomeDir()
+	if err == nil {
+		return path
+	}
+	user, err := user.Current()
+	if err == nil {
+		return user.HomeDir
+	}
+	path, _ = filepath.Abs("~")
+	return path
+}
+
+func makeLogDir()  {
+	home := getHomeDir()
+	dirPath = home + "/collctionlogs"
+	os.MkdirAll(dirPath, 0766)
+}
+
+func GetLogDir() string {
+	makeLogDir()
+	return dirPath
+}