mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +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`.
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package interceptor
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"go.temporal.io/server/common/headers"
|
|
"go.temporal.io/version/check"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type SDKVersionInterceptor struct {
|
|
sync.RWMutex
|
|
sdkInfoSet map[check.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[check.SDKInfo]struct{}),
|
|
versionChecker: headers.NewDefaultVersionChecker(),
|
|
maxSetSize: defaultMaxSetSize,
|
|
}
|
|
}
|
|
|
|
// Intercept a grpc request
|
|
func (vi *SDKVersionInterceptor) Intercept(
|
|
ctx context.Context,
|
|
req interface{},
|
|
info *grpc.UnaryServerInfo,
|
|
handler grpc.UnaryHandler,
|
|
) (interface{}, 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 := check.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() []check.SDKInfo {
|
|
vi.Lock()
|
|
currSet := vi.sdkInfoSet
|
|
vi.sdkInfoSet = make(map[check.SDKInfo]struct{})
|
|
vi.Unlock()
|
|
|
|
sdkInfo := make([]check.SDKInfo, 0, len(currSet))
|
|
for k := range currSet {
|
|
sdkInfo = append(sdkInfo, k)
|
|
}
|
|
return sdkInfo
|
|
}
|