wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
40 lines
684 B
Go
40 lines
684 B
Go
package nats
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/nats-io/nats.go"
|
|
"go-micro.dev/v6/registry"
|
|
)
|
|
|
|
type natsWatcher struct {
|
|
sub *nats.Subscription
|
|
wo registry.WatchOptions
|
|
}
|
|
|
|
func (n *natsWatcher) Next() (*registry.Result, error) {
|
|
var result *registry.Result
|
|
for {
|
|
m, err := n.sub.NextMsg(time.Minute)
|
|
if err != nil && err == nats.ErrTimeout {
|
|
continue
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := json.Unmarshal(m.Data, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(n.wo.Service) > 0 && result.Service.Name != n.wo.Service {
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (n *natsWatcher) Stop() {
|
|
_ = n.sub.Unsubscribe()
|
|
}
|