db

type AlterTableAddInfo struct {
	Keyspace string
	Table    string
	ToAdd    []*gocql.ColumnMetadata
}

type AlterTableDropInfo struct {
	Keyspace string
	Table    string
	ToDrop   []string
}

type ColumnOrder struct {
	Column string
	Order  string
}

type CreateKeyspaceInfo struct {
	Name        string
	DCReplicas  map[string]int
	IfNotExists bool
}

type CreateTableInfo struct {
	Keyspace       string
	Table          string
	PartitionKeys  []*gocql.ColumnMetadata
	ClusteringKeys []*gocql.ColumnMetadata
	Values         []*gocql.ColumnMetadata
	IfNotExists    bool
}

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

type DeleteInfo struct {
	Keyspace    string
	Table       string
	Columns     []string
	QueryParams []interface{}
	IfCondition []types.ConditionItem
	IfExists    bool
}

type DropKeyspaceInfo struct {
	Name     string
	IfExists bool
}

type DropTableInfo struct {
	Keyspace string
	Table    string
	IfExists bool
}

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

type InsertInfo struct {
	Keyspace    string
	Table       string
	Columns     []string
	QueryParams []interface{}
	IfNotExists bool
	TTL         int
}

type QueryOptions struct {
	UserOrRole        string
	Consistency       gocql.Consistency
	SerialConsistency gocql.SerialConsistency
	PageState         []byte
	PageSize          int
	Context           context.Context
}

type ResultMock struct {
	mock.Mock
}

type ResultSet interface {
	PageState() []byte
	Values() []map[string]interface{}
}

type SelectInfo struct {
	Keyspace string
	Table    string
	Where    []types.ConditionItem
	Options  *types.QueryOptions
	OrderBy  []ColumnOrder
}

type Session interface {
	// Execute executes a statement without returning row results
	Execute(query string, options *QueryOptions, values ...interface{}) error

	// ExecuteIterSimple executes a statement and returns iterator to the result set
	ExecuteIter(query string, options *QueryOptions, values ...interface{}) (ResultSet, error)

	// ChangeSchema executes a schema change query and waits for schema agreement
	ChangeSchema(query string, options *QueryOptions) error

	//TODO: Extract metadata methods from interface into another interface
	KeyspaceMetadata(keyspaceName string) (*gocql.KeyspaceMetadata, error)
}

type SessionMock struct {
	mock.Mock
}

type UpdateInfo struct {
	Keyspace    string
	Table       *gocql.TableMetadata
	Columns     []string
	QueryParams []interface{}
	IfCondition []types.ConditionItem
	IfExists    bool
	TTL         int
}