Skip to content

Commit df54d12

Browse files
committed
fixes requests memory leak
1 parent c3de02b commit df54d12

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

dongtai_agent_python/api/openapi.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ def __init__(self, setting):
2727

2828
self.agent_id = 0
2929
self.report_queue = 0
30-
self.session = requests.session()
3130

3231
self.headers = {
3332
'Authorization': 'Token ' + self.config.get('iast', {}).get('server', {}).get('token', ''),
@@ -47,9 +46,10 @@ def get(self, url, params=None):
4746
full_url = self.base_url + url
4847

4948
try:
50-
res = self.session.get(full_url, timeout=20, headers=self.headers, params=params)
49+
res = requests.get(full_url, timeout=20, headers=self.headers, params=params)
5150
resp = bytes.decode(res.content, 'utf-8')
5251
resp = json.loads(resp)
52+
res.close()
5353
except Exception as e:
5454
logger.error("get " + url + " failed: " + str(e) + "\n" + traceback.format_exc())
5555
resp = {}
@@ -62,9 +62,10 @@ def post(self, url, body):
6262

6363
try:
6464
body_data = json.dumps(body)
65-
res = self.session.post(full_url, timeout=20, headers=self.headers, data=body_data)
65+
res = requests.post(full_url, timeout=20, headers=self.headers, data=body_data)
6666
resp = bytes.decode(res.content, 'utf-8')
6767
resp = json.loads(resp)
68+
res.close()
6869
except Exception as e:
6970
logger.error("post " + url + " failed: " + str(e) + "\n" + traceback.format_exc())
7071
resp = {}
@@ -79,10 +80,11 @@ def report(self, url, body):
7980

8081
body_data = gzip.compress(stream_data.encode('utf-8'))
8182
try:
82-
res = self.session.post(api_url, data=body_data, timeout=20, headers=self.headers)
83+
res = requests.post(api_url, data=body_data, timeout=20, headers=self.headers)
8384
logger.debug(res.content)
8485
resp = bytes.decode(res.content, 'utf-8')
8586
resp = json.loads(resp)
87+
res.close()
8688
except Exception as e:
8789
logger.error("report failed: " + str(e) + "\n" + traceback.format_exc())
8890
resp = {}

0 commit comments

Comments
 (0)