Method QueryConn

backend/plugin/db/dynamodb/dynamodb.go:167–201

QueryConn queries a SQL statement in a given connection. The result.Rows.Values can be nil in DynamoDB, which means the column is not set in the row.

(ctx context.Context, _ *sql.Conn, statement string, queryContext db.QueryContext)

// QueryConn queries a SQL statement in a given connection.
// The result.Rows.Values can be nil in DynamoDB, which means the column is not set in the row.
func (d *Driver) QueryConn(ctx context.Context, _ *sql.Conn, statement string, queryContext db.QueryContext) ([]*v1pb.QueryResult, error) {
	if queryContext.Explain {
		return nil, errors.New("DynamoDB does not support EXPLAIN")
	}

	singleSQLs, err := base.SplitMultiSQL(storepb.Engine_DYNAMODB, statement)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to split multi statement")
	}
	singleSQLs = base.FilterEmptyStatements(singleSQLs)
	if len(singleSQLs) == 0 {
		return nil, nil
	}

	var results []*v1pb.QueryResult
	for _, singleSQL := range singleSQLs {
		startTime := time.Now()
		result, err := d.querySinglePartiQL(ctx, singleSQL.Text, queryContext)
		stop := false
		if err != nil {
			result = &v1pb.QueryResult{
				Error: err.Error(),
			}
			stop = true
		}
		result.Latency = durationpb.New(time.Since(startTime))
		result.Statement = statement
		result.RowsCount = int64(len(result.Rows))
		results = append(results, result)
		if stop {
			break
		}
	}
	return results, nil
}

type dynamodbQueryResultMeta struct {
	columnType string