-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
43 lines (38 loc) · 1.42 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import logging
import argparse
import json
from server.thread_per_client_server import ThreadPerClient
from server.thread_per_request_server import ThreadPerRequest
from server.purely_sync_server import PurelySync
from utils.general_utils import settings_analyzer,settings_preparer
from settings import settings_map
FORMAT = "%(asctime)s %(levelname)s %(name)s %(funcName)s %(message)s"
logging.basicConfig(datefmt='%H:%M:%S',level=logging.DEBUG,format=FORMAT)
parser = argparse.ArgumentParser()
parser.add_argument('--settings','-s',type=int)
parser.add_argument('--type','-t',type=str)
parser.add_argument('--port','-p',type=int,default=9999)
args = parser.parse_args()
def main() -> None:
type_to_server_mapping = {
'1':ThreadPerClient,
'2':ThreadPerRequest,
'3':PurelySync,
'ThreadPerClient':ThreadPerClient,
'ThreadPerRequest':ThreadPerRequest,
'PurelySync':PurelySync,
'tpc':ThreadPerClient,
'tpr':ThreadPerRequest,
'ps':PurelySync
}
settings = settings_analyzer(settings_preparer(settings_map[args.settings]))
server_impl = type_to_server_mapping[args.type]
print(json.dumps(settings,default=str,sort_keys=True, indent=2))
server = server_impl(settings,port=args.port)
try:
server.start_loop()
except KeyboardInterrupt:
print("\nStopping server :(")
server.stop_loop()
if __name__ == "__main__":
main()