Skip to content

Commit 1c57833

Browse files
committed
feat(database/socket): Add host attribute to set the socket listen address
1 parent c0b1950 commit 1c57833

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/powerapi/database/socket_db.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,10 @@ class SocketDB(BaseDB):
6868
Database that act as a server that expose a socket where data source will push data
6969
"""
7070

71-
def __init__(self, report_type: Type[Report], port: int):
71+
def __init__(self, report_type: Type[Report], host: str, port: int):
7272
BaseDB.__init__(self, report_type, is_async=True)
7373
self.queue = None
74+
self.host = host
7475
self.port = port
7576
self.server = None
7677

@@ -79,7 +80,7 @@ async def connect(self):
7980
Connect to the socket database.
8081
"""
8182
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)
8384

8485
async def disconnect(self):
8586
"""

0 commit comments

Comments
 (0)