0.10/tools/pdi_plugin/src/org/apache/oodt/filemgrget/OODTProcesses.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.oodt.filemgrget;

import com.google.gson.Gson;
import org.apache.oodt.cas.filemgr.datatransfer.DataTransfer;
import org.apache.oodt.cas.filemgr.datatransfer.LocalDataTransferFactory;
import org.apache.oodt.cas.filemgr.structs.Product;
import org.apache.oodt.cas.filemgr.structs.ProductPage;
import org.apache.oodt.cas.filemgr.structs.ProductType;
import org.apache.oodt.cas.filemgr.structs.exceptions.CatalogException;
import org.apache.oodt.cas.filemgr.structs.exceptions.DataTransferException;
import org.apache.oodt.cas.filemgr.system.XmlRpcFileManagerClient;
import org.apache.oodt.cas.metadata.Metadata;

import java.io.File;
import java.io.IOException;
import java.util.*;

/**
 * Copyright 2014 OSBI Ltd
 * <p/>
 * 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
 * <p/>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p/>
 * 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.
 */
public class OODTProcesses {


    public Map<String, Map<String, String>> getAllProducts(OODTConfig config, String productTypeName) throws Exception {
        XmlRpcFileManagerClient client = config.getXMLRpcClient();
        ProductType type = client.getProductTypeByName(productTypeName);
        if (type == null) {
            throw new Exception("FileManager returned null ProductType");
        }
        ProductPage firstPage = client.getFirstPage(type);
        if (firstPage == null) {
            throw new Exception("FileManager returned null product page");
        }

        Map<String, Map<String, String>> o = new HashMap<String, Map<String, String>>();

        for (int pid = 0; pid < firstPage.getTotalPages(); pid++) {
            if (pid > 0) {
                firstPage = client.getNextPage(type, firstPage);
            }
            for (Product p : firstPage.getPageProducts()) {
                Metadata met = client.getMetadata(p);

                Map<String, String> h = new HashMap<String, String>();
                h.put("name", p.getProductName());
                h.put("type", p.getProductType().getName());
                h.put("structure", p.getProductStructure());
                h.put("transferstatus", p.getTransferStatus());
                Gson g = new Gson();
                String json = g.toJson(met.getHashtable());
                h.put("metadata", json);
                o.put(p.getProductId(), h);

            }

        }
        return o;

    }


    public String getProductByID(OODTConfig config, String id) throws CatalogException, IOException, DataTransferException {
        Product product = config.getXMLRpcClient().getProductById(id);
        product.setProductReferences(config.getXMLRpcClient().getProductReferences(product));
        LocalDataTransferFactory ldtf = new LocalDataTransferFactory();
        DataTransfer dt = ldtf.createDataTransfer();
        String rand = UUID.randomUUID().toString();
        File theDir = new File("/tmp/oodt/" + rand);
        boolean mkdir = false;
        if (!theDir.exists()) {
            mkdir = theDir.mkdir();
        }
        if (mkdir) {
            dt.retrieveProduct(product, new File("/tmp/oodt/" + rand));
            return "/tmp/oodt/" + rand + "/" + product.getProductName();
        } else {
            return null;
        }
    }

    public String getProductByName(OODTConfig config, String name) throws CatalogException, IOException, DataTransferException {
        Product product = config.getXMLRpcClient().getProductByName(name);
        product.setProductReferences(config.getXMLRpcClient().getProductReferences(product));
        LocalDataTransferFactory ldtf = new LocalDataTransferFactory();
        DataTransfer dt = ldtf.createDataTransfer();
        String rand = UUID.randomUUID().toString();
        File theDir = new File("/tmp/oodt/" + rand);
        boolean mkdir = false;
        if (!theDir.exists()) {
            mkdir = theDir.mkdir();
        }
        if (mkdir) {
            dt.retrieveProduct(product, new File("/tmp/oodt/" + rand));
            return "/tmp/oodt/" + rand + "/" + product.getProductName();
        } else {
            return null;
        }

    }

}