different allocators by types

Author: asmyasnikov

internal/value/exp/allocator/allocator.go

@@ -21,7 +21,7 @@ func (a *Allocator) Value() (v *Ydb.Value) {
 	return new(Ydb.Value)
 }
 
-func (a *Allocator) TypedValue() (v *Ydb.TypedValue) {
+func (a *Allocator) Typed() (v *Ydb.TypedValue) {
 	return new(Ydb.TypedValue)
 }
 
@@ -57,51 +57,51 @@ func (a *Allocator) TypeTuple() (v *Ydb.Type_TupleType) {
 	return new(Ydb.Type_TupleType)
 }
 
-func (a *Allocator) EmptyTypeList() (v *Ydb.Type_EmptyListType) {
+func (a *Allocator) TypeEmptyList() (v *Ydb.Type_EmptyListType) {
 	return new(Ydb.Type_EmptyListType)
 }
 
 func (a *Allocator) TypeOptional() (v *Ydb.Type_OptionalType) {
 	return new(Ydb.Type_OptionalType)
 }
 
-func (a *Allocator) BoolValue() (v *Ydb.Value_BoolValue) {
+func (a *Allocator) Bool() (v *Ydb.Value_BoolValue) {
 	return new(Ydb.Value_BoolValue)
 }
 
-func (a *Allocator) BytesValue() (v *Ydb.Value_BytesValue) {
+func (a *Allocator) Bytes() (v *Ydb.Value_BytesValue) {
 	return new(Ydb.Value_BytesValue)
 }
 
-func (a *Allocator) Int32Value() (v *Ydb.Value_Int32Value) {
+func (a *Allocator) Int32() (v *Ydb.Value_Int32Value) {
 	return new(Ydb.Value_Int32Value)
 }
 
-func (a *Allocator) Int64Value() (v *Ydb.Value_Int64Value) {
+func (a *Allocator) Int64() (v *Ydb.Value_Int64Value) {
 	return new(Ydb.Value_Int64Value)
 }
 
-func (a *Allocator) Uint32Value() (v *Ydb.Value_Uint32Value) {
+func (a *Allocator) Uint32() (v *Ydb.Value_Uint32Value) {
 	return new(Ydb.Value_Uint32Value)
 }
 
-func (a *Allocator) FloatValue() (v *Ydb.Value_FloatValue) {
+func (a *Allocator) Float() (v *Ydb.Value_FloatValue) {
 	return new(Ydb.Value_FloatValue)
 }
 
-func (a *Allocator) DoubleValue() (v *Ydb.Value_DoubleValue) {
+func (a *Allocator) Double() (v *Ydb.Value_DoubleValue) {
 	return new(Ydb.Value_DoubleValue)
 }
 
-func (a *Allocator) Uint64Value() (v *Ydb.Value_Uint64Value) {
+func (a *Allocator) Uint64() (v *Ydb.Value_Uint64Value) {
 	return new(Ydb.Value_Uint64Value)
 }
 
-func (a *Allocator) TextValue() (v *Ydb.Value_TextValue) {
+func (a *Allocator) Text() (v *Ydb.Value_TextValue) {
 	return new(Ydb.Value_TextValue)
 }
 
-func (a *Allocator) Low128Value() (v *Ydb.Value_Low_128) {
+func (a *Allocator) Low128() (v *Ydb.Value_Low_128) {
 	return new(Ydb.Value_Low_128)
 }
 
@@ -121,7 +121,7 @@ func (a *Allocator) TypeStruct() (v *Ydb.Type_StructType) {
 	return new(Ydb.Type_StructType)
 }
 
-func (a *Allocator) NestedValue() (v *Ydb.Value_NestedValue) {
+func (a *Allocator) Nested() (v *Ydb.Value_NestedValue) {
 	return new(Ydb.Value_NestedValue)
 }
 

internal/value/exp/allocator/allocator_bool_value_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type boolAllocator struct {
+	allocations []*Ydb.Value_BoolValue
+}
+
+func (a *boolAllocator) Bool() (v *Ydb.Value_BoolValue) {
+	v = boolPool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *boolAllocator) free() {
+	for _, v := range a.allocations {
+		*v = Ydb.Value_BoolValue{}
+		boolPool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}

internal/value/exp/allocator/allocator_bytes_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type bytesAllocator struct {
+	allocations []*Ydb.Value_BytesValue
+}
+
+func (a *bytesAllocator) Bytes() (v *Ydb.Value_BytesValue) {
+	v = bytesPool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *bytesAllocator) free() {
+	for _, v := range a.allocations {
+		*v = Ydb.Value_BytesValue{}
+		bytesPool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}

internal/value/exp/allocator/allocator_decimal_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type decimalAllocator struct {
+	allocations []*Ydb.DecimalType
+}
+
+func (a *decimalAllocator) Decimal() (v *Ydb.DecimalType) {
+	v = decimalPool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *decimalAllocator) free() {
+	for _, v := range a.allocations {
+		v.Reset()
+		decimalPool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}

internal/value/exp/allocator/allocator_dict_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type dictAllocator struct {
+	allocations []*Ydb.DictType
+}
+
+func (a *dictAllocator) Dict() (v *Ydb.DictType) {
+	v = dictPool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *dictAllocator) free() {
+	for _, v := range a.allocations {
+		v.Reset()
+		dictPool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}

internal/value/exp/allocator/allocator_double_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type doubleAllocator struct {
+	allocations []*Ydb.Value_DoubleValue
+}
+
+func (a *doubleAllocator) Double() (v *Ydb.Value_DoubleValue) {
+	v = doublePool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *doubleAllocator) free() {
+	for _, v := range a.allocations {
+		*v = Ydb.Value_DoubleValue{}
+		doublePool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}

internal/value/exp/allocator/allocator_float_go1.18.go

@@ -0,0 +1,24 @@
+//go:build go1.18
+// +build go1.18
+
+package allocator
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb"
+
+type floatAllocator struct {
+	allocations []*Ydb.Value_FloatValue
+}
+
+func (a *floatAllocator) Float() (v *Ydb.Value_FloatValue) {
+	v = floatPool.Get()
+	a.allocations = append(a.allocations, v)
+	return v
+}
+
+func (a *floatAllocator) free() {
+	for _, v := range a.allocations {
+		*v = Ydb.Value_FloatValue{}
+		floatPool.Put(v)
+	}
+	a.allocations = a.allocations[:0]
+}