feat: add inc-hash-agg min/max bool type (openGemini#590)

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

Author: chen19991212

engine/executor/chunk.go

@@ -381,6 +381,8 @@ func (c *ChunkImpl) AppendIntervalFullRows(start, step int64, num int, tag Chunk
 			c.Column(i).AppendIntegerValues(make([]int64, num))
 		case influxql.Float:
 			c.Column(i).AppendFloatValues(make([]float64, num))
+		case influxql.Boolean:
+			c.Column(i).AppendBooleanValues(make([]bool, num))
 		default:
 			panic("unsupported the data type for AppendIntervalFullRows")
 		}

engine/executor/column.gen.go

@@ -69,7 +69,7 @@ type Column interface {
 		Append{{.Name}}Value({{.Type}})
 		Append{{.Name}}Values([]{{.Type}})
 		Set{{.Name}}Values([]{{.Type}})
-        {{- if or (eq .Name "Float") (eq .Name "Integer")}}
+        {{- if or (eq .Name "Float") (eq .Name "Integer") (eq .Name "Boolean")}}
         Update{{.Name}}ValueFast(v {{.Type}}, row int)
         {{- end}}
     {{- else }}
@@ -289,7 +289,8 @@ func (c *ColumnImpl) Set{{.Name}}Values(values []{{.Type}}) {
 	c.{{.name}}Values = values
 }
 
-{{- if or (eq .Name "Float") (eq .Name "Integer")}}
+{{- if or (eq .Name "Float") (eq .Name "Integer") (eq .Name "Boolean")}}
+
 func (c *ColumnImpl) Update{{.Name}}ValueFast(v {{.Type}}, row int) {
 	c.{{.name}}Values[row] = v
 }

engine/executor/column.gen.go.tmpl

@@ -110,6 +110,35 @@ func UpdateHashFloatMinFast(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, sr
 	dstColumn.UpdateFloatValueFast(minV, dstRow)
 }
 
+func UpdateHashBooleanMinSlow(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, srcRow int) {
+	srcColumn, dstColumn := srcChunk.Column(srcCol), dstChunk.Column(dstCol)
+	isNil := srcColumn.IsNilV2(srcRow)
+	if isNil {
+		return
+	}
+	srcRow = srcColumn.GetValueIndexV2(srcRow)
+	sv := srcColumn.BooleanValue(srcRow)
+	dv := dstColumn.BooleanValue(dstRow)
+	if !dv {
+		return
+	}
+	if !sv {
+		dstColumn.UpdateBooleanValueFast(sv, dstRow)
+	}
+}
+
+func UpdateHashBooleanMinFast(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, srcRow int) {
+	dstColumn := dstChunk.Column(dstCol)
+	sv := srcChunk.Column(srcCol).BooleanValue(srcRow)
+	dv := dstColumn.BooleanValue(dstRow)
+	if !dv {
+		return
+	}
+	if !sv {
+		dstColumn.UpdateBooleanValueFast(sv, dstRow)
+	}
+}
+
 func rewriteInterMinColumnFunc(chunk Chunk, srcCol, lastNum int) {
 	column := chunk.Column(srcCol)
 	lastIndex := len(column.IntegerValues()) - 1
@@ -134,6 +163,18 @@ func rewriteFloatMinColumnFunc(chunk Chunk, srcCol, lastNum int) {
 	}
 }
 
+func rewriteBooleanMinColumnFunc(chunk Chunk, srcCol, lastNum int) {
+	column := chunk.Column(srcCol)
+	lastIndex := len(column.BooleanValues()) - 1
+	firstIndex := lastIndex - lastNum + 1
+	if firstIndex < 0 {
+		return
+	}
+	for i := lastIndex; i >= firstIndex; i-- {
+		column.UpdateBooleanValueFast(true, i)
+	}
+}
+
 func UpdateHashInterMaxSlow(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, srcRow int) {
 	srcColumn, dstColumn := srcChunk.Column(srcCol), dstChunk.Column(dstCol)
 	isNil := srcColumn.IsNilV2(srcRow)
@@ -188,6 +229,35 @@ func UpdateHashFloatMaxFast(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, sr
 	dstColumn.UpdateFloatValueFast(maxV, dstRow)
 }
 
+func UpdateHashBooleanMaxSlow(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, srcRow int) {
+	srcColumn, dstColumn := srcChunk.Column(srcCol), dstChunk.Column(dstCol)
+	isNil := srcColumn.IsNilV2(srcRow)
+	if isNil {
+		return
+	}
+	srcRow = srcColumn.GetValueIndexV2(srcRow)
+	sv := srcColumn.BooleanValue(srcRow)
+	dv := dstColumn.BooleanValue(dstRow)
+	if dv {
+		return
+	}
+	if sv {
+		dstColumn.UpdateBooleanValueFast(sv, dstRow)
+	}
+}
+
+func UpdateHashBooleanMaxFast(dstChunk, srcChunk Chunk, dstCol, srcCol, dstRow, srcRow int) {
+	dstColumn := dstChunk.Column(dstCol)
+	sv := srcChunk.Column(srcCol).BooleanValue(srcRow)
+	dv := dstColumn.BooleanValue(dstRow)
+	if dv {
+		return
+	}
+	if sv {
+		dstColumn.UpdateBooleanValueFast(sv, dstRow)
+	}
+}
+
 func rewriteInterMaxColumnFunc(chunk Chunk, srcCol, lastNum int) {
 	column := chunk.Column(srcCol)
 	lastIndex := len(column.IntegerValues()) - 1
@@ -211,3 +281,15 @@ func rewriteFloatMaxColumnFunc(chunk Chunk, srcCol, lastNum int) {
 		column.UpdateFloatValueFast(-math.MaxFloat64, i)
 	}
 }
+
+func rewriteBooleanMaxColumnFunc(chunk Chunk, srcCol, lastNum int) {
+	column := chunk.Column(srcCol)
+	lastIndex := len(column.BooleanValues()) - 1
+	firstIndex := lastIndex - lastNum + 1
+	if firstIndex < 0 {
+		return
+	}
+	for i := lastIndex; i >= firstIndex; i-- {
+		column.UpdateBooleanValueFast(false, i)
+	}
+}

engine/executor/inc_hash_agg_funcs.go

@@ -522,6 +522,10 @@ func (trans *IncHashAggTransform) buildIncMinAggFuncs(i int) {
 		trans.rewriteColumnFunc[srcCol] = rewriteFloatMinColumnFunc
 		trans.incAggFuncs[srcCol][SlowFuncIdx] = UpdateHashFloatMinSlow
 		trans.incAggFuncs[srcCol][FastFuncIdx] = UpdateHashFloatMinFast
+	case influxql.Boolean:
+		trans.rewriteColumnFunc[srcCol] = rewriteBooleanMinColumnFunc
+		trans.incAggFuncs[srcCol][SlowFuncIdx] = UpdateHashBooleanMinSlow
+		trans.incAggFuncs[srcCol][FastFuncIdx] = UpdateHashBooleanMinFast
 	default:
 		panic(fmt.Sprintf("unsupport data type %s for min", srcType))
 	}
@@ -544,6 +548,10 @@ func (trans *IncHashAggTransform) buildIncMaxAggFuncs(i int) {
 		trans.rewriteColumnFunc[srcCol] = rewriteFloatMaxColumnFunc
 		trans.incAggFuncs[srcCol][SlowFuncIdx] = UpdateHashFloatMaxSlow
 		trans.incAggFuncs[srcCol][FastFuncIdx] = UpdateHashFloatMaxFast
+	case influxql.Boolean:
+		trans.rewriteColumnFunc[srcCol] = rewriteBooleanMaxColumnFunc
+		trans.incAggFuncs[srcCol][SlowFuncIdx] = UpdateHashBooleanMaxSlow
+		trans.incAggFuncs[srcCol][FastFuncIdx] = UpdateHashBooleanMaxFast
 	default:
 		panic(fmt.Sprintf("unsupport data type %s for max", srcType))
 	}