bucket

type Bucket struct {
	Id               int       `json:"id"`
	PartitionsEnding time.Time `json:"partitionsEnding" gorm:"type:timestamptz"`
	Created          time.Time `json:"created" gorm:"type:timestamptz;default:now()"`
	Orgs             []*Org    `json:"orgs,omitempty"`
}

type BucketIndexSelector func(org *Org, bucketCount int) int

type BucketMaintenanceOptions struct {
	Repository                    PartitionedRepository
	LeaderElectorClass            string
	MaintenanceFrequency          time.Duration
	PartitionPreparationThreshold time.Duration
}

type CachingOption func(*CachingRepository)

type CachingRepository struct {
	PartitionedRepository
	// contains filtered or unexported fields
}

type Config struct {
	OrgBucketCount                int
	PartitionPreparationThreshold time.Duration
}

type CopyOrgRequest struct {
	SourceBucket      int       `json:"sourceBucket"`
	DestinationBucket int       `json:"destinationBucket"`
	StartTime         time.Time `json:"startTime"`
	EndTime           time.Time `json:"endTime"`
}

type MoveOrgRequest struct {
	SourceBucket      int       `json:"sourceBucket"`
	DestinationBucket int       `json:"destinationBucket"`
	StartTime         time.Time `json:"startTime"`
	EndTime           time.Time `json:"endTime"`
}

type Org struct {
	Name     string    `json:"name" gorm:"primaryKey"`
	Created  time.Time `json:"created" gorm:"type:timestamptz;default:now()"`
	BucketId int       `json:"bucketId"`
	Bucket   *Bucket   `json:"bucket,omitempty"`
}

type OrgResolver interface {
	// ResolveOrg returns (false, nil) only when the org definitively does not exist.
	// Any other failure must be reported as an error so the caller can retry later.
	ResolveOrg(ctx context.Context, name string) (bool, error)
}

type PartitionedRepository interface {
	checkpoints.CheckpointRepository
	database.Repository

	// Org Management
	CreateOrg(org string) error
	GetOrg(name string) (*Org, error)
	ListOrgs(filter string) ([]*Org, error)
	FilterOrgs(orgs []*Org, filter string) ([]*Org, error)
	ListBucketOrgs(bucket *Bucket) ([]*Org, error)
	ListOrgsByName(names []string) ([]*Org, error)

	// Bucket Management
	GetBucket(id int) (*Bucket, error)
	ListBuckets() ([]*Bucket, error)
	EnsureBucketPartitions(b *Bucket, startTime time.Time, years int) error
	MoveOrgToBucket(org *Org, bucket *Bucket) error

	// Data Operations
	CopyOrgData(org *Org, startTime time.Time, endTime time.Time, oldBucket *Bucket, newBucket *Bucket) error
	ScrubOrgData(org *Org, startTime time.Time, endTime time.Time, oldBucket *Bucket) error
	AuditBucket(startTime time.Time, endTime time.Time, oldBucket *Bucket, newBucket *Bucket, orgFilter string) ([]any, error)
}

type PostgresqlPartitionedRepository struct {
	checkpoints.CheckpointRepository
	// contains filtered or unexported fields
}

type SchemaHandler interface {
	// EnsureSchema creates all necessary tables, indexes, and schema structures for a bucket
	EnsureSchema(db *gorm.DB, bucket *Bucket) error

	// EnsurePartitions creates time-based partitions for a bucket's tables
	EnsurePartitions(db *gorm.DB, bucket *Bucket, startTime time.Time, years int) (nextTime time.Time, err error)

	// CopyOrgData copies all data for an organization from one bucket to another within a time range
	CopyOrgData(db *gorm.DB, org *Org, startTime time.Time, endTime time.Time, oldBucket *Bucket, newBucket *Bucket) error

	// ScrubOrgData deletes all data for an organization from a bucket within a time range
	ScrubOrgData(db *gorm.DB, org *Org, startTime time.Time, endTime time.Time, bucket *Bucket) error

	// AuditBuckets executes a query to audit differences between two buckets and returns data-model-specific results
	AuditBuckets(db *gorm.DB, oldBucket *Bucket, newBucket *Bucket, startTime time.Time, endTime time.Time, orgNames []string) ([]any, error)
}

type ScrubOrgRequest struct {
	DestinationBucket int       `json:"destinationBucket"`
	StartTime         time.Time `json:"startTime"`
	EndTime           time.Time `json:"endTime"`
}