def classify_images(type):
X_test, y_test = load_mnist('Data/', kind='t10k')
if type == "google":
message_broker = MessageBroker("google")
# Listening to print out classes
consumer = message_broker.generate_consumer(type='google', topic_name='classes', gcallback=callback)
# Sending images
publisher = message_broker.generate_producer(type='google')
for img in X_test:
print("Sending an image.")
message = img.tobytes()
message_broker.publish_message(type='google', producer_instance=publisher, topic_name='images', value=message)
time.sleep(10)
try:
consumer.result()
except KeyboardInterrupt:
consumer.cancel()
if type == 'kafka':
# Sending images
message_broker = MessageBroker(type='kafka')
for img in X_test:
print("Sending an image.")
producer = message_broker.generate_producer(type="kafka")
message_broker.publish_message(producer_instance=producer, topic_name="images", key='parsed',
value=img.tobytes(), type='kafka')
time.sleep(10)
def google_testtopic():
# Tests run on a test topic for simple communication testing of the adapter
#Consuming messages
message_broker = MessageBroker(type='google')
#consumer = message_broker.generate_consumer(type='google',topic_name='test',gcallback=callback)
#try:
# consumer.result()
#except KeyboardInterrupt:
# consumer.cancel()
#Message publishing
publisher = message_broker.generate_producer(type='google')
message_broker.publish_message(type='google',producer_instance=publisher,topic_name='test',value=b"test?")
