Method writeQueryRows

rest/functions_api.go:112–151

Subroutine to write N1QL query results to a response

(rows sgbucket.QueryResultIterator)

// Subroutine to write N1QL query results to a response
func (h *handler) writeQueryRows(rows sgbucket.QueryResultIterator) error {
	// Use iterator to write results one at a time to the response:
	defer func() {
		if rows != nil {
			_ = rows.Close()
		}
	}()

	// Write the query results to the response, as a JSON array of objects.
	h.setHeader("Content-Type", "application/json")
	var err error
	if _, err = h.response.Write([]byte(`[`)); err != nil {
		return err
	}
	first := true
	var row interface{}
	for rows.Next(h.ctx(), &row) {
		if first {
			first = false
		} else {
			if _, err = h.response.Write([]byte(`,`)); err != nil {
				return err
			}
		}
		if err = h.addJSON(row); err != nil {
			return err
		}
		// The iterator streams results as the query engine produces them, so this loop may take most of the query's time; check for timeout after each iteration:
		if err = db.CheckTimeout(h.ctx()); err != nil {
			return err
		}
	}
	err = rows.Close()
	rows = nil // prevent 'defer' from closing again
	if err != nil {
		return err
	}
	_, err = h.response.Write([]byte("]\n"))
	return err
}