Restlet条件路由到不同的资源

时间:2015-04-26 10:40:42

标签: java restlet

我有一个POST请求,我想根据正文内容分配给不同的Resources

如果正文包含非空的令牌"token":"1q2w3e4r5t",那么我想将请求转发给TokenedResource,,否则将其转发给NonTokenResource

我认为使用过滤器(@beforeHandle),但过滤器返回的唯一指示是CONTINUESTOP ......

有什么建议吗?

1 个答案:

答案 0 :(得分:2)

为此,您需要扩展Restlet路由。后者负责根据计算得分确定哪条路线匹配特定请求。

特别是路由涉及两个您需要覆盖的类:RouterTemplateRoute。路由器1负责创建模板路由。如果要提供自定义模板路由,则需要提供自定义路由器。在自定义模板路径中,您可以实现自己的路线得分算法。

以下是自定义路由器的实现:

public class CustomRouter extends InternalRouter {
    public CustomRouter(Context context) {
        super(context);
        setFinderClass(CustomFinder.class);
    }

    protected TemplateRoute createRoute(String uriPattern,
                Restlet target, int matchingMode) {
        CustomTemplateRoute result = new CustomTemplateRoute(
                                    this, uriPattern, target);

        result.getTemplate().setMatchingMode(matchingMode);
        result.setMatchingQuery(getDefaultMatchingQuery());
        return result;
    }
}

为什么我们需要使用方法setFinderClass指定自定义查找器,这一点并不明显。 Restlet中的查找程序负责实例化每个请求的服务器资源。默认实现(类Finder)的问题在于您无法访问关联的类(targetClass)。如果您需要知道它(似乎是这种情况),您必须提供自己的实现。我们稍后会关注这一点。

使用此类,以下是在应用程序类中附加服务器资源的方法:

@Override
public Restlet createInboundRoot() {
    Router router = new CustomRouter(getContext());

    router.attach("/test", TokenedResource.class);
    router.attach("/test", NonTokenResource.class);

    return router;
}

我们必须将它们放在同一条道路上。

以下是自定义模板路由的实现:

public class CustomTemplateRoute extends TemplateRoute {
    public CustomTemplateRoute(Restlet next) {
        super(next);
    }

    public CustomTemplateRoute(Router router,
                  String uriTemplate, Restlet next) {
        super(router, uriTemplate, next);
    }

    public CustomTemplateRoute(Router router,
                  Template template, Restlet next) {
        super(router, template, next);
    }

    @Override
    public float score(Request request, Response response) {
        float result = super.score(request, response);
        (...)
        return result;
    }
}

在方法score中,如果我们处于对类TokenedResource的强制请求的情况下,我们将增加分数,否则减少。这允许让Restlet为正确的情况选择正确的服务器资源。

在继续之前,让我们提供CustomFinder类的内容:

public class CustomFinder extends Finder {
    private Class<? extends ServerResource> targetClass;

    public CustomFinder() {
        super();
    }

    public CustomFinder(Context context) {
        super(context);
    }

    public CustomFinder(Context context,
            Class<? extends ServerResource> targetClass) {
        super(context, targetClass);
        this.targetClass = targetClass;
    }
}

以下是一种实现自定义处理以计算得分的方法:

@Override
public float score(Request request, Response response) {
    float result = super.score(request, response);

    if (isTokenedServerResource()) {
        boolean containsToken = containsToken(request);
        if (containsToken) {
            return result + 0.1f;
        } else {
            return result - 0.1f;
        }
    }

    return result;
}

方法isTokenedServerResource检查与路由关联的服务器资源(如果有)是否为类TokenedResource。在这种情况下,我们查看有效负载以查看它是否实际包含方法containsToken的令牌。

以下是方法isTokenedServerResource的示例内容:

private boolean isTokenedServerResource() {
    if (getNext() instanceof CustomFinder) {
        CustomFinder finder = (CustomFinder) getNext();
        if (MyServerResource1.class.isAssignableFrom(finder.getTargetClass())) {
            return true;
        }
    }
    return false;
}

以下是方法containsToken的示例内容。它假设使用了JSON内容,我们使用Jackson来解析它。

private boolean containsToken(Request request) {
    try {
        Representation repr = request.getEntity();
        String content = repr.getText();

        ObjectMapper objectMapper = new ObjectMapper();
        Map<String, Object> jsonContent = objectMapper.readValue(
                                  content, Map.class);

        StringRepresentation sRepr = new StringRepresentation(
                              content, repr.getMediaType());
        request.setEntity(sRepr);

        if (jsonContent.get("token") != null) {
            return true;
        }
    } catch (Exception ex) {
        (...)
    }
    return false;
}

需要注意的一点是,我们需要在请求中再次设置请求实体(我们在这里使用StringRepresentation),因为默认情况下,实体内容只能读取一次。

希望它可以帮到你, 亨利