com.redhat.poc.jdg.bankofchina.function.TestCase413RemoteMultiThreads.java Source code

Introduction

Here is the source code for com.redhat.poc.jdg.bankofchina.function.TestCase413RemoteMultiThreads.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.redhat.poc.jdg.bankofchina.function;

import com.opencsv.CSVReader;
import com.redhat.poc.jdg.bankofchina.model.UserBaseInfo;
import java.io.FileReader;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.Options;
import org.infinispan.client.hotrod.RemoteCache;
import org.infinispan.client.hotrod.RemoteCacheManager;
import org.infinispan.client.hotrod.configuration.ConfigurationBuilder;
import org.infinispan.client.hotrod.impl.transport.tcp.RoundRobinBalancingStrategy;
import org.infinispan.commons.api.BasicCache;

/**
 * ? (client-server mode, 2JDG,??)
 * ?/opt/jbshome/data/userinfo/csv
 * PutCache
 *
 * @author maping
 */
public class TestCase413RemoteMultiThreads extends Thread {

    private static final String CSV_FILE_PATH = "/opt/jbshome/data/userinfo/";
    private static final String JDG_SERVERS = "jdg.servers";
    private static final String JDG_HOST = "jdg.host";
    private static final String HOTROD_PORT = "jdg.hotrod.port";
    private static final String PROPERTIES_FILE = "case413.jdg.properties";
    private static final String CACHE_NAME = "case413_cache";
    private static int csvFileStart = 591;
    private static int csvFileEnd = 594;
    private int fileName;

    public TestCase413RemoteMultiThreads(int fileName) {
        this.fileName = fileName;
    }

    public static void main(String[] args) throws Exception {
        CommandLine commandLine;
        Options options = new Options();
        options.addOption("s", true, "The start csv file number option");
        options.addOption("e", true, "The end csv file number option");
        BasicParser parser = new BasicParser();
        parser.parse(options, args);
        commandLine = parser.parse(options, args);
        if (commandLine.getOptions().length > 0) {
            if (commandLine.hasOption("s")) {
                String start = commandLine.getOptionValue("s");
                if (start != null && start.length() > 0) {
                    csvFileStart = Integer.parseInt(start);
                }
            }
            if (commandLine.hasOption("e")) {
                String end = commandLine.getOptionValue("e");
                if (end != null && end.length() > 0) {
                    csvFileEnd = Integer.parseInt(end);
                }
            }
        }

        System.out.println(
                "%%%%%%%%%  csv ?, ?, ?(ms),"
                        + new Date().getTime());
        for (int i = csvFileStart; i <= csvFileEnd; i++) {
            new TestCase413RemoteMultiThreads(i).start();
        }
    }

    @Override
    public void run() {
        ConfigurationBuilder builder = new ConfigurationBuilder();
        builder.addServers(jdgProperty(JDG_SERVERS));
        builder.pingOnStartup(true);
        builder.balancingStrategy(RoundRobinBalancingStrategy.class);
        RemoteCacheManager cacheManager = new RemoteCacheManager(builder.build());
        final RemoteCache<String, UserBaseInfo> userinfoCache = cacheManager.getCache(CACHE_NAME);
        LoadUserBaseInfo(userinfoCache, fileName);
    }

    private static void LoadUserBaseInfo(BasicCache cache, int fileNumber) {
        try {
            //System.out.println("####### " + Thread.currentThread().getName() + " :   " + fileNumber + ".csv ?. ");
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
            CSVReader reader = new CSVReader(new FileReader(CSV_FILE_PATH + fileNumber + ".csv"));
            String[] nextLine;
            int count = 0;
            while ((nextLine = reader.readNext()) != null) {
                if (count == 0) {
                } else {
                    // nextLine[] is an array of values from the line
                    String userId = nextLine[0];
                    String imsi = nextLine[1];
                    String msisdn = nextLine[2];
                    String homeCity = nextLine[3];
                    String homeCountry = nextLine[4];
                    Date inureTime = sdf.parse(nextLine[5]);
                    Date expireTime = sdf.parse(nextLine[6]);
                    String testFlag = nextLine[7];
                    cache.put(userId, new UserBaseInfo(userId, imsi, msisdn, homeCity, homeCountry, inureTime,
                            expireTime, testFlag));
                }
                count++;
            }
            System.out.println("#######, " + Thread.currentThread().getName() + ", : ?  " + fileNumber
                    + ".csv ?,  " + count + "?, ?(ms), "
                    + new Date().getTime());

        } catch (Exception ex) {
            Logger.getLogger(TestCase413RemoteMultiThreads.class.getName()).log(Level.SEVERE, null, ex);
        }
    }

    public static String jdgProperty(String name) {
        Properties props = new Properties();
        try {
            props.load(TestCase413RemoteMultiThreads.class.getClassLoader().getResourceAsStream(PROPERTIES_FILE));
        } catch (IOException ioe) {
            throw new RuntimeException(ioe);
        }
        return props.getProperty(name);
    }
}