Merge pull request #35 from chris-hatton/std-function-callback-support

Add support for using std::function callbacks instead of C-pointers

Author: aentinger

Diff for: src/MqttClient.cpp

@@ -105,7 +105,11 @@ MqttClient::~MqttClient()
   }
 }
 
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+void MqttClient::onMessage(MessageCallback callback)
+#else
 void MqttClient::onMessage(void(*callback)(int))
+#endif
 {
   _onMessage = callback;
 }
@@ -562,7 +566,11 @@ void MqttClient::poll()
             _rxState = MQTT_CLIENT_RX_STATE_READ_PUBLISH_PAYLOAD;
 
             if (_onMessage) {
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+              _onMessage(this,_rxLength);
+#else
               _onMessage(_rxLength);
+#endif
 
               if (_rxLength == 0) {
                 _rxState = MQTT_CLIENT_RX_STATE_READ_TYPE;
@@ -585,7 +593,11 @@ void MqttClient::poll()
           _rxState = MQTT_CLIENT_RX_STATE_READ_PUBLISH_PAYLOAD;
 
           if (_onMessage) {
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+            _onMessage(this,_rxLength);
+#else
             _onMessage(_rxLength);
+#endif
           }
 
           if (_rxLength == 0) {

Diff for: src/MqttClient.h

@@ -32,17 +32,26 @@
 #define MQTT_BAD_USER_NAME_OR_PASSWORD      4
 #define MQTT_NOT_AUTHORIZED                 5
 
+// Make this definition in your application code to use std::functions for onMessage callbacks instead of C-pointers:
+// #define MQTT_CLIENT_STD_FUNCTION_CALLBACK
+
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+#include <functional>
+#endif
+
 class MqttClient : public Client {
 public:
   MqttClient(Client* client);
   MqttClient(Client& client);
   virtual ~MqttClient();
 
-
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+  typedef std::function<void(MqttClient *client, int messageSize)> MessageCallback;
+  void onMessage(MessageCallback callback);
+#else
   inline void setClient(Client& client) { _client = &client; }
-
-
   void onMessage(void(*)(int));
+#endif
 
   int parseMessage();
   String messageTopic() const;
@@ -134,7 +143,11 @@ class MqttClient : public Client {
 private:
   Client* _client;
 
+#ifdef MQTT_CLIENT_STD_FUNCTION_CALLBACK
+  MessageCallback _onMessage;
+#else
   void (*_onMessage)(int);
+#endif
 
   String _id;
   String _username;