package relation

import (
	"context"
	"time"
)

const (
	ObjectPutModelTableName = "object_put"
)

type ObjectPutModel struct {
	PutID         string     `gorm:"column:put_id;primary_key"`
	Hash          string     `gorm:"column:hash"`
	Path          string     `gorm:"column:path"`
	Name          string     `gorm:"column:name"`
	ContentType   string     `gorm:"column:content_type"`
	ObjectSize    int64      `gorm:"column:object_size"`
	FragmentSize  int64      `gorm:"column:fragment_size"`
	Complete      bool       `gorm:"column:complete"`
	ValidTime     *time.Time `gorm:"column:valid_time"`
	EffectiveTime time.Time  `gorm:"column:effective_time"`
	CreateTime    time.Time  `gorm:"column:create_time"`
}

func (ObjectPutModel) TableName() string {
	return ObjectPutModelTableName
}

type ObjectPutModelInterface interface {
	NewTx(tx any) ObjectPutModelInterface
	Create(ctx context.Context, m []*ObjectPutModel) error
	Take(ctx context.Context, putID string) (*ObjectPutModel, error)
	SetCompleted(ctx context.Context, putID string) error
	FindExpirationPut(ctx context.Context, expirationTime time.Time, num int) ([]*ObjectPutModel, error)
	DelPut(ctx context.Context, ids []string) error
}