config validate as struct method

HerodotusDev · Jan 9, 2024 · 2000bb6 · 2000bb6
1 parent 4f5eb8f
commit 2000bb6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/proof_of_work/config.cairo b/src/proof_of_work/config.cairo
@@ -7,7 +7,10 @@ struct ProofOfWorkConfig {
     n_bits: u8,
 }
 
-fn proof_of_work_config_validate(config: ProofOfWorkConfig) {
-    assert(config.n_bits.into() >= MIN_PROOF_OF_WORK_BITS, 'value proof of work bits to low');
-    assert(config.n_bits.into() <= MIN_PROOF_OF_WORK_BITS, 'value proof of work bits to big');
+#[generate_trait]
+impl ProofOfWorkConfigImpl of ProofOfWorkConfigTrait {
+    fn config_validate(ref self: ProofOfWorkConfig) {
+        assert(self.n_bits.into() >= MIN_PROOF_OF_WORK_BITS, 'value proof of work bits to low');
+        assert(self.n_bits.into() <= MAX_PROOF_OF_WORK_BITS, 'value proof of work bits to big');
+    }
 }