EEOP_AGG

higuoxing · Sep 1, 2024 · 79413bb · 79413bb
1 parent aef932a
commit 79413bb
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/asmjit_expr.cc b/asmjit_expr.cc
@@ -1690,7 +1690,23 @@ bool AsmJitCompileExpr(ExprState *State) {
       break;
     }
     case EEOP_AGG_PLAIN_PERGROUP_NULLCHECK: {
-      todo();
+      int jumpnull = Op->d.agg_plain_pergroup_nullcheck.jumpnull;
+
+      /*
+       * pergroup_allaggs = aggstate->all_pergroups
+       * [op->d.agg_plain_pergroup_nullcheck.setoff];
+       */
+      x86::Gp v_aggstatep = emit_load_parent_from_ExprState(Jitcc, Expression);
+      x86::Gp v_allpergroupsp =
+          emit_load_all_pergroups_from_AggState(Jitcc, v_aggstatep);
+      x86::Gp v_pergroup_allaggs =
+          Jitcc.newUIntPtr("v_pergroup_allaggs.uintptr");
+      EmitLoadFromArray(Jitcc, v_allpergroupsp,
+                        Op->d.agg_plain_pergroup_nullcheck.setoff,
+                        v_pergroup_allaggs, sizeof(Datum));
+      Jitcc.cmp(v_pergroup_allaggs, jit::imm(0));
+      Jitcc.je(L_Opblocks[jumpnull]);
+      break;
     }
 
     case EEOP_AGG_PLAIN_TRANS_INIT_STRICT_BYVAL:

diff --git a/jit_types_info.inc b/jit_types_info.inc
@@ -2,6 +2,7 @@
 TYPES_INFO(ExprState, Datum, resvalue, UIntPtr)
 TYPES_INFO(ExprState, bool, resnull, Int8)
 TYPES_INFO(ExprState, TupleTableSlot *, resultslot, UIntPtr)
+TYPES_INFO(ExprState, struct PlanState *, parent, UIntPtr)
 TYPES_INFO(ExprContext, TupleTableSlot *, ecxt_scantuple, UIntPtr)
 TYPES_INFO(ExprContext, TupleTableSlot *, ecxt_innertuple, UIntPtr)
 TYPES_INFO(ExprContext, TupleTableSlot *, ecxt_outertuple, UIntPtr)
@@ -25,3 +26,4 @@ TYPES_INFO(HeapTupleHeaderData, uint16, t_infomask, UInt16)
 TYPES_INFO(HeapTupleHeaderData, uint8, t_hoff, UInt8)
 TYPES_INFO(HeapTupleHeaderData, bits8 *, t_bits, UIntPtr)
 TYPES_INFO(FunctionCallInfoBaseData, bool, isnull, Int8)
+TYPES_INFO(AggState, AggStatePerGroup *, all_pergroups, UIntPtr)