dailz/hpc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(store): add TaskStore CRUD and batch query methods for files and blobs

Browse Source
This commit is contained in:
dailz
2026-04-15 21:30:51 +08:00
parent d46a784efb
commit acf8c1d62b
7 changed files with 520 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/store/file_store.go
View File

@@ -86,6 +86,15 @@ func (s *FileStore) CountByBlobSHA256(ctx context.Context, blobSHA256 string) (i
return count, err
}
func (s *FileStore) GetByIDs(ctx context.Context, ids []int64) ([]model.File, error) {
var files []model.File
if len(ids) == 0 {
return files, nil
}
err := s.db.WithContext(ctx).Where("id IN ?", ids).Find(&files).Error
return files, err
}
func (s *FileStore) GetBlobSHA256ByID(ctx context.Context, id int64) (string, error) {
var file model.File
err := s.db.WithContext(ctx).Select("blob_sha256").First(&file, id).Error