Skip to content

Commit ce8991c

Browse files
committed
Made fake server into gevent server
1 parent 2e355cf commit ce8991c

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

fake_server/fake_server.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,13 @@
88
paramiko repository
99
"""
1010

11-
import multiprocessing
11+
# import multiprocessing
12+
import gevent
13+
from gevent import monkey
14+
monkey.patch_all()
1215
import os
1316
import socket
14-
# import threading
15-
from threading import Event
17+
from gevent.event import Event
1618
import sys
1719
import traceback
1820
import logging
@@ -91,10 +93,10 @@ def listen(cmd_req_response, sock, fail_auth = False):
9193
logger.error('*** Listen failed: %s' % (str(e),))
9294
traceback.print_exc()
9395
return
94-
accept_thread = multiprocessing.Process(target=handle_ssh_connection,
95-
args=(cmd_req_response, sock,),
96-
kwargs={'fail_auth' : fail_auth},)
97-
accept_thread.start()
96+
accept_thread = gevent.spawn(handle_ssh_connection,
97+
cmd_req_response, sock,
98+
fail_auth=fail_auth)
99+
# accept_thread.start()
98100
return accept_thread
99101

100102
def _handle_ssh_connection(cmd_req_response, t, client, addr, fail_auth = False):
@@ -148,7 +150,7 @@ def handle_ssh_connection(cmd_req_response, sock, fail_auth = False):
148150
sock = make_socket('127.0.0.1')
149151
server = listen({'fake' : 'fake response' + os.linesep}, sock)
150152
try:
151-
server.join()
153+
server.get()
152154
except KeyboardInterrupt:
153155
sys.exit(0)
154156

0 commit comments

Comments
 (0)