package repo

import (
	"context"

	"github.com/tidbcloud/dataflow-service/app/repo/entity"
	"github.com/tidbcloud/pkgv2/bizerr"
)

type (
	BillingRecordFilterOp func(qs entity.BillingRecordQuerySet) entity.BillingRecordQuerySet
)

type BillingRecordRepo interface {
	Load(ctx context.Context, id uint64) (*entity.BillingRecord, bizerr.BizError)
	Save(ctx context.Context, record *entity.BillingRecord) bizerr.BizError
	List(ctx context.Context, ops ...BillingRecordFilterOp) ([]entity.BillingRecord, bizerr.BizError)
	Get(ctx context.Context, ops ...BillingRecordFilterOp) (*entity.BillingRecord, bizerr.BizError)
	Count(ctx context.Context, ops ...BillingRecordFilterOp) (int64, bizerr.BizError)
}