reflex.ReflexScriptDataHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Reflex Show documentation
null
There is a newer version: 3.0.4
/**
 * The MIT License (MIT)
 *
 * Copyright (c) 2011-2016 Incapture Technologies LLC
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package reflex;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import rapture.common.RaptureURI;
import rapture.common.api.ScriptingApi;
import rapture.common.impl.jackson.JacksonUtil;
import rapture.common.model.DocumentWithMeta;
import rapture.common.uri.URIParser;
import rapture.util.StringUtil;
import rapture.util.TypeUtil;
import reflex.node.KernelExecutor;
import reflex.value.ReflexValue;

public class ReflexScriptDataHandler extends AbstractReflexDataHandler implements IReflexScriptHandler {

    public ReflexScriptDataHandler(ScriptingApi api) {
        super(api);
    }

    @Override
    public List> batchPullData(List displayNames) {
        // Run in batches of 50
        List content = new ArrayList();
        List> batches = StringUtil.getBatches(displayNames, 50);
        for (List batch : batches) {
            List batchDisplayNames = new ArrayList();
            for (String disp : batch) {
                disp = URIParser.convertDocURI(disp);
                batchDisplayNames.add(disp);
            }
            Map contents = api.getDoc().getDocs(batchDisplayNames);
            content.addAll(contents.values());
        }
        List> ret = new ArrayList>();
        for (String c : content) {
            ret.add((Map) JacksonUtil.getMapFromJson(c));
        }
        return ret;
    }

    @Override
    public List batchPushData(List displayNames, List> datas) {
        List> displayBatches = TypeUtil.getBatches(displayNames, 50);
        List>> dataBatches = TypeUtil.getBatches(datas, 50);
        List ret = new ArrayList();
        for (int i = 0; i < displayBatches.size(); i++) {
            List batchDisplayNames = displayBatches.get(i);
            List> dataBatch = dataBatches.get(i);
            List parsedDisplayNames = new ArrayList();
            List contents = new ArrayList();
            int point = 0;
            for (String disp : batchDisplayNames) {
                disp = URIParser.convertDocURI(disp);
                parsedDisplayNames.add(disp);
                contents.add(JacksonUtil.jsonFromObject(dataBatch.get(point)));
                point++;
            }
            api.getDoc().putDocs(parsedDisplayNames, contents);
            ret.addAll(parsedDisplayNames);
        }
        return ret;
    }

    @Override
    public Map pullData(String raptureURI) {
        raptureURI = URIParser.convertDocURI(raptureURI);
        String content = api.getDoc().getDoc(raptureURI);
        if (content == null) {
            return null;
        } else {
            return (Map) JacksonUtil.getMapFromJson(content);
        }
    }

    @Override
    public String pushData(String raptureURI, Map data) {
        raptureURI = URIParser.convertDocURI(raptureURI);
        data = KernelExecutor.convert(data);
        System.out.println("Data is " + data);
        try {
            String content = JacksonUtil.jsonFromObject(data);
            String ret = api.getDoc().putDoc(raptureURI, content);
            return ret;
        } catch (Exception e) {
            e.printStackTrace();
            return "";
        }
    }

    @Override
    public String getScript(String name) {
        name = URIParser.convertDocURI(name);
        return api.getScript().getScript(name).getScript();
    }

    @Override
    public void remove(String raptureURI) {
        api.getDoc().deleteDoc(raptureURI);
    }

    @Override
    public Map metaPullData(String raptureURI) {
        raptureURI = URIParser.convertDocURI(raptureURI);
        DocumentWithMeta content = api.getDoc().getDocAndMeta(raptureURI);
        if (content == null) {
            return null;
        } else {
            Map ret = new HashMap();
            ret.put("meta", JacksonUtil.getHashFromObject(content.getMetaData()));
            ret.put("content", (Map) JacksonUtil.getMapFromJson(content.getContent()));
            return ret;
        }
    }

    @Override
    public String rawPushData(String raptureURI, String content) {
        raptureURI = URIParser.convertDocURI(raptureURI);
        String ret = api.getDoc().putDoc(raptureURI, content);
        return ret;

    }

    @Override
    public List batchPushRawData(List displayNames, List datas) {
        List> displayBatches = TypeUtil.getBatches(displayNames, 50);
        List> dataBatches = TypeUtil.getBatches(datas, 50);
        List ret = new ArrayList();
        for (int i = 0; i < displayBatches.size(); i++) {
            List batchDisplayNames = displayBatches.get(i);
            List dataBatch = dataBatches.get(i);
            List parsedDisplayNames = new ArrayList();
            for (String disp : batchDisplayNames) {
                disp = URIParser.convertDocURI(disp);
                parsedDisplayNames.add(disp);
            }
            api.getDoc().putDocs(parsedDisplayNames, dataBatch);
            ret.addAll(parsedDisplayNames);
        }
        return ret;
    }

    @Override
    public boolean hasCapability() {
        return true;
    }

    private ScriptingApi originalApi;
    
    @Override
    public void switchApi(ScriptingApi api) {
       this.api = api;
    }

    @Override
    public void resetApi() {
        this.api = originalApi;
    }
}