interfaces

type ConsumerCallback func(Messages, ConsumerCallbackIsDone)

type ConsumerCallbackIsDone struct {
	Done       context.CancelFunc
	EndRequest func()
}

type Database interface {
	Init(logger.Logger) Database
	SetTracer(tracing.Tracing)
	LoadRabbitMQ(string, RabbitMQProviderConfig) RabbitMQ
	LoadSQL(string, SQLConfig) SQL
	LoadRedis(string, RedisProviderConfig) Redis
	LoadEmbedded(EmbeddedOptions) Embedded
	LoadKafka(string, KafkaProviderConfig) Kafka
	LoadMongo(tag string, config MongoProviderConfig) Mongo
}

type Embedded interface {
	DB() *nutsdb.DB
	Set(bucket string, key string, value interface{}, encoding Encoding) (err error)
	Get(bucket string, key string, value interface{}, encoding Encoding) (err error)
	Delete(bucket string, key string) (err error)
	GetAllKeys(bucket string) (keys []string, err error)
}

type EmbeddedOptions struct {
	Directory string
}

type Encoder struct {
	Raw      io.Reader
	Encoding Encoding
	// contains filtered or unexported fields
}

type Encoding int

type Kafka interface {
	Consumer(KafkaOptions, ConsumerCallback)
	Producer(ProducerIsReady)
	Push(ctx context.Context, id string, options KafkaOptions, body interface{}, cb ConsumerCallback) error
}

type KafkaOptions struct {
	Topic          string
	RegistryValue  string
	Group          string
	SchemeVersion  int
	SchemeID       int
	Encoding       Encoding
	MultipleThread bool
	Limiter        int
}

type KafkaProviderConfig struct {
	Enable           bool   `yaml:"enable" default:"false" desc:"config:kafka:enable"`
	Host             string `yaml:"host" default:"127.0.0.1:9092" desc:"config:kafka:host"`
	Registry         string `yaml:"registry" default:"" desc:"config:kafka:registry"`
	Username         string `yaml:"username" default:""  desc:"config:kafka:username"`
	Password         string `yaml:"password" default:"" desc:"config:kafka:password"`
	SecurityProtocol string `yaml:"securityProtocol" default:"SASL_SSL"  desc:"config:kafka:securityProtocol"`
	Mechanisms       string `yaml:"mechanisms" default:"PLAIN"  desc:"config:kafka:mechanisms"`
	Debug            string `yaml:"debug" default:"consumer"  desc:"config:kafka:debug"`
}

type Messages interface {
	Exchange() string
	RoutingKey() string
	Decode(interface{}) error
	SetContext(context.Context)
	Context() context.Context
}

type Mongo interface {
	Client() *mongo.Client
	Start() (err error)
	UseEncryption(encLib encryption.Encryption) Mongo
	ODM() MongoODM
	WithMonitor(enable bool) Mongo
	GetConfig() MongoProviderConfig
}

type MongoIndex struct {
	Collection         string
	Field              string
	Unique             bool
	Compound           bool
	PrimaryKey         bool
	Desc               bool
	ExpiredAfterSecond int32
}

type MongoODM interface {
	Error() error
	Collection(model interface{}, name string) MongoODM
	NativeCollection() *mongo.Collection
	Index() MongoODM
	Create(ctx context.Context, data interface{}) *MongoTransaction
	Upsert(ctx context.Context, data interface{}, matchKey []string) *MongoTransaction
	Filter(filter map[string]interface{}) MongoODM
	Find(ctx context.Context, dest interface{}) *MongoTransaction
	FindRaw(ctx context.Context) (results []bson.Raw, cc *MongoTransaction)
	Drop(ctx context.Context) MongoODM
	Update(ctx context.Context, data map[string]interface{}) *MongoTransaction
	Delete(ctx context.Context) *MongoTransaction
	SetSort(sortBy string, desc bool) MongoODM
	SetLimit(limit int64) MongoODM
	SetSkip(skip int64) MongoODM
	Count(ctx context.Context) (res int64, err error)
}

type MongoProviderConfig struct {
	Name              string `yaml:"name" default:"" desc:"config:mongo:name"`
	Enable            bool   `yaml:"enable" default:"false" desc:"config:mongo:enable"`
	Uri               string `yaml:"uri" default:"" desc:"config:mongo:uri"`
	Host              string `yaml:"host" default:"127.0.0.1" desc:"config:mongo:host"`
	Port              int    `yaml:"port" default:"27017" desc:"config:mongo:port"`
	UserName          string `yaml:"username" default:"root" desc:"config:mongo:username"`
	Password          string `yaml:"password" default:"root" desc:"config:mongo:password"`
	Database          string `yaml:"database" default:"sample_database" desc:"config:mongo:database"`
	Options           string `yaml:"options" default:"" desc:"config:mongo:options"`
	TimeoutConnection int    `yaml:"timeoutConnection" default:"5000" desc:"config:mongo:timeoutConnection"`
	AutoReconnect     bool   `yaml:"autoReconnect" default:"false"  desc:"config:autoReconnect"`
	StartInterval     int    `yaml:"startInterval" default:"2"  desc:"config:startInterval"`
	MaxError          int    `yaml:"maxError" default:"5"  desc:"config:maxError"`
}

type MongoTransaction struct {
	Error        error
	InsertedIDs  []primitive.ObjectID
	EffectedIDs  []primitive.ObjectID
	DeletedCount int64
}

type ORM interface {
	AutoMigrate(dst ...interface{}) (err error)
	Model(dst interface{}) OrmModel
}

type OrmModel interface {
	Debug() OrmModel
	WithContext(ctx context.Context) OrmModel
	Drop() error
	Truncate() error
	Limit(limit int) OrmModel
	Offset(offset int) OrmModel
	Order(query ...clause.OrderByColumn) OrmModel
	Equal(query ...interface{}) OrmModel
	NotEqual(query ...interface{}) OrmModel
	Or(query ...interface{}) OrmModel
	Create(data interface{}) (err error)
	Upsert(data interface{}) (err error)
	Update(data map[string]interface{}) (err error)
	Find(dst interface{}) error
	First(dst interface{}) error
	FindRaw() (res []interface{}, err error)
	Delete() (rowEffected int64, err error)
	Where(query interface{}, args ...interface{}) OrmModel
}

type ProducerIsReady func()

type Provider int

type RabbitMQ interface {
	Consumer(RabbitMQOptions, ConsumerCallback)
	Producer(RabbitMQOptions)
	Push(ctx context.Context,
		id, exchange,
		key string,
		body interface{},
		cb ConsumerCallback) error
}

type RabbitMQOptions struct {
	Exchange     string
	ExchangeType string
	RoutingKey   string
	Durable      bool
	AutoDeleted  bool
	NoWait       bool
	Encoding     Encoding
}

type RabbitMQProviderConfig struct {
	Enable              bool   `yaml:"enable" default:"false" desc:"config:rabbitmq:enable"`
	Host                string `yaml:"host" default:"127.0.0.1" desc:"config:rabbitmq:host"`
	Port                int    `yaml:"port" default:"5672" desc:"config:rabbitmq:port"`
	Username            string `yaml:"username" default:"guest"  desc:"config:rabbitmq:username"`
	Password            string `yaml:"password" default:"guest" desc:"config:rabbitmq:password"`
	ReconnectDuration   int    `yaml:"reconnectDuration" default:"5" desc:"config:rabbitmq:reconnectDuration"`
	DedicatedConnection bool   `yaml:"dedicatedConnection" default:"false" desc:"config:rabbitmq:dedicatedConnection"`
}

type Redis interface {
	Init() error
	GetClient() (pool radix.Client)
	GetConfig() RedisProviderConfig
}

type RedisProviderConfig struct {
	Name          string `yaml:"name" default:"" desc:"config:redis:name"`
	Enable        bool   `yaml:"enable" default:"false" desc:"config:redis:enable"`
	Host          string `yaml:"host" default:"127.0.0.1" desc:"config:redis:host"`
	Port          int    `yaml:"port" default:"6379" desc:"config:redis:port"`
	Password      string `yaml:"password" default:"" desc:"config:redis:password"`
	Pool          int    `yaml:"pool" default:"10" desc:"config:redis:pool"`
	AutoReconnect bool   `yaml:"autoReconnect" default:"false"  desc:"config:autoReconnect"`
	StartInterval int    `yaml:"startInterval" default:"2"  desc:"config:startInterval"`
	MaxError      int    `yaml:"maxError" default:"5"  desc:"config:maxError"`
}

type SQL interface {
	Config() SQLConfig
	UseEncryption(encLib encryption.Encryption) SQL
	Gorm() *gorm.DB
	ORM() ORM
	Start() error
	Close() error
	GetEncryption() (encLib encryption.Encryption)
}

type SQLConfig struct {
	Name               string `yaml:"name" default:"" desc:"config:sql:name"`
	Driver             string `yaml:"driver" default:"mysql" desc:"config:sql:enable"`
	Enable             bool   `yaml:"enable" default:"false" desc:"config:sql:enable"`
	Host               string `yaml:"host" default:"127.0.0.1" desc:"config:sql:host"`
	Port               int    `yaml:"port" default:"3306" desc:"config:sql:port"`
	Username           string `yaml:"username" default:"root"  desc:"config:sql:username"`
	Password           string `yaml:"password" default:"root" desc:"config:sql:password"`
	Database           string `yaml:"database" default:"sample_database" desc:"config:sql:database"`
	Options            string `yaml:"options" default:"" desc:"config:sql:options"`
	Connection         string `yaml:"connection" default:"" desc:"config:sql:connection"`
	AutoReconnect      bool   `yaml:"autoReconnect" default:"false"  desc:"config:autoReconnect"`
	StartInterval      int    `yaml:"startInterval" default:"2"  desc:"config:startInterval"`
	MaxError           int    `yaml:"maxError" default:"5"  desc:"config:maxError"`
	DBPath             string `yaml:"dbPath" default:"./sqlite.db" desc:"config:sql:dbPath"`
	SsLMode            string `yaml:"sslMode" default:"disable" desc:"config:sql:sslMode"`
	TimeoutConnection  int    `yaml:"timeoutConnection" default:"5000" desc:"config:sql:timeoutConnection"`
	TimeoutTransaction int    `yaml:"timeoutTransaction" default:"5000" desc:"config:sql:timeoutTransaction"`
}