Client.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.cxf.performance.complex_type.client;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.xml.datatype.DatatypeConfigurationException;
import javax.xml.datatype.DatatypeFactory;
import javax.xml.namespace.QName;
import javax.xml.ws.Holder;


import org.apache.cxf.frontend.ClientProxy;
import org.apache.cxf.transport.http.HTTPConduit;
import org.apache.cxf.transports.http.configuration.HTTPClientPolicy;

import org.apache.cxf.pat.internal.TestCaseBase;
import org.apache.cxf.pat.internal.TestResult;
import org.apache.cxf.cxf.performance.DocPortType;
import org.apache.cxf.cxf.performance.DocPortTypeWrapped;
import org.apache.cxf.cxf.performance.RPCPortType;
import org.apache.cxf.cxf.performance.PerfService;
import org.apache.cxf.cxf.performance.types.ColourEnum;
import org.apache.cxf.cxf.performance.types.NestedComplexType;
import org.apache.cxf.cxf.performance.types.NestedComplexTypeSeq;
import org.apache.cxf.cxf.performance.types.SimpleStruct;
//import org.apache.cxf.cxf.performance.server.Server;
import org.apache.cxf.performance.complex_type.server.Server;


public final class Client extends TestCaseBase<DocPortType> {
    private static final QName SERVICE_NAME = new QName(
                                                       "http://cxf.apache.org/cxf/performance",
                                                       "PerfService");
    private static final QName PORT_NAME = new QName(
                                                    "http://cxf.apache.org/cxf/performance",
                                                    "DocPortType");
    private PerfService cs;
    private final NestedComplexTypeSeq complexTypeSeq = new NestedComplexTypeSeq();
    private int opid;

    private byte[] inputBase64;
    private String inputString = new String();

    private static int statId;
    private final int asciiCount = 1 * 1024;

    public Client(String[] args, boolean warmup) {
        super("Base TestCase", args, warmup);
        wsdlPath = PerfService.WSDL_LOCATION.toString();
        serviceName = SERVICE_NAME.getLocalPart();
        portName = PORT_NAME.getLocalPart();
        operationName = "echoComplexTypeDoc";
        wsdlNameSpace = "http://cxf.apache.org/cxf/performance";
        amount = 30;
        packetSize = 1;
        usingTime = true;
        numberOfThreads = 4;
    }

    public void processArgs() {
        super.processArgs();
        if (getOperationName().equals("echoStringDoc")) {
            opid = 0;
        } else if (getOperationName().equals("echoBase64Doc")) {
            opid = 1;
        } else if (getOperationName().equals("echoComplexTypeDoc")) {
            opid = 2;
        } else {
            System.out.println("Invalid operation: " + getOperationName());
        }
    }

    public static void main(String args[]) throws Exception {
        //workaround issue of xmlsec logging too much
        Logger.getLogger("org.apache.xml.security.signature.Reference").setLevel(Level.WARNING);

        int threadIdx = -1;
        int servIdx = -1;
        for (int x = 0; x < args.length; x++) {
            if ("-Threads".equals(args[x])) {
                threadIdx = x + 1;
            } else if ("-Server".equals(args[x])) {
                servIdx = x;
                break;
            }
        }
        if (servIdx != -1) {
            String tmp[] = new String[args.length - servIdx];
            System.arraycopy(args, servIdx, tmp, 0, args.length - servIdx);
            Server.main(tmp);

            tmp = new String[servIdx];
            System.arraycopy(args, 0, tmp, 0, servIdx);
            args = tmp;
        }
        List<String> threadList = new ArrayList<>();
        if (threadIdx != -1) {
            String threads[] = args[threadIdx].split(",");
            for (String s : threads) {
                if (s.indexOf("-") != -1) {
                    String s1 = s.substring(0, s.indexOf("-"));
                    String s2 = s.substring(s.indexOf("-") + 1);
                    int i1 = Integer.parseInt(s1);
                    int i2 = Integer.parseInt(s2);
                    for (int x = i1; x <= i2; x++) {
                        threadList.add(Integer.toString(x));
                    }
                } else {
                    threadList.add(s);
                }
            }
        } else {
            threadList.add("1");
        }
        System.out.println(threadList);
        boolean first = true;
        for (String numThreads: threadList) {
            if (threadIdx != -1) {
                args[threadIdx] = numThreads;
            }
            System.out.println(Arrays.asList(args));
            Client client = new Client(args, first);
            first = false;
            client.initialize();


            client.run();

            List results = client.getTestResults();
            TestResult testResult = null;

            double rt = 0.0;
            double tp = 0.0;
            for (Iterator iter = results.iterator(); iter.hasNext();) {
                testResult = (TestResult)iter.next();
                System.out.println("Throughput " + testResult.getThroughput());
                System.out.println("AVG Response Time " + testResult.getAvgResponseTime());
                rt += testResult.getAvgResponseTime();
                tp += testResult.getThroughput();
            }
            rt *= 1000;
            rt /= (double)results.size();

            System.out.println("Total(" + numThreads + "):  " + tp + " tps     " + rt + " ms");

            System.out.println();
            System.out.println();
            System.out.println();
        }
        System.out.println("cxf client is going to shutdown!");
        System.exit(0);
    }

    private SimpleStruct getSimpleStruct() throws DatatypeConfigurationException {
        SimpleStruct ss = new SimpleStruct();
        ss.setVarFloat(Float.MAX_VALUE);
        ss.setVarShort(Short.MAX_VALUE);
        ss.setVarByte(Byte.MAX_VALUE);
        ss.setVarDecimal(new BigDecimal("3.1415926"));
        ss.setVarDouble(Double.MAX_VALUE);
        ss.setVarString("1234567890!@#$%^&*()abcdefghijk");
        ss.setVarAttrString("1234567890!@#$%^&*()abcdefghijk");
        ss.setVarDateTime(DatatypeFactory.newInstance().newXMLGregorianCalendar(2005, 12, 3, 0, 0, 9, 0, 0));
        return ss;
    }


    private NestedComplexType createComplexType() {
        NestedComplexType complexType = new NestedComplexType();
        complexType.setVarString("#12345ABc");
        complexType.setVarUByte((short)255);
        complexType.setVarUnsignedLong(new BigInteger("13691056728"));
        complexType.setVarFloat(Float.MAX_VALUE);
        complexType.setVarQName(new QName("http://cxf.apache.org", "return"));
        try {
            complexType.setVarStruct(getSimpleStruct());
        } catch (DatatypeConfigurationException e) {
            e.printStackTrace();
        }

        complexType.setVarEnum(ColourEnum.RED);
        byte[] binary = new byte[256];
        for (int jdx = 0; jdx < 256; jdx++) {
            binary[jdx] = (byte)(jdx - 128);
        }
        complexType.setVarHexBinary(binary);
        complexType.setVarBase64Binary(binary);
        return complexType;
    }

    public void initTestData() {
        NestedComplexType ct = createComplexType();
        for (int i = 0; i < packetSize; i++) {
            complexTypeSeq.getItem().add(ct);
        }
        // init String and Binary
        String temp = "abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*()_+?><[]/0123456789";
        inputBase64 = new byte[1024];
        for (int idx = 0; idx < 4; idx++) {
            for (int jdx = 0; jdx < 256; jdx++) {
                inputBase64[idx * 256 + jdx] = (byte)(jdx - 128);
            }
        }

        StringBuilder builder = new StringBuilder(packetSize * 1024);
        builder.append(inputString);
        for (int i = 0; i < asciiCount / temp.length() * packetSize; i++) {
            builder.append(temp);
        }
        inputString = builder.toString();
    }

    public void doJob(DocPortType port) {
        try {
            switch (opid) {
            case 0:
                port.echoStringDoc(inputString);
                break;
            case 1:
                port.echoBase64Doc(inputBase64);
                break;
            case 2:
                int id = ++statId;
                Holder<Integer> i = new Holder<Integer>();
                port.echoComplexTypeDoc(complexTypeSeq, id, i);
                if (id != i.value) {
                    System.out.println(id + " != " + i.value);
                }
                break;
            default:
                port.echoComplexTypeDoc(complexTypeSeq, 0, new Holder<Integer>());
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public synchronized DocPortType getPort() {
        try {
            URL wsdl = null;
            if (wsdlPath.startsWith("file:")
                || wsdlPath.startsWith("http://")
                || wsdlPath.startsWith("https://")) {
                wsdl = new URL(wsdlPath);
            } else {
                wsdl = new URL("file://" + wsdlPath);
            }
            cs = new PerfService(wsdl, SERVICE_NAME);
        } catch (MalformedURLException e) {
            e.printStackTrace();
        }
        DocPortType port = cs.getSoapHttpDocLitPort();
        /*
        org.apache.cxf.endpoint.Client client = ClientProxy.getClient(port);
        HTTPConduit http = (HTTPConduit) client.getConduit();

        HTTPClientPolicy httpClientPolicy = new HTTPClientPolicy();
        //httpClientPolicy.setAllowChunking(false);

        http.setClient(httpClientPolicy);
        */
        return port;
    }

    public void printUsage() {
        System.out.println("Syntax is: Client [-WSDL wsdllocation] [-PacketSize packetnumber] ");
    }
}