Refactor

awslabs · Nov 9, 2024 · a2c47fb · a2c47fb
1 parent 58fc2b6
commit a2c47fb
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/modules/llrt_assert/src/lib.rs b/modules/llrt_assert/src/lib.rs
@@ -31,13 +31,14 @@ fn assert(ctx: Ctx, value: Value, message: Opt<Value>) -> Result<()> {
     }
 
     if let Some(obj) = message.0 {
-        if let Some(msg) = obj.as_string() {
-            let msg = msg.to_string().unwrap();
-            return Err(Exception::throw_message(&ctx, &msg));
-        }
-        if let Some(err) = obj.as_exception() {
-            return Err(err.clone().throw());
-        }
+        match obj.type_of() {
+            Type::String => {
+                let msg = obj.as_string().unwrap().to_string().unwrap();
+                return Err(Exception::throw_message(&ctx, &msg));
+            },
+            Type::Exception => return Err(obj.as_exception().cloned().unwrap().throw()),
+            _ => {},
+        };
     }
 
     Err(Exception::throw_message(