feat(lazer-protocol): Record publisher count for aggregated price feeds (#2225)

* Add publisher count to aggregated price feed data

* Add publisher count to parsed payloads

* Change count to u16

* Bump version and change count to option of u16

Author: darunrs

lazer/Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "pyth-lazer-protocol"
-version = "0.2.0"
+version = "0.2.1"
 edition = "2021"
 description = "Pyth Lazer SDK - protocol types."
 license = "Apache-2.0"

lazer/sdk/rust/protocol/Cargo.toml

@@ -32,13 +32,15 @@ pub enum PayloadPropertyValue {
     Price(Option<Price>),
     BestBidPrice(Option<Price>),
     BestAskPrice(Option<Price>),
+    PublisherCount(Option<u16>),
 }
 
 #[derive(Debug, Clone, Default)]
 pub struct AggregatedPriceFeedData {
     pub price: Option<Price>,
     pub best_bid_price: Option<Price>,
     pub best_ask_price: Option<Price>,
+    pub publisher_count: Option<u16>,
 }
 
 pub const PAYLOAD_FORMAT_MAGIC: u32 = 2479346549;
@@ -67,6 +69,9 @@ impl PayloadData {
                             PriceFeedProperty::BestAskPrice => {
                                 PayloadPropertyValue::BestAskPrice(feed.best_ask_price)
                             }
+                            PriceFeedProperty::PublisherCount => {
+                                PayloadPropertyValue::PublisherCount(feed.publisher_count)
+                            }
                         })
                         .collect(),
                 })
@@ -96,6 +101,10 @@ impl PayloadData {
                         writer.write_u8(PriceFeedProperty::BestAskPrice as u8)?;
                         write_option_price::<BO>(&mut writer, *price)?;
                     }
+                    PayloadPropertyValue::PublisherCount(count) => {
+                        writer.write_u8(PriceFeedProperty::PublisherCount as u8)?;
+                        write_option_u16::<BO>(&mut writer, *count)?;
+                    }
                 }
             }
         }
@@ -134,6 +143,8 @@ impl PayloadData {
                     PayloadPropertyValue::BestBidPrice(read_option_price::<BO>(&mut reader)?)
                 } else if property == PriceFeedProperty::BestAskPrice as u8 {
                     PayloadPropertyValue::BestAskPrice(read_option_price::<BO>(&mut reader)?)
+                } else if property == PriceFeedProperty::PublisherCount as u8 {
+                    PayloadPropertyValue::PublisherCount(read_option_u16::<BO>(&mut reader)?)
                 } else {
                     bail!("unknown property");
                 };
@@ -161,6 +172,18 @@ fn read_option_price<BO: ByteOrder>(mut reader: impl Read) -> std::io::Result<Op
     Ok(value.map(Price))
 }
 
+fn write_option_u16<BO: ByteOrder>(
+    mut writer: impl Write,
+    value: Option<u16>,
+) -> std::io::Result<()> {
+    writer.write_u16::<BO>(value.unwrap_or(0))
+}
+
+fn read_option_u16<BO: ByteOrder>(mut reader: impl Read) -> std::io::Result<Option<u16>> {
+    let value = reader.read_u16::<BO>()?;
+    Ok(Some(value))
+}
+
 pub const BINARY_UPDATE_FORMAT_MAGIC: u32 = 1937213467;
 
 pub const PARSED_FORMAT_MAGIC: u32 = 2584795844;

lazer/sdk/rust/protocol/src/payload.rs

@@ -131,6 +131,7 @@ pub enum PriceFeedProperty {
     Price,
     BestBidPrice,
     BestAskPrice,
+    PublisherCount,
     // More fields may be added later.
 }
 
@@ -392,6 +393,9 @@ pub struct ParsedFeedPayload {
     #[serde(with = "crate::serde_str::option_price")]
     #[serde(default)]
     pub best_ask_price: Option<Price>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    #[serde(default)]
+    pub publisher_count: Option<u16>,
     // More fields may be added later.
 }
 
@@ -406,6 +410,7 @@ impl ParsedFeedPayload {
             price: None,
             best_bid_price: None,
             best_ask_price: None,
+            publisher_count: None,
         };
         for &property in properties {
             match property {
@@ -418,6 +423,9 @@ impl ParsedFeedPayload {
                 PriceFeedProperty::BestAskPrice => {
                     output.best_ask_price = data.best_ask_price;
                 }
+                PriceFeedProperty::PublisherCount => {
+                    output.publisher_count = data.publisher_count;
+                }
             }
         }
         output
@@ -429,6 +437,7 @@ impl ParsedFeedPayload {
             price: data.price,
             best_bid_price: data.best_bid_price,
             best_ask_price: data.best_ask_price,
+            publisher_count: data.publisher_count,
         }
     }
 }