JCS-pub/common/pkgs/db/model/model.go

package model

import (
	"fmt"
	"reflect"
	"time"

	cdssdk "gitlink.org.cn/cloudream/common/sdks/storage"
	"gitlink.org.cn/cloudream/common/utils/serder"
	stgmod "gitlink.org.cn/cloudream/storage/common/models"
)

// TODO 可以考虑逐步迁移到cdssdk中。迁移思路：数据对象应该包含的字段都迁移到cdssdk中，内部使用的一些特殊字段则留在这里

type Storage struct {
	StorageID cdssdk.StorageID `db:"StorageID" json:"storageID"`
	Name      string           `db:"Name" json:"name"`
	NodeID    cdssdk.NodeID    `db:"NodeID" json:"nodeID"`
	Directory string           `db:"Directory" json:"directory"`
	State     string           `db:"State" json:"state"`
}

type NodeDelay struct {
	SourceNodeID      int64 `db:"SourceNodeID"`
	DestinationNodeID int64 `db:"DestinationNodeID"`
	DelayInMs         int   `db:"DelayInMs"`
}

type User struct {
	UserID   cdssdk.UserID `db:"UserID" json:"userID"`
	Password string        `db:"PassWord" json:"password"`
}

type UserBucket struct {
	UserID   cdssdk.UserID   `db:"UserID" json:"userID"`
	BucketID cdssdk.BucketID `db:"BucketID" json:"bucketID"`
}

type UserNode struct {
	UserID cdssdk.UserID `db:"UserID" json:"userID"`
	NodeID cdssdk.NodeID `db:"NodeID" json:"nodeID"`
}

type UserStorage struct {
	UserID    cdssdk.UserID    `db:"UserID" json:"userID"`
	StorageID cdssdk.StorageID `db:"StorageID" json:"storageID"`
}

type Bucket = cdssdk.Bucket

type Package = cdssdk.Package

type Object = cdssdk.Object

// 由于Object的Redundancy字段是interface，所以不能直接将查询结果scan成Object，必须先scan成TempObject，
// 再.ToObject()转成Object
type TempObject struct {
	cdssdk.Object
	Redundancy RedundancyWarpper `db:"Redundancy"`
}

func (o *TempObject) ToObject() cdssdk.Object {
	obj := o.Object
	obj.Redundancy = o.Redundancy.Value
	return obj
}

type RedundancyWarpper struct {
	Value cdssdk.Redundancy
}

func (o *RedundancyWarpper) Scan(src interface{}) error {
	data, ok := src.([]uint8)
	if !ok {
		return fmt.Errorf("unknow src type: %v", reflect.TypeOf(data))
	}

	red, err := serder.JSONToObjectEx[cdssdk.Redundancy](data)
	if err != nil {
		return err
	}

	o.Value = red
	return nil
}

type ObjectBlock = stgmod.ObjectBlock

type Cache struct {
	FileHash   string        `db:"FileHash" json:"fileHash"`
	NodeID     cdssdk.NodeID `db:"NodeID" json:"nodeID"`
	CreateTime time.Time     `db:"CreateTime" json:"createTime"`
	Priority   int           `db:"Priority" json:"priority"`
}

const (
	StoragePackageStateNormal   = "Normal"
	StoragePackageStateDeleted  = "Deleted"
	StoragePackageStateOutdated = "Outdated"
)

// Storage当前加载的Package
type StoragePackage struct {
	StorageID cdssdk.StorageID `db:"StorageID" json:"storageID"`
	PackageID cdssdk.PackageID `db:"PackageID" json:"packageID"`
	UserID    cdssdk.UserID    `db:"UserID" json:"userID"`
	State     string           `db:"State" json:"state"`
}

type StoragePackageLog struct {
	StorageID  cdssdk.StorageID `db:"StorageID" json:"storageID"`
	PackageID  cdssdk.PackageID `db:"PackageID" json:"packageID"`
	UserID     cdssdk.UserID    `db:"UserID" json:"userID"`
	CreateTime time.Time        `db:"CreateTime" json:"createTime"`
}

type Location struct {
	LocationID cdssdk.LocationID `db:"LocationID" json:"locationID"`
	Name       string            `db:"Name" json:"name"`
}