Make requested CU amount and price in upd_price transactions configurable in pythd

Author: tompntn

pc/manager.cpp

@@ -14,6 +14,9 @@ using namespace pc;
 #define PC_MAX_BATCH          8
 // Flush partial batches if not completed within 400 ms.
 #define PC_FLUSH_INTERVAL       (400L*PC_NSECS_IN_MSEC)
+// Compute units requested per price update instruction
+// The biggest instruction appears to be about ~10300 CUs, so we overestimate by 100%.
+#define PC_UPD_PRICE_COMPUTE_UNITS 20000
 
 ///////////////////////////////////////////////////////////////////////////
 // manager_sub
@@ -76,6 +79,8 @@ manager::manager()
   is_pub_( false ),
   cmt_( commitment::e_confirmed ),
   max_batch_( PC_MAX_BATCH ),
+  requested_upd_price_cu_units_( PC_UPD_PRICE_COMPUTE_UNITS ),
+  requested_upd_price_cu_price_( 0UL ),
   sreq_{ { commitment::e_processed } },
   secondary_{ nullptr },
   is_secondary_( false )
@@ -189,6 +194,22 @@ int64_t manager::get_publish_interval() const
   return pub_int_ / PC_NSECS_IN_MSEC;
 }
 
+void manager::set_requested_upd_price_cu_units( unsigned cu_units ) {
+  requested_upd_price_cu_units_ = cu_units;
+}
+
+unsigned manager::get_requested_upd_price_cu_units() const {
+  return requested_upd_price_cu_units_;
+}
+
+void manager::set_requested_upd_price_cu_price( unsigned cu_price ) {
+  requested_upd_price_cu_price_ = cu_price;
+}
+
+unsigned manager::get_requested_upd_price_cu_price() const {
+  return requested_upd_price_cu_price_;
+}
+
 void manager::set_max_batch_size( unsigned batch_size )
 {
   max_batch_ = batch_size;

pc/manager.hpp

@@ -100,6 +100,14 @@ namespace pc
     void set_publish_interval( int64_t mill_secs );
     int64_t get_publish_interval() const;
 
+    // override the amount of requested CU units per upd_price transaction
+    void set_requested_upd_price_cu_units( unsigned cu_units );
+    unsigned get_requested_upd_price_cu_units() const;
+
+    // override the price per CU for upd_price transaction
+    void set_requested_upd_price_cu_price( unsigned cu_price );
+    unsigned get_requested_upd_price_cu_price() const;
+
     // override the default maximum number of price updates to send in a batch
     void set_max_batch_size( unsigned batch_size );
     unsigned get_max_batch_size() const;
@@ -275,6 +283,8 @@ namespace pc
     tx_parser    txp_;      // handle unexpected errors
     commitment   cmt_;      // account get/subscribe commitment
     unsigned     max_batch_;// maximum number of price updates that can be sent in a single batch
+    unsigned     requested_upd_price_cu_units_; // amount of requested CU units per upd_price transaction
+    unsigned     requested_upd_price_cu_price_; // price per CU for upd_price transaction
 
     // requests
     rpc::get_slot              sreq_[1]; // slot subscription

pcapps/pythd.cpp

@@ -72,6 +72,10 @@ int usage()
   std::cerr << "  -d" << std::endl;
   std::cerr << "     Turn on debug logging. Can also toggle this on/off via "
                "kill -s SIGUSR1 <pid>\n" << std::endl;
+  std::cerr << "  -u" << std::endl;
+  std::cerr << "     Number of compute units requested by each upd_price transaction (default 20000)" << std::endl;
+  std::cerr << "  -v" << std::endl;
+  std::cerr << "     Price per compute unit for each upd_price transaction, in micro lamports (default 31250)" << std::endl;
   return 1;
 }
 
@@ -104,6 +108,8 @@ int main(int argc, char **argv)
   int pyth_port = get_port();
   int opt = 0;
   int pub_int = 1000;
+  unsigned cu_units = 20000;
+  unsigned cu_price = 31250;
   unsigned max_batch_size = 0;
   bool do_wait = true, do_tx = true, do_ws = true, do_debug = false;
   while( (opt = ::getopt(argc,argv, "r:s:t:p:i:k:w:c:l:m:b:dnxhz" )) != -1 ) {
@@ -123,6 +129,8 @@ int main(int argc, char **argv)
       case 'x': do_tx = false; break;
       case 'z': do_ws = false; break;
       case 'd': do_debug = true; break;
+      case 'u': cu_units = strtoul(optarg, NULL, 0); break;
+      case 'v': cu_price = strtoul(optarg, NULL, 0); break;
       default: return usage();
     }
   }
@@ -153,6 +161,8 @@ int main(int argc, char **argv)
   mgr.set_do_capture( !cap_file.empty() );
   mgr.set_commitment( cmt );
   mgr.set_publish_interval( pub_int );
+  mgr.set_requested_upd_price_cu_units( cu_units );
+  mgr.set_requested_upd_price_cu_price( cu_price );
 
   bool do_secondary = !secondary_rpc_host.empty();
   if ( do_secondary ) {