1.4.5.2

EdgeAPI/internal/db/models/file_chunk_dao.go

@@ -0,0 +1,94 @@
+package models
+
+import (
+	"errors"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
+)
+
+type FileChunkDAO dbs.DAO
+
+func NewFileChunkDAO() *FileChunkDAO {
+	return dbs.NewDAO(&FileChunkDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeFileChunks",
+			Model:  new(FileChunk),
+			PkName: "id",
+		},
+	}).(*FileChunkDAO)
+}
+
+var SharedFileChunkDAO *FileChunkDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedFileChunkDAO = NewFileChunkDAO()
+	})
+}
+
+// CreateFileChunk 创建文件Chunk
+func (this *FileChunkDAO) CreateFileChunk(tx *dbs.Tx, fileId int64, data []byte) (int64, error) {
+	var op = NewFileChunkOperator()
+	op.FileId = fileId
+	op.Data = data
+	err := this.Save(tx, op)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(op.Id), nil
+}
+
+// FindAllFileChunks 列出所有的文件Chunk
+func (this *FileChunkDAO) FindAllFileChunks(tx *dbs.Tx, fileId int64) (result []*FileChunk, err error) {
+	_, err = this.Query(tx).
+		Attr("fileId", fileId).
+		AscPk().
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// FindAllFileChunkIds 读取文件的所有片段ID
+func (this *FileChunkDAO) FindAllFileChunkIds(tx *dbs.Tx, fileId int64) ([]int64, error) {
+	ones, err := this.Query(tx).
+		Attr("fileId", fileId).
+		AscPk().
+		ResultPk().
+		FindAll()
+	if err != nil {
+		return nil, err
+	}
+	result := []int64{}
+	for _, one := range ones {
+		result = append(result, int64(one.(*FileChunk).Id))
+	}
+	return result, nil
+}
+
+// DeleteFileChunks 删除以前的文件
+func (this *FileChunkDAO) DeleteFileChunks(tx *dbs.Tx, fileId int64) error {
+	if fileId <= 0 {
+		return errors.New("invalid fileId")
+	}
+	_, err := this.Query(tx).
+		Attr("fileId", fileId).
+		Delete()
+	return err
+}
+
+// FindFileChunk 根据ID查找片段
+func (this *FileChunkDAO) FindFileChunk(tx *dbs.Tx, chunkId int64) (*FileChunk, error) {
+	one, err := this.Query(tx).
+		Pk(chunkId).
+		Find()
+	if err != nil {
+		return nil, err
+	}
+	if one == nil {
+		return nil, nil
+	}
+	return one.(*FileChunk), nil
+}