overwrite function patterns

danilopedraza · Sep 24, 2024 · e5e5304 · e5e5304
1 parent a051070
commit e5e5304
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/object.rs b/core/src/object.rs
@@ -1084,7 +1084,7 @@ impl Callable for Function {
     }
 }
 
-#[derive(Clone, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
+#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
 pub enum FunctionPatternKind {
     Memoized,
     NotMemoized,
@@ -1099,6 +1099,13 @@ pub struct PatternFunction {
 
 impl PatternFunction {
     pub fn add_pattern(&mut self, args: &[ASTNode], value: &ASTNode, kind: FunctionPatternKind) {
+        for i in 0..self.patterns.len() {
+            let (_, other_args, _) = &self.patterns[i];
+            if args == other_args {
+                self.patterns[i] = (kind, args.to_owned(), value.to_owned());
+            }
+        }
+
         if self.patterns.is_empty() {
             self.params = args.len();
         } else {