com.weibo.rill.flow.interfaces.utils.HttpUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rill-flow-interfaces Show documentation

There is a newer version: 0.1.18

package com.weibo.rill.flow.interfaces.utils;

import com.weibo.rill.flow.interfaces.model.http.HttpParameter;
import com.weibo.rill.flow.interfaces.model.resource.Resource;
import com.weibo.rill.flow.interfaces.model.strategy.DispatchInfo;

import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

public class HttpUtil {

    public static HttpParameter functionRequestParams(Resource resource, DispatchInfo dispatchInfo) {
        String executionId = dispatchInfo.getExecutionId();
        Map input = dispatchInfo.getInput();
        String taskInfoName = dispatchInfo.getTaskInfo().getTask().getName();

        return functionRequestParams(executionId, taskInfoName, resource, input);
    }

    public static HttpParameter functionRequestParams(String executionId, String taskInfoName, Resource resource, Map input) {
        HttpParameter httpParameter = buildRequestParams(executionId, input);
        Map queryParams = httpParameter.getQueryParams();
        queryParams.put("name", taskInfoName);
        Map body = httpParameter.getBody();
        httpParameter.setBody(Optional.ofNullable((Map) body.get("data")).orElse(new HashMap<>()));
        return httpParameter;
    }

    public static HttpParameter buildRequestParams(String executionId, Map input) {
        Map queryParams = new HashMap<>();
        Map body = new HashMap<>();
        Map functionInput = new HashMap<>();
        Map callback = new HashMap<>();

        queryParams.put("execution_id", executionId);
        body.put("data", functionInput);
        body.put("group_id", executionId);
        Optional.ofNullable(input).ifPresent(inputMap -> inputMap.forEach((key, value) -> {
            if (key.startsWith("query_params_") && value instanceof Map) {
                queryParams.putAll((Map) value);
            } else if (key.startsWith("request_config_") && value instanceof Map) {
                body.putAll((Map) value);
            } else if (key.startsWith("request_callback_") && value instanceof Map) {
                callback.putAll((Map) value);
            } else {
                functionInput.put(key, value);
            }
        }));

        return HttpParameter.builder().queryParams(queryParams).body(body).callback(callback).build();
    }
}