mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## 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`.
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination connections_mock.go
|
|
|
|
package history
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"go.temporal.io/server/api/historyservice/v1"
|
|
"go.temporal.io/server/common/membership"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type (
|
|
clientConnection struct {
|
|
historyClient historyservice.HistoryServiceClient
|
|
grpcConn *grpc.ClientConn
|
|
}
|
|
|
|
rpcAddress string
|
|
|
|
connectionPoolImpl struct {
|
|
mu struct {
|
|
sync.RWMutex
|
|
conns map[rpcAddress]clientConnection
|
|
}
|
|
|
|
historyServiceResolver membership.ServiceResolver
|
|
rpcFactory RPCFactory
|
|
}
|
|
|
|
// RPCFactory is a subset of the [go.temporal.io/server/common/rpc.RPCFactory] interface to make testing easier.
|
|
RPCFactory interface {
|
|
CreateHistoryGRPCConnection(rpcAddress string) *grpc.ClientConn
|
|
}
|
|
|
|
connectionPool interface {
|
|
getOrCreateClientConn(addr rpcAddress) clientConnection
|
|
getAllClientConns() []clientConnection
|
|
resetConnectBackoff(clientConnection)
|
|
}
|
|
)
|
|
|
|
func newConnectionPool(
|
|
historyServiceResolver membership.ServiceResolver,
|
|
rpcFactory RPCFactory,
|
|
) *connectionPoolImpl {
|
|
c := &connectionPoolImpl{
|
|
historyServiceResolver: historyServiceResolver,
|
|
rpcFactory: rpcFactory,
|
|
}
|
|
c.mu.conns = make(map[rpcAddress]clientConnection)
|
|
return c
|
|
}
|
|
|
|
func (c *connectionPoolImpl) getOrCreateClientConn(addr rpcAddress) clientConnection {
|
|
c.mu.RLock()
|
|
cc, ok := c.mu.conns[addr]
|
|
c.mu.RUnlock()
|
|
if ok {
|
|
return cc
|
|
}
|
|
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
|
|
if cc, ok = c.mu.conns[addr]; ok {
|
|
return cc
|
|
}
|
|
grpcConn := c.rpcFactory.CreateHistoryGRPCConnection(string(addr))
|
|
cc = clientConnection{
|
|
historyClient: historyservice.NewHistoryServiceClient(grpcConn),
|
|
grpcConn: grpcConn,
|
|
}
|
|
|
|
c.mu.conns[addr] = cc
|
|
return cc
|
|
}
|
|
|
|
func (c *connectionPoolImpl) getAllClientConns() []clientConnection {
|
|
hostInfos := c.historyServiceResolver.Members()
|
|
|
|
var clientConns []clientConnection
|
|
for _, hostInfo := range hostInfos {
|
|
cc := c.getOrCreateClientConn(rpcAddress(hostInfo.GetAddress()))
|
|
clientConns = append(clientConns, cc)
|
|
}
|
|
|
|
return clientConns
|
|
}
|
|
|
|
func (c *connectionPoolImpl) resetConnectBackoff(cc clientConnection) {
|
|
cc.grpcConn.ResetConnectBackoff()
|
|
}
|