chore: import upstream snapshot with attribution

This commit is contained in:
wehub-resource-sync
2026-07-13 12:08:39 +08:00
commit a0df89c693
5252 changed files with 523444 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
package logoutcontext
import (
"context"
)
const LogoutPrefix = "logout-"
func GetContext(token string) context.Context {
return GetService(logoutToken(token)).GetLogoutCtx()
}
func Cancel(token string) {
GetService(logoutToken(token)).Cancel()
RemoveService(logoutToken(token))
}
func logoutToken(token string) string {
return LogoutPrefix + token
}
+106
View File
@@ -0,0 +1,106 @@
package logoutcontext
import (
"context"
"sync"
"testing"
"github.com/stretchr/testify/require"
)
func TestGetContext_ReturnsActiveContext(t *testing.T) {
t.Parallel()
token := "token-get-context-active"
defer Cancel(token)
ctx := GetContext(token)
require.NoError(t, ctx.Err())
}
func TestCancel_CancelsContext(t *testing.T) {
t.Parallel()
token := "token-cancel"
ctx := GetContext(token)
require.NoError(t, ctx.Err())
Cancel(token)
require.ErrorIs(t, ctx.Err(), context.Canceled)
}
func TestCancel_RemovesService(t *testing.T) {
t.Parallel()
token := "token-cancel-removes"
first := GetContext(token)
Cancel(token)
second := GetContext(token)
defer Cancel(token)
require.ErrorIs(t, first.Err(), context.Canceled)
require.NoError(t, second.Err())
require.NotEqual(t, first, second)
}
func TestGetService_ReturnsSameServiceForSameToken(t *testing.T) {
t.Parallel()
token := logoutToken("token-same-service")
defer RemoveService(token)
s1 := GetService(token)
s2 := GetService(token)
require.Same(t, s1, s2)
}
func TestGetService_ReturnsDistinctServicesForDifferentTokens(t *testing.T) {
t.Parallel()
tokenA := logoutToken("token-distinct-a")
tokenB := logoutToken("token-distinct-b")
defer RemoveService(tokenA)
defer RemoveService(tokenB)
sA := GetService(tokenA)
sB := GetService(tokenB)
require.NotSame(t, sA, sB)
}
func TestGetService_ConcurrentAccess(t *testing.T) {
t.Parallel()
const goroutines = 50
token := logoutToken("token-concurrent")
defer RemoveService(token)
var wg sync.WaitGroup
services := make([]*Service, goroutines)
for i := range goroutines {
wg.Add(1)
go func(i int) {
defer wg.Done()
services[i] = GetService(token)
}(i)
}
wg.Wait()
for i := 1; i < goroutines; i++ {
require.Same(t, services[0], services[i])
}
}
func TestLogoutToken_AddsPrefix(t *testing.T) {
t.Parallel()
result := logoutToken("abc123")
require.Equal(t, LogoutPrefix+"abc123", result)
}
+28
View File
@@ -0,0 +1,28 @@
package logoutcontext
import (
"context"
)
type (
Service struct {
ctx context.Context
cancel context.CancelFunc
}
)
func NewService() *Service {
ctx, cancel := context.WithCancel(context.Background())
return &Service{
ctx: ctx,
cancel: cancel,
}
}
func (s *Service) Cancel() {
s.cancel()
}
func (s *Service) GetLogoutCtx() context.Context {
return s.ctx
}
+34
View File
@@ -0,0 +1,34 @@
package logoutcontext
import "sync"
type (
ServiceFactory struct {
mu sync.Mutex
services map[string]*Service
}
)
var serviceFactory = ServiceFactory{
services: make(map[string]*Service),
}
func GetService(token string) *Service {
serviceFactory.mu.Lock()
defer serviceFactory.mu.Unlock()
service, ok := serviceFactory.services[token]
if !ok {
service = NewService()
serviceFactory.services[token] = service
}
return service
}
func RemoveService(token string) {
serviceFactory.mu.Lock()
defer serviceFactory.mu.Unlock()
delete(serviceFactory.services, token)
}