Merge pull request #9923 from EOSIO/huangminghuang/get-producers-fix

get_producers unittest

libraries/chain/backing_store/db_key_value_format.cpp

@@ -290,4 +290,31 @@ namespace eosio { namespace chain { namespace backing_store { namespace db_key_v
       bytes composite_key = kt ? create_prefix_type_key(scope, table, *kt) : create_prefix_key(scope, table);
       return create_full_key(composite_key, code);
    }
+
+   eosio::session::shared_bytes create_full_primary_key(const eosio::session::shared_bytes& prefix, uint64_t primary_key) {
+      const auto prefix_size = detail::prefix_size<eosio::session::shared_bytes>();
+      b1::chain_kv::bytes composite_key;
+      constexpr static auto type_size = sizeof(key_type);
+      static_assert( type_size == 1, "" ); // this changing will break check below of prefix.back()
+      if (prefix.size() == prefix_size + type_size) {
+         const key_type kt = static_cast<key_type>(prefix[prefix.size() - 1]);
+         EOS_ASSERT(kt == key_type::primary, bad_composite_key_exception,
+                    "DB intrinsic create_full_primary_key should only be called with a prefix for a primary key, this has a prefix for type: ${type}",
+		    ("type", static_cast<uint64_t>(kt)));
+         composite_key.reserve(detail::key_size(key_type::primary));
+      }
+      else {
+         EOS_ASSERT(prefix.size() == prefix_size, bad_composite_key_exception,
+                    "DB intrinsic create_full_primary_key should only be called with a prefix of code, scope, and table and optionally a primary type indicator");
+         composite_key.reserve(type_size + detail::key_size(key_type::primary));
+	 composite_key.push_back(static_cast<char>(key_type::primary));
+      }
+
+      b1::chain_kv::append_key(composite_key, primary_key);
+      return eosio::session::make_shared_bytes<std::string_view, 2>({std::string_view{prefix.data(),
+                                                                                      prefix.size()},
+                                                                     std::string_view{composite_key.data(),
+								                      composite_key.size()}});
+
+   }
 }}}} // namespace eosio::chain::backing_store::db_key_value_format

libraries/chain/include/eosio/chain/backing_store/db_key_value_format.hpp

@@ -565,6 +565,7 @@ namespace chain { namespace backing_store { namespace db_key_value_format {
 
    eosio::session::shared_bytes create_full_primary_key(name code, name scope, name table, uint64_t primary_key);
    eosio::session::shared_bytes create_full_prefix_key(name code, name scope, name table, std::optional<key_type> kt = std::optional<key_type>{});
+   eosio::session::shared_bytes create_full_primary_key(const eosio::session::shared_bytes& prefix, uint64_t primary_key);
 
    template<typename Key>
    eosio::session::shared_bytes create_full_secondary_key(name code, name scope, name table, const Key& sec_key, uint64_t primary_key) {

libraries/chain/include/eosio/chain/combined_database.hpp

@@ -11,6 +11,8 @@
 
 #include <eosio/chain/account_object.hpp>
 #include <eosio/chain/backing_store.hpp>
+#include <eosio/chain/backing_store/chain_kv_payer.hpp>
+#include <eosio/chain/backing_store/db_key_value_format.hpp>
 #include <eosio/chain/block_summary_object.hpp>
 #include <eosio/chain/code_object.hpp>
 #include <eosio/chain/contract_table_objects.hpp>
@@ -134,6 +136,36 @@ namespace eosio { namespace chain {
       auto &get_kv_undo_stack(void) const { return kv_undo_stack; }
       backing_store_type get_backing_store() const { return backing_store; }
 
+      template<typename Lambda>
+      bool get_primary_key_data(name code, name scope, name table, uint64_t primary_key, Lambda&& process_data) const {
+         if (backing_store == backing_store_type::CHAINBASE) {
+            const auto* t_id = db.find<chain::table_id_object, chain::by_code_scope_table>( boost::make_tuple( code, scope, table ) );
+            if ( !t_id ) {
+               return false;
+            }
+
+            const auto* obj = db.find<chain::key_value_object, chain::by_scope_primary>(boost::make_tuple(t_id->id, primary_key) );
+
+            if (obj) {
+               return process_data(obj->payer, obj->value.data(), obj->value.size());
+            }
+         }
+         else {
+            using namespace eosio::chain;
+            EOS_ASSERT(backing_store == backing_store_type::ROCKSDB,
+                     chain::contract_table_query_exception,
+                     "Support for configured backing_store has not been added");
+            auto full_primary_key = chain::backing_store::db_key_value_format::create_full_primary_key(code, scope, table, primary_key);
+            auto session = get_kv_undo_stack()->top();
+            auto value = session.read(full_primary_key);
+            if (value) {
+               backing_store::payer_payload pp{value->data(), value->size()};
+               return process_data(pp.payer, pp.value, pp.value_size);
+            }
+         }
+	 return false;
+      }
+
     private:
       void add_contract_tables_to_snapshot(const snapshot_writer_ptr& snapshot) const;
       void read_contract_tables_from_snapshot(const snapshot_reader_ptr& snapshot);