Metacat

package edu.ucsb.nceas.metacat.dataone;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;

import java.util.concurrent.Future;

import org.apache.commons.io.FileUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dataone.client.auth.AuthTokenSession;

import org.dataone.client.exception.ClientSideException;

import org.dataone.client.v2.MNode;

import org.dataone.client.v2.impl.MultipartMNode;

import org.dataone.service.types.v1.Identifier;

import org.dataone.service.types.v2.SystemMetadata;

import org.dataone.service.util.TypeMarshaller;

import org.junit.Test;

import edu.ucsb.nceas.MCTestCase;

import edu.ucsb.nceas.metacat.properties.PropertyService;

public class MultiThreadClientIT extends MCTestCase {

    //need to configure this.

    private static String tokenPath = "/Users/tao/curl-files/token";

    protected static Log log = LogFactory.getLog(MultiThreadClientIT.class);

    private static ExecutorService executor = Executors.newFixedThreadPool(5);

    private static int number_of_objects = 30;

    private static String mnURL = null;

    static {

        try {

            mnURL = PropertyService.getProperty("test.contextUrl")+"/d1/mn";

        } catch (Exception e) {

            log.error("MultiThreadClientIT - can't find the test.contexturl from the metacat.porperties file.", e);

        }

    }

    @Test 

    public void testConstructUsingRegisteredCertificate() throws IOException, ClientSideException, Exception {

        String token = FileUtils.readFileToString(new File(tokenPath), "UTF-8");

        final AuthTokenSession tokenSession = new AuthTokenSession(token);

        final MNode node = new MultipartMNode(tokenSession.getMultipartRestClient(), mnURL, tokenSession);

        Runnable runner = new Runnable() {

            @Override

            public void run() {

                try {

                    double random = Math.random()*10000;

                    long round = Math.round(random);

                    long time = System.nanoTime();

                    String identifier ="test"+time+round;

                    Identifier pid = new Identifier();

                    pid.setValue(identifier);

                    FileInputStream sysFile = new FileInputStream("test/sysmeta-noaa.xml");

                    SystemMetadata sysmeta = TypeMarshaller.unmarshalTypeFromStream(SystemMetadata.class, sysFile);

                    sysmeta.setIdentifier(pid);

                    FileInputStream object = new FileInputStream("test/sciencemetadata-noaa.xml");

                    node.create(tokenSession, pid, object, sysmeta);

                    System.out.println("after sending -------"+identifier);

                    object.close();

                    sysFile.close();

                } catch (Exception e) {

                    log.error("Error running: " + e.getMessage(), e);

                    //throw new RuntimeException(e.getMessage(), e);

                }

            }

        };

        // submit the task, and that's it

        Future future = null;

        for(int i=0; i<number_of_objects; i++) {

            future = executor.submit(runner);

        }

        while (!future.isDone()) {

            Thread.sleep(5000);

        }

        Thread.sleep(20000);

    }

}