no message

tomlee0201 · Oct 6, 2017 · fd806f3 · fd806f3
1 parent 092e608
commit fd806f3
Show file tree

Hide file tree

Showing 7 changed files with 30 additions and 4 deletions.
diff --git a/MQTTDemo/IMLib/MQTTClient/NetworkService.mm b/MQTTDemo/IMLib/MQTTClient/NetworkService.mm
@@ -150,6 +150,7 @@ - (void) createMars {
   mars::app::SetCallback(mars::app::AppCallBack::Instance());
   mars::stn::setConnectionStatusCallback(new CSCB(self));
   mars::stn::setReceiveMessageCallback(new RPCB(self));
+
   mars::baseevent::OnCreate();
 }
 

diff --git a/MQTTDemo/IMLib/mars.framework/Headers/stn/stn.h b/MQTTDemo/IMLib/mars.framework/Headers/stn/stn.h
@@ -536,6 +536,10 @@ extern void (*ReportDnsProfile)(const DnsProfile& _dns_profile);
 
       extern void setConnectionStatusCallback(ConnectionStatusCallback *callback);
       extern void setReceiveMessageCallback(ReceiveMessageCallback *callback);
+
+        extern void setRefreshUserInfoCallback(GetUserInfoCallback *callback);
+        extern void setRefreshGroupInfoCallback(GetGroupInfoCallback *callback);
+
       extern ConnectionStatus getConnectionStatus();
 
 extern int (*sendMessage)(TMessage &tmsg, SendMessageCallback *callback);

diff --git a/mars/stn/mqtt/MessageDB.cpp b/mars/stn/mqtt/MessageDB.cpp
@@ -611,13 +611,12 @@ namespace mars {
 
         class TGetGroupInfoCallback : public GetGroupInfoCallback {
             void onSuccess(std::list<TGroupInfo> groupInfoList) {
-                if(StnCallBack::Instance()->m_getGroupInfoCB) {
-                    StnCallBack::Instance()->m_getGroupInfoCB->onSuccess(groupInfoList);
-                }
-
                 for (std::list<TGroupInfo>::iterator it = groupInfoList.begin(); it != groupInfoList.end(); it++) {
                     MessageDB::Instance()->InsertGroupInfo(*it);
                 }
+                if(StnCallBack::Instance()->m_getGroupInfoCB) {
+                    StnCallBack::Instance()->m_getGroupInfoCB->onSuccess(groupInfoList);
+                }
                 delete this;
             }
             void onFalure(int errorCode) {

diff --git a/mars/stn/mqtt/stn_callback.cc b/mars/stn/mqtt/stn_callback.cc
@@ -39,6 +39,15 @@ namespace mars {
         StnCallBack::Instance()->setReceiveMessageCallback(callback);
       }
 
+        void setRefreshUserInfoCallback(GetUserInfoCallback *callback) {
+            StnCallBack::Instance()->setGetUserInfoCallback(callback);
+        }
+
+        void setRefreshGroupInfoCallback(GetGroupInfoCallback *callback) {
+            StnCallBack::Instance()->setGetGroupInfoCallback(callback);
+        }
+
+
       ConnectionStatus getConnectionStatus() {
         return StnCallBack::Instance()->getConnectionStatus();
       }

diff --git a/mars/stn/mqtt/stn_callback.h b/mars/stn/mqtt/stn_callback.h
@@ -36,6 +36,7 @@ namespace mars {
 
 
         class TGetGroupInfoCallback;
+        class RefreshUserInfoCallback;
 class StnCallBack : public Callback,  PullingMessageCallback {
 
 private:
@@ -88,6 +89,7 @@ class StnCallBack : public Callback,  PullingMessageCallback {
     void onDBOpened();
 
     friend class TGetGroupInfoCallback;
+    friend class RefreshUserInfoCallback;
 private:
     static StnCallBack* instance_;
     void PullMessage(int64_t head);

diff --git a/mars/stn/stn.cc b/mars/stn/stn.cc
@@ -134,9 +134,16 @@ MQTTTask::MQTTTask(MQTT_MSG_TYPE type) : Task(), type(type) {
                 for (std::list<const TUserInfo>::const_iterator it = userInfoList.begin(); it != userInfoList.end(); ++it) {
                     MessageDB::Instance()->InsertUserInfoOrReplace(*it);
                 }
+
+                if(StnCallBack::Instance()->m_getUserInfoCB) {
+                    StnCallBack::Instance()->m_getUserInfoCB->onSuccess(userInfoList);
+                }
                 delete this;
             }
             void onFalure(int errorCode) {
+                if(StnCallBack::Instance()->m_getUserInfoCB) {
+                    StnCallBack::Instance()->m_getUserInfoCB->onFalure(errorCode);
+                }
                 delete this;
             }
 

diff --git a/mars/stn/stn.h b/mars/stn/stn.h
@@ -536,6 +536,10 @@ extern void (*ReportDnsProfile)(const DnsProfile& _dns_profile);
 
       extern void setConnectionStatusCallback(ConnectionStatusCallback *callback);
       extern void setReceiveMessageCallback(ReceiveMessageCallback *callback);
+
+        extern void setRefreshUserInfoCallback(GetUserInfoCallback *callback);
+        extern void setRefreshGroupInfoCallback(GetGroupInfoCallback *callback);
+
       extern ConnectionStatus getConnectionStatus();
 
 extern int (*sendMessage)(TMessage &tmsg, SendMessageCallback *callback);