nautilus_hyperliquid/websocket/

handler.rs

// -------------------------------------------------------------------------------------------------
//  Copyright (C) 2015-2026 Nautech Systems Pty Ltd. All rights reserved.
//  https://nautechsystems.io
//
//  Licensed under the GNU Lesser General Public License Version 3.0 (the "License");
//  You may not use this file except in compliance with the License.
//  You may obtain a copy of the License at https://www.gnu.org/licenses/lgpl-3.0.en.html
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
// -------------------------------------------------------------------------------------------------

//! WebSocket message handler for Hyperliquid.

use std::{
    collections::{BTreeSet, VecDeque},
    future::Future,
    sync::{
        Arc,
        atomic::{AtomicBool, Ordering},
    },
};

use ahash::{AHashMap, AHashSet};
use nautilus_common::cache::fifo::FifoCache;
use nautilus_core::{AtomicTime, Params, nanos::UnixNanos, time::get_atomic_clock_realtime};
use nautilus_model::{
    data::{BarType, CustomData, Data, DataType},
    identifiers::{AccountId, InstrumentId},
    instruments::{Instrument, InstrumentAny},
    types::Price,
};
use nautilus_network::{
    RECONNECTED,
    error::SendError,
    retry::{RetryError, RetryManager, create_websocket_retry_manager},
    websocket::{SubscriptionState, WebSocketClient},
};
use rust_decimal::Decimal;
use tokio_tungstenite::tungstenite::Message;
use tokio_util::sync::CancellationToken;
use ustr::Ustr;

use super::{
    client::{AssetContextDataType, CloidCache},
    enums::HyperliquidWsChannel,
    error::HyperliquidWsError,
    messages::{
        CandleData, ExecutionReport, HyperliquidWsMessage, HyperliquidWsRequest, NautilusWsMessage,
        PostRequest, SubscriptionRequest, WsActiveAssetCtxData, WsAllDexsAssetCtxsData,
        WsUserEventData,
    },
    parse::{
        parse_ws_asset_context, parse_ws_candle, parse_ws_fill_report, parse_ws_open_interest,
        parse_ws_order_book_deltas, parse_ws_order_book_depth10, parse_ws_order_status_report,
        parse_ws_public_trade, parse_ws_quote_tick, parse_ws_trade_tick, parse_ws_twap_history_row,
        parse_ws_twap_slice_fill,
    },
    post::PostRouter,
    rate_limits::WebSocketRateLimits,
    trades::TradeStreamUses,
};
use crate::{
    common::consts::HEARTBEAT_INTERVAL,
    data_types::{
        HyperliquidAllDexsAssetCtxs, HyperliquidAllMids, HyperliquidDexAssetCtx,
        HyperliquidImpactPrices,
    },
};

const HEARTBEAT_MESSAGE: &str = r#"{"method":"ping"}"#;

/// Commands sent from the outer client to the inner message handler.
#[derive(Debug)]
#[expect(
    clippy::large_enum_variant,
    reason = "Commands are ephemeral and immediately consumed"
)]
#[allow(private_interfaces)]
pub enum HandlerCommand {
    /// Set the WebSocketClient for the handler to use.
    SetClient(WebSocketClient),
    /// Disconnect the WebSocket connection.
    Disconnect,
    /// Subscribe to the given subscriptions.
    Subscribe {
        subscriptions: Vec<SubscriptionRequest>,
    },
    /// Unsubscribe from the given subscriptions.
    Unsubscribe {
        subscriptions: Vec<SubscriptionRequest>,
    },
    /// Resubscribes without interleaving handler input between the two sends.
    Resubscribe { subscription: SubscriptionRequest },
    /// Send a WebSocket post request.
    Post {
        id: u64,
        request: PostRequest,
        deadline: tokio::time::Instant,
        cancellation_token: CancellationToken,
    },
    /// Initialize the instruments cache with the given instruments.
    InitializeInstruments(Vec<InstrumentAny>),
    /// Update a single instrument in the cache.
    UpdateInstrument(InstrumentAny),
    /// Add a bar type mapping for candle parsing.
    AddBarType { key: String, bar_type: BarType },
    /// Remove a bar type mapping.
    RemoveBarType { key: String },
    /// Update asset context subscriptions for a coin.
    UpdateAssetContextSubs {
        coin: Ustr,
        data_types: AHashSet<AssetContextDataType>,
    },
    /// Update the logical consumers of a `trades` stream for a coin.
    UpdateTradeSubs { coin: Ustr, uses: TradeStreamUses },
    /// Cache the ordered instrument IDs needed to normalize `allDexsAssetCtxs`.
    CacheAllDexAssetCtxsInstrumentIds(AHashMap<Ustr, Vec<Option<InstrumentId>>>),
    /// Cache spot fill coin mappings for instrument lookup.
    CacheSpotFillCoins(AHashMap<Ustr, Ustr>),
    /// Flag whether the `l2Book` stream for `coin` should also be emitted
    /// as [`NautilusWsMessage::Depth10`] snapshots.
    SetDepth10Sub { coin: Ustr, subscribed: bool },
}

#[derive(Default)]
struct AssetContextCaches {
    mark_price: AHashMap<Ustr, Decimal>,
    index_price: AHashMap<Ustr, Decimal>,
    funding_rate: AHashMap<Ustr, Decimal>,
    open_interest: AHashMap<Ustr, Decimal>,
}

impl AssetContextCaches {
    fn clear(&mut self, coin: Ustr, data_type: AssetContextDataType) {
        match data_type {
            AssetContextDataType::MarkPrice => {
                self.mark_price.remove(&coin);
            }
            AssetContextDataType::IndexPrice => {
                self.index_price.remove(&coin);
            }
            AssetContextDataType::FundingRate => {
                self.funding_rate.remove(&coin);
            }
            AssetContextDataType::OpenInterest => {
                self.open_interest.remove(&coin);
            }
        }
    }

    fn clear_removed(
        &mut self,
        coin: Ustr,
        previous_data_types: Option<&AHashSet<AssetContextDataType>>,
        next_data_types: &AHashSet<AssetContextDataType>,
    ) {
        let Some(previous_data_types) = previous_data_types else {
            return;
        };

        for data_type in previous_data_types {
            if !next_data_types.contains(data_type) {
                self.clear(coin, *data_type);
            }
        }
    }
}

#[derive(Debug)]
struct AllMidsDataTypeCache {
    dexes: BTreeSet<Option<String>>,
    projected: Vec<DataType>,
}

impl Default for AllMidsDataTypeCache {
    fn default() -> Self {
        let mut cache = Self {
            dexes: BTreeSet::new(),
            projected: Vec::new(),
        };
        cache.rebuild();
        cache
    }
}

impl AllMidsDataTypeCache {
    fn apply(&mut self, subscription: &SubscriptionRequest, subscribed: bool) {
        let SubscriptionRequest::AllMids { dex } = subscription else {
            return;
        };
        let changed = if subscribed {
            self.dexes.insert(dex.clone())
        } else {
            self.dexes.remove(dex)
        };

        if changed {
            self.rebuild();
        }
    }

    fn as_slice(&self) -> &[DataType] {
        &self.projected
    }

    fn rebuild(&mut self) {
        self.projected.clear();
        if self.dexes.is_empty() {
            self.projected
                .push(DataType::new("HyperliquidAllMids", None, None));
            return;
        }

        self.projected.extend(self.dexes.iter().map(|dex| {
            let metadata = dex.as_ref().map(|dex| {
                let mut metadata = Params::new();
                metadata.insert("dex".to_owned(), serde_json::Value::String(dex.clone()));
                metadata
            });
            DataType::new("HyperliquidAllMids", metadata, None)
        }));
    }
}

pub(super) struct FeedHandler {
    clock: &'static AtomicTime,
    signal: Arc<AtomicBool>,
    client: Option<WebSocketClient>,
    cmd_rx: tokio::sync::mpsc::UnboundedReceiver<HandlerCommand>,
    raw_rx: tokio::sync::mpsc::UnboundedReceiver<Message>,
    cmd_closed: bool,
    raw_closed: bool,
    out_tx: tokio::sync::mpsc::UnboundedSender<NautilusWsMessage>,
    account_id: Option<AccountId>,
    subscriptions: SubscriptionState,
    all_mids_data_types: AllMidsDataTypeCache,
    post_router: Arc<PostRouter>,
    rate_limits: Arc<WebSocketRateLimits>,
    client_id: u64,
    heartbeat: tokio::time::Interval,
    retry_manager: RetryManager<HyperliquidWsError>,
    retry_manager_post: RetryManager<PostSendError>,
    message_buffer: VecDeque<NautilusWsMessage>,
    instruments: AHashMap<Ustr, InstrumentAny>,
    cloid_cache: CloidCache,
    bar_types_cache: AHashMap<String, BarType>,
    bar_cache: AHashMap<String, CandleData>,
    asset_context_subs: AHashMap<Ustr, AHashSet<AssetContextDataType>>,
    trade_subs: AHashMap<Ustr, TradeStreamUses>,
    all_dex_asset_ctxs_instrument_ids: AHashMap<Ustr, Vec<Option<InstrumentId>>>,
    depth10_subs: AHashSet<Ustr>,
    processed_trade_ids: FifoCache<u64, 10_000>,
    processed_public_trade_ids: FifoCache<(Ustr, u64), 10_000>,
    asset_context_caches: AssetContextCaches,
}

impl FeedHandler {
    /// Creates a new [`FeedHandler`] instance.
    #[allow(
        clippy::too_many_arguments,
        reason = "constructs the handler from independent runtime channels and caches"
    )]
    pub(super) fn new(
        signal: Arc<AtomicBool>,
        cmd_rx: tokio::sync::mpsc::UnboundedReceiver<HandlerCommand>,
        raw_rx: tokio::sync::mpsc::UnboundedReceiver<Message>,
        out_tx: tokio::sync::mpsc::UnboundedSender<NautilusWsMessage>,
        account_id: Option<AccountId>,
        subscriptions: SubscriptionState,
        cloid_cache: CloidCache,
        post_router: Arc<PostRouter>,
        rate_limits: Arc<WebSocketRateLimits>,
        client_id: u64,
    ) -> Self {
        let heartbeat = tokio::time::interval_at(
            tokio::time::Instant::now() + HEARTBEAT_INTERVAL,
            HEARTBEAT_INTERVAL,
        );
        Self {
            clock: get_atomic_clock_realtime(),
            signal,
            client: None,
            cmd_rx,
            raw_rx,
            cmd_closed: false,
            raw_closed: false,
            out_tx,
            account_id,
            subscriptions,
            all_mids_data_types: AllMidsDataTypeCache::default(),
            post_router,
            rate_limits,
            client_id,
            heartbeat,
            retry_manager: create_websocket_retry_manager(),
            retry_manager_post: create_websocket_retry_manager(),
            message_buffer: VecDeque::new(),
            instruments: AHashMap::new(),
            cloid_cache,
            bar_types_cache: AHashMap::new(),
            bar_cache: AHashMap::new(),
            asset_context_subs: AHashMap::new(),
            trade_subs: AHashMap::new(),
            all_dex_asset_ctxs_instrument_ids: AHashMap::new(),
            depth10_subs: AHashSet::new(),
            processed_trade_ids: FifoCache::new(),
            processed_public_trade_ids: FifoCache::new(),
            asset_context_caches: AssetContextCaches::default(),
        }
    }

    /// Send a message to the output channel.
    pub(super) fn send(&self, msg: NautilusWsMessage) -> Result<(), String> {
        self.out_tx
            .send(msg)
            .map_err(|e| format!("Failed to send message: {e}"))
    }

    /// Check if the handler has received a stop signal.
    pub(super) fn is_stopped(&self) -> bool {
        self.signal.load(Ordering::Relaxed)
    }

    async fn send_with_retry(&self, payload: String) -> anyhow::Result<()> {
        if let Some(client) = &self.client {
            let rate_key = self.rate_limits.message_key();
            self.retry_manager
                .invocation(
                    "websocket_send",
                    || {
                        let payload = payload.clone();
                        async move {
                            client
                                .send_text(payload, Some(std::slice::from_ref(&rate_key)))
                                .await
                                .map_err(|e| {
                                    HyperliquidWsError::ClientError(format!("Send failed: {e}"))
                                })
                        }
                    },
                    should_retry_hyperliquid_error,
                    |e| create_hyperliquid_timeout_error(e.to_string()),
                )
                .execute()
                .await
                .map_err(|e| anyhow::anyhow!("{e}"))
        } else {
            Err(anyhow::anyhow!("No WebSocket client available"))
        }
    }

    pub(super) async fn next(&mut self) -> Option<NautilusWsMessage> {
        if let Some(msg) = self.message_buffer.pop_front() {
            return Some(msg);
        }

        loop {
            if self.raw_closed && self.cmd_rx.is_empty() {
                log::debug!("Handler shutting down: input stream closed");
                return None;
            }

            tokio::select! {
                cmd = self.cmd_rx.recv(), if !self.cmd_closed => {
                    let Some(cmd) = cmd else {
                        self.cmd_closed = true;
                        continue;
                    };

                    match cmd {
                        HandlerCommand::SetClient(client) => {
                            log::debug!("Setting WebSocket client in handler");
                            self.client = Some(client);
                        }
                        HandlerCommand::Disconnect => {
                            log::debug!("Handler received disconnect command");

                            if let Some(ref client) = self.client {
                                self.rate_limits.acquire_message().await;
                                client.disconnect().await;
                            }
                            self.signal.store(true, Ordering::SeqCst);
                            return None;
                        }
                        HandlerCommand::Subscribe { subscriptions } => {
                            self.subscribe(subscriptions).await;
                        }
                        HandlerCommand::Unsubscribe { subscriptions } => {
                            self.unsubscribe(subscriptions).await;
                        }
                        HandlerCommand::Resubscribe { subscription } => {
                            self.unsubscribe(vec![subscription.clone()]).await;
                            self.subscribe(vec![subscription]).await;
                        }
                        HandlerCommand::Post {
                            id,
                            request,
                            deadline,
                            cancellation_token,
                        } => {
                            if cancellation_token.is_cancelled()
                                || tokio::time::Instant::now() >= deadline
                            {
                                self.post_router
                                    .cancel_registration(id, &cancellation_token)
                                    .await;
                                continue;
                            }

                            let request = HyperliquidWsRequest::Post { id, request };
                            match serde_json::to_string(&request) {
                                Ok(payload) => {
                                    log::debug!("Sending post payload: id={id}");
                                    let result = if let Some(client) = &self.client {
                                        let rate_key = self.rate_limits.message_key();
                                        send_post_with_retry(
                                            &self.retry_manager_post,
                                            deadline,
                                            &cancellation_token,
                                            || {
                                                let payload = payload.clone();
                                                async move {
                                                    let connection_epoch =
                                                        client.connection_epoch();
                                                    client
                                                        .send_text_on_connection(
                                                            payload,
                                                            Some(std::slice::from_ref(&rate_key)),
                                                            connection_epoch,
                                                        )
                                                        .await
                                                        .map_err(PostSendError::Transport)
                                                }
                                            },
                                        )
                                        .await
                                        .map_err(|e| anyhow::anyhow!("{e}"))
                                    } else {
                                        Err(anyhow::anyhow!("No WebSocket client available"))
                                    };

                                    if let Err(e) = result {
                                        log::error!("Error sending post request id={id}: {e}");
                                        self.post_router
                                            .cancel_registration(id, &cancellation_token)
                                            .await;
                                    }
                                }
                                Err(e) => {
                                    log::error!("Error serializing post request id={id}: {e}");
                                    self.post_router
                                        .cancel_registration(id, &cancellation_token)
                                        .await;
                                }
                            }
                        }
                        HandlerCommand::InitializeInstruments(instruments) => {
                            for inst in instruments {
                                let coin = inst.raw_symbol().inner();
                                self.instruments.insert(coin, inst);
                            }
                        }
                        HandlerCommand::UpdateInstrument(inst) => {
                            let coin = inst.raw_symbol().inner();
                            self.instruments.insert(coin, inst);
                        }
                        HandlerCommand::AddBarType { key, bar_type } => {
                            self.bar_types_cache.insert(key, bar_type);
                        }
                        HandlerCommand::RemoveBarType { key } => {
                            self.bar_types_cache.remove(&key);
                            self.bar_cache.remove(&key);
                        }
                        HandlerCommand::UpdateAssetContextSubs { coin, data_types } => {
                            let previous_data_types = self.asset_context_subs.get(&coin).cloned();
                            self.asset_context_caches.clear_removed(
                                coin,
                                previous_data_types.as_ref(),
                                &data_types,
                            );

                            if data_types.is_empty() {
                                self.asset_context_subs.remove(&coin);
                            } else {
                                self.asset_context_subs.insert(coin, data_types);
                            }
                        }
                        HandlerCommand::UpdateTradeSubs { coin, uses } => {
                            if uses.is_empty() {
                                self.trade_subs.remove(&coin);
                            } else {
                                self.trade_subs.insert(coin, uses);
                            }
                        }
                        HandlerCommand::CacheAllDexAssetCtxsInstrumentIds(mappings) => {
                            // merge by dex to match the client cache, so a partial
                            // mapping keeps the entries of dexes it did not cover
                            self.all_dex_asset_ctxs_instrument_ids.extend(mappings);
                        }
                        HandlerCommand::CacheSpotFillCoins(_) => {
                            // No longer needed - raw_symbol now contains the proper format
                        }
                        HandlerCommand::SetDepth10Sub { coin, subscribed } => {
                            if subscribed {
                                self.depth10_subs.insert(coin);
                            } else {
                                self.depth10_subs.remove(&coin);
                            }
                        }
                    }
                }

                raw_msg = self.raw_rx.recv(), if !self.raw_closed => {
                    let Some(raw_msg) = raw_msg else {
                        self.raw_closed = true;
                        continue;
                    };

                    match raw_msg {
                        Message::Text(text) => {
                            if text == RECONNECTED {
                                log::info!("Received RECONNECTED sentinel");
                                return Some(NautilusWsMessage::Reconnected);
                            }

                            match serde_json::from_str::<HyperliquidWsMessage>(&text) {
                                Ok(msg) => {
                                    if let HyperliquidWsMessage::Post { data } = msg {
                                        self.post_router.complete(data).await;
                                        continue;
                                    }

                                    if let HyperliquidWsMessage::SubscriptionResponse { data } = &msg {
                                        let key = subscription_to_key(&data.subscription);
                                        match data.method.as_str() {
                                            "subscribe" => self.subscriptions.confirm_subscribe(&key),
                                            "unsubscribe" => {
                                                let was_pending = self
                                                    .subscriptions
                                                    .pending_unsubscribe_topics()
                                                    .iter()
                                                    .any(|topic| topic == &key);
                                                self.subscriptions.confirm_unsubscribe(&key);

                                                if was_pending {
                                                    self.rate_limits.release_subscription(
                                                        self.client_id,
                                                        &key,
                                                    );
                                                }
                                            }
                                            method => {
                                                log::warn!(
                                                    "Unknown subscription response method: {method}"
                                                );
                                            }
                                        }
                                        continue;
                                    }

                                    let ts_init = self.clock.get_time_ns();

                                    let nautilus_msgs = Self::parse_to_nautilus_messages(
                                        msg,
                                        &self.instruments,
                                        &self.cloid_cache,
                                        &self.bar_types_cache,
                                        self.account_id,
                                        ts_init,
                                        &self.asset_context_subs,
                                        &self.trade_subs,
                                        &self.depth10_subs,
                                        &mut self.processed_trade_ids,
                                        &mut self.processed_public_trade_ids,
                                        &mut self.asset_context_caches,
                                        &mut self.bar_cache,
                                        &self.all_dex_asset_ctxs_instrument_ids,
                                        self.all_mids_data_types.as_slice(),
                                    );

                                    if !nautilus_msgs.is_empty() {
                                        let mut iter = nautilus_msgs.into_iter();
                                        let first = iter.next().unwrap();
                                        self.message_buffer.extend(iter);
                                        return Some(first);
                                    }
                                }
                                Err(e) => {
                                    log::error!("Error parsing WebSocket message: {e}, text: {text}");
                                }
                            }
                        }
                        Message::Ping(data) => {
                            if let Some(ref client) = self.client {
                                self.rate_limits.acquire_message().await;

                                if let Err(e) = client.send_pong(data.to_vec()).await {
                                    log::error!("Error sending pong: {e}");
                                }
                            }
                        }
                        Message::Close(_) => {
                            log::debug!("Received WebSocket close frame");
                            return None;
                        }
                        _ => {}
                    }
                }

                _ = self.heartbeat.tick() => {
                    if self.client.as_ref().is_some_and(WebSocketClient::is_active)
                        && let Err(e) = self.send_with_retry(
                            HEARTBEAT_MESSAGE.to_string(),
                        ).await
                    {
                        log::error!("Error sending WebSocket heartbeat: {e}");
                    }
                }

            }
        }
    }

    async fn subscribe(&mut self, subscriptions: Vec<SubscriptionRequest>) {
        for subscription in subscriptions {
            let key = subscription_to_key(&subscription);
            self.subscriptions.mark_subscribe(&key);

            if let Err(e) = self
                .rate_limits
                .reserve_subscription(self.client_id, &subscription)
            {
                log::error!("Cannot subscribe to {key}: {e}");
                self.subscriptions.mark_unsubscribe(&key);
                self.subscriptions.confirm_unsubscribe(&key);
                continue;
            }

            self.all_mids_data_types.apply(&subscription, true);
            let request = HyperliquidWsRequest::Subscribe { subscription };
            match serde_json::to_string(&request) {
                Ok(payload) => {
                    log::debug!("Sending subscribe payload ({} bytes)", payload.len());
                    if let Err(e) = self.send_with_retry(payload).await {
                        log::error!("Error subscribing to {key}: {e}");
                        self.subscriptions.mark_failure(&key);
                    }
                }
                Err(e) => {
                    log::error!("Error serializing subscription for {key}: {e}");
                    self.subscriptions.mark_failure(&key);
                }
            }
        }
    }

    async fn unsubscribe(&mut self, subscriptions: Vec<SubscriptionRequest>) {
        for subscription in subscriptions {
            let key = subscription_to_key(&subscription);
            self.subscriptions.mark_unsubscribe(&key);
            self.all_mids_data_types.apply(&subscription, false);

            let request = HyperliquidWsRequest::Unsubscribe { subscription };
            match serde_json::to_string(&request) {
                Ok(payload) => {
                    log::debug!("Sending unsubscribe payload ({} bytes)", payload.len());
                    if let Err(e) = self.send_with_retry(payload).await {
                        log::error!("Error unsubscribing from {key}: {e}");
                    }
                }
                Err(e) => {
                    log::error!("Error serializing unsubscription for {key}: {e}");
                }
            }
        }
    }

    #[expect(clippy::too_many_arguments)]
    fn parse_to_nautilus_messages(
        msg: HyperliquidWsMessage,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        cloid_cache: &CloidCache,
        bar_types: &AHashMap<String, BarType>,
        account_id: Option<AccountId>,
        ts_init: UnixNanos,
        asset_context_subs: &AHashMap<Ustr, AHashSet<AssetContextDataType>>,
        trade_subs: &AHashMap<Ustr, TradeStreamUses>,
        depth10_subs: &AHashSet<Ustr>,
        processed_trade_ids: &mut FifoCache<u64, 10_000>,
        processed_public_trade_ids: &mut FifoCache<(Ustr, u64), 10_000>,
        asset_context_caches: &mut AssetContextCaches,
        bar_cache: &mut AHashMap<String, CandleData>,
        all_dex_asset_ctxs_instrument_ids: &AHashMap<Ustr, Vec<Option<InstrumentId>>>,
        all_mids_data_types: &[DataType],
    ) -> Vec<NautilusWsMessage> {
        let mut result = Vec::new();

        match msg {
            HyperliquidWsMessage::OrderUpdates { data } => {
                if let Some(account_id) = account_id
                    && let Some(msg) = Self::handle_order_updates(
                        &data,
                        instruments,
                        cloid_cache,
                        account_id,
                        ts_init,
                    )
                {
                    result.push(msg);
                }
            }
            HyperliquidWsMessage::UserEvents { data } | HyperliquidWsMessage::User { data } => {
                // Process fills from userEvents channel (userFills channel is redundant)
                match data {
                    WsUserEventData::Fills { fills } => {
                        log::debug!("Received {} fill(s) from userEvents channel", fills.len());
                        for fill in &fills {
                            log::debug!(
                                "Fill: oid={}, coin={}, side={:?}, sz={}, px={}",
                                fill.oid,
                                fill.coin,
                                fill.side,
                                fill.sz,
                                fill.px
                            );
                        }

                        if let Some(account_id) = account_id {
                            log::debug!("Processing fills with account_id={account_id}");

                            if let Some(msg) = Self::handle_user_fills(
                                &fills,
                                instruments,
                                cloid_cache,
                                account_id,
                                ts_init,
                                processed_trade_ids,
                            ) {
                                log::debug!("Successfully created fill message");
                                result.push(msg);
                            } else {
                                log::debug!("handle_user_fills returned None (no new fills)");
                            }
                        } else {
                            log::warn!("Cannot process fills: account_id is None");
                        }
                    }
                    WsUserEventData::Liquidation { liquidation } => {
                        log::warn!(
                            "Liquidation event: lid={}, liquidator={}, liquidated_user={}, ntl_pos={}, account_value={}",
                            liquidation.lid,
                            liquidation.liquidator,
                            liquidation.liquidated_user,
                            liquidation.liquidated_ntl_pos,
                            liquidation.liquidated_account_value,
                        );
                    }
                    _ => {
                        log::debug!("Received non-fill user event: {data:?}");
                    }
                }
            }
            HyperliquidWsMessage::UserFills { data } => {
                // UserFills channel is redundant with userEvents, but handle it for
                // backwards compatibility if explicitly subscribed
                if let Some(account_id) = account_id
                    && let Some(msg) = Self::handle_user_fills(
                        &data.fills,
                        instruments,
                        cloid_cache,
                        account_id,
                        ts_init,
                        processed_trade_ids,
                    )
                {
                    result.push(msg);
                }
            }
            HyperliquidWsMessage::Trades { data } => {
                result.extend(Self::handle_trades(
                    &data,
                    instruments,
                    trade_subs,
                    processed_public_trade_ids,
                    ts_init,
                ));
            }
            HyperliquidWsMessage::AllMids { data } => {
                let mut mids = std::collections::HashMap::with_capacity(
                    data.mids.len().min(instruments.len()),
                );

                for (coin, mid_str) in &data.mids {
                    if let Some(instrument) = instruments.get(coin) {
                        match mid_str.parse::<Price>() {
                            Ok(price) => {
                                mids.insert(instrument.id(), price);
                            }
                            Err(e) => {
                                log::warn!("Failed to parse mid price for {coin}: {e}");
                            }
                        }
                    } else {
                        log::debug!("No instrument found for coin: {coin}");
                    }
                }

                if !mids.is_empty() {
                    // Take instead of clone on the last subscriber
                    let last_idx = all_mids_data_types.len().saturating_sub(1);
                    for (i, data_type) in all_mids_data_types.iter().enumerate() {
                        let mids_for_this = if i == last_idx {
                            std::mem::take(&mut mids)
                        } else {
                            mids.clone()
                        };
                        let all_mids = HyperliquidAllMids::new(mids_for_this, ts_init, ts_init);
                        result.push(NautilusWsMessage::CustomData(Data::Custom(
                            CustomData::new(Arc::new(all_mids), data_type.clone()),
                        )));
                    }
                }
            }
            HyperliquidWsMessage::AllDexsAssetCtxs { data } => {
                if let Some(msg) = Self::handle_all_dexs_asset_ctxs(
                    data,
                    all_dex_asset_ctxs_instrument_ids,
                    ts_init,
                ) {
                    result.push(msg);
                }
            }
            HyperliquidWsMessage::Bbo { data } => {
                if let Some(msg) = Self::handle_bbo(&data, instruments, ts_init) {
                    result.push(msg);
                }
            }
            HyperliquidWsMessage::L2Book { data } => {
                result.extend(Self::handle_l2_book(
                    &data,
                    instruments,
                    depth10_subs,
                    ts_init,
                ));
            }
            HyperliquidWsMessage::Candle { data } => {
                if let Some(msg) =
                    Self::handle_candle(&data, instruments, bar_types, bar_cache, ts_init)
                {
                    result.push(msg);
                }
            }
            HyperliquidWsMessage::ActiveAssetCtx { data }
            | HyperliquidWsMessage::ActiveSpotAssetCtx { data } => {
                result.extend(Self::handle_asset_context(
                    &data,
                    instruments,
                    asset_context_subs,
                    asset_context_caches,
                    ts_init,
                ));
            }
            HyperliquidWsMessage::UserTwapHistory { data } => {
                result.extend(Self::handle_user_twap_history(&data, instruments, ts_init));
            }
            HyperliquidWsMessage::UserTwapSliceFills { data } => {
                result.extend(Self::handle_user_twap_slice_fills(
                    &data,
                    instruments,
                    ts_init,
                ));
            }
            HyperliquidWsMessage::Error { data } => {
                log::warn!("Received error from Hyperliquid WebSocket: {data}");
            }
            // Ignore other message types (subscription confirmations, etc)
            _ => {}
        }

        result
    }

    fn handle_order_updates(
        data: &[super::messages::WsOrderData],
        instruments: &AHashMap<Ustr, InstrumentAny>,
        cloid_cache: &CloidCache,
        account_id: AccountId,
        ts_init: UnixNanos,
    ) -> Option<NautilusWsMessage> {
        let mut exec_reports = Vec::new();

        for order_update in data {
            let instrument = instruments.get(&order_update.order.coin);

            if let Some(instrument) = instrument {
                match parse_ws_order_status_report(order_update, instrument, account_id, ts_init) {
                    Ok(mut report) => {
                        // Resolve cloid to real client_order_id if cached
                        if let Some(cloid) = &order_update.order.cloid {
                            let cloid_ustr = Ustr::from(cloid.as_str());
                            let resolved = cloid_cache.lock().get(&cloid_ustr).copied();

                            if let Some(real_client_order_id) = resolved {
                                log::debug!("Resolved cloid {cloid} -> {real_client_order_id}");
                                report.client_order_id = Some(real_client_order_id);
                            }
                        }
                        exec_reports.push(ExecutionReport::Order(report));
                    }
                    Err(e) => {
                        log::error!("Error parsing order update: {e}");
                    }
                }
            } else {
                log::debug!("No instrument found for coin: {}", order_update.order.coin);
            }
        }

        if exec_reports.is_empty() {
            None
        } else {
            Some(NautilusWsMessage::ExecutionReports(exec_reports))
        }
    }

    fn handle_user_fills(
        fills: &[super::messages::WsFillData],
        instruments: &AHashMap<Ustr, InstrumentAny>,
        cloid_cache: &CloidCache,
        account_id: AccountId,
        ts_init: UnixNanos,
        processed_trade_ids: &mut FifoCache<u64, 10_000>,
    ) -> Option<NautilusWsMessage> {
        let mut exec_reports = Vec::new();

        for fill in fills {
            if processed_trade_ids.contains(&fill.tid) {
                log::debug!("Skipping duplicate fill: tid={}", fill.tid);
                continue;
            }

            let instrument = instruments.get(&fill.coin);

            if let Some(instrument) = instrument {
                log::debug!("Found instrument for fill coin={}", fill.coin);
                match parse_ws_fill_report(fill, instrument, account_id, ts_init) {
                    Ok(mut report) => {
                        // Mark processed only after successful parse
                        processed_trade_ids.add(fill.tid);

                        if let Some(cloid) = &fill.cloid {
                            let cloid_ustr = Ustr::from(cloid.as_str());
                            let resolved = cloid_cache.lock().get(&cloid_ustr).copied();

                            if let Some(real_client_order_id) = resolved {
                                log::debug!(
                                    "Resolved fill cloid {cloid} -> {real_client_order_id}"
                                );
                                report.client_order_id = Some(real_client_order_id);
                            }
                        }
                        log::debug!(
                            "Parsed fill report: venue_order_id={:?}, trade_id={:?}",
                            report.venue_order_id,
                            report.trade_id
                        );
                        exec_reports.push(ExecutionReport::Fill(report));
                    }
                    Err(e) => {
                        log::error!("Error parsing fill: {e}");
                    }
                }
            } else {
                // Not marked as processed so fill is retried if instrument loads later
                log::warn!("No instrument found for fill coin={}", fill.coin);
            }
        }

        if exec_reports.is_empty() {
            None
        } else {
            Some(NautilusWsMessage::ExecutionReports(exec_reports))
        }
    }

    fn handle_trades(
        data: &[super::messages::WsTradeData],
        instruments: &AHashMap<Ustr, InstrumentAny>,
        trade_subs: &AHashMap<Ustr, TradeStreamUses>,
        processed_public_trade_ids: &mut FifoCache<(Ustr, u64), 10_000>,
        ts_init: UnixNanos,
    ) -> Vec<NautilusWsMessage> {
        let mut trade_ticks = Vec::new();
        let mut public_trades = Vec::new();

        for trade in data {
            if let Some(instrument) = instruments.get(&trade.coin) {
                let uses = trade_subs.get(&trade.coin).copied().unwrap_or_default();

                if uses.ticks {
                    match parse_ws_trade_tick(trade, instrument, ts_init) {
                        Ok(tick) => trade_ticks.push(tick),
                        Err(e) => {
                            log::error!("Error parsing trade tick: {e}");
                        }
                    }
                }

                if uses.public_trades {
                    let trade_key = (trade.coin, trade.tid);
                    if processed_public_trade_ids.contains(&trade_key) {
                        log::debug!(
                            "Skipping replayed public trade: coin={}, tid={}",
                            trade.coin,
                            trade.tid
                        );
                        continue;
                    }

                    match parse_ws_public_trade(trade, instrument, ts_init) {
                        Ok(trade) => {
                            processed_public_trade_ids.add(trade_key);
                            public_trades.push(trade);
                        }
                        Err(e) => {
                            log::error!("Error parsing public trade: {e}");
                        }
                    }
                }
            } else {
                log::debug!("No instrument found for coin: {}", trade.coin);
            }
        }

        let mut result = Vec::with_capacity(1 + public_trades.len());
        if !trade_ticks.is_empty() {
            result.push(NautilusWsMessage::Trades(trade_ticks));
        }
        result.extend(public_trades.into_iter().map(|trade| {
            let instrument_id = trade.instrument_id;
            NautilusWsMessage::CustomData(Data::Custom(CustomData::new(
                Arc::new(trade),
                Self::public_trade_data_type(instrument_id),
            )))
        }));
        result
    }

    fn handle_bbo(
        data: &super::messages::WsBboData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        ts_init: UnixNanos,
    ) -> Option<NautilusWsMessage> {
        if let Some(instrument) = instruments.get(&data.coin) {
            match parse_ws_quote_tick(data, instrument, ts_init) {
                Ok(quote_tick) => Some(NautilusWsMessage::Quote(quote_tick)),
                Err(e) => {
                    log::error!("Error parsing quote tick: {e}");
                    None
                }
            }
        } else {
            log::debug!("No instrument found for coin: {}", data.coin);
            None
        }
    }

    fn handle_l2_book(
        data: &super::messages::WsBookData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        depth10_subs: &AHashSet<Ustr>,
        ts_init: UnixNanos,
    ) -> Vec<NautilusWsMessage> {
        let mut out = Vec::new();

        let Some(instrument) = instruments.get(&data.coin) else {
            log::debug!("No instrument found for coin: {}", data.coin);
            return out;
        };

        match parse_ws_order_book_deltas(data, instrument, ts_init) {
            Ok(deltas) => out.push(NautilusWsMessage::Deltas(deltas)),
            Err(e) => log::error!("Error parsing order book deltas: {e}"),
        }

        if depth10_subs.contains(&data.coin) {
            match parse_ws_order_book_depth10(data, instrument, ts_init) {
                Ok(depth) => out.push(NautilusWsMessage::Depth10(Box::new(depth))),
                Err(e) => log::error!("Error parsing order book depth10: {e}"),
            }
        }

        out
    }

    fn handle_candle(
        data: &CandleData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        bar_types: &AHashMap<String, BarType>,
        bar_cache: &mut AHashMap<String, CandleData>,
        ts_init: UnixNanos,
    ) -> Option<NautilusWsMessage> {
        let key = format!("candle:{}:{}", data.s, data.i);

        let mut closed_bar = None;

        if let Some(cached) = bar_cache.get(&key) {
            // Emit cached bar when close_time changes, indicating the previous period closed
            if cached.close_time != data.close_time {
                log::debug!(
                    "Bar period changed for {}: prev_close_time={}, new_close_time={}",
                    data.s,
                    cached.close_time,
                    data.close_time
                );
                closed_bar = Some(cached.clone());
            }
        }

        bar_cache.insert(key.clone(), data.clone());

        if let Some(closed_data) = closed_bar {
            if let Some(bar_type) = bar_types.get(&key) {
                if let Some(instrument) = instruments.get(&data.s) {
                    match parse_ws_candle(&closed_data, instrument, bar_type, ts_init) {
                        Ok(bar) => return Some(NautilusWsMessage::Candle(bar)),
                        Err(e) => {
                            log::error!("Error parsing closed candle: {e}");
                        }
                    }
                } else {
                    log::debug!("No instrument found for coin: {}", data.s);
                }
            } else {
                log::debug!("No bar type found for key: {key}");
            }
        }

        None
    }

    fn handle_asset_context(
        data: &WsActiveAssetCtxData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        asset_context_subs: &AHashMap<Ustr, AHashSet<AssetContextDataType>>,
        asset_context_caches: &mut AssetContextCaches,
        ts_init: UnixNanos,
    ) -> Vec<NautilusWsMessage> {
        let mut result = Vec::new();

        let coin = match data {
            WsActiveAssetCtxData::Perp { coin, .. } => coin,
            WsActiveAssetCtxData::Spot { coin, .. } => coin,
        };

        if let Some(instrument) = instruments.get(coin) {
            let (mark_px, oracle_px, funding, open_interest) = match data {
                WsActiveAssetCtxData::Perp { ctx, .. } => (
                    &ctx.shared.mark_px,
                    Some(&ctx.oracle_px),
                    Some(&ctx.funding),
                    Some(&ctx.open_interest),
                ),
                WsActiveAssetCtxData::Spot { ctx, .. } => (&ctx.shared.mark_px, None, None, None),
            };

            let mark_changed = asset_context_caches.mark_price.get(coin) != Some(mark_px);
            let index_changed =
                oracle_px.is_some_and(|px| asset_context_caches.index_price.get(coin) != Some(px));
            let funding_changed = funding
                .is_some_and(|rate| asset_context_caches.funding_rate.get(coin) != Some(rate));
            let open_interest_changed = open_interest
                .is_some_and(|value| asset_context_caches.open_interest.get(coin) != Some(value));

            let subscribed_types = asset_context_subs.get(coin);

            if mark_changed || index_changed || funding_changed {
                match parse_ws_asset_context(data, instrument, ts_init) {
                    Ok((mark_price, index_price, funding_rate)) => {
                        if mark_changed
                            && subscribed_types
                                .is_some_and(|s| s.contains(&AssetContextDataType::MarkPrice))
                        {
                            asset_context_caches.mark_price.insert(*coin, *mark_px);
                            result.push(NautilusWsMessage::MarkPrice(mark_price));
                        }

                        if index_changed
                            && subscribed_types
                                .is_some_and(|s| s.contains(&AssetContextDataType::IndexPrice))
                        {
                            if let Some(px) = oracle_px {
                                asset_context_caches.index_price.insert(*coin, *px);
                            }

                            if let Some(index) = index_price {
                                result.push(NautilusWsMessage::IndexPrice(index));
                            }
                        }

                        if funding_changed
                            && subscribed_types
                                .is_some_and(|s| s.contains(&AssetContextDataType::FundingRate))
                        {
                            if let Some(rate) = funding {
                                asset_context_caches.funding_rate.insert(*coin, *rate);
                            }

                            if let Some(funding) = funding_rate {
                                result.push(NautilusWsMessage::FundingRate(funding));
                            }
                        }
                    }
                    Err(e) => {
                        log::error!("Error parsing asset context: {e}");
                    }
                }
            }

            if let Some(value) = open_interest
                && open_interest_changed
                && subscribed_types.is_some_and(|s| s.contains(&AssetContextDataType::OpenInterest))
            {
                match parse_ws_open_interest(*value, instrument, ts_init) {
                    Ok(open_interest_data) => {
                        asset_context_caches.open_interest.insert(*coin, *value);

                        let data_type =
                            Self::open_interest_data_type(open_interest_data.instrument_id);
                        result.push(NautilusWsMessage::CustomData(Data::Custom(
                            CustomData::new(Arc::new(open_interest_data), data_type),
                        )));
                    }
                    Err(e) => {
                        log::error!("Error parsing open interest: {e}");
                    }
                }
            }
        } else {
            log::debug!("No instrument found for coin: {coin}");
        }

        result
    }

    fn handle_all_dexs_asset_ctxs(
        data: WsAllDexsAssetCtxsData,
        all_dex_asset_ctxs_instrument_ids: &AHashMap<Ustr, Vec<Option<InstrumentId>>>,
        ts_init: UnixNanos,
    ) -> Option<NautilusWsMessage> {
        let mut entries = Vec::new();

        for (dex, ctxs) in data.ctxs {
            let dex_key = Ustr::from(dex.as_str());
            let Some(instrument_ids) = all_dex_asset_ctxs_instrument_ids.get(&dex_key) else {
                log::warn!("Missing Hyperliquid allDexsAssetCtxs mapping for dex='{dex}'");
                continue;
            };

            if ctxs.len() != instrument_ids.len() {
                // Mapping is rebuilt on each instrument refresh, instrument request, and data
                // client connect, so a count change means the universe drifted since the last
                // build and positional alignment can no longer be trusted until the next one.
                log::warn!(
                    "Hyperliquid allDexsAssetCtxs count mismatch for dex='{dex}': received {} contexts but cached {} instrument IDs (the next instrument refresh, instrument request, or data client connect rebuilds the mapping)",
                    ctxs.len(),
                    instrument_ids.len()
                );
            }

            for (index, ctx) in ctxs.into_iter().enumerate() {
                let Some(Some(instrument_id)) = instrument_ids.get(index).copied() else {
                    log::warn!(
                        "Missing Hyperliquid allDexsAssetCtxs instrument mapping for dex='{dex}' index={index}"
                    );
                    continue;
                };

                match Self::normalize_all_dex_asset_ctx_entry(&dex, instrument_id, ctx) {
                    Ok(entry) => entries.push(entry),
                    Err(e) => {
                        log::warn!(
                            "Failed to normalize Hyperliquid allDexsAssetCtxs entry dex='{dex}' index={index}: {e}"
                        );
                    }
                }
            }
        }

        if entries.is_empty() {
            return None;
        }

        let payload = HyperliquidAllDexsAssetCtxs::new(entries, ts_init, ts_init);
        let data_type = DataType::new("HyperliquidAllDexsAssetCtxs", None, None);
        Some(NautilusWsMessage::CustomData(Data::Custom(
            CustomData::new(Arc::new(payload), data_type),
        )))
    }

    fn normalize_all_dex_asset_ctx_entry(
        dex: &str,
        instrument_id: InstrumentId,
        ctx: super::messages::PerpsAssetCtx,
    ) -> anyhow::Result<HyperliquidDexAssetCtx> {
        let mark_price = Price::from_decimal(ctx.shared.mark_px).map_err(anyhow::Error::msg)?;
        let oracle_price = Price::from_decimal(ctx.oracle_px).map_err(anyhow::Error::msg)?;
        let prev_day_price =
            Price::from_decimal(ctx.shared.prev_day_px).map_err(anyhow::Error::msg)?;
        let mid_price = ctx
            .shared
            .mid_px
            .map(|value| Price::from_decimal(value).map_err(anyhow::Error::msg))
            .transpose()?;
        let funding_rate = ctx.funding;
        let open_interest = ctx.open_interest;
        let premium = ctx.premium;
        let day_ntl_volume = ctx.shared.day_ntl_vlm;
        let day_base_volume = ctx
            .shared
            .day_base_vlm
            .ok_or_else(|| anyhow::anyhow!("missing dayBaseVlm"))?;
        let impact_prices = match ctx.shared.impact_pxs {
            Some(values) => match values.as_slice() {
                [bid, ask] => Some(HyperliquidImpactPrices {
                    bid: bid.parse::<Price>().map_err(anyhow::Error::msg)?,
                    ask: ask.parse::<Price>().map_err(anyhow::Error::msg)?,
                }),
                other => {
                    anyhow::bail!("expected 2 impact prices, received {}", other.len());
                }
            },
            None => None,
        };

        Ok(HyperliquidDexAssetCtx {
            dex: dex.to_string(),
            instrument_id,
            mark_price,
            oracle_price,
            prev_day_price,
            mid_price,
            impact_prices,
            funding_rate,
            open_interest,
            premium,
            day_ntl_volume,
            day_base_volume,
        })
    }

    fn open_interest_data_type(instrument_id: InstrumentId) -> DataType {
        let mut metadata = Params::new();
        metadata.insert(
            "instrument_id".to_string(),
            serde_json::Value::String(instrument_id.to_string()),
        );
        DataType::new(
            "HyperliquidOpenInterest",
            Some(metadata),
            Some(instrument_id.to_string()),
        )
    }

    fn public_trade_data_type(instrument_id: InstrumentId) -> DataType {
        let mut metadata = Params::new();
        metadata.insert(
            "instrument_id".to_string(),
            serde_json::Value::String(instrument_id.to_string()),
        );
        DataType::new(
            "HyperliquidPublicTrade",
            Some(metadata),
            Some(instrument_id.to_string()),
        )
    }

    fn handle_user_twap_history(
        data: &super::messages::WsUserTwapHistoryData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        ts_init: UnixNanos,
    ) -> Vec<NautilusWsMessage> {
        let is_snapshot = data.is_snapshot.unwrap_or(false);
        let mut result = Vec::with_capacity(data.history.len());

        for row in &data.history {
            let instrument = instruments.get(&row.state.coin);
            match parse_ws_twap_history_row(row, &data.user, is_snapshot, instrument, ts_init) {
                Ok(payload) => {
                    let user = payload.user.clone();
                    result.push(NautilusWsMessage::CustomData(Data::Custom(
                        CustomData::new(Arc::new(payload), Self::twap_history_data_type(&user)),
                    )));
                }
                Err(e) => {
                    log::error!("Error parsing TWAP history row: {e}");
                }
            }
        }

        result
    }

    fn handle_user_twap_slice_fills(
        data: &super::messages::WsUserTwapSliceFillsData,
        instruments: &AHashMap<Ustr, InstrumentAny>,
        ts_init: UnixNanos,
    ) -> Vec<NautilusWsMessage> {
        let is_snapshot = data.is_snapshot.unwrap_or(false);
        let mut result = Vec::with_capacity(data.twap_slice_fills.len());

        for item in &data.twap_slice_fills {
            let instrument = instruments.get(&item.fill.coin);
            match parse_ws_twap_slice_fill(item, &data.user, is_snapshot, instrument, ts_init) {
                Ok(payload) => {
                    let user = payload.user.clone();
                    result.push(NautilusWsMessage::CustomData(Data::Custom(
                        CustomData::new(Arc::new(payload), Self::twap_slice_fill_data_type(&user)),
                    )));
                }
                Err(e) => {
                    log::error!("Error parsing TWAP slice fill: {e}");
                }
            }
        }

        result
    }

    fn twap_history_data_type(user: &str) -> DataType {
        let mut metadata = Params::new();
        metadata.insert(
            "user".to_string(),
            serde_json::Value::String(user.to_string()),
        );
        DataType::new(
            "HyperliquidTwapHistory",
            Some(metadata),
            Some(user.to_string()),
        )
    }

    fn twap_slice_fill_data_type(user: &str) -> DataType {
        let mut metadata = Params::new();
        metadata.insert(
            "user".to_string(),
            serde_json::Value::String(user.to_string()),
        );
        DataType::new(
            "HyperliquidTwapSliceFill",
            Some(metadata),
            Some(user.to_string()),
        )
    }
}

#[derive(Debug, thiserror::Error)]
enum PostSendError {
    #[error(transparent)]
    Transport(SendError),
    #[error(transparent)]
    Retry(RetryError),
    #[error("Post deadline expired")]
    Deadline,
}

async fn send_post_with_retry<F, Fut>(
    retry_manager: &RetryManager<PostSendError>,
    deadline: tokio::time::Instant,
    cancellation_token: &CancellationToken,
    send: F,
) -> Result<(), PostSendError>
where
    F: Fn() -> Fut + Clone,
    Fut: Future<Output = Result<(), PostSendError>>,
{
    let invocation = retry_manager
        .invocation(
            "websocket_post_send",
            || {
                let send = send.clone();
                async move { send_post_before_deadline(deadline, cancellation_token, send).await }
            },
            should_retry_post_send,
            PostSendError::Retry,
        )
        .cancellation_token(cancellation_token)
        .execute();
    tokio::pin!(invocation);

    tokio::select! {
        biased;
        () = tokio::time::sleep_until(deadline) => {
            Err(PostSendError::Deadline)
        }
        result = &mut invocation => result,
    }
}

async fn send_post_before_deadline<F, Fut>(
    deadline: tokio::time::Instant,
    cancellation_token: &CancellationToken,
    send: F,
) -> Result<(), PostSendError>
where
    F: FnOnce() -> Fut,
    Fut: Future<Output = Result<(), PostSendError>>,
{
    if cancellation_token.is_cancelled() {
        return Err(PostSendError::Retry(RetryError::Canceled));
    }

    if tokio::time::Instant::now() >= deadline {
        return Err(PostSendError::Deadline);
    }

    send().await
}

fn should_retry_post_send(error: &PostSendError) -> bool {
    matches!(
        error,
        PostSendError::Transport(SendError::Timeout | SendError::ConnectionChanged)
    )
}

pub(super) fn subscription_to_key(sub: &SubscriptionRequest) -> String {
    match sub {
        SubscriptionRequest::AllMids { dex } => {
            if let Some(dex_name) = dex {
                format!("{}:{dex_name}", HyperliquidWsChannel::AllMids.as_str())
            } else {
                HyperliquidWsChannel::AllMids.as_str().to_string()
            }
        }
        SubscriptionRequest::AllDexsAssetCtxs => {
            HyperliquidWsChannel::AllDexsAssetCtxs.as_str().to_string()
        }
        SubscriptionRequest::Notification { user } => {
            format!("{}:{user}", HyperliquidWsChannel::Notification.as_str())
        }
        SubscriptionRequest::WebData2 { user } => {
            format!("{}:{user}", HyperliquidWsChannel::WebData2.as_str())
        }
        SubscriptionRequest::Candle { coin, interval } => {
            format!(
                "{}:{coin}:{}",
                HyperliquidWsChannel::Candle.as_str(),
                interval.as_str()
            )
        }
        SubscriptionRequest::L2Book { coin, .. } => {
            format!("{}:{coin}", HyperliquidWsChannel::L2Book.as_str())
        }
        SubscriptionRequest::Trades { coin } => {
            format!("{}:{coin}", HyperliquidWsChannel::Trades.as_str())
        }
        SubscriptionRequest::OrderUpdates { user } => {
            format!("{}:{user}", HyperliquidWsChannel::OrderUpdates.as_str())
        }
        SubscriptionRequest::UserEvents { user } => {
            format!("{}:{user}", HyperliquidWsChannel::UserEvents.as_str())
        }
        SubscriptionRequest::UserFills { user, .. } => {
            format!("{}:{user}", HyperliquidWsChannel::UserFills.as_str())
        }
        SubscriptionRequest::UserFundings { user } => {
            format!("{}:{user}", HyperliquidWsChannel::UserFundings.as_str())
        }
        SubscriptionRequest::UserNonFundingLedgerUpdates { user } => {
            format!(
                "{}:{user}",
                HyperliquidWsChannel::UserNonFundingLedgerUpdates.as_str()
            )
        }
        SubscriptionRequest::ActiveAssetCtx { coin } => {
            format!("{}:{coin}", HyperliquidWsChannel::ActiveAssetCtx.as_str())
        }
        SubscriptionRequest::ActiveSpotAssetCtx { coin } => {
            format!(
                "{}:{coin}",
                HyperliquidWsChannel::ActiveSpotAssetCtx.as_str()
            )
        }
        SubscriptionRequest::ActiveAssetData { user, coin } => {
            format!(
                "{}:{user}:{coin}",
                HyperliquidWsChannel::ActiveAssetData.as_str()
            )
        }
        SubscriptionRequest::UserTwapSliceFills { user } => {
            format!(
                "{}:{user}",
                HyperliquidWsChannel::UserTwapSliceFills.as_str()
            )
        }
        SubscriptionRequest::UserTwapHistory { user } => {
            format!("{}:{user}", HyperliquidWsChannel::UserTwapHistory.as_str())
        }
        SubscriptionRequest::Bbo { coin } => {
            format!("{}:{coin}", HyperliquidWsChannel::Bbo.as_str())
        }
    }
}

/// Determines whether a Hyperliquid WebSocket error should trigger a retry.
pub(crate) fn should_retry_hyperliquid_error(error: &HyperliquidWsError) -> bool {
    match error {
        HyperliquidWsError::TungsteniteError(_) => true,
        HyperliquidWsError::ClientError(msg) => {
            let msg_lower = msg.to_lowercase();
            msg_lower.contains("timeout")
                || msg_lower.contains("timed out")
                || msg_lower.contains("connection")
                || msg_lower.contains("network")
        }
        _ => false,
    }
}

/// Creates a timeout error for Hyperliquid retry logic.
pub(crate) fn create_hyperliquid_timeout_error(msg: String) -> HyperliquidWsError {
    HyperliquidWsError::ClientError(msg)
}

#[cfg(test)]
mod tests {
    use std::{
        sync::{
            Arc,
            atomic::{AtomicBool, AtomicUsize, Ordering},
        },
        time::Duration,
    };

    use ahash::{AHashMap, AHashSet};
    use log::{Level, LevelFilter, Log, Metadata, Record};
    use nautilus_common::cache::fifo::FifoCacheMap;
    use nautilus_core::nanos::UnixNanos;
    use nautilus_model::{
        data::Data,
        identifiers::{ClientOrderId, InstrumentId, Symbol},
        instruments::{CryptoPerpetual, Instrument, InstrumentAny},
        types::{Currency, Price, Quantity},
    };
    use nautilus_network::{
        error::SendError,
        retry::{RetryConfig, RetryError, RetryManager},
        websocket::SubscriptionState,
    };
    use parking_lot::Mutex;
    use rstest::rstest;
    use rust_decimal::Decimal;
    use rust_decimal_macros::dec;
    use serde_json::json;
    use tokio_tungstenite::tungstenite::Message;
    use tokio_util::sync::CancellationToken;
    use ustr::Ustr;

    use super::{
        super::{
            client::{AssetContextDataType, CLOID_CACHE_CAPACITY, CloidCache},
            messages::{
                HyperliquidWsRequest, NautilusWsMessage, PerpsAssetCtx, PostRequest,
                SharedAssetCtx, SpotAssetCtx, SubscriptionRequest, WsActiveAssetCtxData,
                WsAllDexsAssetCtxsData, WsBookData, WsLevelData,
            },
            post::PostRouter,
            rate_limits::WebSocketRateLimits,
        },
        AllMidsDataTypeCache, AssetContextCaches, FeedHandler, HandlerCommand, PostSendError,
        send_post_before_deadline, send_post_with_retry, should_retry_post_send,
    };
    use crate::{
        common::consts::{
            HYPERLIQUID_VENUE, HYPERLIQUID_WS_MESSAGES_PER_MINUTE, HYPERLIQUID_WS_SUBSCRIPTIONS_MAX,
        },
        data_types::{HyperliquidAllDexsAssetCtxs, HyperliquidOpenInterest},
    };

    const SECRET_MARKER: &str = "OUTBOUND_SECRET_MARKER";

    struct OutboundLogCapture {
        messages: Mutex<Vec<String>>,
    }

    static OUTBOUND_LOG_CAPTURE: OutboundLogCapture = OutboundLogCapture {
        messages: Mutex::new(Vec::new()),
    };

    impl OutboundLogCapture {
        fn clear(&self) {
            self.messages.lock().clear();
        }

        fn messages(&self) -> Vec<String> {
            self.messages.lock().clone()
        }
    }

    impl Log for OutboundLogCapture {
        fn enabled(&self, metadata: &Metadata<'_>) -> bool {
            metadata.level() == Level::Debug
                && metadata.target() == "nautilus_hyperliquid::websocket::handler"
        }

        fn log(&self, record: &Record<'_>) {
            if self.enabled(record.metadata()) {
                let message = record.args().to_string();
                if message.starts_with("Sending ") {
                    self.messages.lock().push(message);
                }
            }
        }

        fn flush(&self) {}
    }

    #[rstest]
    fn all_mids_cache_projects_subscriptions_without_scanning_every_websocket_message() {
        let mut cache = AllMidsDataTypeCache::default();

        assert_eq!(cache.as_slice().len(), 1);
        assert!(cache.as_slice()[0].metadata().is_none());

        cache.apply(
            &SubscriptionRequest::AllMids {
                dex: Some("xyz".to_owned()),
            },
            true,
        );
        assert_eq!(cache.as_slice().len(), 1);
        assert_eq!(
            cache.as_slice()[0]
                .metadata()
                .and_then(|metadata| metadata.get_str("dex")),
            Some("xyz"),
        );

        cache.apply(&SubscriptionRequest::AllMids { dex: None }, true);
        assert_eq!(cache.as_slice().len(), 2);

        cache.apply(
            &SubscriptionRequest::AllMids {
                dex: Some("xyz".to_owned()),
            },
            false,
        );
        assert_eq!(cache.as_slice().len(), 1);
        assert_eq!(cache.as_slice()[0].type_name(), "HyperliquidAllMids");
        assert!(cache.as_slice()[0].metadata().is_none());

        cache.apply(&SubscriptionRequest::AllMids { dex: None }, false);
        assert_eq!(cache.as_slice().len(), 1);
        assert_eq!(cache.as_slice()[0].type_name(), "HyperliquidAllMids");
        assert!(cache.as_slice()[0].metadata().is_none());
    }

    fn btc_perp() -> InstrumentAny {
        InstrumentAny::CryptoPerpetual(
            CryptoPerpetual::builder()
                .instrument_id(InstrumentId::new(
                    Symbol::new("BTC-PERP"),
                    *HYPERLIQUID_VENUE,
                ))
                .raw_symbol(Symbol::new("BTC-PERP"))
                .base_currency(Currency::from("BTC"))
                .quote_currency(Currency::from("USDC"))
                .settlement_currency(Currency::from("USDC"))
                .is_inverse(false)
                .price_precision(2)
                .size_precision(3)
                .price_increment(Price::from("0.01"))
                .size_increment(Quantity::from("0.001"))
                .ts_event(UnixNanos::default())
                .ts_init(UnixNanos::default())
                .build()
                .unwrap(),
        )
    }

    fn one_level_book() -> WsBookData {
        WsBookData {
            coin: Ustr::from("BTC"),
            levels: [
                vec![WsLevelData {
                    px: dec!(100.00),
                    sz: dec!(1.0),
                    n: 1,
                }],
                vec![WsLevelData {
                    px: dec!(100.01),
                    sz: dec!(1.0),
                    n: 1,
                }],
            ],
            time: 1_700_000_000_000,
        }
    }

    fn btc_active_spot_asset_ctx() -> WsActiveAssetCtxData {
        WsActiveAssetCtxData::Spot {
            coin: Ustr::from("BTC"),
            ctx: SpotAssetCtx {
                shared: SharedAssetCtx {
                    day_ntl_vlm: dec!(1000000.0),
                    prev_day_px: dec!(49000.0),
                    mark_px: dec!(50000.0),
                    mid_px: Some(dec!(50001.0)),
                    impact_pxs: None,
                    day_base_vlm: Some(dec!(100.0)),
                },
                circulating_supply: dec!(19000000.0),
            },
        }
    }

    fn btc_active_asset_ctx(open_interest: Decimal) -> WsActiveAssetCtxData {
        WsActiveAssetCtxData::Perp {
            coin: Ustr::from("BTC"),
            ctx: PerpsAssetCtx {
                shared: SharedAssetCtx {
                    day_ntl_vlm: dec!(1000000.0),
                    prev_day_px: dec!(49000.0),
                    mark_px: dec!(50000.0),
                    mid_px: Some(dec!(50001.0)),
                    impact_pxs: Some(vec!["50000.0".to_string(), "50002.0".to_string()]),
                    day_base_vlm: Some(dec!(100.0)),
                },
                funding: dec!(0.0001),
                open_interest,
                oracle_px: dec!(50005.0),
                premium: Some(dec!(-0.0001)),
            },
        }
    }

    fn sample_all_dexs_asset_ctxs() -> WsAllDexsAssetCtxsData {
        let raw = include_str!("../../test_data/ws_all_dexs_asset_ctxs.json");
        let msg: super::super::messages::HyperliquidWsMessage =
            serde_json::from_str(raw).expect("expected valid allDexsAssetCtxs fixture");

        let super::super::messages::HyperliquidWsMessage::AllDexsAssetCtxs { data } = msg else {
            panic!("expected allDexsAssetCtxs fixture message");
        };

        let default_entry = data
            .ctxs
            .iter()
            .find(|(dex, _)| dex.is_empty())
            .and_then(|(dex, ctxs)| ctxs.first().cloned().map(|ctx| (dex.clone(), vec![ctx])))
            .expect("expected default dex sample");
        let xyz_entry = data
            .ctxs
            .iter()
            .find(|(dex, _)| dex == "xyz")
            .and_then(|(dex, ctxs)| ctxs.first().cloned().map(|ctx| (dex.clone(), vec![ctx])))
            .expect("expected xyz dex sample");

        WsAllDexsAssetCtxsData {
            ctxs: vec![default_entry, xyz_entry],
        }
    }

    #[tokio::test]
    async fn post_send_failure_cancels_router_waiter() {
        let signal = Arc::new(AtomicBool::new(false));
        let (cmd_tx, cmd_rx) = tokio::sync::mpsc::unbounded_channel();
        let (raw_tx, raw_rx) = tokio::sync::mpsc::unbounded_channel();
        let (out_tx, _out_rx) = tokio::sync::mpsc::unbounded_channel();
        let post_router = PostRouter::new();
        let cloid_cache: CloidCache = Arc::new(Mutex::new(FifoCacheMap::<
            Ustr,
            ClientOrderId,
            CLOID_CACHE_CAPACITY,
        >::new()));
        let mut handler = FeedHandler::new(
            signal,
            cmd_rx,
            raw_rx,
            out_tx,
            None,
            SubscriptionState::new(':'),
            cloid_cache,
            Arc::clone(&post_router),
            Arc::new(WebSocketRateLimits::new()),
            1,
        );

        let id = 99;
        let cancellation_token = CancellationToken::new();
        let rx = post_router
            .register_with_cancellation(id, &cancellation_token)
            .await
            .unwrap();

        let task = tokio::spawn(async move { handler.next().await });

        cmd_tx
            .send(HandlerCommand::Post {
                id,
                request: PostRequest::Info {
                    payload: json!({"type": "userRateLimit", "user": "0x123"}),
                },
                deadline: tokio::time::Instant::now() + Duration::from_secs(1),
                cancellation_token,
            })
            .unwrap();
        drop(cmd_tx);
        drop(raw_tx);

        let closed = tokio::time::timeout(Duration::from_millis(100), rx)
            .await
            .expect("post waiter should close without waiting for post timeout");
        assert!(closed.is_err(), "post router cancel must close the waiter");
        let _rx = post_router
            .register(id)
            .await
            .expect("post id should be reusable after cancellation");
        assert!(task.await.unwrap().is_none());
    }

    fn retry_manager_with_backoff() -> RetryManager<PostSendError> {
        RetryManager::new(RetryConfig {
            max_retries: 1,
            initial_delay_ms: 1_000,
            max_delay_ms: 1_000,
            backoff_factor: 1.0,
            jitter_ms: 0,
            operation_timeout_ms: None,
            immediate_first: false,
            max_elapsed_ms: None,
        })
    }

    #[rstest]
    #[tokio::test(start_paused = true)]
    async fn expired_post_deadline_prevents_first_send() {
        let cancellation_token = CancellationToken::new();
        let sends = Arc::new(AtomicUsize::new(0));
        let send_count = Arc::clone(&sends);

        let error = send_post_before_deadline(
            tokio::time::Instant::now(),
            &cancellation_token,
            move || {
                let send_count = Arc::clone(&send_count);
                async move {
                    send_count.fetch_add(1, Ordering::SeqCst);
                    Ok(())
                }
            },
        )
        .await
        .unwrap_err();

        assert!(matches!(error, PostSendError::Deadline));
        assert_eq!(sends.load(Ordering::SeqCst), 0);
    }

    #[rstest]
    #[tokio::test(start_paused = true)]
    async fn post_deadline_while_waiting_for_message_quota_prevents_send() {
        let manager = retry_manager_with_backoff();
        let cancellation_token = CancellationToken::new();
        let limits = Arc::new(WebSocketRateLimits::new());
        let rate_key = limits.message_key();
        for _ in 0..HYPERLIQUID_WS_MESSAGES_PER_MINUTE {
            assert!(limits.messages.check_key(&rate_key).is_ok());
        }
        let sends = Arc::new(AtomicUsize::new(0));
        let send_count = Arc::clone(&sends);
        let send_limits = Arc::clone(&limits);
        let deadline = tokio::time::Instant::now() + Duration::from_millis(10);

        let error = send_post_with_retry(&manager, deadline, &cancellation_token, move || {
            let send_count = Arc::clone(&send_count);
            let send_limits = Arc::clone(&send_limits);
            async move {
                send_limits.acquire_message().await;
                send_count.fetch_add(1, Ordering::SeqCst);
                Ok(())
            }
        })
        .await
        .unwrap_err();

        assert!(matches!(error, PostSendError::Deadline));
        assert_eq!(sends.load(Ordering::SeqCst), 0);
    }

    #[rstest]
    #[tokio::test(start_paused = true)]
    async fn post_deadline_during_backoff_prevents_retry() {
        let manager = retry_manager_with_backoff();
        let cancellation_token = CancellationToken::new();
        let sends = Arc::new(AtomicUsize::new(0));
        let send_count = Arc::clone(&sends);
        let deadline = tokio::time::Instant::now() + Duration::from_millis(100);

        let error = send_post_with_retry(&manager, deadline, &cancellation_token, move || {
            let send_count = Arc::clone(&send_count);
            async move {
                send_count.fetch_add(1, Ordering::SeqCst);
                Err(PostSendError::Transport(SendError::Timeout))
            }
        })
        .await
        .unwrap_err();

        assert!(matches!(error, PostSendError::Deadline));
        assert_eq!(sends.load(Ordering::SeqCst), 1);
    }

    #[rstest]
    #[tokio::test(start_paused = true)]
    async fn post_deadline_after_send_starts_preserves_unknown_outcome() {
        let manager = retry_manager_with_backoff();
        let cancellation_token = CancellationToken::new();
        let sends = Arc::new(AtomicUsize::new(0));
        let send_count = Arc::clone(&sends);
        let deadline = tokio::time::Instant::now() + Duration::from_millis(100);

        let error = send_post_with_retry(&manager, deadline, &cancellation_token, move || {
            let send_count = Arc::clone(&send_count);
            async move {
                send_count.fetch_add(1, Ordering::SeqCst);
                std::future::pending::<Result<(), PostSendError>>().await
            }
        })
        .await
        .unwrap_err();

        assert!(matches!(error, PostSendError::Deadline));
        assert_eq!(sends.load(Ordering::SeqCst), 1);
    }

    #[rstest]
    #[tokio::test(start_paused = true)]
    async fn post_cancellation_stops_started_send() {
        let manager = retry_manager_with_backoff();
        let cancellation_token = CancellationToken::new();
        let task_cancellation_token = cancellation_token.clone();
        let started = Arc::new(tokio::sync::Notify::new());
        let task_started = Arc::clone(&started);
        let sends = Arc::new(AtomicUsize::new(0));
        let task_sends = Arc::clone(&sends);
        let deadline = tokio::time::Instant::now() + Duration::from_secs(60);
        let task = tokio::spawn(async move {
            send_post_with_retry(&manager, deadline, &task_cancellation_token, move || {
                let task_started = Arc::clone(&task_started);
                let task_sends = Arc::clone(&task_sends);
                async move {
                    task_sends.fetch_add(1, Ordering::SeqCst);
                    task_started.notify_one();
                    std::future::pending::<Result<(), PostSendError>>().await
                }
            })
            .await
        });

        started.notified().await;
        cancellation_token.cancel();
        let error = task.await.unwrap().unwrap_err();

        assert!(matches!(error, PostSendError::Retry(RetryError::Canceled)));
        assert_eq!(sends.load(Ordering::SeqCst), 1);
    }

    #[rstest]
    #[case(SendError::Timeout, true)]
    #[case(SendError::ConnectionChanged, true)]
    #[case(SendError::WriteTimeout, false)]
    #[case(SendError::BrokenPipe("transport failed".to_string()), false)]
    #[case(SendError::Closed, false)]
    #[case(SendError::InvalidInput("invalid payload".to_string()), false)]
    fn post_send_retries_only_before_writing(#[case] error: SendError, #[case] expected: bool) {
        assert_eq!(
            should_retry_post_send(&PostSendError::Transport(error)),
            expected
        );
    }

    #[rstest]
    #[tokio::test]
    async fn outbound_subscription_logs_omit_payload_bodies() {
        log::set_logger(&OUTBOUND_LOG_CAPTURE).expect("test logger already installed");
        log::set_max_level(LevelFilter::Debug);

        let signal = Arc::new(AtomicBool::new(false));
        let (cmd_tx, cmd_rx) = tokio::sync::mpsc::unbounded_channel();
        let (raw_tx, raw_rx) = tokio::sync::mpsc::unbounded_channel();
        let (out_tx, _out_rx) = tokio::sync::mpsc::unbounded_channel();
        let post_router = PostRouter::new();
        let cloid_cache: CloidCache = Arc::new(Mutex::new(FifoCacheMap::<
            Ustr,
            ClientOrderId,
            CLOID_CACHE_CAPACITY,
        >::new()));
        let mut handler = FeedHandler::new(
            signal,
            cmd_rx,
            raw_rx,
            out_tx,
            None,
            SubscriptionState::new(':'),
            cloid_cache,
            post_router,
            Arc::new(WebSocketRateLimits::new()),
            1,
        );
        let subscription = SubscriptionRequest::Notification {
            user: SECRET_MARKER.to_string(),
        };
        let subscribe_len = serde_json::to_string(&HyperliquidWsRequest::Subscribe {
            subscription: subscription.clone(),
        })
        .unwrap()
        .len();
        let unsubscribe_len = serde_json::to_string(&HyperliquidWsRequest::Unsubscribe {
            subscription: subscription.clone(),
        })
        .unwrap()
        .len();
        OUTBOUND_LOG_CAPTURE.clear();

        cmd_tx
            .send(HandlerCommand::Subscribe {
                subscriptions: vec![subscription.clone()],
            })
            .unwrap();
        cmd_tx
            .send(HandlerCommand::Unsubscribe {
                subscriptions: vec![subscription],
            })
            .unwrap();
        drop(cmd_tx);
        drop(raw_tx);

        assert!(handler.next().await.is_none());

        let messages = OUTBOUND_LOG_CAPTURE.messages();

        assert!(
            messages
                .iter()
                .all(|message| !message.contains(SECRET_MARKER)),
            "outbound logs exposed the secret marker: {messages:?}"
        );
        assert!(
            messages
                .iter()
                .any(|message| message
                    == &format!("Sending subscribe payload ({subscribe_len} bytes)")),
            "subscribe metadata missing or inaccurate: {messages:?}"
        );
        assert!(
            messages.iter().any(|message| {
                message == &format!("Sending unsubscribe payload ({unsubscribe_len} bytes)")
            }),
            "unsubscribe metadata missing or inaccurate: {messages:?}"
        );
    }

    #[tokio::test]
    async fn unsubscribe_confirmation_releases_shared_reservation() {
        let signal = Arc::new(AtomicBool::new(false));
        let (cmd_tx, cmd_rx) = tokio::sync::mpsc::unbounded_channel();
        let (raw_tx, raw_rx) = tokio::sync::mpsc::unbounded_channel();
        let (out_tx, _out_rx) = tokio::sync::mpsc::unbounded_channel();
        let subscriptions = SubscriptionState::new(':');
        let limits = Arc::new(WebSocketRateLimits::new());
        let released = SubscriptionRequest::Trades {
            coin: Ustr::from("RELEASED"),
        };
        let released_key = super::subscription_to_key(&released);
        subscriptions.mark_subscribe(&released_key);
        subscriptions.confirm_subscribe(&released_key);
        subscriptions.mark_unsubscribe(&released_key);
        assert!(limits.reserve_subscription(1, &released).unwrap());

        for index in 1..HYPERLIQUID_WS_SUBSCRIPTIONS_MAX {
            assert!(
                limits
                    .reserve_subscription(
                        1,
                        &SubscriptionRequest::Trades {
                            coin: Ustr::from(&format!("COIN-{index}")),
                        },
                    )
                    .unwrap()
            );
        }
        assert!(
            limits
                .reserve_subscription(
                    2,
                    &SubscriptionRequest::Trades {
                        coin: Ustr::from("BEFORE-ACK"),
                    },
                )
                .is_err()
        );

        let cloid_cache: CloidCache = Arc::new(Mutex::new(FifoCacheMap::<
            Ustr,
            ClientOrderId,
            CLOID_CACHE_CAPACITY,
        >::new()));
        let mut handler = FeedHandler::new(
            signal,
            cmd_rx,
            raw_rx,
            out_tx,
            None,
            subscriptions,
            cloid_cache,
            PostRouter::new(),
            Arc::clone(&limits),
            1,
        );
        raw_tx
            .send(Message::Text(
                json!({
                    "channel": "subscriptionResponse",
                    "data": {
                        "method": "unsubscribe",
                        "subscription": {
                            "type": "trades",
                            "coin": "RELEASED",
                        },
                    },
                })
                .to_string()
                .into(),
            ))
            .unwrap();
        drop(raw_tx);
        drop(cmd_tx);

        assert!(handler.next().await.is_none());
        assert!(
            limits
                .reserve_subscription(
                    2,
                    &SubscriptionRequest::Trades {
                        coin: Ustr::from("AFTER-ACK"),
                    },
                )
                .unwrap()
        );
    }

    #[rstest]
    fn handle_l2_book_emits_deltas_only_when_not_in_depth10_subs() {
        let mut instruments = AHashMap::new();
        instruments.insert(Ustr::from("BTC"), btc_perp());
        let depth10_subs = AHashSet::<Ustr>::new();

        let msgs = FeedHandler::handle_l2_book(
            &one_level_book(),
            &instruments,
            &depth10_subs,
            UnixNanos::default(),
        );

        assert_eq!(msgs.len(), 1);
        assert!(matches!(msgs[0], NautilusWsMessage::Deltas(_)));
    }

    #[rstest]
    fn handle_l2_book_emits_deltas_and_depth10_when_coin_in_subs() {
        let mut instruments = AHashMap::new();
        instruments.insert(Ustr::from("BTC"), btc_perp());
        let mut depth10_subs = AHashSet::<Ustr>::new();
        depth10_subs.insert(Ustr::from("BTC"));

        let msgs = FeedHandler::handle_l2_book(
            &one_level_book(),
            &instruments,
            &depth10_subs,
            UnixNanos::default(),
        );

        assert_eq!(msgs.len(), 2);
        assert!(matches!(msgs[0], NautilusWsMessage::Deltas(_)));
        assert!(matches!(msgs[1], NautilusWsMessage::Depth10(_)));
    }

    #[rstest]
    fn handle_l2_book_returns_empty_when_instrument_unknown() {
        let instruments = AHashMap::<Ustr, InstrumentAny>::new();
        let depth10_subs = AHashSet::<Ustr>::new();

        let msgs = FeedHandler::handle_l2_book(
            &one_level_book(),
            &instruments,
            &depth10_subs,
            UnixNanos::default(),
        );

        assert!(msgs.is_empty());
    }

    #[rstest]
    fn handle_asset_context_emits_open_interest_custom_data_when_subscribed() {
        let instrument = btc_perp();
        let instrument_id = instrument.id();
        let mut instruments = AHashMap::new();
        instruments.insert(Ustr::from("BTC"), instrument);

        let mut asset_context_subs = AHashMap::new();
        asset_context_subs.insert(
            Ustr::from("BTC"),
            AHashSet::from_iter([AssetContextDataType::OpenInterest]),
        );

        let mut asset_context_caches = AssetContextCaches::default();

        let msgs = FeedHandler::handle_asset_context(
            &btc_active_asset_ctx(dec!(100000.0)),
            &instruments,
            &asset_context_subs,
            &mut asset_context_caches,
            UnixNanos::default(),
        );

        assert_eq!(msgs.len(), 1);

        match &msgs[0] {
            NautilusWsMessage::CustomData(Data::Custom(custom)) => {
                let open_interest = custom
                    .data
                    .as_any()
                    .downcast_ref::<HyperliquidOpenInterest>()
                    .expect("expected HyperliquidOpenInterest");
                assert_eq!(open_interest.instrument_id, instrument_id);
                assert_eq!(open_interest.open_interest.to_string(), "100000.0");
                assert_eq!(
                    custom
                        .data_type
                        .metadata()
                        .and_then(|metadata| metadata.get_str("instrument_id"))
                        .map(ToString::to_string),
                    Some(instrument_id.to_string()),
                );
            }
            other => panic!("unexpected message type: {other:?}"),
        }
    }

    #[rstest]
    fn handle_all_dexs_asset_ctxs_emits_normalized_custom_data() {
        let mapping = AHashMap::from_iter([
            (
                Ustr::from(""),
                vec![Some(InstrumentId::from("BTC-USD-PERP.HYPERLIQUID"))],
            ),
            (
                Ustr::from("xyz"),
                vec![Some(InstrumentId::from("xyz:XYZ100-USD-PERP.HYPERLIQUID"))],
            ),
        ]);

        let msg = FeedHandler::handle_all_dexs_asset_ctxs(
            sample_all_dexs_asset_ctxs(),
            &mapping,
            UnixNanos::default(),
        )
        .expect("expected custom data");

        match msg {
            NautilusWsMessage::CustomData(Data::Custom(custom)) => {
                let payload = custom
                    .data
                    .as_any()
                    .downcast_ref::<HyperliquidAllDexsAssetCtxs>()
                    .expect("expected HyperliquidAllDexsAssetCtxs");
                assert_eq!(payload.entries.len(), 2);
                assert_eq!(
                    payload.entries[0].instrument_id,
                    InstrumentId::from("BTC-USD-PERP.HYPERLIQUID")
                );
                assert_eq!(payload.entries[1].dex, "xyz");
                assert_eq!(
                    payload.entries[1].instrument_id,
                    InstrumentId::from("xyz:XYZ100-USD-PERP.HYPERLIQUID")
                );
                assert_eq!(payload.entries[0].mark_price.to_string(), "77562.0");
                assert_eq!(payload.entries[1].day_base_volume.to_string(), "5135.2458");
            }
            other => panic!("expected custom data, found {other:?}"),
        }
    }

    #[rstest]
    fn handle_all_dexs_asset_ctxs_preserves_index_alignment_when_mappings_are_missing() {
        let data = WsAllDexsAssetCtxsData {
            ctxs: vec![(
                String::new(),
                vec![
                    PerpsAssetCtx {
                        shared: SharedAssetCtx {
                            day_ntl_vlm: dec!(1516669192.1953897476),
                            prev_day_px: dec!(76317.0),
                            mark_px: dec!(77562.0),
                            mid_px: Some(dec!(77558.5)),
                            impact_pxs: Some(vec!["77558.0".to_string(), "77559.0".to_string()]),
                            day_base_vlm: Some(dec!(19707.77457)),
                        },
                        funding: dec!(-0.0000015186),
                        open_interest: dec!(27353.17682),
                        oracle_px: dec!(77605.0),
                        premium: Some(dec!(-0.0005927453)),
                    },
                    PerpsAssetCtx {
                        shared: SharedAssetCtx {
                            day_ntl_vlm: dec!(591989409.9392402172),
                            prev_day_px: dec!(2094.6),
                            mark_px: dec!(2123.7),
                            mid_px: Some(dec!(2123.95)),
                            impact_pxs: Some(vec!["2123.65".to_string(), "2124.0".to_string()]),
                            day_base_vlm: Some(dec!(281686.8234999999)),
                        },
                        funding: dec!(0.0000125),
                        open_interest: dec!(605822.2557999999),
                        oracle_px: dec!(2124.6),
                        premium: Some(dec!(-0.0002824061)),
                    },
                ],
            )],
        };

        let mapping = AHashMap::from_iter([(
            Ustr::from(""),
            vec![None, Some(InstrumentId::from("ETH-USD-PERP.HYPERLIQUID"))],
        )]);

        let msg = FeedHandler::handle_all_dexs_asset_ctxs(data, &mapping, UnixNanos::default())
            .expect("expected custom data");

        match msg {
            NautilusWsMessage::CustomData(Data::Custom(custom)) => {
                let payload = custom
                    .data
                    .as_any()
                    .downcast_ref::<HyperliquidAllDexsAssetCtxs>()
                    .expect("expected HyperliquidAllDexsAssetCtxs");
                assert_eq!(payload.entries.len(), 1);
                assert_eq!(
                    payload.entries[0].instrument_id,
                    InstrumentId::from("ETH-USD-PERP.HYPERLIQUID")
                );
                assert_eq!(payload.entries[0].mark_price.to_string(), "2123.7");
            }
            other => panic!("expected custom data, found {other:?}"),
        }
    }

    #[rstest]
    fn handle_asset_context_skips_open_interest_for_spot_payload() {
        let instrument = btc_perp();
        let mut instruments = AHashMap::new();
        instruments.insert(Ustr::from("BTC"), instrument);

        let mut asset_context_subs = AHashMap::new();
        asset_context_subs.insert(
            Ustr::from("BTC"),
            AHashSet::from_iter([AssetContextDataType::OpenInterest]),
        );

        let mut asset_context_caches = AssetContextCaches::default();

        let msgs = FeedHandler::handle_asset_context(
            &btc_active_spot_asset_ctx(),
            &instruments,
            &asset_context_subs,
            &mut asset_context_caches,
            UnixNanos::default(),
        );

        assert!(msgs.is_empty());
        assert!(asset_context_caches.open_interest.is_empty());
    }

    #[rstest]
    fn handle_asset_context_suppresses_unchanged_open_interest() {
        let instrument = btc_perp();
        let mut instruments = AHashMap::new();
        instruments.insert(Ustr::from("BTC"), instrument);

        let mut asset_context_subs = AHashMap::new();
        asset_context_subs.insert(
            Ustr::from("BTC"),
            AHashSet::from_iter([AssetContextDataType::OpenInterest]),
        );

        let mut asset_context_caches = AssetContextCaches::default();

        let first = FeedHandler::handle_asset_context(
            &btc_active_asset_ctx(dec!(100000.0)),
            &instruments,
            &asset_context_subs,
            &mut asset_context_caches,
            UnixNanos::default(),
        );
        let second = FeedHandler::handle_asset_context(
            &btc_active_asset_ctx(dec!(100000.0)),
            &instruments,
            &asset_context_subs,
            &mut asset_context_caches,
            UnixNanos::default(),
        );

        assert_eq!(first.len(), 1);
        assert!(second.is_empty());
    }

    #[rstest]
    fn asset_context_caches_clear_removed_data_types() {
        let coin = Ustr::from("BTC");
        let mut caches = AssetContextCaches::default();
        caches.mark_price.insert(coin, dec!(98455.5));
        caches.index_price.insert(coin, dec!(98460.0));
        caches.funding_rate.insert(coin, dec!(0.0001));
        caches.open_interest.insert(coin, dec!(1500.0));

        let previous_data_types = AHashSet::from_iter([
            AssetContextDataType::MarkPrice,
            AssetContextDataType::IndexPrice,
            AssetContextDataType::FundingRate,
            AssetContextDataType::OpenInterest,
        ]);
        let next_data_types = AHashSet::from_iter([
            AssetContextDataType::MarkPrice,
            AssetContextDataType::FundingRate,
        ]);

        caches.clear_removed(coin, Some(&previous_data_types), &next_data_types);

        assert_eq!(caches.mark_price.get(&coin).copied(), Some(dec!(98455.5)));
        assert!(caches.index_price.get(&coin).is_none());
        assert_eq!(caches.funding_rate.get(&coin).copied(), Some(dec!(0.0001)));
        assert!(caches.open_interest.get(&coin).is_none());
    }
}