Generate max-size constants for dynamic fields

For example, for the struct: ``` type Log struct { Data []Uint256 `json:"data" ssz-max:"6000"` } ``` Will be generated the following constant in `log_encoding.go`: `const LogMaxDataSize = 6000` Signed-off-by: Mikhail Sherstennikov <[email protected]>
NilFoundation · Nov 27, 2024 · f18caf3 · f18caf3
1 parent 181ec7e
commit f18caf3
Show file tree

Hide file tree

Showing 3 changed files with 120 additions and 4 deletions.
diff --git a/spectests/structs_encoding.go b/spectests/structs_encoding.go
diff --git a/sszgen/generator/size.go b/sszgen/generator/size.go
@@ -19,16 +19,29 @@ func (e *env) size(name string, v *Value) string {
 		{{.dynamic}}
 		{{end}}
 		return
-	}`
+	}
+	{{.fieldsMaxSizes}}
+	`
 
 	str := execTmpl(tmpl, map[string]interface{}{
-		"name":    name,
-		"fixed":   v.fixedSize(),
-		"dynamic": v.sizeContainer("size", true),
+		"name":           name,
+		"fixed":          v.fixedSize(),
+		"dynamic":        v.sizeContainer("size", true),
+		"fieldsMaxSizes": v.fieldsMaxSizes(name),
 	})
 	return appendObjSignature(str, v)
 }
 
+func (v *Value) fieldsMaxSizes(name string) string {
+	out := []string{}
+	for _, v := range v.o {
+		if !v.isFixed() {
+			out = append(out, fmt.Sprintf("const %sMax%sSize = %d", name, v.name, v.s))
+		}
+	}
+	return strings.Join(out, "\n")
+}
+
 func (v *Value) fixedSize() uint64 {
 	switch v.t {
 	case TypeVector:

diff --git a/tests/codetrie_encoding.go b/tests/codetrie_encoding.go