mongoutil

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

type AdminOptions struct {
	ConnectionTimeout time.Duration
	Username          string
	Password          string
	AuthDatabase      string
}

type IAdminConnections interface {
	// Get returns a client for the given address
	Get(addr string) (IMongoClient, error)

	// GetAll returns all clients
	GetAll() map[string]IMongoClient

	// Add adds a new client for the given address
	Add(addr string) error

	// Remove removes the client for the given address
	Remove(addr string) error

	// Reset closes and removes all connections
	Reset()
}

type IMongoClient interface {

	// Disconnect closes the connection to MongoDB
	Disconnect(ctx context.Context) error

	// GetReplicaSetStatus returns the replica set status
	GetReplicaSetStatus(ctx context.Context) (*ReplicaSetStatus, error)

	// GetReplicaSetConfig returns the replica set configuration
	GetReplicaSetConfig(ctx context.Context) (*ReplicaSetConfig, error)

	// InitiateReplicaSet initializes a new replica set
	InitiateReplicaSet(ctx context.Context, config *ReplicaSetConfig) error

	// AddMemberToReplicaSet adds a new member to the replica set
	AddMemberToReplicaSet(ctx context.Context, host string, port int) error

	// RemoveMemberFromReplicaSet removes a member from the replica set
	RemoveMemberFromReplicaSet(ctx context.Context, host string, port int) error

	// StepDown forces the primary to step down
	StepDown(ctx context.Context, stepDownSecs int) error

	// IsMaster checks if the current node is the primary
	IsMaster(ctx context.Context) (*IsMasterResult, error)
}

type IReplicaSetAdmin interface {
	// Connections returns the connection map of all clients
	Connections() IAdminConnections

	// Close the admin connections
	Close()

	// GetReplicaSetInfos return the ReplicaSetInfos for all nodes
	GetReplicaSetInfos(ctx context.Context, memberCount int) *ReplicaSetInfos

	// InitiateReplicaSet initializes a new replica set
	InitiateReplicaSet(ctx context.Context, addr, replicaSetName string, members []ReplicaSetConfigMember) error

	// AddMember adds a new member to the replica set
	AddMember(ctx context.Context, primaryAddr, memberHost string, memberPort int) error

	// RemoveMember removes a member from the replica set
	RemoveMember(ctx context.Context, primaryAddr, memberHost string, memberPort int) error

	// StepDown forces the primary to step down
	StepDown(ctx context.Context, primaryAddr string, stepDownSecs int) error
}

type IsMasterResult struct {
	IsMaster                     bool      `bson:"ismaster"`
	IsSecondary                  bool      `bson:"secondary"`
	MaxBsonObjectSize            int       `bson:"maxBsonObjectSize"`
	MaxMessageSizeBytes          int       `bson:"maxMessageSizeBytes"`
	MaxWriteBatchSize            int       `bson:"maxWriteBatchSize"`
	LocalTime                    time.Time `bson:"localTime"`
	LogicalSessionTimeoutMinutes int       `bson:"logicalSessionTimeoutMinutes"`
	ConnectionId                 int       `bson:"connectionId"`
	MinWireVersion               int       `bson:"minWireVersion"`
	MaxWireVersion               int       `bson:"maxWireVersion"`
	ReadOnly                     bool      `bson:"readOnly"`
	OK                           float64   `bson:"ok"`
	SetName                      string    `bson:"setName,omitempty"`
	SetVersion                   int       `bson:"setVersion,omitempty"`
	Hosts                        []string  `bson:"hosts,omitempty"`
	Primary                      string    `bson:"primary,omitempty"`
	Me                           string    `bson:"me,omitempty"`
}

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

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

type ReplicaSetConfig struct {
	Id       string                    `bson:"_id"`
	Version  int                       `bson:"version"`
	Members  []ReplicaSetConfigMember  `bson:"members"`
	Settings *ReplicaSetConfigSettings `bson:"settings,omitempty"`
}

type ReplicaSetConfigMember struct {
	ID                 int     `bson:"_id"`
	Host               string  `bson:"host"`
	Priority           float64 `bson:"priority"`
	Hidden             bool    `bson:"hidden,omitempty"`
	ArbiterOnly        bool    `bson:"arbiterOnly,omitempty"`
	Votes              int     `bson:"votes,omitempty"`
	SecondaryDelaySecs int     `bson:"secondaryDelaySecs,omitempty"`
	BuildIndexes       bool    `bson:"buildIndexes,omitempty"`
}

type ReplicaSetConfigSettings struct {
	ChainingAllowed         bool                      `bson:"chainingAllowed,omitempty"`
	HeartbeatIntervalMillis int                       `bson:"heartbeatIntervalMillis,omitempty"`
	HeartbeatTimeoutSecs    int                       `bson:"heartbeatTimeoutSecs,omitempty"`
	ElectionTimeoutMillis   int                       `bson:"electionTimeoutMillis,omitempty"`
	CatchUpTimeoutMillis    int                       `bson:"catchUpTimeoutMillis,omitempty"`
	GetLastErrorModes       map[string]map[string]int `bson:"getLastErrorModes,omitempty"`
	GetLastErrorDefaults    map[string]interface{}    `bson:"getLastErrorDefaults,omitempty"`
	ReplicaSetId            primitive.ObjectID        `bson:"replicaSetId,omitempty"`
}

type ReplicaSetInfoStatus string

type ReplicaSetInfos struct {
	Infos  map[string]*ReplicaSetNode
	Status ReplicaSetInfoStatus
}

type ReplicaSetNode struct {
	Host        string
	Port        int
	ID          int
	State       string
	Role        string
	Health      int
	Priority    float64
	Hidden      bool
	ArbiterOnly bool
	Votes       int
}

type ReplicaSetStatus struct {
	Set     string                   `bson:"set"`
	Date    interface{}              `bson:"date"`
	MyState int                      `bson:"myState"`
	Term    int                      `bson:"term,omitempty"`
	Members []ReplicaSetStatusMember `bson:"members"`
}

type ReplicaSetStatusMember struct {
	ID                int         `bson:"_id"`
	Name              string      `bson:"name"`
	Health            int         `bson:"health"`
	State             int         `bson:"state"`
	StateStr          string      `bson:"stateStr"`
	Uptime            int         `bson:"uptime,omitempty"`
	OptimeDate        interface{} `bson:"optimeDate,omitempty"`
	ConfigVersion     int         `bson:"configVersion,omitempty"`
	LastHeartbeat     interface{} `bson:"lastHeartbeat,omitempty"`
	LastHeartbeatRecv interface{} `bson:"lastHeartbeatRecv,omitempty"`
	PingMs            int         `bson:"pingMs,omitempty"`
}