cassandra

type Interface interface {
	CreateKeyspaces(ksc ...*KeyspaceConfig) error
	NewSession(cfg *SessionConfig) (*Session, error)
	NewBatchWriter(session *Session, c *WriterConfig, builderCreator func() StatementBuilder) writer.Writer
}

type KeyspaceConfig struct {
	Name        string                    `file:"name" env:"CASSANDRA_KEYSPACE"`
	Auto        bool                      `file:"auto"`
	Replication KeyspaceReplicationConfig `file:"replication"`
}

type KeyspaceReplicationConfig struct {
	Class  string `file:"class" default:"SimpleStrategy"`
	Factor int32  `file:"factor" default:"2"`
}

type ReconnectionConfig struct {
	Enable        bool          `file:"enable" default:"true"`
	CheckInterval time.Duration `file:"check_interval" default:"10m"`
	CheckTimeout  time.Duration `file:"check_timeout" default:"60s"`
}

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

type SessionConfig struct {
	Keyspace     KeyspaceConfig     `file:"keyspace"`
	Consistency  string             `file:"consistency" default:"LOCAL_ONE"`
	Reconnection ReconnectionConfig `file:"reconnection"`
}

type StatementBuilder interface {
	GetStatement(data interface{}) (string, []interface{}, error)
}

type WriterConfig struct {
	Parallelism uint64 `file:"parallelism" default:"4" desc:"parallelism"`
	Batch       struct {
		SizeBytes int           `file:"size_bytes" desc:"cassandra batch failed size bytes"`
		Size      uint64        `file:"size" default:"100" desc:"batch size"`
		Timeout   time.Duration `file:"timeout" default:"30s" desc:"timeout to flush buffer for batch write"`
	} `file:"batch"`
	Retry int `file:"retry" desc:"retry if fail to write"`
}