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).
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package interceptor
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/hanzoai/tasks/common/headers"
|
|
"github.com/hanzoai/tasks/common/versioninfo"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type SDKVersionInterceptor struct {
|
|
sync.RWMutex
|
|
sdkInfoSet map[versioninfo.SDKInfo]struct{}
|
|
versionChecker headers.VersionChecker
|
|
maxSetSize int
|
|
}
|
|
|
|
const defaultMaxSetSize = 100
|
|
|
|
// NewSDKVersionInterceptor creates a new SDKVersionInterceptor with default max set size
|
|
func NewSDKVersionInterceptor() *SDKVersionInterceptor {
|
|
return &SDKVersionInterceptor{
|
|
sdkInfoSet: make(map[versioninfo.SDKInfo]struct{}),
|
|
versionChecker: headers.NewDefaultVersionChecker(),
|
|
maxSetSize: defaultMaxSetSize,
|
|
}
|
|
}
|
|
|
|
// Intercept a grpc request
|
|
func (vi *SDKVersionInterceptor) Intercept(
|
|
ctx context.Context,
|
|
req any,
|
|
info *grpc.UnaryServerInfo,
|
|
handler grpc.UnaryHandler,
|
|
) (any, error) {
|
|
sdkName, sdkVersion := headers.GetClientNameAndVersion(ctx)
|
|
if sdkName != "" && sdkVersion != "" {
|
|
vi.RecordSDKInfo(sdkName, sdkVersion)
|
|
if err := vi.versionChecker.ClientSupported(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return handler(ctx, req)
|
|
}
|
|
|
|
// RecordSDKInfo records name and version tuple in memory
|
|
func (vi *SDKVersionInterceptor) RecordSDKInfo(name, version string) {
|
|
info := versioninfo.SDKInfo{Name: name, Version: version}
|
|
|
|
vi.RLock()
|
|
overCap := len(vi.sdkInfoSet) >= vi.maxSetSize
|
|
_, found := vi.sdkInfoSet[info]
|
|
vi.RUnlock()
|
|
|
|
if !overCap && !found {
|
|
vi.Lock()
|
|
vi.sdkInfoSet[info] = struct{}{}
|
|
vi.Unlock()
|
|
}
|
|
}
|
|
|
|
// GetAndResetSDKInfo gets all recorded name, version tuples and resets internal records
|
|
func (vi *SDKVersionInterceptor) GetAndResetSDKInfo() []versioninfo.SDKInfo {
|
|
vi.Lock()
|
|
currSet := vi.sdkInfoSet
|
|
vi.sdkInfoSet = make(map[versioninfo.SDKInfo]struct{})
|
|
vi.Unlock()
|
|
|
|
sdkInfo := make([]versioninfo.SDKInfo, 0, len(currSet))
|
|
for k := range currSet {
|
|
sdkInfo = append(sdkInfo, k)
|
|
}
|
|
return sdkInfo
|
|
}
|