refactor concurrency: replace atomic package usage with sync/atomic for improved clarity and thread safety

Author: ehsannm

std/gateways/fasthttp/bundle.go

@@ -49,7 +49,7 @@ type bundle struct {
 	wsUpgrade     websocket.FastHTTPUpgrader
 	wsRoutes      map[string]*routeData
 	wsEndpoint    string
-	wsNextID      uint64
+	wsNextID      atomic.Uint64
 	predicateKey  string
 	rpcInFactory  kit.IncomingRPCFactory
 	rpcOutFactory kit.OutgoingRPCFactory
@@ -244,7 +244,7 @@ func (b *bundle) wsHandler(ctx *fasthttp.RequestCtx) {
 		func(conn *websocket.Conn) {
 			wsc := &wsConn{
 				kv:            map[string]string{},
-				id:            atomic.AddUint64(&b.wsNextID, 1),
+				id:            b.wsNextID.Add(1),
 				clientIP:      realip.FromRequest(ctx),
 				c:             conn,
 				rpcOutFactory: b.rpcOutFactory,

std/gateways/fastws/gateway.go

@@ -17,7 +17,7 @@ type gateway struct {
 	utils.SpinLock
 
 	b      *bundle
-	nextID uint64
+	nextID atomic.Uint64
 	conns  map[uint64]*wsConn
 }
 
@@ -51,7 +51,7 @@ func (gw *gateway) OnShutdown(_ gnet.Engine) {}
 
 func (gw *gateway) OnOpen(c gnet.Conn) (out []byte, action gnet.Action) {
 	wsc := newWebsocketConn(
-		atomic.AddUint64(&gw.nextID, 1),
+		gw.nextID.Add(1),
 		c,
 		gw.b.rpcOutFactory,
 		gw.b.writeMode,

stub/stub_ws.go

@@ -31,7 +31,7 @@ type WebsocketCtx struct {
 
 	pendingMtx     sync.Mutex
 	pending        map[string]chan kit.IncomingRPCContainer
-	lastActivity   uint32
+	lastActivity   atomic.Uint32
 	disconnectChan chan struct{}
 
 	// fasthttp entities
@@ -116,11 +116,11 @@ func (wCtx *WebsocketCtx) Reconnect(ctx context.Context) error {
 }
 
 func (wCtx *WebsocketCtx) setActivity() {
-	atomic.StoreUint32(&wCtx.lastActivity, uint32(utils.TimeUnix()))
+	wCtx.lastActivity.Store(uint32(utils.TimeUnix()))
 }
 
 func (wCtx *WebsocketCtx) getActivity() int64 {
-	return int64(atomic.LoadUint32(&wCtx.lastActivity))
+	return int64(wCtx.lastActivity.Load())
 }
 
 func (wCtx *WebsocketCtx) watchdog(c *websocket.Conn) {