Method QueryConn

backend/plugin/db/mongodb/mongodb.go:179–197

QueryConn queries a SQL statement in a given connection.

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

// QueryConn queries a SQL statement in a given connection.
func (d *Driver) QueryConn(ctx context.Context, _ *sql.Conn, statement string, queryContext db.QueryContext) ([]*v1pb.QueryResult, error) {
	if queryContext.Explain {
		return nil, errors.New("MongoDB does not support EXPLAIN")
	}

	statement = strings.Trim(statement, " \t\n\r\f;")
	startTime := time.Now()

	gmClient := gomongo.NewClient(d.client)
	var gmOpts []gomongo.ExecuteOption
	if queryContext.Limit > 0 {
		gmOpts = append(gmOpts, gomongo.WithMaxRows(int64(queryContext.Limit)))
	}
	result, err := gmClient.Execute(ctx, d.databaseName, statement, gmOpts...)
	if err != nil {
		return nil, err
	}
	return d.convertGomongoResult(result, statement, startTime), nil
}

func (*Driver) convertGomongoResult(res *gomongo.Result, statement string, startTime time.Time) []*v1pb.QueryResult {
	rows := []*v1pb.QueryRow{}