mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
118 lines
2.5 KiB
Go
118 lines
2.5 KiB
Go
package static
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/dgryski/go-farm"
|
|
"github.com/hanzoai/tasks/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() {
|
|
}
|