com.mellanox.jxio.helloworld.HelloClient.java Source code

Introduction

Here is the source code for com.mellanox.jxio.helloworld.HelloClient.java

Source

/*
 ** Copyright (C) 2013 Mellanox Technologies
 **
 ** 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:
 **
 ** 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 com.mellanox.jxio.helloworld;

import java.io.UnsupportedEncodingException;
import java.net.URI;
import java.net.URISyntaxException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import java.io.IOException;

import com.mellanox.jxio.*;

public class HelloClient {

    private final static Log LOG = LogFactory.getLog(HelloClient.class.getCanonicalName());
    private final MsgPool mp;
    private final EventQueueHandler eqh;
    private ClientSession client;
    public int exitStatus = 1;

    public static void main(String[] args) {
        if (args.length < 2) {
            usage();
            return;
        }

        final String serverhostname = args[0];
        final int port = Integer.parseInt(args[1]);

        URI uri = null;
        try {
            uri = new URI("rdma://" + serverhostname + ":" + port + "/");
        } catch (URISyntaxException e) {
            e.printStackTrace();
            return;
        }

        HelloClient client = new HelloClient();
        client.connect(uri);
        client.run();

        LOG.info("Client is releasing JXIO resources and exiting");
        client.releaseResources();
        System.exit(client.exitStatus);
    }

    HelloClient() {
        this.eqh = new EventQueueHandler(null);
        this.mp = new MsgPool(256, 100, 100);
    }

    public void connect(URI uri) {
        LOG.info("Try to establish a new session to '" + uri + "'");
        this.client = new ClientSession(eqh, uri, new MyClientCallbacks(this));

        Msg msg = this.mp.getMsg();
        try {
            msg.getOut().put("Hello Server".getBytes("UTF-8"));
        } catch (UnsupportedEncodingException e) {
            // Just suppress the exception handling in this demo code
        }
        try {
            client.sendRequest(msg);
        } catch (IOException e) {
            //all exceptions thrown extend IOException
            LOG.error(e.toString());
        }
    }

    public void run() {
        // block for JXIO incoming event
        eqh.run();
    }

    public void releaseResources() {
        mp.deleteMsgPool();
        eqh.close();
    }

    public static void usage() {
        LOG.info("Usage: ./runHelloServer.sh <SERVER_IPADDR> [<PORT>]");
    }

    class MyClientCallbacks implements ClientSession.Callbacks {
        private final HelloClient client;

        MyClientCallbacks(HelloClient client) {
            this.client = client;
        }

        public void onSessionEstablished() {
            LOG.info("[SUCCESS] Session established! Hurray !");
        }

        public void onResponse(Msg msg) {
            LOG.info("[SUCCESS] Got a message! Bring the champagne!");

            // Read reply message String
            byte ch;
            StringBuffer buffer = new StringBuffer();
            while (msg.getIn().hasRemaining() && ((ch = msg.getIn().get()) > -1)) {
                buffer.append((char) ch);
            }
            LOG.info("msg is: '" + buffer.toString() + "'");

            msg.returnToParentPool();

            LOG.info("Closing the session...");
            this.client.client.close();

            exitStatus = 0; // Success, we got our message response back
        }

        public void onSessionEvent(EventName event, EventReason reason) {
            String str = "[EVENT] Got event " + event + " because of " + reason;
            if (event == EventName.SESSION_CLOSED) { // normal exit
                LOG.info(str);
            } else {
                this.client.exitStatus = 1; // Failure on any kind of error
                LOG.error(str);
            }
            this.client.eqh.stop();
        }

        public void onMsgError(Msg msg, EventReason reason) {
            LOG.info("[ERROR] onMsgErrorCallback. reason=" + reason);
            if (reason == EventReason.MSG_FLUSHED) {
                LOG.info("[STATUS] getIsClosing() = " + this.client.client.getIsClosing());
            }
            msg.returnToParentPool();
            this.client.exitStatus = 1; // Failure on any kind of error
            System.exit(exitStatus);
        }
    }
}