add define statement for equity (#6)

[bytom/equity.git] / compiler / checks.go

diff --git a/compiler/checks.go b/compiler/checks.go
index 15879e2..bd2c524 100644 (file)
--- a/compiler/checks.go
+++ b/compiler/checks.go
@@ -48,6 +48,8 @@ func requireAllParamsUsedInClause(params []*Param, clause *Clause) error {
                used := false
                for _, stmt := range clause.statements {
                        switch s := stmt.(type) {
+                       case *defineStatement:
+                               used = references(s.expr, p.Name)
                        case *verifyStatement:
                                used = references(s.expr, p.Name)
                        case *lockStatement: