Documentation
¶
Index ¶
- Constants
- func ShardIDByGuild(guildID snowflake.ID, shardCount int) int
- type ConfigOpt
- func WithAutoScaling(autoScaling bool) ConfigOpt
- func WithCloseHandler(closeHandler gateway.CloseHandlerFunc) ConfigOpt
- func WithDefault() ConfigOpt
- func WithDefaultIdentifyRateLimiterConfigOpt(opts ...gateway.IdentifyRateLimiterConfigOpt) ConfigOpt
- func WithGatewayConfigOpts(opts ...gateway.ConfigOpt) ConfigOpt
- func WithGatewayCreateFunc(gatewayCreateFunc gateway.CreateFunc) ConfigOpt
- func WithIdentifyRateLimiter(rateLimiter gateway.IdentifyRateLimiter) ConfigOpt
- func WithIdentifyRateLimiterConfigOpt(opts ...gateway.IdentifyRateLimiterConfigOpt) ConfigOpt
- func WithLogger(logger *slog.Logger) ConfigOpt
- func WithShardCount(shardCount int) ConfigOpt
- func WithShardIDs(shardIDs ...int) ConfigOpt
- func WithShardIDsWithStates(shards map[int]ShardState) ConfigOpt
- func WithShardSplitCount(shardSplitCount int) ConfigOpt
- type ShardManager
- type ShardState
Constants ¶
const DefaultShardSplitCount = 2
DefaultShardSplitCount is the default count a shard should be split into when it needs re-sharding.
Variables ¶
This section is empty.
Functions ¶
Types ¶
type ConfigOpt ¶
type ConfigOpt func(config *config)
ConfigOpt is a type alias for a function that takes a config and is used to configure your Server.
func WithAutoScaling ¶
WithAutoScaling sets whether the ShardManager should automatically re-shard shards if they are too large. This is disabled by default.
func WithCloseHandler ¶
func WithCloseHandler(closeHandler gateway.CloseHandlerFunc) ConfigOpt
WithCloseHandler sets the function which is called when a gateway.Gateway is closed and could not reconnect (or auto reconnecting is disabled). If auto-scaling is enabled and the shard was closed due to gateway.CloseEventCodeShardingRequired, the closeHandler will not be called.
func WithDefault ¶
func WithDefault() ConfigOpt
WithDefault returns a ConfigOpt that sets the default values for the ShardManager.
func WithDefaultIdentifyRateLimiterConfigOpt ¶
func WithDefaultIdentifyRateLimiterConfigOpt(opts ...gateway.IdentifyRateLimiterConfigOpt) ConfigOpt
WithDefaultIdentifyRateLimiterConfigOpt lets you configure the default gateway.IdentifyRateLimiter used by the ShardManager and prepend the options to the existing ones.
func WithGatewayConfigOpts ¶
WithGatewayConfigOpts lets you configure the gateway.Gateway created by the ShardManager.
func WithGatewayCreateFunc ¶
func WithGatewayCreateFunc(gatewayCreateFunc gateway.CreateFunc) ConfigOpt
WithGatewayCreateFunc sets the function which is used by the ShardManager to create a new gateway.Gateway.
func WithIdentifyRateLimiter ¶
func WithIdentifyRateLimiter(rateLimiter gateway.IdentifyRateLimiter) ConfigOpt
WithIdentifyRateLimiter lets you inject your own RateLimiter into the ShardManager.
func WithIdentifyRateLimiterConfigOpt ¶
func WithIdentifyRateLimiterConfigOpt(opts ...gateway.IdentifyRateLimiterConfigOpt) ConfigOpt
WithIdentifyRateLimiterConfigOpt lets you configure the default gateway.IdentifyRateLimiter used by the ShardManager.
func WithLogger ¶
WithLogger sets the logger of the ShardManager.
func WithShardCount ¶
WithShardCount sets the shard count of the ShardManager.
func WithShardIDs ¶
WithShardIDs sets the shardIDs the ShardManager should manage.
func WithShardIDsWithStates ¶
func WithShardIDsWithStates(shards map[int]ShardState) ConfigOpt
WithShardIDsWithStates sets the shardIDs and their ShardState the ShardManager should manage.
func WithShardSplitCount ¶
WithShardSplitCount sets the count a shard should be split into if it is too large. This is only used if AutoScaling is enabled.
type ShardManager ¶
type ShardManager interface {
// Open opens all configured shards.
Open(ctx context.Context)
// Close closes all shards.
Close(ctx context.Context)
// OpenShard opens a specific shard.
OpenShard(ctx context.Context, shardID int) error
// ResumeShard resumes a specific shard with the given sessionID and sequence.
ResumeShard(ctx context.Context, shardID int, state ShardState) error
// CloseShard closes a specific shard.
CloseShard(ctx context.Context, shardID int)
// ShardByGuildID returns the gateway.Gateway for the shard that contains the given guild.
ShardByGuildID(guildId snowflake.ID) gateway.Gateway
// Shard returns the gateway.Gateway for the given shard ID.
Shard(shardID int) gateway.Gateway
// Shards returns all shards. This function is thread-safe.
Shards() iter.Seq[gateway.Gateway]
}
ShardManager manages multiple gateway.Gateway connections. For more information on sharding see: https://discord.com/developers/docs/topics/gateway#sharding
func New ¶
func New(token string, eventHandlerFunc gateway.EventHandlerFunc, opts ...ConfigOpt) ShardManager
New creates a new default ShardManager with the given token, eventHandlerFunc and ConfigOpt(s).
type ShardState ¶
type ShardState struct {
// SessionID is the session ID of the shard. This is used to resume the shard.
SessionID string
// Sequence is the sequence number of the shard. This is used to resume the shard.
Sequence int
// ResumeURL is the resume url to use for the shard. This is used to resume the shard.
ResumeURL string
}
ShardState is used to tell a gateway.Gateway managed by the ShardManager which session & sequence it should use when starting the shard. This is useful for resuming shards when using the ShardManager.
Source Files