generic and specific avro examples (#1381)

* generic and specific avro examples
moved protobuf to a similiar folder

* removed empty __init__.py

Author: emasab

examples/Makefile

@@ -1,5 +1,5 @@
-user_pb2.py: user.proto
-	protoc -I=. --python_out=. ./user.proto;
+user_pb2.py: protobuf/user.proto
+	cd protobuf && protoc -I=. --python_out=. ./user.proto;
 
 clean:
-	rm -f $(TARGET_DIR)/*_pb2.py
+	rm -f $(TARGET_DIR)/protobuf/*_pb2.py

examples/avro-cli.py

@@ -16,25 +16,17 @@
 #
 
 import argparse
+import os
 from uuid import uuid4
 
 from six.moves import input
 
 from confluent_kafka import avro
 
 # Parse Schema used for serializing User class
-record_schema = avro.loads("""
-    {
-        "namespace": "confluent.io.examples.serialization.avro",
-        "name": "User",
-        "type": "record",
-        "fields": [
-            {"name": "name", "type": "string"},
-            {"name": "favorite_number", "type": "int"},
-            {"name": "favorite_color", "type": "string"}
-        ]
-    }
-""")
+path = os.path.realpath(os.path.dirname(__file__))
+with open(f"{path}/avro/user_specific.avsc") as f:
+    record_schema = avro.loads(f.read())
 
 
 class User(object):

examples/avro/user_generic.avsc

@@ -0,0 +1,18 @@
+{
+    "name": "User",
+    "type": "record",
+    "fields": [
+        {
+            "name": "name",
+            "type": "string"
+        },
+        {
+            "name": "favorite_number",
+            "type": "long"
+        },
+        {
+            "name": "favorite_color",
+            "type": "string"
+        }
+    ]
+}

examples/avro/user_specific.avsc

@@ -0,0 +1,19 @@
+{
+    "namespace": "confluent.io.examples.serialization.avro",
+    "name": "User",
+    "type": "record",
+    "fields": [
+        {
+            "name": "name",
+            "type": "string"
+        },
+        {
+            "name": "favorite_number",
+            "type": "long"
+        },
+        {
+            "name": "favorite_color",
+            "type": "string"
+        }
+    ]
+}

examples/avro_consumer.py

@@ -20,6 +20,7 @@
 # This is a simple example of the SerializingProducer using Avro.
 #
 import argparse
+import os
 
 from confluent_kafka import DeserializingConsumer
 from confluent_kafka.schema_registry import SchemaRegistryClient
@@ -66,19 +67,16 @@ def dict_to_user(obj, ctx):
 
 def main(args):
     topic = args.topic
+    is_specific = args.specific == "true"
 
-    schema_str = """
-    {
-        "namespace": "confluent.io.examples.serialization.avro",
-        "name": "User",
-        "type": "record",
-        "fields": [
-            {"name": "name", "type": "string"},
-            {"name": "favorite_number", "type": "int"},
-            {"name": "favorite_color", "type": "string"}
-        ]
-    }
-    """
+    if is_specific:
+        schema = "user_specific.avsc"
+    else:
+        schema = "user_generic.avsc"
+
+    path = os.path.realpath(os.path.dirname(__file__))
+    with open(f"{path}/avro/{schema}") as f:
+        schema_str = f.read()
 
     sr_conf = {'url': args.schema_registry}
     schema_registry_client = SchemaRegistryClient(sr_conf)
@@ -110,8 +108,8 @@ def main(args):
                       "\tfavorite_number: {}\n"
                       "\tfavorite_color: {}\n"
                       .format(msg.key(), user.name,
-                              user.favorite_color,
-                              user.favorite_number))
+                              user.favorite_number,
+                              user.favorite_color))
         except KeyboardInterrupt:
             break
 
@@ -129,5 +127,7 @@ def main(args):
                         help="Topic name")
     parser.add_argument('-g', dest="group", default="example_serde_avro",
                         help="Consumer group")
+    parser.add_argument('-p', dest="specific", default="true",
+                        help="Avro specific record")
 
     main(parser.parse_args())

examples/avro_producer.py

@@ -20,6 +20,7 @@
 # This is a simple example of the SerializingProducer using Avro.
 #
 import argparse
+import os
 from uuid import uuid4
 
 from six.moves import input
@@ -99,19 +100,17 @@ def delivery_report(err, msg):
 
 def main(args):
     topic = args.topic
+    is_specific = args.specific == "true"
+
+    if is_specific:
+        schema = "user_specific.avsc"
+    else:
+        schema = "user_generic.avsc"
+
+    path = os.path.realpath(os.path.dirname(__file__))
+    with open(f"{path}/avro/{schema}") as f:
+        schema_str = f.read()
 
-    schema_str = """
-    {
-        "namespace": "confluent.io.examples.serialization.avro",
-        "name": "User",
-        "type": "record",
-        "fields": [
-            {"name": "name", "type": "string"},
-            {"name": "favorite_number", "type": "int"},
-            {"name": "favorite_color", "type": "string"}
-        ]
-    }
-    """
     schema_registry_conf = {'url': args.schema_registry}
     schema_registry_client = SchemaRegistryClient(schema_registry_conf)
 
@@ -158,5 +157,7 @@ def main(args):
                         help="Schema Registry (http(s)://host[:port]")
     parser.add_argument('-t', dest="topic", default="example_serde_avro",
                         help="Topic name")
+    parser.add_argument('-p', dest="specific", default="true",
+                        help="Avro specific record")
 
     main(parser.parse_args())

examples/json_consumer.py

@@ -116,8 +116,8 @@ def main(args):
                       "\tfavorite_number: {}\n"
                       "\tfavorite_color: {}\n"
                       .format(msg.key(), user.name,
-                              user.favorite_color,
-                              user.favorite_number))
+                              user.favorite_number,
+                              user.favorite_color))
         except KeyboardInterrupt:
             break
 

examples/user.proto renamed to examples/protobuf/user.proto

@@ -31,8 +31,8 @@
 #
 import argparse
 
-# Protobuf generated class; resides at ./user_pb2.py
-import user_pb2
+# Protobuf generated class; resides at ./protobuf/user_pb2.py
+import protobuf.user_pb2 as user_pb2
 from confluent_kafka import DeserializingConsumer
 from confluent_kafka.schema_registry.protobuf import ProtobufDeserializer
 from confluent_kafka.serialization import StringDeserializer