mongodb

type Config struct {
	URL      string
	Database string
}

type DBError struct {
	// contains filtered or unexported fields
}

type Entity interface {
	SetCreateTime(time time.Time)
	SetUpdateTime(time time.Time)
	PrimaryKey() string
	TableName() string
	ShortTableName() string
	Index() map[string]interface{}
}

type FilterOptions struct {
	Queries    []FuzzyQueryOption
	In         []InQueryOption
	IsNotExist []IsNotExistQueryOption
	WithIn     []WithInTimeQueryOption
	Gte        []Gte
	Lte        []Lte
}

type FuzzyQueryOption struct {
	Key   string
	Query string
}

type Gte struct {
	Key  string
	Time time.Time
}

type InQueryOption struct {
	Key    string
	Values []string
}

type IsNotExistQueryOption struct {
	Key string
}

type Iterator struct {
	// contains filtered or unexported fields
}

type ListOptions struct {
	FilterOptions
	Page     int
	PageSize int
	SortBy   []SortOption
}

type Lte struct {
	Key  string
	Time time.Time
}

type MongoDB interface {
	// Add adds entity to database, Name() and TableName() can't return zero value.
	Add(ctx context.Context, entity Entity) error

	// BatchAdd will adds batched entities to database, Name() and TableName() can't return zero value.
	BatchAdd(ctx context.Context, entities []Entity) error

	// Put will update entity to database, Name() and TableName() can't return zero value.
	Put(ctx context.Context, entity Entity) error

	// Delete entity from database, Name() and TableName() can't return zero value.
	Delete(ctx context.Context, entity Entity) error

	// Get entity from database, Name() and TableName() can't return zero value.
	Get(ctx context.Context, entity Entity) error

	// List entities from database, TableName() can't return zero value, if no matches, it will return a zero list without error.
	List(ctx context.Context, query Entity, options *ListOptions) ([]Entity, error)

	// Count entities from database, TableName() can't return zero value.
	Count(ctx context.Context, entity Entity, options *FilterOptions) (int64, error)

	// IsExist Name() and TableName() can't return zero value.
	IsExist(ctx context.Context, entity Entity) (bool, error)
}

type SortOption struct {
	Key   string
	Order SortOrder
}

type SortOrder int

type WithInTimeQueryOption struct {
	Key       string
	StartTime time.Time
	EndTime   time.Time
}