impl TryFrom<U256> for Chain (gakonst#1247)

joleeee · May 11, 2022 · 98fc8c8 · 98fc8c8
1 parent 847110a
commit 98fc8c8
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/ethers-core/src/types/chain.rs b/ethers-core/src/types/chain.rs
@@ -2,7 +2,7 @@ use serde::Deserialize;
 use thiserror::Error;
 
 use core::convert::TryFrom;
-use std::{default, fmt, str::FromStr};
+use std::{convert::TryInto, default, fmt, str::FromStr};
 
 use crate::types::U256;
 
@@ -128,6 +128,17 @@ impl TryFrom<u64> for Chain {
     }
 }
 
+impl TryFrom<U256> for Chain {
+    type Error = ParseChainError;
+
+    fn try_from(chain: U256) -> Result<Chain, Self::Error> {
+        if chain.bits() > 64 {
+            return Err(ParseChainError(chain.to_string()))
+        }
+        chain.as_u64().try_into()
+    }
+}
+
 impl FromStr for Chain {
     type Err = ParseChainError;
     fn from_str(chain: &str) -> Result<Self, Self::Err> {