org.hyperledger.fabric.sdk.Endpoint Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fabric-java-sdk Show documentation
This is a fork of Hyperledger Fabric fabric-sdk-java
There is a newer version: 1.1
/*
 *  Copyright 2016 DTCC, Fujitsu Australia Software Technology - All Rights Reserved.
 *
 *  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 org.hyperledger.fabric.sdk;

import java.util.Properties;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import javax.net.ssl.SSLException;

import io.grpc.ManagedChannelBuilder;
import io.grpc.netty.GrpcSslContexts;
import io.grpc.netty.NettyChannelBuilder;
import io.netty.handler.ssl.SslContext;
import io.netty.util.internal.StringUtil;

public class Endpoint {

	private String addr;
	private int port;
	private ManagedChannelBuilder channelBuilder = null;

	public Endpoint(String url, String pem) {
		Properties purl = parseUrl(url);
	    String protocol = purl.getProperty("protocol");
	    this.addr = purl.getProperty("host");
	    this.port = Integer.parseInt(purl.getProperty("port"));

	    if (protocol.equalsIgnoreCase("grpc")) {//
	        this.channelBuilder = ManagedChannelBuilder.forAddress(addr, port)
	        		.usePlaintext(true);
	    } else if (protocol.equalsIgnoreCase("grpcs")) {
	    	if (StringUtil.isNullOrEmpty(pem)) {
	    		// use root certificate
	    		this.channelBuilder = ManagedChannelBuilder.forAddress(addr, port);
	    	} else {
	    		try {
		    		SslContext sslContext = GrpcSslContexts.forClient().trustManager(new java.io.File(pem)).build();
			    	this.channelBuilder = NettyChannelBuilder.forAddress(addr, port)
			    			.sslContext(sslContext);
	    		} catch(SSLException sslex) {
	    			throw new RuntimeException(sslex);
	    		}
	    	}
	    } else {
	        throw new RuntimeException("invalid protocol: " + protocol);
	    }
	}

	public ManagedChannelBuilder getChannelBuilder() {
		return this.channelBuilder;
	}

	public String getHost() {
		return this.addr;
	}

	public int getPort() {
		return this.port;
	}
	private Properties parseUrl(String url) {
		if (StringUtil.isNullOrEmpty(url)) {
			throw new RuntimeException("URL cannot be null or empty");
		}

		Properties props = new Properties();
		Pattern p = Pattern.compile("([^:]+)[:]//([^:]+)[:]([0-9]+)", Pattern.CASE_INSENSITIVE);
		Matcher m = p.matcher(url);
		if (m.matches()) {
			props.setProperty("protocol", m.group(1));
			props.setProperty("host", m.group(2));
			props.setProperty("port", m.group(3));
		} else {
			throw new RuntimeException("URL must be of the format protocol://host:port");
		}

		// TODO: allow all possible formats of the URL
		return props;
	}
}