json: conditionaly use orjson or json

maki-nage · Sep 21, 2023 · cad0e67 · cad0e67
1 parent 5f7cda2
commit cad0e67
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/rxsci/container/json.py b/rxsci/container/json.py
@@ -1,6 +1,10 @@
 import typing
-import orjson as json
-
+serialization_as_string = True
+try:
+    import orjson as json
+    serialization_as_string = False
+except Exception:
+    import json
 import rx
 import rx.operators as ops
 import rxsci.io.file as file
@@ -90,7 +94,9 @@ def _dump(source):
         def on_subscribe(observer, scheduler):
 
             def on_next(i):
-                line = json.dumps(i).decode()
+                line = json.dumps(i)
+                if serialization_as_string is False:
+                    line = line.decode()
                 line += newline
                 observer.on_next(line)