db

type BatchOperation interface {
	Query(stmt string, args ...interface{})
	Size() int
}

type BatchWrapper struct {
	*gocql.Batch
}

type CacheChangeNotifier interface {
	Notify(tableName string, changedKey string, operation OperationType)
}

type CacheInfo struct {
	Stats *CacheStats
	// contains filtered or unexported fields
}

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

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

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

type CacheStats struct {
	DaoRefreshTime time.Time     `json:"daoRefreshTime"`
	CacheSize      int           `json:"cacheSize"`
	RequestCount   uint64        `json:"requestCount"`
	NonAbsentCount int           `json:"nonAbsentCount"`
	EvictionCount  uint64        `json:"evictionCount"`
	HitRate        float64       `json:"hitRate"`
	MissRate       float64       `json:"missRate"`
	TotalLoadTime  time.Duration `json:"totalLoadTime"`
}

type CachedListingDao interface {
	GetOne(tableName string, rowKey string, key2 interface{}) (interface{}, error)
	SetOne(tableName string, rowKey string, key2 interface{}, entity interface{}) error
	DeleteOne(tableName string, rowKey string, key2 interface{}) error
	DeleteAll(tableName string, rowKey string) error
	GetAll(tableName string, rowKey string) ([]interface{}, error)
	GetAllAsMap(tableName string, rowKey string, key2List []interface{}) (map[interface{}]interface{}, error)
	GetKeys(tableName string) ([]TwoKeys, error)
	GetKey2AsList(tableName string, rowKey string) ([]interface{}, error)
}

type CachedSimpleDao interface {
	GetOne(tableName string, rowKey string) (interface{}, error)
	GetOneFromCacheOnly(tableName string, rowKey string) (interface{}, error)
	SetOne(tableName string, rowKey string, entity interface{}) error
	DeleteOne(tableName string, rowKey string) error
	GetAllByKeys(tableName string, rowKeys []string) ([]interface{}, error)
	GetAllAsList(tableName string, maxResults int) ([]interface{}, error)
	GetAllAsMap(tableName string) (map[interface{}]interface{}, error)
	GetKeys(tableName string) ([]interface{}, error)
	RefreshAll(tableName string) error
	RefreshOne(tableName string, rowKey string) error
}

type CassandraClient struct {
	*gocql.Session
	*gocql.ClusterConfig
	SleepTime          int32
	ConcurrentQueries  chan bool
	LocalDc            string
	DeviceKeyspace     string
	DevicePodTableName string
	TestOnly           bool
	Connection_type    string
	// contains filtered or unexported fields
}

type CassandraConnector interface {
	NewCassandraClient(conf *configuration.Config, testOnly bool) (*CassandraClient, error)
}

type ChangedData struct {
	ColumnName     gocql.UUID    `json:"columnName"`
	CfName         string        `json:"cfName"`
	ChangedKey     string        `json:"changedKey"`
	Operation      OperationType `json:"operation"`
	DaoId          int32         `json:"DAOid"`
	ValidCacheSize int32         `json:"validCacheSize"`
	UserName       string        `json:"userName"`
	ServerOriginId string        `json:"serverOriginId"`
}

type CompressingDataDao interface {
	GetOne(tableName string, rowKey string) (interface{}, error)
	SetOne(tableName string, rowKey string, value []byte) error
	DeleteOne(tableName string, rowKey string) error
	GetAllByKeys(tableName string, rowKeys []string) ([]interface{}, error)
	GetAllAsList(tableName string, continueOnError bool) ([]interface{}, error)
	GetAllAsMap(tableName string, continueOnError bool) (map[string]interface{}, error)
	GetKeys(tableName string) []string
}

type DatabaseClient interface {
	SetUp() error
	TearDown() error
	Close() error
	Sleep()

	// Xconf
	QueryXconfDataRows(query string, queryParams ...string) ([]map[string]interface{}, error)
	ModifyXconfData(query string, queryParameters ...string) error

	// Batch operations
	NewBatch(batchType int) BatchOperation
	ExecuteBatch(batch BatchOperation) error

	SetXconfData(tableName string, rowKey string, value []byte, ttl int) error
	GetXconfData(tableName string, rowKey string) ([]byte, error)
	GetAllXconfDataByKeys(tableName string, rowKeys []string) [][]byte
	GetAllXconfKeys(tableName string) []string
	GetAllXconfDataAsList(tableName string, maxResults int) [][]byte
	GetAllXconfDataAsMap(tableName string, maxResults int) map[string][]byte
	DeleteXconfData(tableName string, rowKey string) error
	DeleteAllXconfData(tableName string) error

	// Xconf TwoKeys
	GetAllXconfData(tableName string, rowKey string) [][]byte
	GetAllXconfDataTwoKeysRange(tableName string, rowKey interface{}, key2FieldName string, rangeInfo *RangeInfo) [][]byte
	GetAllXconfDataTwoKeysAsMap(tableName string, rowKey string, key2FieldName string, key2List []interface{}) map[interface{}][]byte
	SetXconfDataTwoKeys(tableName string, rowKey interface{}, key2FieldName string, key2 interface{}, value []byte, ttl int) error
	GetXconfDataTwoKeys(tableName string, rowKey string, key2FieldName string, key2 interface{}) ([]byte, error)
	DeleteXconfDataTwoKeys(tableName string, rowKey string, key2FieldName string, key2 interface{}) error
	GetAllXconfTwoKeys(tableName string, key2FieldName string) []TwoKeys
	GetAllXconfKey2s(tableName string, rowKey string, key2FieldName string) []interface{}
	// Xconf compressed data
	SetXconfCompressedData(tableName string, rowKey string, values [][]byte, ttl int) error
	GetXconfCompressedData(tableName string, rowKey string) ([]byte, error)
	GetAllXconfCompressedDataAsMap(tableName string) map[string][]byte

	// Pod table lookup estbMac from pod serialNum
	GetEcmMacFromPodTable(string) (string, error)

	// not found
	IsDbNotFound(error) bool

	// Penetration Metrics
	GetPenetrationMetrics(macAddress string) (map[string]interface{}, error)
	SetPenetrationMetrics(penetrationmetrics *PenetrationMetrics) error
	SetFwPenetrationMetrics(*FwPenetrationMetrics) error
	GetFwPenetrationMetrics(string) (*FwPenetrationMetrics, error)
	SetRfcPenetrationMetrics(pMetrics *RfcPenetrationMetrics, is304FromPrecook bool) error
	GetRfcPenetrationMetrics(string) (*RfcPenetrationMetrics, error)
	UpdateFwPenetrationMetrics(map[string]string) error
	GetEstbIp(string) (string, error)

	SetRecookingStatus(module string, partitionId string, state int) error
	GetRecookingStatus(module string, partitionId string) (int, time.Time, error)
	CheckFinalRecookingStatus(module string) (bool, time.Time, error)

	// XPC precook reference data
	SetPrecookDataInXPC(RfcPrecookHash string, RfcPrecookPayload []byte) error
	GetPrecookDataFromXPC(RfcPrecookHash string) ([]byte, string, error)

	// Locks
	AcquireLock(lockName string, lockedBy string, ttlSeconds int) error
	ReleaseLock(lockName string, lockedBy string) error
	GetLockInfo(lockName string) (map[string]interface{}, error)
}

type DefaultCassandraConnection struct {
	Connection_type string
}

type DistributedLock struct {
	DistributedLockSettings
	// contains filtered or unexported fields
}

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

type FwPenetrationMetrics struct {
	EstbMac                 string
	Partner                 string
	Model                   string
	FwFilename              string
	FwVersion               string
	FwReportedVersion       string
	FwAdditionalVersionInfo string
	FwAppliedRule           string
	FwTs                    int64
	ClientCertExpiry        string
	RecoveryCertExpiry      string
}

type GroupServiceCacheDao interface {
	GetGroupServiceFeatureTags(cacheKey string) map[string]string
	SetGroupServiceFeatureTags(cacheKey string, tags map[string]string) error
	DeleteGroupServiceFeatureTags(cacheKey string) error
}

type GroupServiceCacheDaoImpl struct{}

type ListingDao interface {
	GetOne(tableName string, rowKey string, key2 interface{}) (interface{}, error)
	SetOne(tableName string, rowKey interface{}, key2 interface{}, value []byte) error
	DeleteOne(tableName string, rowKey string, key2 interface{}) error
	DeleteAll(tableName string, rowKey string) error
	GetAll(tableName string, rowKey string) ([]interface{}, error)
	GetAllAsList(tableName string) ([]interface{}, error)
	GetAllAsMap(tableName string, rowKey string, key2List []interface{}) (map[interface{}]interface{}, error)
	GetRange(tableName string, rowKey interface{}, rangeInfo *RangeInfo) ([]interface{}, error)
	GetKeys(tableName string) ([]TwoKeys, error)
	GetKey2AsList(tableName string, rowKey string) ([]interface{}, error)
}

type OperationType string

type PenetrationMetrics struct {
	EstbMac                 string
	Partner                 string
	Model                   string
	FwVersion               string
	FwReportedVersion       string
	FwAdditionalVersionInfo string
	FwAppliedRule           string
	FwTs                    time.Time
	RfcAppliedRules         string
	RfcFeatures             string
	RfcTs                   time.Time
}

type RangeInfo struct {
	StartValue interface{}
	EndValue   interface{}
}

type RecookingStatus struct {
	AppName     string    `json:"appName"`
	PartitionId string    `json:"partitionId"`
	State       int       `json:"state"`
	UpdatedTime time.Time `json:"updatedTime"`
}

type RfcPenetrationMetrics struct {
	EstbMac              string
	EcmMac               string
	SerialNum            string
	Partner              string
	Model                string
	RfcPartner           string
	TitanPartner         string
	RfcModel             string
	RfcFwReportedVersion string
	RfcAppliedRules      string
	RfcAccountMgmt       string
	RfcFeatures          string
	RfcTs                int64
	RfcAccountHash       string
	RfcAccountId         string
	TitanAccountId       string
	RfcEnv               string
	RfcApplicationType   string
	RfcExperience        string
	RfcTimeZone          string
	RfcConfigsetHash     string
	RfcQueryParams       string
	RfcTags              string
	RfcEstbIp            string
	RfcPostProc          string
	ClientCertExpiry     string
	RecoveryCertExpiry   string
}

type SimpleDao interface {
	Modify(query string, queryParams ...string) error
	Query(query string, queryParams ...string) ([]map[string]interface{}, error)
	GetOne(tableName string, rowKey string) (interface{}, error)
	SetOne(tableName string, rowKey string, value []byte) error
	DeleteOne(tableName string, rowKey string) error
	GetAllByKeys(tableName string, rowKeys []string) ([]interface{}, error)
	GetAllAsList(tableName string, maxResults int) ([]interface{}, error)
	GetAllAsMap(tableName string, maxResults int) (map[string]interface{}, error)
	GetKeys(tableName string) []string
	GetAllAsMapRaw(tableName string, maxResults int) (map[string]json.RawMessage, error)

	// Batch operations
	NewBatch(batchType int) BatchOperation
	ExecuteBatch(batch BatchOperation) error
}

type Statistics struct {
	CacheMap map[string]CacheStats `json:"Statistics"`
}

type TableInfo struct {
	TableName       string
	ConstructorFunc func() interface{} // model/struct constructor function
	Compress        bool               // data is compressed
	Split           bool               // data is split into multiple chunks
	CacheData       bool               // specifies whether to cache the data
	TTL             int                // TTL for the data
	Key2FieldName   string             // column name for listing table, e.g. Logs2, XconfChangedKeys4
	DaoId           int32              // Xconf DAO ID
}

type TwoKeys struct {
	Key  string
	Key2 interface{}
}