From 1fd0ddb6c14b08e2313ebe5152115406f19aa773 Mon Sep 17 00:00:00 2001
From: Patrick <dreamlike.sky@foxmail.com>
Date: Tue, 10 Mar 2020 15:05:50 +0800
Subject: [PATCH] modify rest
---
protocol/rest/rest_protocol.go | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/protocol/rest/rest_protocol.go b/protocol/rest/rest_protocol.go
index f4f12e415..fe147e25a 100644
--- a/protocol/rest/rest_protocol.go
+++ b/protocol/rest/rest_protocol.go
@@ -55,8 +55,8 @@ type RestProtocol struct {
func NewRestProtocol() *RestProtocol {
return &RestProtocol{
BaseProtocol: protocol.NewBaseProtocol(),
- serverMap: make(map[string]rest_interface.RestServer),
- clientMap: make(map[rest_interface.RestOptions]rest_interface.RestClient),
+ serverMap: make(map[string]rest_interface.RestServer, 8),
+ clientMap: make(map[rest_interface.RestOptions]rest_interface.RestClient, 8),
}
}
@@ -105,13 +105,13 @@ func (rp *RestProtocol) getServer(url common.URL, serverType string) rest_interf
panic("[RestProtocol]" + url.ServiceKey() + "is not existing")
}
rp.serverLock.Lock()
+ defer rp.serverLock.Unlock()
restServer, ok = rp.serverMap[url.Location]
if !ok {
restServer = extension.GetNewRestServer(serverType)
restServer.Start(url)
rp.serverMap[url.Location] = restServer
}
- rp.serverLock.Unlock()
return restServer
}
@@ -121,12 +121,12 @@ func (rp *RestProtocol) getClient(restOptions rest_interface.RestOptions, client
return restClient
}
rp.clientLock.Lock()
+ defer rp.clientLock.Unlock()
restClient, ok = rp.clientMap[restOptions]
if !ok {
restClient = extension.GetNewRestClient(clientType, &restOptions)
rp.clientMap[restOptions] = restClient
}
- rp.clientLock.Unlock()
return restClient
}
--
GitLab