get data

99e8021d · 赵威 · dabe6dd9 · 99e8021d
Commit 99e8021d authored Aug 07, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

online_prediction.py src/online_prediction.py +3 -3

No files found.
--- a/src/online_prediction.py
+++ b/src/online_prediction.py
 from kafka import KafkaConsumer
+import json


 def consume_kafka():
@@ -9,10 +10,9 @@ def consume_kafka():

    consumer.subscribe(["gm-portrait-update-device"])

-    message = consumer.consume_data()
-
-    for msg in message:
+    for msg in consumer:
        print(msg.value)
+        print(json.loads(msg.value), "\n")


 if __name__ == "__main__":