Fixed invalid BOOST_ASSET.

When client want to restore serialized messages, endpoint might keep the message that has the same packet_id. In this case, overwrite store_.
redboltz · May 8, 2019 · ff9a294 · ff9a294
1 parent 58d6681
commit ff9a294
Showing 1 changed file with 65 additions and 6 deletions.
diff --git a/include/mqtt/endpoint.hpp b/include/mqtt/endpoint.hpp
@@ -7476,7 +7476,23 @@ class endpoint : public std::enable_shared_from_this<endpoint<Socket, Mutex, Loc
                 std::move(msg),
                 std::move(life_keeper)
             );
-            BOOST_ASSERT(ret.second);
+            // When client want to restore serialized messages,
+            // endpoint might keep the message that has the same packet_id.
+            // In this case, overwrite store_.
+            if (!ret.second) {
+                store_.modify(
+                    ret.first,
+                    [&] (auto& e) {
+                        e = store(
+                            packet_id,
+                            qos == qos::at_least_once ? control_packet_type::puback
+                                                      : control_packet_type::pubrec,
+                            std::move(msg),
+                            std::move(life_keeper)
+                        );
+                    }
+                );
+            }
         }
     }
 
@@ -7494,7 +7510,21 @@ class endpoint : public std::enable_shared_from_this<endpoint<Socket, Mutex, Loc
                 control_packet_type::pubcomp,
                 std::move(msg)
             );
-            BOOST_ASSERT(ret.second);
+            // When client want to restore serialized messages,
+            // endpoint might keep the message that has the same packet_id.
+            // In this case, overwrite store_.
+            if (!ret.second) {
+                store_.modify(
+                    ret.first,
+                    [&] (auto& e) {
+                        e = store(
+                            packet_id,
+                            control_packet_type::pubcomp,
+                            std::move(msg)
+                        );
+                    }
+                );
+            }
         }
     }
 
@@ -7544,7 +7574,23 @@ class endpoint : public std::enable_shared_from_this<endpoint<Socket, Mutex, Loc
                 std::move(msg),
                 std::move(life_keeper)
             );
-            BOOST_ASSERT(ret.second);
+            // When client want to restore serialized messages,
+            // endpoint might keep the message that has the same packet_id.
+            // In this case, overwrite store_.
+            if (!ret.second) {
+                store_.modify(
+                    ret.first,
+                    [&] (auto& e) {
+                        e = store(
+                            packet_id,
+                            qos == qos::at_least_once ? control_packet_type::puback
+                                                      : control_packet_type::pubrec,
+                            std::move(msg),
+                            std::move(life_keeper)
+                        );
+                    }
+                );
+            }
         }
     }
 
@@ -7560,10 +7606,23 @@ class endpoint : public std::enable_shared_from_this<endpoint<Socket, Mutex, Loc
             auto ret = store_.emplace(
                 packet_id,
                 control_packet_type::pubcomp,
-                std::move(msg),
-                []{}
+                std::move(msg)
             );
-            BOOST_ASSERT(ret.second);
+            // When client want to restore serialized messages,
+            // endpoint might keep the message that has the same packet_id.
+            // In this case, overwrite store_.
+            if (!ret.second) {
+                store_.modify(
+                    ret.first,
+                    [&] (auto& e) {
+                        e = store(
+                            packet_id,
+                            control_packet_type::pubcomp,
+                            std::move(msg)
+                        );
+                    }
+                );
+            }
         }
     }