cluster-req-demo/src/main/java/com/fr/plugin/cluster/req/demo/cluster/ClusterReqDemoForwardProvider.java

package com.fr.plugin.cluster.req.demo.cluster;

import com.fr.plugin.cluster.req.demo.TaskInfoManager;
import com.fr.plugin.cluster.req.demo.URIConstants;
import com.fr.stable.fun.impl.AbstractClusterForwardProvider;
import com.fr.web.utils.WebUtils;
import org.jetbrains.annotations.Nullable;

import javax.servlet.http.HttpServletRequest;

public class ClusterReqDemoForwardProvider extends AbstractClusterForwardProvider {

    @Override
    public @Nullable String getTargetNodeID(String key) {
        return TaskInfoManager.getTargetNodeId(key);
    }

    @Override
    public String getKey(HttpServletRequest request) {
        return WebUtils.getHTTPRequestParameter(request, "id");
    }

    @Override
    public boolean accept(HttpServletRequest request) {
        String requestUri = request.getRequestURI();
        return requestUri.contains(URIConstants.CREATE) || requestUri.contains(URIConstants.SEARCH);
    }
}