File tree 1 file changed +3
-2
lines changed
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -68,9 +68,10 @@ class SocketDB(BaseDB):
68
68
Database that act as a server that expose a socket where data source will push data
69
69
"""
70
70
71
- def __init__ (self , report_type : Type [Report ], port : int ):
71
+ def __init__ (self , report_type : Type [Report ], host : str , port : int ):
72
72
BaseDB .__init__ (self , report_type , is_async = True )
73
73
self .queue = None
74
+ self .host = host
74
75
self .port = port
75
76
self .server = None
76
77
@@ -79,7 +80,7 @@ async def connect(self):
79
80
Connect to the socket database.
80
81
"""
81
82
self .queue = asyncio .Queue ()
82
- self .server = await asyncio .start_server (self ._gen_server_callback (), host = '127.0.0.1' , port = self .port )
83
+ self .server = await asyncio .start_server (self ._gen_server_callback (), host = self . host , port = self .port )
83
84
84
85
async def disconnect (self ):
85
86
"""
You can’t perform that action at this time.
0 commit comments