com.redhat.poc.jdg.bankofchina.performance.TestCase532RemoteMultiThreads.java Source code

Introduction

Here is the source code for com.redhat.poc.jdg.bankofchina.performance.TestCase532RemoteMultiThreads.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.performance;

import com.opencsv.CSVReader;
import com.redhat.poc.jdg.bankofchina.model.UserBaseInfo;
import java.io.FileReader;
import java.io.IOException;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Properties;
import java.util.Random;
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;

/**
 * ? (client-server mode, 2JDG,??)
 *
 * @author maping
 */
public class TestCase532RemoteMultiThreads 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 = "case412.jdg.properties";
    private static final String CACHE_NAME = "case412_cache";
    private static DecimalFormat df = new DecimalFormat(",###");
    private static int writeThreadNumbers = 50;
    private static int threadWriteTimes = 40000;
    private static List<String> userIdList = new ArrayList<String>();
    private static int randomPoolSize;
    private static int randomPoolSizeByThread;
    private int randomStartIndexByThread;

    public TestCase532RemoteMultiThreads(int randomStartIndex) {
        this.randomStartIndexByThread = randomStartIndex;
    }

    public static void main(String[] args) throws Exception {
        CommandLine commandLine;
        Options options = new Options();
        options.addOption("n", true, "The read thread numbers option");
        options.addOption("t", true, "The thread read times option");
        BasicParser parser = new BasicParser();
        parser.parse(options, args);
        commandLine = parser.parse(options, args);
        if (commandLine.getOptions().length > 0) {
            if (commandLine.hasOption("n")) {
                String numbers = commandLine.getOptionValue("n");
                if (numbers != null && numbers.length() > 0) {
                    writeThreadNumbers = Integer.parseInt(numbers);
                }
            }
            if (commandLine.hasOption("t")) {
                String times = commandLine.getOptionValue("t");
                if (times != null && times.length() > 0) {
                    threadWriteTimes = Integer.parseInt(times);
                }
            }
        }
        System.out.println();
        System.out.println("%%%%%%%%%  userid.csv  %%%%%%%%%");
        LoadUserIdList();
        randomPoolSize = userIdList.size();
        System.out.println(
                "%%%%%%%%% ? userid.csv ,  " + randomPoolSize + " ?? %%%%%%%%%");
        randomPoolSizeByThread = randomPoolSize / writeThreadNumbers;

        System.out.println();
        System.out.println(
                "%%%%%%%%% ?, " + writeThreadNumbers + ", , ??, "
                        + threadWriteTimes + ", ?, ?(ms)," + new Date().getTime());

        for (int i = 0; i < writeThreadNumbers; i++) {
            new TestCase532RemoteMultiThreads(i * randomPoolSizeByThread).start();
        }
    }

    @Override
    public void run() {
        RandomWrite(randomStartIndexByThread);
        //System.out.println("%%%%%%%%%  " + Thread.currentThread().getName() + " ?: " + randomStartIndexByThread);
    }

    private static void RandomWrite(int randomStart) {
        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);

        for (int i = 0; i < threadWriteTimes; i++) {
            // random write
            int rand = new Random().nextInt(randomPoolSizeByThread) + randomStart;
            String randomUserId = userIdList.get(rand);
            userinfoCache.put(randomUserId, new UserBaseInfo(randomUserId));
            //            System.out.println("#######  " + Thread.currentThread().getName() + " ? " + (rand + 1) + " ??, "
            //                    + " : \n" + userinfoCache.get(randomUserId));
        }
        System.out.println("%%%%%%%%%  " + Thread.currentThread().getName() + " ?(ms), "
                + new Date().getTime() + ", %%%%%%%%% ");
    }

    private static void LoadUserIdList() throws Exception {
        CSVReader reader = new CSVReader(new FileReader(CSV_FILE_PATH + "userid.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];
                userIdList.add(userId);
            }
            count++;
        }
    }

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