Method handleFunction

db/blip_connected_client.go:91–152

FUNCTION: Handles a Connected-Client "function" request. - Function name is in the "name" property. - Arguments (optional) are JSON-encoded in the body.

(rq *blip.Message)

// - Function name is in the "name" property.
// - Arguments (optional) are JSON-encoded in the body.
func (bh *blipHandler) handleFunction(rq *blip.Message) error {
	name, found := rq.Properties[QueryName]
	if !found {
		return base.HTTPErrorf(http.StatusBadRequest, "Missing 'name'")
	}

	requestParams, err := bh.parseJsonBody(rq, bh.db.Options.UserFunctions.MaxRequestSize)
	if err != nil {
		return err
	}

	bh.logEndpointEntry(rq.Profile(), fmt.Sprintf("name: %s", name))
	return WithTimeout(bh.loggingCtx, bh.db.UserFunctionTimeout, func(ctx context.Context) error {
		// Call the function:
		fn, err := bh.db.GetUserFunction(ctx, name, requestParams, true)
		if err != nil {
			return err
		}

		if iter, err := fn.Iterate(); err != nil {
			return err
		} else if iter != nil {
			// Write each iterated result to the response:
			defer func() {
				if iter != nil {
					_ = iter.Close()
				}
			}()
			var out bytes.Buffer
			enc := base.JSONEncoder(&out)
			var row interface{}
			for iter.Next(bh.loggingCtx, &row) {
				if err = enc.Encode(row); err != nil { // always ends with a newline
					return err
				}
				if err = CheckTimeout(ctx); err != nil {
					return err
				}
			}
			err = iter.Close()
			iter = nil
			if err != nil {
				return err
			}
			response := rq.Response()
			response.SetCompressed(true)
			response.SetJSONBodyAsBytes(out.Bytes())
			return nil

		} else {
			// Write the single result to the response:
			result, err := fn.Run(bh.loggingCtx)
			if err != nil {
				return err
			}
			response := rq.Response()
			response.SetCompressed(true)
			_ = response.SetJSONBody(result)