Merge pull request #465 from GolosChain/464-get-all-content-replies

Add API social_network::get_all_content_replies. #464

Author: kotbegemot

plugins/social_network/include/golos/plugins/social_network/social_network.hpp

@@ -22,6 +22,7 @@ namespace golos {
             DEFINE_API_ARGS(get_content,                           msg_pack, discussion)
             DEFINE_API_ARGS(get_trending_tags,                     msg_pack, std::vector<tag_api_object>)
             DEFINE_API_ARGS(get_content_replies,                   msg_pack, std::vector<discussion>)
+            DEFINE_API_ARGS(get_all_content_replies,               msg_pack, std::vector<discussion>)
             DEFINE_API_ARGS(get_tags_used_by_author,               msg_pack, tags_used_by_author_r)
             DEFINE_API_ARGS(get_discussions_by_payout,             msg_pack, std::vector<discussion>)
             DEFINE_API_ARGS(get_discussions_by_trending,           msg_pack, std::vector<discussion>)
@@ -160,6 +161,7 @@ namespace golos {
                  **/
                 (get_discussions_by_promoted)
                                 (get_content_replies)
+                                (get_all_content_replies)
                                 /**
                                     *  This method is used to fetch all posts/comments by start_author that occur after before_date and start_permlink with up to limit being returned.
                                     *

plugins/social_network/social_network.cpp

@@ -182,7 +182,17 @@ namespace golos {
                         Args... args
                 ) const;
 
-                std::vector<discussion> get_content_replies(std::string author, std::string permlink) const;
+                void select_content_replies(
+                    std::vector<discussion>& result, const std::string &author, const std::string &permlink
+                ) const;
+
+                std::vector<discussion> get_content_replies(
+                    const std::string &author, const std::string &permlink
+                ) const;
+
+                std::vector<discussion> get_all_content_replies(
+                    const std::string &author, const std::string &permlink
+                ) const;
 
                 std::vector<tag_api_object> get_trending_tags(std::string after, uint32_t limit) const;
 
@@ -325,20 +335,31 @@ namespace golos {
                 }
             }
 
-            std::vector<discussion> social_network_t::impl::get_content_replies(std::string author, std::string permlink) const {
+            void social_network_t::impl::select_content_replies(
+                std::vector<discussion> &result, const std::string &author, const std::string &permlink
+            ) const {
                 account_name_type acc_name = account_name_type(author);
                 const auto &by_permlink_idx = database().get_index<comment_index>().indices().get<by_parent>();
                 auto itr = by_permlink_idx.find(boost::make_tuple(acc_name, permlink));
-                std::vector<discussion> result;
-                while (itr != by_permlink_idx.end() && itr->parent_author == author &&
-                       to_string(itr->parent_permlink) == permlink) {
+                while (
+                    itr != by_permlink_idx.end() &&
+                    itr->parent_author == author &&
+                    !strcmp(itr->parent_permlink.c_str(), permlink.c_str())
+                ) {
                     discussion push_discussion(*itr);
                     push_discussion.active_votes = get_active_votes(author, permlink);
 
                     result.emplace_back(*itr);
                     set_pending_payout(result.back());
                     ++itr;
                 }
+            }
+
+            std::vector<discussion> social_network_t::impl::get_content_replies(
+                const std::string &author, const std::string &permlink
+            ) const {
+                std::vector<discussion> result;
+                select_content_replies(result, author, permlink);
                 return result;
             }
 
@@ -351,6 +372,27 @@ namespace golos {
                 });
             }
 
+            std::vector<discussion> social_network_t::impl::get_all_content_replies(
+                const std::string &author, const std::string &permlink
+            ) const {
+                std::vector<discussion> result;
+                select_content_replies(result, author, permlink);
+                for (std::size_t i = 0; i < result.size(); ++i) {
+                    if (result[i].children > 0) {
+                        select_content_replies(result, result[i].author, result[i].permlink);
+                    }
+                }
+                return result;
+            }
+
+            DEFINE_API(social_network_t, get_all_content_replies) {
+                CHECK_ARG_SIZE(2)
+                auto author = args.args->at(0).as<string>();
+                auto permlink = args.args->at(1).as<string>();
+                return pimpl->database().with_read_lock([&]() {
+                    return pimpl->get_all_content_replies(author, permlink);
+                });
+            }
             boost::multiprecision::uint256_t to256(const fc::uint128_t &t) {
                 boost::multiprecision::uint256_t result(t.high_bits());
                 result <<= 65;