com.redhat.lightblue.mediator.SimpleFindImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lightblue-core-crud Show documentation
There is a newer version: 2.18.0

/*
 Copyright 2013 Red Hat, Inc. and/or its affiliates.

 This file is part of lightblue.

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see .
 */
package com.redhat.lightblue.mediator;

import java.util.ArrayList;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.node.JsonNodeFactory;

import com.redhat.lightblue.crud.CRUDController;
import com.redhat.lightblue.crud.CRUDFindRequest;
import com.redhat.lightblue.crud.CRUDFindResponse;
import com.redhat.lightblue.crud.DocCtx;
import com.redhat.lightblue.crud.Factory;
import com.redhat.lightblue.crud.ListDocumentStream;
import com.redhat.lightblue.crud.ExplainQuerySupport;

import com.redhat.lightblue.metadata.EntityMetadata;

import com.redhat.lightblue.util.JsonDoc;
import com.redhat.lightblue.util.Path;

public class SimpleFindImpl implements Finder {

    private static final Logger LOGGER = LoggerFactory.getLogger(SimpleFindImpl.class);

    private final CRUDController controller;

    public SimpleFindImpl(EntityMetadata md,
                          Factory factory) {
        this.controller = factory.getCRUDController(md);
        LOGGER.debug("Controller for {}:{}", md.getName(), controller.getClass().getName());
    }

    @Override
    public CRUDFindResponse find(OperationContext ctx,
                                 CRUDFindRequest req) {
        CRUDFindResponse result = controller.find(ctx,
                req.getQuery(),
                req.getProjection(),
                req.getSort(),
                req.getFrom(),
                req.getTo());
        return result;
    }

    @Override
    public void explain(OperationContext ctx,
                        CRUDFindRequest req) {
        if(controller instanceof ExplainQuerySupport) {
            JsonDoc doc=new JsonDoc(ctx.getFactory().getNodeFactory().objectNode());
            if(req.getQuery()!=null)
                doc.modify(new Path("request.query"),req.getQuery().toJson(),true);
            if(req.getProjection()!=null) 
                doc.modify(new Path("request.projection"),req.getProjection().toJson(),true);
            if(req.getSort()!=null)
                doc.modify(new Path("request.sort"),req.getSort().toJson(),true);
            if(req.getFrom()!=null)
                doc.modify(new Path("request.from"),JsonNodeFactory.instance.numberNode(req.getFrom()),true);
            if(req.getTo()!=null)
                doc.modify(new Path("request.to"),JsonNodeFactory.instance.numberNode(req.getTo()),true);
            ((ExplainQuerySupport)controller).explain(ctx,
                                                      req.getQuery(),
                                                      req.getProjection(),
                                                      req.getSort(),
                                                      req.getFrom(),
                                                      req.getTo(),
                                                      doc);
            LOGGER.debug("Adding explain doc:{}",doc);
            ArrayList l=new ArrayList<>(1);
            l.add(new DocCtx(doc));
            ctx.setDocumentStream(new ListDocumentStream(l));
        }
    }
}