dan9i10/xray-core
1
0
Fork 0
mirror of https://github.com/XTLS/Xray-core.git synced 2026-05-08 14:13:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimize with double-checked locking for better concurrency

Browse Source 
Co-authored-by: RPRX <63339210+RPRX@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-01-09 10:35:22 +00:00
parent 385867e82b
commit 52f7f3d174
1 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
proxy/wireguard/server.go
View File

@@ -83,18 +83,26 @@ func (*Server) Network() []net.Network {
// Process implements proxy.Inbound.
func (s *Server) Process(ctx context.Context, network net.Network, conn stat.Connection, dispatcher routing.Dispatcher) error {
// Use RWMutex to safely handle concurrent access to routing info
// Use double-checked locking to safely handle concurrent access to routing info
// Only update if not set or if dispatcher is different
s.infoMutex.Lock()
if s.info.dispatcher == nil || s.info.dispatcher != dispatcher {
s.info = routingInfo{
ctx: ctx,
dispatcher: dispatcher,
inboundTag: session.InboundFromContext(ctx),
contentTag: session.ContentFromContext(ctx),
// First check without write lock for better concurrency
s.infoMutex.RLock()
needsUpdate := s.info.dispatcher == nil || s.info.dispatcher != dispatcher
s.infoMutex.RUnlock()
if needsUpdate {
s.infoMutex.Lock()
// Double-check after acquiring write lock
if s.info.dispatcher == nil || s.info.dispatcher != dispatcher {
s.info = routingInfo{
ctx: ctx,
dispatcher: dispatcher,
inboundTag: session.InboundFromContext(ctx),
contentTag: session.ContentFromContext(ctx),
}
}
s.infoMutex.Unlock()
}
s.infoMutex.Unlock()
ep, err := s.bindServer.ParseEndpoint(conn.RemoteAddr().String())
if err != nil {