整理代码

Author: pangdogs

ec/ictx/context.go

@@ -47,8 +47,7 @@ type Context interface {
 
 type iContext interface {
 	init(parentCtx context.Context, autoRecover bool, reportError chan error)
-	setPaired(v bool) bool
-	getPaired() bool
+	getPaired() *atomic.Bool
 	returnTerminated()
 }
 
@@ -107,12 +106,8 @@ func (ctx *ContextBehavior) init(parentCtx context.Context, autoRecover bool, re
 	ctx.terminated = async.MakeAsyncRet()
 }
 
-func (ctx *ContextBehavior) setPaired(v bool) bool {
-	return ctx.paired.CompareAndSwap(!v, v)
-}
-
-func (ctx *ContextBehavior) getPaired() bool {
-	return ctx.paired.Load()
+func (ctx *ContextBehavior) getPaired() *atomic.Bool {
+	return &ctx.paired
 }
 
 func (ctx *ContextBehavior) returnTerminated() {

ec/ictx/unsafe_context.go

@@ -21,6 +21,7 @@ package ictx
 
 import (
 	"context"
+	"sync/atomic"
 )
 
 // Deprecated: UnsafeContext 访问上下文内部方法
@@ -39,13 +40,8 @@ func (u _UnsafeContext) Init(parentCtx context.Context, autoRecover bool, report
 	u.init(parentCtx, autoRecover, reportError)
 }
 
-// SetPaired 设置配对标记
-func (u _UnsafeContext) SetPaired(v bool) bool {
-	return u.setPaired(v)
-}
-
 // GetPaired 获取配对标记
-func (u _UnsafeContext) GetPaired() bool {
+func (u _UnsafeContext) GetPaired() *atomic.Bool {
 	return u.getPaired()
 }
 

runtime.go

@@ -158,7 +158,7 @@ func (rt *RuntimeBehavior) init(rtCtx runtime.Context, options RuntimeOptions) {
 		exception.Panicf("%w: %w: rtCtx is nil", ErrRuntime, ErrArgs)
 	}
 
-	if !ictx.UnsafeContext(rtCtx).SetPaired(true) {
+	if !ictx.UnsafeContext(rtCtx).GetPaired().CompareAndSwap(false, true) {
 		exception.Panicf("%w: context already paired", ErrRuntime)
 	}
 

service.go

@@ -91,7 +91,7 @@ func (svc *ServiceBehavior) init(svcCtx service.Context, options ServiceOptions)
 		exception.Panicf("%w: %w: svcCtx is nil", ErrService, ErrArgs)
 	}
 
-	if !ictx.UnsafeContext(svcCtx).SetPaired(true) {
+	if !ictx.UnsafeContext(svcCtx).GetPaired().CompareAndSwap(false, true) {
 		exception.Panicf("%w: context already paired", ErrService)
 	}