JCS-pub/common/pkgs/db2/storage.go

package db2

import (
	"fmt"

	cdssdk "gitlink.org.cn/cloudream/common/sdks/storage"
	"gitlink.org.cn/cloudream/storage/common/pkgs/db/model"
)

type StorageDB struct {
	*DB
}

func (db *DB) Storage() *StorageDB {
	return &StorageDB{DB: db}
}

func (db *StorageDB) GetByID(ctx SQLContext, stgID cdssdk.StorageID) (model.Storage, error) {
	var stg model.Storage
	err := ctx.Table("Storage").First(&stg, stgID).Error
	return stg, err
}

func (db *StorageDB) BatchGetAllStorageIDs(ctx SQLContext, start int, count int) ([]cdssdk.StorageID, error) {
	var ret []cdssdk.StorageID
	err := ctx.Table("Storage").Select("StorageID").Find(ret).Limit(count).Offset(start).Error
	return ret, err
}

func (db *StorageDB) IsAvailable(ctx SQLContext, userID cdssdk.UserID, storageID cdssdk.StorageID) (bool, error) {
	rows, err := ctx.Table("Storage").Select("Storage.StorageID").
		Joins("inner join UserStorage on Storage.StorageID = UserStorage.StorageID").
		Where("UserID = ? and StorageID = ?", userID, storageID).Rows()
	if err != nil {
		return false, fmt.Errorf("execute sql: %w", err)
	}
	defer rows.Close()

	return rows.Next(), nil
}

func (db *StorageDB) GetUserStorage(ctx SQLContext, userID cdssdk.UserID, storageID cdssdk.StorageID) (model.Storage, error) {
	var stg model.Storage
	err := ctx.Table("Storage").Select("Storage.*").
		Joins("inner join UserStorage on Storage.StorageID = UserStorage.StorageID").
		Where("UserID = ? and StorageID = ?", userID, storageID).First(&stg).Error

	return stg, err
}

func (db *StorageDB) GetUserStorageByName(ctx SQLContext, userID cdssdk.UserID, name string) (model.Storage, error) {
	var stg model.Storage
	err := ctx.Table("Storage").Select("Storage.*").
		Joins("inner join UserStorage on Storage.StorageID = UserStorage.StorageID").
		Where("UserID = ? and Name = ?", userID, name).First(&stg).Error

	return stg, err
}

// func (db *StorageDB) ChangeState(ctx SQLContext, storageID cdssdk.StorageID, state string) error {
// 	return ctx.Table("Storage").Where("StorageID = ?", storageID).Update("State", state).Error
// }