27 lines
447 B
27 lines
447 B
package consul
|
|
|
|
import (
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"github.com/go-kratos/kratos/v2/registry"
|
|
)
|
|
|
|
type serviceSet struct {
|
|
serviceName string
|
|
watcher map[*watcher]struct{}
|
|
services *atomic.Value
|
|
lock sync.RWMutex
|
|
}
|
|
|
|
func (s *serviceSet) broadcast(ss []*registry.ServiceInstance) {
|
|
s.services.Store(ss)
|
|
s.lock.RLock()
|
|
defer s.lock.RUnlock()
|
|
for k := range s.watcher {
|
|
select {
|
|
case k.event <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
|