refactor(agent): rename adapter->state (#124)

* refactor(agent): move adapter out of pythd

* refactor(agent): fix adapter imports

* refactor(agent): rename adapter->state

Author: Reisen

src/agent.rs

@@ -69,15 +69,14 @@ pub mod metrics;
 pub mod pythd;
 pub mod remote_keypair_loader;
 pub mod solana;
+pub mod state;
 pub mod store;
 use {
     self::{
         config::Config,
-        pythd::{
-            adapter::notifier,
-            api::rpc,
-        },
+        pythd::api::rpc,
         solana::network,
+        state::notifier,
     },
     anyhow::Result,
     futures_util::future::join_all,
@@ -122,9 +121,8 @@ impl Agent {
         let (secondary_keypair_loader_tx, secondary_keypair_loader_rx) = mpsc::channel(10);
 
         // Create the Pythd Adapter.
-        let adapter = Arc::new(
-            pythd::adapter::Adapter::new(self.config.pythd_adapter.clone(), logger.clone()).await,
-        );
+        let adapter =
+            Arc::new(state::State::new(self.config.pythd_adapter.clone(), logger.clone()).await);
 
         // Spawn the primary network
         jhs.extend(network::spawn_network(
@@ -197,6 +195,7 @@ pub mod config {
             pythd,
             remote_keypair_loader,
             solana::network,
+            state,
         },
         anyhow::Result,
         config as config_rs,
@@ -216,7 +215,7 @@ pub mod config {
         pub primary_network:       network::Config,
         pub secondary_network:     Option<network::Config>,
         #[serde(default)]
-        pub pythd_adapter:         pythd::adapter::Config,
+        pub pythd_adapter:         state::Config,
         #[serde(default)]
         pub pythd_api_server:      pythd::api::rpc::Config,
         #[serde(default)]

src/agent/dashboard.rs

@@ -1,20 +1,20 @@
 use {
     super::{
-        pythd::adapter::{
+        solana::{
+            network::Network,
+            oracle::PriceEntry,
+        },
+        state::{
             global::GlobalStore,
             local::{
                 LocalStore,
                 PriceInfo,
             },
         },
-        solana::{
-            network::Network,
-            oracle::PriceEntry,
-        },
     },
     crate::agent::{
         metrics::MetricsServer,
-        pythd::adapter::global::{
+        state::global::{
             AllAccountsData,
             AllAccountsMetadata,
             PriceAccountMetadata,

src/agent/metrics.rs

@@ -1,7 +1,7 @@
 use {
-    super::pythd::adapter::{
+    super::state::{
         local::PriceInfo,
-        Adapter,
+        State,
     },
     crate::agent::{
         solana::oracle::PriceEntry,
@@ -69,16 +69,12 @@ lazy_static! {
 pub struct MetricsServer {
     pub start_time: Instant,
     pub logger:     Logger,
-    pub adapter:    Arc<Adapter>,
+    pub adapter:    Arc<State>,
 }
 
 impl MetricsServer {
     /// Instantiate a metrics API with a dashboard
-    pub async fn spawn(
-        addr: impl Into<SocketAddr> + 'static,
-        logger: Logger,
-        adapter: Arc<Adapter>,
-    ) {
+    pub async fn spawn(addr: impl Into<SocketAddr> + 'static, logger: Logger, adapter: Arc<State>) {
         let server = MetricsServer {
             start_time: Instant::now(),
             logger,

src/agent/pythd.rs

@@ -1,2 +1 @@
-pub mod adapter;
 pub mod api;

src/agent/pythd/api/rpc.rs

@@ -6,14 +6,14 @@
 
 use {
     super::{
-        super::adapter,
         Conf,
         NotifyPrice,
         NotifyPriceSched,
         Price,
         Pubkey,
         SubscriptionID,
     },
+    crate::agent::state,
     anyhow::{
         anyhow,
         Result,
@@ -120,7 +120,7 @@ async fn handle_connection<S>(
     notify_price_sched_tx_buffer: usize,
     logger: Logger,
 ) where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
     S: Send,
     S: Sync,
     S: 'static,
@@ -168,7 +168,7 @@ async fn handle_next<S>(
     notify_price_sched_rx: &mut mpsc::Receiver<NotifyPriceSched>,
 ) -> Result<()>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     tokio::select! {
         msg = ws_rx.next() => {
@@ -210,7 +210,7 @@ async fn handle<S>(
     msg: Message,
 ) -> Result<()>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     // Ignore control and binary messages
     if !msg.is_text() {
@@ -296,7 +296,7 @@ async fn dispatch_and_catch_error<S>(
     request: &Request<Method, Value>,
 ) -> Response<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     debug!(
         logger,
@@ -436,7 +436,7 @@ pub async fn run<S>(
     adapter: Arc<S>,
     shutdown_rx: broadcast::Receiver<()>,
 ) where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
     S: Send,
     S: Sync,
     S: 'static,
@@ -454,7 +454,7 @@ async fn serve<S>(
     mut shutdown_rx: broadcast::Receiver<()>,
 ) -> Result<()>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
     S: Send,
     S: Sync,
     S: 'static,

src/agent/pythd/api/rpc/get_all_products.rs

@@ -1,11 +1,11 @@
 use {
-    crate::agent::pythd::adapter,
+    crate::agent::state,
     anyhow::Result,
 };
 
 pub async fn get_all_products<S>(adapter: &S) -> Result<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     let products = adapter.get_all_products().await?;
     Ok(serde_json::to_value(products)?)

src/agent/pythd/api/rpc/get_product.rs

@@ -3,7 +3,7 @@ use {
         GetProductParams,
         Method,
     },
-    crate::agent::pythd::adapter,
+    crate::agent::state,
     anyhow::{
         anyhow,
         Result,
@@ -19,7 +19,7 @@ pub async fn get_product<S>(
     request: &Request<Method, Value>,
 ) -> Result<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     let params: GetProductParams = {
         let value = request.params.clone();

src/agent/pythd/api/rpc/get_product_list.rs

@@ -1,11 +1,11 @@
 use {
-    crate::agent::pythd::adapter,
+    crate::agent::state,
     anyhow::Result,
 };
 
 pub async fn get_product_list<S>(adapter: &S) -> Result<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     let product_list = adapter.get_product_list().await?;
     Ok(serde_json::to_value(product_list)?)

src/agent/pythd/api/rpc/subscribe_price.rs

@@ -5,7 +5,7 @@ use {
         SubscribePriceParams,
         SubscribeResult,
     },
-    crate::agent::pythd::adapter,
+    crate::agent::state,
     anyhow::{
         anyhow,
         Result,
@@ -23,7 +23,7 @@ pub async fn subscribe_price<S>(
     request: &Request<Method, Value>,
 ) -> Result<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     let params: SubscribePriceParams = serde_json::from_value(
         request

src/agent/pythd/api/rpc/subscribe_price_sched.rs

@@ -5,7 +5,7 @@ use {
         SubscribePriceSchedParams,
         SubscribeResult,
     },
-    crate::agent::pythd::adapter,
+    crate::agent::state,
     anyhow::{
         anyhow,
         Result,
@@ -23,7 +23,7 @@ pub async fn subscribe_price_sched<S>(
     request: &Request<Method, Value>,
 ) -> Result<serde_json::Value>
 where
-    S: adapter::AdapterApi,
+    S: state::StateApi,
 {
     let params: SubscribePriceSchedParams = serde_json::from_value(
         request