mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-07 01:32:56 +08:00
54 lines
1007 B
Go
54 lines
1007 B
Go
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
func checkNotExist(src string) bool {
|
|
_, err := os.Stat(src)
|
|
return os.IsNotExist(err)
|
|
}
|
|
|
|
func checkPermission(src string) bool {
|
|
_, err := os.Stat(src)
|
|
return os.IsPermission(err)
|
|
}
|
|
|
|
func isNotExistMkDir(src string) error {
|
|
if notExist := checkNotExist(src); notExist == true {
|
|
if err := mkDir(src); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func mkDir(src string) error {
|
|
err := os.MkdirAll(src, os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func mustOpen(fileName, dir string) (*os.File, error) {
|
|
perm := checkPermission(dir)
|
|
if perm == true {
|
|
return nil, fmt.Errorf("permission denied dir: %s", dir)
|
|
}
|
|
|
|
err := isNotExistMkDir(dir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error during make dir %s, err: %s", dir, err)
|
|
}
|
|
|
|
f, err := os.OpenFile(dir+string(os.PathSeparator)+fileName, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0644)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("fail to open file, err: %s", err)
|
|
}
|
|
|
|
return f, nil
|
|
}
|