go_process_manager/internal/app/logic/file.go

package logic

import (
	"fmt"
	"io"
	"os"

	"github.com/lzh-1625/go_process_manager/config"
	"github.com/lzh-1625/go_process_manager/internal/app/model"
	"github.com/lzh-1625/go_process_manager/log"
)

type fileLogic struct{}

var FileLogic = new(fileLogic)

func (f *fileLogic) ReadFileFromPath(path string) (result []byte, err error) {
	fi, err := os.Open(path)
	if err != nil {
		return
	}
	defer fi.Close()
	fileInfo, err := fi.Stat()
	if err != nil {
		return
	}
	if size := float64(fileInfo.Size()) / 1e6; size > config.CF.FileSizeLimit {
		err = fmt.Errorf("写入数据大小%vMB,超过%vMB限制", size, config.CF.FileSizeLimit)
		return
	}
	result, err = io.ReadAll(fi)
	if err != nil {
		return
	}
	log.Logger.Debugw("文件写入成功", "path", path)
	return
}

func (f *fileLogic) UpdateFileData(filePath string, file io.Reader, size int64) error {
	if size := float64(size) / 1e6; size > config.CF.FileSizeLimit {
		return fmt.Errorf("写入数据大小%vMB,超过%vMB限制", size, config.CF.FileSizeLimit)
	}
	fi, err := os.OpenFile(filePath, os.O_RDWR|os.O_TRUNC, 0777)
	if err != nil {
		return err
	}
	defer fi.Close()
	if _, err = io.Copy(fi, file); err != nil {
		return err
	}
	log.Logger.Debugw("文件写入成功", "path", filePath)
	return nil
}

func (f *fileLogic) GetFileAndDirByPath(srcPath string) []model.FileStruct {
	result := []model.FileStruct{}
	files, err := os.ReadDir(srcPath)
	if err != nil {
		return result
	}
	for _, file := range files {
		result = append(result, model.FileStruct{
			Name:  srcPath + file.Name(),
			IsDir: file.IsDir(),
		})
	}
	return result
}

func (f *fileLogic) CreateNewDir(path string, name string) error {
	_, err := os.Create(path + name)
	return err
}

func (f *fileLogic) CreateNewFile(path string, name string) error {
	return os.MkdirAll(path+name, os.ModeDir)
}