Golang gen: add setters for fieldmasks

Makefile

@@ -70,6 +70,7 @@ goldmaster_nocompile: build
 		--generateByteVersions=cases_bytes. \
 		--generateRandomCode \
 		--generateLegacyJsonRead=false \
+		--checkLengthSanity=false \
 		./$(TLS_PATH)/cases.tl
 	@./target/bin/tlgen --language=go --split-internal -v \
 		--copyrightPath=./COPYRIGHT \

cmd/tlgen/main2.go

@@ -71,6 +71,8 @@ func parseFlags(opt *tlcodegen.Gen2Options) {
 		"timestamp of schema (for documentation, TLO version)")
 	flag.StringVar(&opt.SchemaCommit, "schemaCommit", "",
 		"commit of schema (for documentation)")
+	flag.BoolVar(&opt.UseCheckLengthSanity, "checkLengthSanity", true,
+		"enable feature to generate code to check length sanity of arrays (default:true)")
 
 	// C++
 	flag.StringVar(&opt.RootCPPNamespace, "cpp-namespace", "",

internal/tlcodegen/qt_brackets.qtpl

@@ -35,9 +35,11 @@ func {%s= goName %}Read(w []byte, vec *{%s= typeString %} {%s= natDecl %}) (_ []
     if w, err = basictl.NatRead(w, &l); err != nil {
         return w, err
     }
+    {%- if tuple.wr.gen.options.UseCheckLengthSanity -%}
     if err = basictl.CheckLengthSanity(w, l, 4); err != nil {
         return w, err
     }
+    {%- endif -%}
     if uint32(cap(*vec)) < l {
         *vec = make([]{%s= elementTypeString %}, l)
     } else {
@@ -240,9 +242,11 @@ func {%s= goName %}Read(w []byte, m *map[{%s= keyTypeString %}]{%s= valueTypeStr
     if w, err = basictl.NatRead(w, &l); err != nil {
         return w, err
     }
+    {%- if tuple.wr.gen.options.UseCheckLengthSanity -%}
     if err = basictl.CheckLengthSanity(w, l, 4); err != nil {
         return w, err
     }
+    {%- endif -%}
     var data map[{%s= keyTypeString %}]{%s= valueTypeString %}
     if *m == nil {
         if l == 0 {
@@ -460,9 +464,11 @@ func {%s= goName %}Read(w []byte, vec *{%s= typeString %} {%s= natDecl %}) (_ []
     if w, err = basictl.NatRead(w, &l); err != nil {
         return w, err
     }
+    {%- if tuple.wr.gen.options.UseCheckLengthSanity -%}
     if err = basictl.CheckLengthSanity(w, l, 4); err != nil {
         return w, err
     }
+    {%- endif -%}
     if uint32(cap(*vec)) < l {
         *vec = make([]{%s= elementTypeString %}, l)
     } else {
@@ -559,9 +565,11 @@ func {%s= goName %}FillRandom(rg *basictl.RandGenerator, vec *{%s= typeString %}
             {%- endif -%}
 
 func {%s= goName %}Read(w []byte, vec *{%s= typeString %} {%s= natDecl %}) (_ []byte, err error) {
+    {%- if tuple.wr.gen.options.UseCheckLengthSanity -%}
     if err = basictl.CheckLengthSanity(w, nat_n, 4); err != nil {
         return w, err
     }
+    {%- endif -%}
     if uint32(cap(*vec)) < nat_n {
         *vec = make([]{%s= elementTypeString %}, nat_n)
     } else {

internal/tlcodegen/qt_struct.qtpl

@@ -1,6 +1,9 @@
 {% package tlcodegen %}
 
 {% import "fmt" %}
+{% import "github.com/vkcom/tl/internal/utils" %}
+{% import "strings" %}
+{% import "sort" %}
 
 {%- func (struct_ *TypeRWStruct)GenerateCode(bytesVersion bool, directImports *DirectImports) -%}
     {%- code
@@ -794,6 +797,63 @@ func (item *{%s= goName %}) ReadResultJSONWriteResult(r []byte, w []byte) ([]byt
   w, err = item.WriteResult(w, ret)
   return r, w, err
 }
+{%- for fieldId, field := range struct_.Fields -%}
+    {%- if _, affectedBits := struct_.GetFieldNatPropertiesAsUsageMap(fieldId, false, true); len(affectedBits) != 0 -%}
+        {%- code
+            bits := make([]uint32, 0)
+            for i := range affectedBits {
+                bits = append(bits, i)
+            }
+            sort.Slice(bits, func(i, j int) bool {
+                return bits[i] < bits[j]
+            })
+        -%}
+        {%- for _, bit := range bits -%}
+            {%- code
+                bitUsage := affectedBits[bit]
+                affectedTypes := make([]*TypeRWStruct, 0)
+            	for i := range bitUsage.AffectedFields {
+            		affectedTypes = append(affectedTypes, i)
+            	}
+            	sort.Slice(affectedTypes, func(i, j int) bool {
+            		return TypeRWWrapperLessGlobal(affectedTypes[i].wr, affectedTypes[j].wr) > 0
+            	})
+            -%}
+            {%- for _, affectedType := range affectedTypes -%}
+                {%- code
+                    affectedFieldsByBit := bitUsage.AffectedFields[affectedType]
+                    affectedFieldsByBit = utils.SetToSlice(utils.SliceToSet(affectedFieldsByBit))
+                    sort.Slice(affectedFieldsByBit, func(i, j int) bool {
+                        return affectedFieldsByBit[i] < affectedFieldsByBit[j]
+                    })
+                    fieldNames := make([]string, len(affectedFieldsByBit))
+                    fieldNamesForComment := make([]string, len(affectedFieldsByBit))
+
+                    for i := range fieldNames {
+                        fieldNames[i] = affectedType.Fields[affectedFieldsByBit[i]].goName
+                        fieldNamesForComment[i] = "\"" + affectedType.Fields[affectedFieldsByBit[i]].originalName + "\""
+                    }
+
+                    mergedFields := strings.Join(fieldNames, "And")
+                    mergedFieldsForComment := strings.Join(fieldNamesForComment, ", ")
+                    maybeS := ""
+                    if len(affectedFieldsByBit) > 1 {
+                        maybeS = "s"
+                    }
+                -%}
+// Set field{%s= maybeS %} {%s= mergedFieldsForComment %} in "{%s= affectedType.wr.tlName.String() %}" by changing fieldMask "{%s= field.originalName %}"
+func (item *{%s= goName %}) Set{%s= affectedType.wr.goGlobalName %}{%s= mergedFields %}(value bool) {
+    if value {
+        item.{%s= field.goName %} |= 1 << {%d= int(bit) %}
+    } else {
+        item.{%s= field.goName %} &^= 1 << {%d= int(bit) %}
+    }
+}
+
+            {%- endfor -%}
+        {%- endfor -%}
+    {%- endif -%}
+{%- endfor -%}
 
 {%- endfunc -%}
 
@@ -854,6 +914,8 @@ if mask{%s= field.goName %} & (1<<{%v= index %}) != 0 {
             {%- elseif (fieldProps & FieldUsedAsSize) != 0 -%}
 {%s= field.t.TypeRandomCode(bytesVersion, directImports, struct_.wr.ins, "item."+field.goName, formatNatArgs(struct_.Fields, field.natArgs), field.recursive) %}
 item.{%s= field.goName %} = rg.LimitValue(item.{%s= field.goName %})
+            {%- elseif fieldProps == FieldIsNat -%}
+{%s= field.t.TypeRandomCode(bytesVersion, directImports, struct_.wr.ins, "item."+field.goName, formatNatArgs(struct_.Fields, field.natArgs), field.recursive) %}
             {%- endif -%}
         {%- else -%}
 {%s= field.t.TypeRandomCode(bytesVersion, directImports, struct_.wr.ins, "item."+field.goName, formatNatArgs(struct_.Fields, field.natArgs), field.recursive) %}