executor、server、planner: Increase the support of the returning keyword (

#83) * Add returning keyword plan and executor for supporting PG compliant DML (#69) * Support return result set after deletion Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * add returning plan and exectuor modify adapter.go to execute returning separately Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> Co-authored-by: RogueJin <21214103+RogueJin@users.noreply.github.com> * upload delete returning.png * upgrade DCparser and fix lint issues (#70) Signed-off-by: RogueJin <21214103+RogueJin@users.noreply.github.com> * Hacking demo (#71) * Readme: Updated readme and modified the version of TiDB cluster deployed (#50) * privileges: fixed unit test of user authentication Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * fixed error msg and added const Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * executor: Modify encrypted ciphertext in ut Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * planner: Implement some interfaces in the test to solve the problem of build failed Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * planner: Implement some interfaces in the test to solve the problem of build failed Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * update readme * Go Mod, Makefile: Dependency update (#51) * added coverage report to gitignore Signed-off-by: David <davidyangad@gmail.com> * added dev-tmp to skip explaintest Signed-off-by: David <davidyangad@gmail.com> * update DCParser version to v1.3-alpha Signed-off-by: David <davidyangad@gmail.com> * changed dependency requirement for HdrHistogram Signed-off-by: David <davidyangad@gmail.com> * downgraded DCParser version to 1.25-alpha Signed-off-by: David <davidyangad@gmail.com> * Explaintest, Server: Changed Explaintest's driver (#52) * changed driver for explaintest Signed-off-by: David <davidyangad@gmail.com> * improves status handling for server during transaction Signed-off-by: David <davidyangad@gmail.com> * Executor: Added support for setting isolation level in begin (#57) * update DCParser reference Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * added setting isolation level capability in begin Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * adding documents for BEGIN statement Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * readme: update url for build status (#56) Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * executor: uncommented some previously skipped test (#54) * uncommented some previously skipped test Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * wait longer for update version Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * further enlarge test wait time for better stability Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * variable,session: add search_path system variable. (#58) * Fixed unit test bugs Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style: fmt check Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style： fmt check , delete blank lines Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style: fmt check Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Add postgres search_path system variable Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Modifying Comment Information Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Added extra_float_digit system variable Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * fix:Delete unnecessary comments,Change the DCParser version Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * formatting: fixed formatting issues (#59) * fixing formatting issues Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * typo fix Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * fixed a bug that params description will be sent when no params in handleStmtDescription (#61) Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * infoschema , util, autoid: Compatible with some tables in pg_catalog (#63) * Compatible with some tables in pg_catalog Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * add the correct license information and fix unit test Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * expression：PostgreSQL system function support (#64) * Added the logic realization of some system functions. Added unit tests for PostgreSQL system functions. Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * update DCParser version Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * fixed built in function unit tests (#66) Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * core, executor: add the implemet of setparamtype at show, showddl, datasource, limit plan. (#67) * handle the implement of setparamtype at show, showddl or datasource plan Signed-off-by: studiolee <1964773741@qq.com> * handle the implement of setparamtype at limit plan Signed-off-by: studiolee <1964773741@qq.com> * add ut for setparamtype Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> Co-authored-by: studiolee <1964773741@qq.com> * upgrade DCparser and fix lint issues Signed-off-by: RogueJin <21214103+RogueJin@users.noreply.github.com> Co-authored-by: pupillord <51103574+pupillord@users.noreply.github.com> Co-authored-by: AmoebaProtozoa <davidyangad@gmail.com> Co-authored-by: Cue Ray <50295175+Orion7r@users.noreply.github.com> Co-authored-by: David <8039876+AmoebaProtozoa@users.noreply.github.com> Co-authored-by: studiolee <1964773741@qq.com> * merge main into hacking-demo (#79) * Readme: Updated readme and modified the version of TiDB cluster deployed (#50) * privileges: fixed unit test of user authentication Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * fixed error msg and added const Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * executor: Modify encrypted ciphertext in ut Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * planner: Implement some interfaces in the test to solve the problem of build failed Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * planner: Implement some interfaces in the test to solve the problem of build failed Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * update readme * Go Mod, Makefile: Dependency update (#51) * added coverage report to gitignore Signed-off-by: David <davidyangad@gmail.com> * added dev-tmp to skip explaintest Signed-off-by: David <davidyangad@gmail.com> * update DCParser version to v1.3-alpha Signed-off-by: David <davidyangad@gmail.com> * changed dependency requirement for HdrHistogram Signed-off-by: David <davidyangad@gmail.com> * downgraded DCParser version to 1.25-alpha Signed-off-by: David <davidyangad@gmail.com> * Explaintest, Server: Changed Explaintest's driver (#52) * changed driver for explaintest Signed-off-by: David <davidyangad@gmail.com> * improves status handling for server during transaction Signed-off-by: David <davidyangad@gmail.com> * Executor: Added support for setting isolation level in begin (#57) * update DCParser reference Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * added setting isolation level capability in begin Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * adding documents for BEGIN statement Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * readme: update url for build status (#56) Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * executor: uncommented some previously skipped test (#54) * uncommented some previously skipped test Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * wait longer for update version Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * further enlarge test wait time for better stability Signed-off-by: AmoebaProtozoa <davidyangad@gmail.com> * variable,session: add search_path system variable. (#58) * Fixed unit test bugs Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style: fmt check Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style： fmt check , delete blank lines Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * style: fmt check Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Add postgres search_path system variable Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Modifying Comment Information Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * Added extra_float_digit system variable Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * fix:Delete unnecessary comments,Change the DCParser version Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * formatting: fixed formatting issues (#59) * fixing formatting issues Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * typo fix Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * fixed a bug that params description will be sent when no params in handleStmtDescription (#61) Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> * infoschema , util, autoid: Compatible with some tables in pg_catalog (#63) * Compatible with some tables in pg_catalog Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * add the correct license information and fix unit test Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * expression：PostgreSQL system function support (#64) * Added the logic realization of some system functions. Added unit tests for PostgreSQL system functions. Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * update DCParser version Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * fixed built in function unit tests (#66) Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * core, executor: add the implemet of setparamtype at show, showddl, datasource, limit plan. (#67) * handle the implement of setparamtype at show, showddl or datasource plan Signed-off-by: studiolee <1964773741@qq.com> * handle the implement of setparamtype at limit plan Signed-off-by: studiolee <1964773741@qq.com> * add ut for setparamtype Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> Co-authored-by: studiolee <1964773741@qq.com> * Executor: Sort param (#72) * added test suites for sortParamOrder Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * removed magic Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * redo ParamMakerSorter Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * use 0 index Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * added ut case and fixed compatibility with MySQL Signed-off-by: jk <51103574+pupillord@users.noreply.github.com> Co-authored-by: jk <51103574+pupillord@users.noreply.github.com> * Conn: Improved extended query handling (#73) * removed unnecessary write back to client Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * fixed extended query describe Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * Server: Improved Extended Query Protocol (#74) * binary long input Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * binary float support Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * new field to save pg oid Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * Bind binary (#75) * binary long input Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * binary float support Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * new field to save pg oid Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * passed BenchMarkSQL Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * Server: Restructure Parse OID handling (#77) * handle OID and formatCode Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * add pgOID conversion to parse handler Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * handle OID of 0 Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * restructured parse oid handling Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> * removed unnecessary OID structure in bind Signed-off-by: AmoebaProtozoa <8039876+AmoebaProtozoa@users.noreply.github.com> Co-authored-by: pupillord <51103574+pupillord@users.noreply.github.com> Co-authored-by: Cue Ray <50295175+Orion7r@users.noreply.github.com> Co-authored-by: studiolee <1964773741@qq.com> * add DML statement returning compatibility * add insert returning support Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * update go mod Signed-off-by: Orion7r <50295175+Orion7r@users.noreply.github.com> * update go mod Signed-off-by: OrionRay <50295175+Orion7r@users.noreply.github.com> Co-authored-by: pupillord <51103574+pupillord@users.noreply.github.com> Co-authored-by: RogueJin <21214103+RogueJin@users.noreply.github.com> Co-authored-by: AmoebaProtozoa <davidyangad@gmail.com> Co-authored-by: David <8039876+AmoebaProtozoa@users.noreply.github.com> Co-authored-by: studiolee <1964773741@qq.com>
DigitalChinaOpenSource · Nov 30, 2021 · 1ba8365 · 1ba8365
1 parent 8dadaf3
commit 1ba8365
Show file tree

Hide file tree

Showing 25 changed files with 723 additions and 24 deletions.
diff --git a/docs/delete returning.png b/docs/delete returning.png
diff --git a/executor/adapter.go b/executor/adapter.go
@@ -79,6 +79,11 @@ type recordSet struct {
 	stmt       *ExecStmt
 	lastErr    error
 	txnStartTS uint64
+	rows       []chunk.Row
+}
+
+func (a *recordSet) Rows() []chunk.Row {
+	return a.rows
 }
 
 func (a *recordSet) Fields() []*ast.ResultField {
@@ -364,6 +369,9 @@ func (a *ExecStmt) Exec(ctx context.Context) (_ sqlexec.RecordSet, err error) {
 	}
 
 	if handled, result, err := a.handleNoDelay(ctx, e, isPessimistic); handled {
+		//if returningRS != nil {
+		//	return returningRS, err
+		//}
 		return result, err
 	}
 
@@ -439,6 +447,10 @@ type chunkRowRecordSet struct {
 	execStmt *ExecStmt
 }
 
+func (c *chunkRowRecordSet) Rows() []chunk.Row {
+	return c.rows
+}
+
 func (c *chunkRowRecordSet) Fields() []*ast.ResultField {
 	return c.fields
 }
@@ -507,7 +519,7 @@ func (a *ExecStmt) handleNoDelayExecutor(ctx context.Context, e Executor) (sqlex
 		defer span1.Finish()
 		ctx = opentracing.ContextWithSpan(ctx, span1)
 	}
-
+	var returningRS *recordSet
 	// Check if "tidb_snapshot" is set for the write executors.
 	// In history read mode, we can not do write operations.
 	switch e.(type) {
@@ -532,9 +544,57 @@ func (a *ExecStmt) handleNoDelayExecutor(ctx context.Context, e Executor) (sqlex
 	if err != nil {
 		return nil, err
 	}
+
+	// Do returning
+	returningRS = getReturningRecordSet(ctx, e, a)
+
+	if returningRS != nil {
+		return returningRS, err
+	}
 	return nil, err
 }
 
+func getReturningRecordSet(ctx context.Context, e Executor, a *ExecStmt) *recordSet {
+	var rs *recordSet
+	if del, ok := e.(*DeleteExec); ok && del.returning != nil {
+		err := del.returning.Next(ctx, del.AffectedRows())
+		if err != nil {
+			return rs
+		}
+
+		if ret, ok := del.returning.(*ReturningExec); ok {
+			rs = ret.ResultSet
+			rs.stmt = a
+		}
+	}
+
+	if updt, ok := e.(*UpdateExec); ok && updt.returning != nil {
+		err := updt.returning.Next(ctx, updt.AffectedRows())
+		if err != nil {
+			return rs
+		}
+
+		if ret, ok := updt.returning.(*ReturningExec); ok {
+			rs = ret.ResultSet
+			rs.stmt = a
+		}
+	}
+
+	if insert, ok := e.(*InsertExec); ok && insert.returning != nil {
+		err := insert.returning.Next(ctx, insert.evalBuffer4Return.ToRow().Chunk())
+		if err != nil {
+			return rs
+		}
+
+		if ret, ok := insert.returning.(*ReturningExec); ok {
+			rs = ret.ResultSet
+			rs.stmt = a
+			rs.stmt.OutputNames = ret.ReturningFields
+		}
+	}
+	return rs
+}
+
 func (a *ExecStmt) handlePessimisticDML(ctx context.Context, e Executor) error {
 	sctx := a.Ctx
 	// Do not active the transaction here.

diff --git a/executor/builder.go b/executor/builder.go
@@ -230,6 +230,8 @@ func (b *executorBuilder) build(p plannercore.Plan) Executor {
 		return b.buildAdminShowTelemetry(v)
 	case *plannercore.AdminResetTelemetryID:
 		return b.buildAdminResetTelemetryID(v)
+	case *plannercore.PhysicalReturning:
+		return b.buildReturning(v)
 	default:
 		if mp, ok := p.(MockPhysicalPlan); ok {
 			return mp.GetExecutor()
@@ -723,6 +725,12 @@ func (b *executorBuilder) buildInsert(v *plannercore.Insert) Executor {
 		InsertValues: ivs,
 		OnDuplicate:  append(v.OnDuplicate, v.GenCols.OnDuplicates...),
 	}
+
+	if v.ReturningPlan != nil {
+		insert.returning = b.build(v.ReturningPlan)
+		insert.returning.(*ReturningExec).ReturningFields = v.OutputNames()
+	}
+
 	return insert
 }
 
@@ -1708,6 +1716,11 @@ func (b *executorBuilder) buildUpdate(v *plannercore.Update) Executor {
 		tblID2table:               tblID2table,
 		tblColPosInfos:            v.TblColPosInfos,
 	}
+
+	if v.ReturningPlan != nil {
+		updateExec.returning = b.build(v.ReturningPlan)
+	}
+
 	return updateExec
 }
 
@@ -1721,6 +1734,7 @@ func (b *executorBuilder) buildDelete(v *plannercore.Delete) Executor {
 	}
 	b.snapshotTS = b.ctx.GetSessionVars().TxnCtx.GetForUpdateTS()
 	selExec := b.build(v.SelectPlan)
+
 	if b.err != nil {
 		return nil
 	}
@@ -1732,6 +1746,11 @@ func (b *executorBuilder) buildDelete(v *plannercore.Delete) Executor {
 		IsMultiTable:   v.IsMultiTable,
 		tblColPosInfos: v.TblColPosInfos,
 	}
+
+	if v.ReturningPlan != nil {
+		deleteExec.returning = b.build(v.ReturningPlan)
+	}
+
 	return deleteExec
 }
 
@@ -3144,3 +3163,18 @@ func (b *executorBuilder) buildAdminShowTelemetry(v *plannercore.AdminShowTeleme
 func (b *executorBuilder) buildAdminResetTelemetryID(v *plannercore.AdminResetTelemetryID) Executor {
 	return &AdminResetTelemetryIDExec{baseExecutor: newBaseExecutor(b.ctx, v.Schema(), v.ID())}
 }
+
+func (b *executorBuilder) buildReturning(v *plannercore.PhysicalReturning) Executor {
+	childExec := b.build(v.Children()[0])
+	if b.err != nil {
+		return nil
+	}
+
+	returningExec := ReturningExec{
+		baseExecutor: newBaseExecutor(b.ctx, v.Schema(), v.ID(), childExec),
+		schema:       v.Schema(),
+	}
+	executorCounterSortExec.Inc()
+
+	return &returningExec
+}
diff --git a/executor/delete.go b/executor/delete.go
@@ -37,6 +37,9 @@ type DeleteExec struct {
 	// the columns ordinals is present in ordinal range format, @see plannercore.TblColPosInfos
 	tblColPosInfos plannercore.TblColPosInfoSlice
 	memTracker     *memory.Tracker
+	affectedRow    *chunk.Chunk
+
+	returning Executor
 }
 
 // Next implements the Executor Next interface.
@@ -112,6 +115,7 @@ func (e *DeleteExec) deleteSingleTableByChunk(ctx context.Context) error {
 			}
 			rowCount++
 		}
+		e.AddAffectedRows(chk)
 		chk = chunk.Renew(chk, e.maxChunkSize)
 	}
 
@@ -197,10 +201,32 @@ func (e *DeleteExec) Open(ctx context.Context) error {
 	e.memTracker = memory.NewTracker(e.id, -1)
 	e.memTracker.AttachTo(e.ctx.GetSessionVars().StmtCtx.MemTracker)
 
-	return e.children[0].Open(ctx)
+	err := e.children[0].Open(ctx)
+	if err != nil {
+		return err
+	}
+
+	if e.returning != nil {
+		err = e.returning.Open(ctx)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
 }
 
 // tableRowMapType is a map for unique (Table, Row) pair. key is the tableID.
 // the key in map[int64]Row is the joined table handle, which represent a unique reference row.
 // the value in map[int64]Row is the deleting row.
 type tableRowMapType map[int64]map[int64][]types.Datum
+
+// AffectedRows get affected rows.
+func (e *DeleteExec) AffectedRows() *chunk.Chunk {
+	return e.affectedRow
+}
+
+// AddAffectedRows adds affected row.
+func (e *DeleteExec) AddAffectedRows(chk *chunk.Chunk) {
+	e.affectedRow = chk
+}
diff --git a/executor/insert.go b/executor/insert.go
@@ -37,10 +37,12 @@ import (
 // InsertExec represents an insert executor.
 type InsertExec struct {
 	*InsertValues
-	OnDuplicate    []*expression.Assignment
-	evalBuffer4Dup chunk.MutRow
-	curInsertVals  chunk.MutRow
-	row4Update     []types.Datum
+	OnDuplicate       []*expression.Assignment
+	evalBuffer4Dup    chunk.MutRow
+	evalBuffer4Return chunk.MutRow
+	curInsertVals     chunk.MutRow
+	row4Update        []types.Datum
+	returning         Executor
 
 	Priority mysql.PriorityEnum
 }
@@ -312,6 +314,10 @@ func (e *InsertExec) Open(ctx context.Context) error {
 	if !e.allAssignmentsAreConstant {
 		e.initEvalBuffer()
 	}
+	if e.returning != nil {
+		e.initEvalBuffer4Return()
+	}
+
 	return nil
 }
 
@@ -338,6 +344,27 @@ func (e *InsertExec) initEvalBuffer4Dup() {
 	e.row4Update = make([]types.Datum, 0, len(evalBufferTypes))
 }
 
+func (e *InsertExec) initEvalBuffer4Return() {
+	// Use public columns for new row.
+	numCols := len(e.Table.Cols())
+	// Use writable columns for old row for update.
+	numWritableCols := len(e.Table.WritableCols())
+
+	evalBufferTypes := make([]*types.FieldType, 0, numCols+numWritableCols)
+
+	// Append the old row before the new row, to be consistent with "Schema4OnDuplicate" in the "Insert" PhysicalPlan.
+	for _, col := range e.Table.WritableCols() {
+		evalBufferTypes = append(evalBufferTypes, &col.FieldType)
+	}
+	for _, col := range e.Table.Cols() {
+		evalBufferTypes = append(evalBufferTypes, &col.FieldType)
+	}
+	if e.hasExtraHandle {
+		evalBufferTypes = append(evalBufferTypes, types.NewFieldType(mysql.TypeLonglong))
+	}
+	e.evalBuffer4Return = chunk.MutRowFromTypes(evalBufferTypes[numWritableCols:])
+}
+
 // doDupRowUpdate updates the duplicate row.
 func (e *InsertExec) doDupRowUpdate(ctx context.Context, handle int64, oldRow []types.Datum, newRow []types.Datum,
 	cols []*expression.Assignment) ([]types.Datum, bool, int64, error) {

diff --git a/executor/returning.go b/executor/returning.go
@@ -0,0 +1,92 @@
+// Copyright 2021 Digital China Group Co.,Ltd
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package executor
+
+import (
+	"context"
+	"github.com/pingcap/tidb/types"
+	"runtime/trace"
+	"time"
+
+	"github.com/pingcap/tidb/expression"
+	"github.com/pingcap/tidb/util/chunk"
+	"github.com/pingcap/tidb/util/execdetails"
+)
+
+// ReturningExec represents Returning Executor
+type ReturningExec struct {
+	baseExecutor
+
+	Idx             int
+	fetched         bool
+	schema          *expression.Schema
+	ReturningFields types.NameSlice
+	ResultSet       *recordSet
+}
+
+// Open Returning Executor
+func (e *ReturningExec) Open(ctx context.Context) error {
+
+	return e.children[0].Open(ctx)
+}
+
+// Next Returning Executor
+func (e *ReturningExec) Next(ctx context.Context, req *chunk.Chunk) error {
+	return e.fetchRowChunks(ctx, req)
+}
+
+// Close Returning Executor
+func (e *ReturningExec) Close() error {
+	return e.children[0].Close()
+}
+
+// Returning Executor fetchRowChunks
+func (e *ReturningExec) fetchRowChunks(ctx context.Context, req *chunk.Chunk) error {
+	defer func() {
+		e.fetched = true
+	}()
+
+	var stmtDetail *execdetails.StmtExecDetails
+	stmtDetailRaw := ctx.Value(execdetails.StmtExecDetailKey)
+	if stmtDetailRaw != nil {
+		stmtDetail = stmtDetailRaw.(*execdetails.StmtExecDetails)
+	}
+
+	rs := &recordSet{
+		executor: e.base().children[0],
+	}
+	e.ResultSet = rs
+	rs.NewChunk()
+	if req == nil {
+		return nil
+	}
+
+	rowCount := req.NumRows()
+	if rowCount == 0 {
+		return nil
+	}
+	start := time.Now()
+	reg := trace.StartRegion(ctx, "ProcessReturning")
+	iter := chunk.NewIterator4Chunk(req)
+	for chunkRow := iter.Begin(); chunkRow != iter.End(); chunkRow = iter.Next() {
+		rs.rows = append(rs.rows, chunkRow)
+	}
+	e.ResultSet = rs
+	if stmtDetail != nil {
+		stmtDetail.WriteSQLRespDuration += time.Since(start)
+	}
+	reg.End()
+
+	return nil
+}