Add possibility for UDP socket
Sven Mäder committed
Feb 13, 2018 at 20:38 UTC
ed4a9744ed19bc20bc829c9f41f759565b670722
1 file changed
+6
-1
python.d/python_modules/bases/FrameworkServices/SocketService.py
+6
-1
@@ -14,6 +14,7 @@ class SocketService(SimpleService):
14
self.host = 'localhost'
15
self.port = None
16
self.unix_socket = None
17
+ self.dgram_socket = False
18
self.request = ''
19
self.__socket_config = None
20
self.__empty_request = "".encode()
@@ -115,7 +116,11 @@ class SocketService(SimpleService):
116
if self.__socket_config is not None:
117
self._connect2socket()
118
else:
118
- for res in socket.getaddrinfo(self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM):
119
+ if self.dgram_socket:
120
+ sock_type = socket.SOCK_DGRAM
121
+ else:
122
+ sock_type = socket.SOCK_STREAM
123
+ for res in socket.getaddrinfo(self.host, self.port, socket.AF_UNSPEC, sock_type):
124
if self._connect2socket(res):
125
break
126