refactor: use the go generic to rewrite interface of agg iterator (openGemini#580)

Signed-off-by: scuzyj <[email protected]>

Author: scuzyj

engine/executor/agg_func.gen.go

@@ -24,64 +24,6 @@ import (
 	"github.com/openGemini/openGemini/engine/hybridqp"
 )
 
-{{range .}}
-{{- if or (eq .Name "Float") (eq .Name "Integer")}}
-func New{{.Name}}MedianReduce({{.name}}SliceItem *{{.Name}}SliceItem) (int, int64, float64, bool) {
-		length := len({{.name}}SliceItem.value)
-		if length == 0 {
-			return -1, {{.Nil}}, {{.Nil}}, true
-		}
-		if length == 1 {
-		    return -1,  {{.name}}SliceItem.time[0], float64({{.name}}SliceItem.value[0]), false
-		}
-
-		sort.Stable({{.name}}SliceItem)
-
-        if length % 2 == 0 {
-            lowvalue, highvalue := {{.name}}SliceItem.value[length/2-1], {{.name}}SliceItem.value[length/2]
-            return -1, {{.name}}SliceItem.time[length/2-1], float64(lowvalue) + float64(highvalue - lowvalue)/2, false
-        }
-		return -1, {{.name}}SliceItem.time[length/2], float64({{.name}}SliceItem.value[length/2]), false
-}
-{{- end}}
-{{end}}
-
-{{range .}}
-{{- if or (eq .Name "Float") (eq .Name "Integer") (eq .Name "String")}}
-func New{{.Name}}ModeReduce({{.Name}}SliceItem *{{.Name}}SliceItem) (int, int64, float64, bool) {
-	length := len({{.Name}}SliceItem.value)
-	start := 0
-	end := length-1
-	if length == 0 {
-		return 0, 0, 0, true
-	}
-
-	sort.Stable({{.Name}}SliceItem)
-	curri := start
-	currFreq := 0
-	currValue := {{.Name}}SliceItem.value[start]
-	modei := start
-	modeFreq := 0
-	for i := start; i <= end; i++ {
-		if {{.Name}}SliceItem.value[i] != currValue{
-			currFreq = 1
-			currValue = {{.Name}}SliceItem.value[i]
-			curri = i
-			continue
-		}
-		currFreq++
-		if modeFreq > currFreq || (modeFreq == currFreq && {{.Name}}SliceItem.time[curri] > {{.Name}}SliceItem.time[modei]){
-			continue
-		}
-		modeFreq = currFreq
-		modei = curri
-	}
-	return modei, 0, 0, false
-}
-{{- end}}
-{{end}}
-
-
 func NewBooleanModeReduce(BooleanSliceItem *BooleanSliceItem) (int, int64, float64, bool) {
 	length := len(BooleanSliceItem.value)
 	if length == 0 {
@@ -114,23 +56,6 @@ func NewBooleanModeReduce(BooleanSliceItem *BooleanSliceItem) (int, int64, float
 	return falsei, 0, 0, false
 }
 
-
-{{range .}}
-func {{.Name}}CountReduce(c Chunk, ordinal, start, end int) (int, int64, bool) {
-	var count int64
-	if c.Column(ordinal).NilCount() == 0 {
-		// fast path
-		count = int64(end - start)
-		return start, count, count == 0
-	}
-
-	// slow path
-	vs, ve := c.Column(ordinal).GetRangeValueIndexV2(start, end)
-	count = int64(ve - vs)
-	return start, count, count == 0
-}
-{{end}}
-
 func IntegerCountMerge(prevPoint, currPoint *IntegerPoint) {
 	if currPoint.isNil {
 		return
@@ -1146,28 +1071,6 @@ func {{.Name}}IrateMerge(prevPoints [2]*{{.Name}}Point, interval *hybridqp.Inter
 {{- end}}
 {{end}}
 
-{{range .}}
-func {{.Name}}AbsentReduce(c Chunk, ordinal, start, end int) (int, int64, bool) {
-	var count int64
-	if c.Column(ordinal).NilCount() == 0 {
-		// fast path
-		count = int64(end - start)
-		if count > 0 {
-			return start, 1, false
-		}
-		return start, 0, true
-	}
-
-	// slow path
-	vs, ve := c.Column(ordinal).GetRangeValueIndexV2(start, end)
-	count = int64(ve - vs)
-	if count > 0 {
-		return start, 1, false
-	}
-	return start, 0, true
-}
-{{end}}
-
 func IntegerAbsentMerge(prevPoint, currPoint *IntegerPoint) {
 	if prevPoint.isNil && currPoint.isNil {
 		prevPoint.isNil = true

engine/executor/agg_func.gen.go.tmpl

@@ -114,3 +114,37 @@ func NewModeReduce[T util.ExceptBool](si *SliceItem[T]) (int, int64, float64, bo
 	}
 	return modei, 0, 0, false
 }
+
+func CountReduce(c Chunk, ordinal, start, end int) (int, int64, bool) {
+	var count int64
+	if c.Column(ordinal).NilCount() == 0 {
+		// fast path
+		count = int64(end - start)
+		return start, count, count == 0
+	}
+
+	// slow path
+	vs, ve := c.Column(ordinal).GetRangeValueIndexV2(start, end)
+	count = int64(ve - vs)
+	return start, count, count == 0
+}
+
+func AbsentReduce(c Chunk, ordinal, start, end int) (int, int64, bool) {
+	var count int64
+	if c.Column(ordinal).NilCount() == 0 {
+		// fast path
+		count = int64(end - start)
+		if count > 0 {
+			return start, 1, false
+		}
+		return start, 0, true
+	}
+
+	// slow path
+	vs, ve := c.Column(ordinal).GetRangeValueIndexV2(start, end)
+	count = int64(ve - vs)
+	if count > 0 {
+		return start, 1, false
+	}
+	return start, 0, true
+}

engine/executor/agg_func.go

@@ -394,22 +394,22 @@ func NewCountRoutineImpl(inRowDataType, outRowDataType hybridqp.RowDataType, opt
 	switch dataType {
 	case influxql.Integer:
 		return NewRoutineImpl(
-			NewIntegerColIntegerIterator(IntegerCountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewIntegerColIntegerIterator(CountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.Float:
 		return NewRoutineImpl(
-			NewFloatColIntegerIterator(FloatCountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewFloatColIntegerIterator(CountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.String:
 		return NewRoutineImpl(
-			NewStringColIntegerIterator(StringCountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewStringColIntegerIterator(CountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.Boolean:
 		return NewRoutineImpl(
-			NewBooleanColIntegerIterator(BooleanCountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewBooleanColIntegerIterator(CountReduce, IntegerCountMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	default:
@@ -929,22 +929,22 @@ func NewAbsentRoutineImpl(inRowDataType, outRowDataType hybridqp.RowDataType, op
 	switch dataType {
 	case influxql.Integer:
 		return NewRoutineImpl(
-			NewIntegerColIntegerIterator(IntegerAbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewIntegerColIntegerIterator(AbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.Float:
 		return NewRoutineImpl(
-			NewFloatColIntegerIterator(FloatAbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewFloatColIntegerIterator(AbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.String:
 		return NewRoutineImpl(
-			NewStringColIntegerIterator(StringAbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewStringColIntegerIterator(AbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	case influxql.Boolean:
 		return NewRoutineImpl(
-			NewBooleanColIntegerIterator(BooleanAbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
+			NewBooleanColIntegerIterator(AbsentReduce, IntegerAbsentMerge, isSingleCall, inOrdinal, outOrdinal,
 				nil, nil),
 			inOrdinal, outOrdinal), nil
 	default:

engine/executor/call_processor.go

@@ -97,25 +97,25 @@ func NewSlidingWindowCountRoutineImpl(
 	case influxql.Integer:
 		return NewRoutineImpl(
 			NewIntegerSlidingWindowIntegerIterator(
-				IntegerCountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
+				CountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
 				inOrdinal, outOrdinal, slidingNum),
 			inOrdinal, outOrdinal)
 	case influxql.Float:
 		return NewRoutineImpl(
 			NewIntegerSlidingWindowIntegerIterator(
-				FloatCountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
+				CountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
 				inOrdinal, outOrdinal, slidingNum),
 			inOrdinal, outOrdinal)
 	case influxql.String:
 		return NewRoutineImpl(
 			NewIntegerSlidingWindowIntegerIterator(
-				StringCountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
+				CountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
 				inOrdinal, outOrdinal, slidingNum),
 			inOrdinal, outOrdinal)
 	case influxql.Boolean:
 		return NewRoutineImpl(
 			NewBooleanSlidingWindowIntegerIterator(
-				BooleanCountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
+				CountReduce, IntegerCountMerge, IntegerSlidingWindowMergeFunc,
 				inOrdinal, outOrdinal, slidingNum),
 			inOrdinal, outOrdinal)
 	default: