windy/go-caatsm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Update agent guidelines and improve documentation structure. Refactor AGENTS.md to streamline commands and code style guidelines, enhancing clarity and usability. Update README.md with refined NATS consumer configuration details and observability metrics. Modify .gitignore to exclude dynamically generated Prometheus target files. Enhance configuration files for development and production environments, ensuring consistency and clarity in settings.

Browse Source
This commit is contained in:
windyboy
2025-11-19 13:07:09 +08:00
parent 06fc9cb9e0
commit c687fdcde8
39 changed files with 1112 additions and 3193 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/infra/nats/consumer_manager.go
+2 -43
View File
@@ -59,53 +59,12 @@ func (cm *ConsumerManager) EnsureConsumer(config *nats.ConsumerConfig) error {
return nil
}
// RecoverResources attempts to recreate the stream and consumer in dev/test environments
func (cm *ConsumerManager) RecoverResources(streamManager *StreamManager, consumerConfig *nats.ConsumerConfig) error {
if cm.js == nil {
return fmt.Errorf("jetstream context is nil")
}
// Ensure stream exists (dev/test may auto-create, prod will error).
if err := streamManager.EnsureStream(); err != nil {
return fmt.Errorf("ensure stream %s: %w", cm.streamName, err)
}
// Ensure durable consumer exists and is properly bound.
if err := cm.EnsureConsumer(consumerConfig); err != nil {
return fmt.Errorf("ensure consumer %s: %w", cm.consumerName, err)
}
return nil
}
// CreatePullSubscription creates a pull subscription with recovery logic
func (cm *ConsumerManager) CreatePullSubscription() (*nats.Subscription, error) {
return cm.js.PullSubscribe(cm.subject, cm.consumerName, nats.Bind(cm.streamName, cm.consumerName))
}
// CreatePullSubscriptionWithRecovery creates a pull subscription and attempts recovery if needed
// CreatePullSubscriptionWithRecovery creates a pull subscription
func (cm *ConsumerManager) CreatePullSubscriptionWithRecovery(streamManager *StreamManager, consumerConfig *nats.ConsumerConfig) (*nats.Subscription, error) {
sub, err := cm.CreatePullSubscription()
if err == nil {
return sub, nil
}
if isJetStreamResourceNotFound(err) && isDevLikeEnv() && shouldBootstrapStream() {
cm.logger.Warn("PullSubscribe failed due to missing JetStream resources; attempting to recreate",
zap.Error(err),
zap.String("stream", cm.streamName),
zap.String("consumer", cm.consumerName),
)
if recErr := cm.RecoverResources(streamManager, consumerConfig); recErr != nil {
return nil, fmt.Errorf("failed to recover JetStream resources: %w", recErr)
}
// Retry subscription after successful recovery.
sub, err = cm.CreatePullSubscription()
if err != nil {
return nil, fmt.Errorf("failed to create pull subscription after recovery: %w", err)
}
return sub, nil
}
return nil, fmt.Errorf("failed to create pull subscription: %w", err)
return cm.CreatePullSubscription()
}