Lobster: Support required fields

google · Mar 9, 2024 · 960cd4d · 960cd4d
1 parent 6ff9e90
commit 960cd4d
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/idl_gen_lobster.cpp b/src/idl_gen_lobster.cpp
@@ -145,11 +145,17 @@ class LobsterGenerator : public BaseGenerator {
           code += def + "() -> " + name + ":\n        ";
           code += "return " + name + "{ buf_, pos_ + " + offsets + " }\n";
         } else {
-          code += def + "() -> " + name + "?:\n        ";
+          code += def + "() -> " + name;
+          if (!field.IsRequired()) code += "?";
+          code += ":\n        ";
           code += std::string("let o = flatbuffers.field_") +
                   (field.value.type.struct_def->fixed ? "struct" : "table") +
-                  "(buf_, pos_, " + offsets + ")\n        return if o: " + name +
-                  " { buf_, o } else: nil\n";
+                  "(buf_, pos_, " + offsets + ")\n        return ";
+          if (field.IsRequired()) {
+            code += name + " { buf_, assert o }\n";
+          } else {
+            code += "if o: " + name + " { buf_, o } else: nil\n";
+          }
         }
         break;
       }