80 lines
1.3 KiB
Go
80 lines
1.3 KiB
Go
|
package file
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"skapp/pkg/utils/json"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Support struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func New() *Support {
|
||
|
|
return &Support{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Support) IsFile(path string) bool {
|
||
|
|
info, err := os.Stat(path)
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
return !info.IsDir()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Support) IsDir(path string) bool {
|
||
|
|
info, err := os.Stat(path)
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
return info.IsDir()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Support) CopyN(w io.Writer, path string, n int64) (int64, error) {
|
||
|
|
if s.IsDir(path) {
|
||
|
|
return 0, errors.New(fmt.Sprintf("%s 为文件夹", path))
|
||
|
|
}
|
||
|
|
|
||
|
|
file, err := os.Open(path)
|
||
|
|
defer func(file *os.File) {
|
||
|
|
_ = file.Close()
|
||
|
|
}(file)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
|
||
|
|
info, _ := file.Stat()
|
||
|
|
if n <= 0 {
|
||
|
|
return io.Copy(w, file)
|
||
|
|
}
|
||
|
|
|
||
|
|
if info.Size() < n {
|
||
|
|
return io.Copy(w, file)
|
||
|
|
} else {
|
||
|
|
return io.CopyN(w, file, n)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Support) SaveJSONFile(filePath string, data interface{}) error {
|
||
|
|
if !filepath.IsAbs(filePath) {
|
||
|
|
root, _ := os.Getwd()
|
||
|
|
filePath = filepath.Join(root, filePath)
|
||
|
|
}
|
||
|
|
|
||
|
|
jsonFile, err := os.OpenFile(filePath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
||
|
|
defer func(scmF *os.File) {
|
||
|
|
_ = scmF.Close()
|
||
|
|
}(jsonFile)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = jsonFile.WriteString(json.Json(data))
|
||
|
|
return err
|
||
|
|
}
|