package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-09-26 20:11:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: sig

Browse Source
This commit is contained in:
sujit
2025-03-29 23:03:27 +05:45
parent b205d29bb5
commit ffaa1a137b
4 changed files with 108 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
consumer.go
View File

@@ -118,6 +118,12 @@ func (c *Consumer) OnMessage(ctx context.Context, msg *codec.Message, conn net.C
log.Printf("Unable to stop consumer: %v", err)
}
return err
case consts.CONSUMER_UPDATE:
err := c.Update(ctx, msg.Payload)
if err != nil {
log.Printf("Unable to update consumer: %v", err)
}
return err
default:
log.Printf("CONSUMER - UNKNOWN_COMMAND ~> %s on %s", msg.Command, msg.Queue)
}
@@ -295,6 +301,21 @@ func (c *Consumer) Pause(ctx context.Context) error {
return c.operate(ctx, consts.CONSUMER_PAUSED, c.pool.Pause)
}
func (c *Consumer) Update(ctx context.Context, payload []byte) error {
var newConfig DynamicConfig
if err := json.Unmarshal(payload, &newConfig); err != nil {
log.Printf("Invalid payload for CONSUMER_UPDATE: %v", err)
return err
}
if c.pool != nil {
if err := c.pool.UpdateConfig(&newConfig); err != nil {
log.Printf("Failed to update pool config: %v", err)
return err
}
}
return c.sendOpsMessage(ctx, consts.CONSUMER_UPDATED)
}
func (c *Consumer) Resume(ctx context.Context) error {
return c.operate(ctx, consts.CONSUMER_RESUMED, c.pool.Resume)
}
@@ -304,10 +325,10 @@ func (c *Consumer) Stop(ctx context.Context) error {
}
func (c *Consumer) operate(ctx context.Context, cmd consts.CMD, poolOperation func()) error {
poolOperation()
if err := c.sendOpsMessage(ctx, cmd); err != nil {
return err
}
poolOperation()
return nil
}