Files
tasks/common/membership/static/service_resolver.go
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

118 lines
2.5 KiB
Go

package static
import (
"sync"
"github.com/dgryski/go-farm"
"go.temporal.io/server/common/membership"
)
// staticResolver is a service resolver that maintains static mapping between services and host info
type staticResolver struct {
mu sync.Mutex
hostInfos []membership.HostInfo
listeners map[string]chan<- *membership.ChangedEvent
hashfunc func([]byte) uint32
}
func newStaticResolver(hosts []string) *staticResolver {
hostInfos := make([]membership.HostInfo, 0, len(hosts))
for _, host := range hosts {
hostInfos = append(hostInfos, membership.NewHostInfoFromAddress(host))
}
return &staticResolver{
hostInfos: hostInfos,
hashfunc: farm.Fingerprint32,
listeners: make(map[string]chan<- *membership.ChangedEvent),
}
}
func (s *staticResolver) start(hosts []string) {
hostInfos := make([]membership.HostInfo, 0, len(hosts))
for _, host := range hosts {
hostInfos = append(hostInfos, membership.NewHostInfoFromAddress(host))
}
event := &membership.ChangedEvent{
HostsAdded: hostInfos,
}
s.mu.Lock()
defer s.mu.Unlock()
s.hostInfos = hostInfos
for _, ch := range s.listeners {
select {
case ch <- event:
default:
}
}
}
func (s *staticResolver) Lookup(key string) (membership.HostInfo, error) {
s.mu.Lock()
defer s.mu.Unlock()
if len(s.hostInfos) == 0 {
return nil, membership.ErrInsufficientHosts
}
hash := int(s.hashfunc([]byte(key)))
idx := hash % len(s.hostInfos)
return s.hostInfos[idx], nil
}
func (s *staticResolver) LookupN(key string, _ int) []membership.HostInfo {
info, err := s.Lookup(key)
if err != nil {
return []membership.HostInfo{}
}
return []membership.HostInfo{info}
}
func (s *staticResolver) AddListener(name string, notifyChannel chan<- *membership.ChangedEvent) error {
s.mu.Lock()
defer s.mu.Unlock()
_, ok := s.listeners[name]
if ok {
return membership.ErrListenerAlreadyExist
}
s.listeners[name] = notifyChannel
return nil
}
func (s *staticResolver) RemoveListener(name string) error {
s.mu.Lock()
defer s.mu.Unlock()
_, ok := s.listeners[name]
if !ok {
return nil
}
delete(s.listeners, name)
return nil
}
func (s *staticResolver) MemberCount() int {
s.mu.Lock()
defer s.mu.Unlock()
return len(s.hostInfos)
}
func (s *staticResolver) AvailableMemberCount() int {
s.mu.Lock()
defer s.mu.Unlock()
return len(s.hostInfos)
}
func (s *staticResolver) Members() []membership.HostInfo {
s.mu.Lock()
defer s.mu.Unlock()
return s.hostInfos
}
func (s *staticResolver) AvailableMembers() []membership.HostInfo {
return s.Members()
}
func (s *staticResolver) RequestRefresh() {
}