Add payment_end_time to worker request

golos-blockchain · Jul 23, 2020 · 21b9581 · 21b9581
1 parent b1be8f6
commit 21b9581
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/plugins/worker_api/include/golos/plugins/worker_api/worker_api_objects.hpp b/plugins/worker_api/include/golos/plugins/worker_api/worker_api_objects.hpp
@@ -37,6 +37,7 @@ namespace golos { namespace plugins { namespace worker_api {
         share_type upvote_total;
         asset paid_out;
         asset paid_out_vests;
+        time_point_sec payment_end_time;
     };
 
     struct by_net_rshares;
@@ -94,7 +95,8 @@ namespace golos { namespace plugins { namespace worker_api {
               upvote_rshares(o.upvote_rshares),
               upvote_total(o.upvote_total),
               paid_out(o.paid_out),
-              paid_out_vests(o.paid_out_vests) {
+              paid_out_vests(o.paid_out_vests),
+              payment_end_time(o.payment_end_time) {
         }
 
         worker_request_api_object() {
@@ -135,6 +137,7 @@ namespace golos { namespace plugins { namespace worker_api {
         asset paid_out;
         asset paid_out_vests;
         asset remaining_payment;
+        time_point_sec payment_end_time;
     };
 
 } } } // golos::plugins::worker_api
@@ -150,5 +153,5 @@ FC_REFLECT((golos::plugins::worker_api::worker_request_api_object),
     (creation_fee)
     (upvotes)(downvotes)
     (stake_rshares)(stake_total)(upvote_rshares)(upvote_total)
-    (paid_out)(paid_out_vests)(remaining_payment)
+    (paid_out)(paid_out_vests)(remaining_payment)(payment_end_time)
 )
diff --git a/plugins/worker_api/worker_api_plugin.cpp b/plugins/worker_api/worker_api_plugin.cpp
@@ -116,6 +116,21 @@ struct post_operation_visitor {
             o.paid_out_vests += op.reward_in_vests_if_vest;
         });
     }
+
+    result_type operator()(const worker_state_operation& op) const {
+        if (op.state != worker_request_state::payment_complete) {
+            return;
+        }
+
+        const auto& post = _db.get_comment(op.author, op.permlink);
+
+        const auto& wrmo_idx = _db.get_index<worker_request_metadata_index, by_post>();
+        auto wrmo_itr = wrmo_idx.find(post.id);
+
+        _db.modify(*wrmo_itr, [&](auto& o) {
+            o.payment_end_time = _db.head_block_time();
+        });
+    }
 };
 
 class worker_api_plugin::worker_api_plugin_impl final {