go

Author: guibescos

runtime/src/bank/pyth_accumulator.rs

@@ -373,18 +373,14 @@ fn post_accumulator_attestation(
 }
 
 pub fn update_v2(bank: &Bank) -> std::result::Result<(), AccumulatorUpdateErrorV1> {
-    let accounts = bank
+    let accounts: Vec<(Pubkey, AccountSharedData)> = bank
         .get_program_accounts(&ORACLE_PUBKEY, &ScanConfig::new(true))
         .map_err(AccumulatorUpdateErrorV1::GetProgramAccounts)?;
 
     let mut any_v1_aggregations = false;
     let mut v2_messages = Vec::new();
 
-    let account_datas = accounts
-    .iter()
-    .map(|(_, account)| account.data().borrow())
-    .collect::<Vec<_>>();
-    let message = pyth_oracle::validator::compute_publisher_caps(account_datas, bank.clock().unix_timestamp);
+    compute_publisher_stake_caps(&accounts, bank.clock().unix_timestamp, &mut v2_messages, true);
 
     for (pubkey, mut account) in accounts {
         let mut price_account_data = account.data().to_owned();
@@ -414,3 +410,20 @@ pub fn update_v2(bank: &Bank) -> std::result::Result<(), AccumulatorUpdateErrorV
 
     update_v1(bank, &v2_messages, any_v1_aggregations)
 }
+
+pub fn compute_publisher_stake_caps(
+    accounts : &Vec<(Pubkey, AccountSharedData)>, 
+    timestamp : i64,
+    messages : &mut Vec<Vec<u8>>,
+    is_active : bool
+) {
+    let account_datas : Vec<&[u8]> = accounts
+        .iter()
+        .map(|(_, account)| account.data().borrow())
+        .collect();
+    let message = pyth_oracle::validator::compute_publisher_caps(account_datas, timestamp);
+
+    if is_active {
+        messages.push(message);
+    }
+}