File tree Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -294,8 +294,17 @@ def on_message(self, *args):
294
294
if (isinstance (message , HiveMessage ) and message .msg_type == HiveMessageType .BINARY ):
295
295
self ._handle_binary (message )
296
296
return
297
- self .emitter .emit ('message' , message ) # raw message
298
- self ._handle_hive_protocol (HiveMessage (** message ))
297
+
298
+ if isinstance (message , HiveMessage ):
299
+ self .emitter .emit ('message' , message .serialize ()) # raw message
300
+ self ._handle_hive_protocol (message )
301
+ elif isinstance (message , str ):
302
+ self .emitter .emit ('message' , message ) # raw message
303
+ self ._handle_hive_protocol (HiveMessage (** json .loads (message )))
304
+ else :
305
+ assert isinstance (message , dict )
306
+ self .emitter .emit ('message' , json .dumps (message , ensure_ascii = False )) # raw message
307
+ self ._handle_hive_protocol (HiveMessage (** message ))
299
308
300
309
def _handle_binary (self , message : HiveMessage ):
301
310
assert message .msg_type == HiveMessageType .BINARY
You can’t perform that action at this time.
0 commit comments