From 89612f0921a647a9b1c4bc62f17f6ad7e1ebdb4a Mon Sep 17 00:00:00 2001
From: RequiemOfSouls <RequiemOfSouls@zklink.org>
Date: Fri, 24 Nov 2023 13:51:21 +0800
Subject: [PATCH] feat: add UpdateGlobalVar in asset_reduced_sub_account func,
 and add tx_type function for ZkLinkTx

---
 types/src/tx_type/zklink_tx.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/types/src/tx_type/zklink_tx.rs b/types/src/tx_type/zklink_tx.rs
index 788fec03..8c495dc6 100644
--- a/types/src/tx_type/zklink_tx.rs
+++ b/types/src/tx_type/zklink_tx.rs
@@ -140,6 +140,24 @@ impl ZkLinkTx {
         }
     }
 
+    /// Check tx format
+    pub fn tx_type(&self) -> ZkLinkTxType {
+        match self {
+            ZkLinkTx::Transfer(_) => ZkLinkTxType::Transfer,
+            ZkLinkTx::Withdraw(_) => ZkLinkTxType::Withdraw,
+            ZkLinkTx::ChangePubKey(_) => ZkLinkTxType::ChangePubKey,
+            ZkLinkTx::ForcedExit(_) => ZkLinkTxType::ForcedExit,
+            ZkLinkTx::OrderMatching(_) => ZkLinkTxType::OrderMatching,
+            ZkLinkTx::Deposit(_) => ZkLinkTxType::Deposit,
+            ZkLinkTx::FullExit(_) => ZkLinkTxType::FullExit,
+            ZkLinkTx::ContractMatching(_) => ZkLinkTxType::ContractMatching,
+            ZkLinkTx::Liquidation(_) => ZkLinkTxType::Liquidation,
+            ZkLinkTx::AutoDeleveraging(_) => ZkLinkTxType::AutoDeleveraging,
+            ZkLinkTx::UpdateGlobalVar(_) => ZkLinkTxType::UpdateGlobalVar,
+            ZkLinkTx::Funding(_) => ZkLinkTxType::Funding,
+        }
+    }
+
     /// Check tx correct
     pub fn is_valid(&self) -> bool {
         match self {
@@ -206,6 +224,7 @@ impl ZkLinkTx {
             ZkLinkTx::ContractMatching(tx) => vec![tx.sub_account_id],
             ZkLinkTx::AutoDeleveraging(tx) => vec![tx.sub_account_id],
             ZkLinkTx::Funding(tx) => vec![tx.sub_account_id],
+            ZkLinkTx::UpdateGlobalVar(tx) => vec![tx.sub_account_id],
             _ => vec![],
         }
     }