RecordScanWriter.java

/*
 * Copyright 2012 NGDATA nv
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.lilyproject.tools.import_.json;

import org.codehaus.jackson.node.ArrayNode;
import org.codehaus.jackson.node.ObjectNode;
import org.lilyproject.repository.api.LRepository;
import org.lilyproject.repository.api.QName;
import org.lilyproject.repository.api.RecordScan;
import org.lilyproject.repository.api.RepositoryException;
import org.lilyproject.tools.import_.json.filters.RecordFilterJsonConverters;
import org.lilyproject.util.json.JsonFormat;

public class RecordScanWriter implements EntityWriter<RecordScan> {
    public static final RecordScanWriter INSTANCE = new RecordScanWriter();

    @Override
    public ObjectNode toJson(RecordScan entity, WriteOptions options, LRepository repository)
            throws RepositoryException, InterruptedException {
        Namespaces namespaces = new NamespacesImpl(options != null ? options.getUseNamespacePrefixes() :
                        NamespacesImpl.DEFAULT_USE_PREFIXES);

        ObjectNode node = toJson(entity, options, namespaces, repository);

        if (namespaces.usePrefixes()) {
            node.put("namespaces", NamespacesConverter.toJson(namespaces));
        }

        return node;
    }

    @Override
    public ObjectNode toJson(RecordScan scan, WriteOptions options, Namespaces namespaces, LRepository repository)
            throws RepositoryException, InterruptedException {

        ObjectNode node = JsonFormat.OBJECT_MAPPER.createObjectNode();

        if (scan.getStartRecordId() != null) {
            node.put("startRecordId", scan.getStartRecordId().toString());
        }

        if (scan.getStopRecordId() != null) {
            node.put("stopRecordId", scan.getStopRecordId().toString());
        }

        if (scan.getRawStartRecordId() != null) {
            node.put("rawStartRecordId", scan.getRawStartRecordId());
        }

        if (scan.getRawStopRecordId() != null) {
            node.put("rawStopRecordId", scan.getRawStopRecordId());
        }

        if (scan.getRecordFilter() != null) {
            node.put("recordFilter", RecordFilterJsonConverters.INSTANCE.toJson(scan.getRecordFilter(),
                    namespaces, repository, RecordFilterJsonConverters.INSTANCE));
        }

        if (scan.getReturnFields() != null) {
            ObjectNode returnFieldsNode = node.putObject("returnFields");
            returnFieldsNode.put("type", scan.getReturnFields().getType().toString());
            if (scan.getReturnFields().getFields() != null) {
                ArrayNode fieldsArray = returnFieldsNode.putArray("fields");
                for (QName name : scan.getReturnFields().getFields()) {
                    fieldsArray.add(QNameConverter.toJson(name, namespaces));
                }
            }
        }

        node.put("caching", scan.getCaching());

        node.put("cacheBlocks", scan.getCacheBlocks());

        return node;
    }
}