Add some util functions.

This commit is contained in:
Yang Luo 2020-07-24 11:45:42 +08:00
parent d6fb56ec6d
commit 32f11f335d
2 changed files with 37 additions and 1 deletions

View File

@ -14,7 +14,10 @@
package util
import "os"
import (
"io/ioutil"
"os"
)
func FileExist(path string) bool {
if _, err := os.Stat(path); os.IsNotExist(err) {
@ -23,3 +26,19 @@ func FileExist(path string) bool {
return true
}
func ListFiles(path string) []string {
res := []string{}
files, err := ioutil.ReadDir(path)
if err != nil {
panic(err)
}
for _, f := range files {
if !f.IsDir() {
res = append(res, f.Name())
}
}
return res
}

View File

@ -15,6 +15,7 @@
package util
import (
"io/ioutil"
"regexp"
"strconv"
)
@ -38,3 +39,19 @@ func IsValidUsername(username string) bool {
}
return true
}
func ReadStringFromPath(path string) string {
data, err := ioutil.ReadFile(path)
if err != nil {
panic(err)
}
return string(data)
}
func WriteStringToPath(s string, path string) {
err := ioutil.WriteFile(path, []byte(s), 0644)
if err != nil {
panic(err)
}
}