com.paremus.dosgi.net.client.ClientRequestSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of com.paremus.dosgi.net Show documentation

An implementation of OSGi Remote Service Admin, enhanced to provide support for OSGi Promises and streaming using OSGi PushStreams.

The newest version!

/*-
 * #%L
 * com.paremus.dosgi.net
 * %%
 * Copyright (C) 2016 - 2019 Paremus Ltd
 * %%
 * Licensed under the Fair Source License, Version 0.9 (the "License");
 * 
 * See the NOTICE.txt file distributed with this work for additional 
 * information regarding copyright ownership. You may not use this file 
 * except in compliance with the License. For usage restrictions see the 
 * LICENSE.txt file distributed with this work
 * #L%
 */
package com.paremus.dosgi.net.client;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.paremus.dosgi.net.message.AbstractRSAMessage;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelOutboundHandlerAdapter;
import io.netty.channel.ChannelPromise;

public class ClientRequestSerializer extends ChannelOutboundHandlerAdapter {

	private static final Logger LOG = LoggerFactory.getLogger(ClientRequestSerializer.class);
	
	private final ClientResponseHandler responseHandler;
	
	public ClientRequestSerializer(ClientResponseHandler responseHandler) {
		this.responseHandler = responseHandler;
	}

	@Override
	public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
		
		@SuppressWarnings("unchecked")
		AbstractRSAMessage invocation = (AbstractRSAMessage) msg;
		
		ClientMessageType callType = invocation.getType();
		
		try {
			/* See Protocol_V1 and Protocol_V2 for header structure */
			ByteBuf buffer = ctx.alloc().ioBuffer();
			invocation.write(buffer, promise);
			
			switch(callType.getAction()) {
			
				case ADD :
					responseHandler.registerInvocation((AbstractClientInvocationWithResult) invocation);
					promise.addListener(f -> {
							if (!f.isSuccess()) {
								responseHandler.unregisterInvocation(invocation.getKey());
							}
						});
					break;
				case REMOVE :
					responseHandler.unregisterInvocation(invocation.getKey());
					break;
				case SKIP :
					break;
				default :
					throw new IllegalArgumentException("An unknown action type " + callType.getAction().name()
						+ " was made on service " + invocation.getServiceId());
			}
			ctx.write(buffer, promise);
		} catch (Exception e) {
			LOG.error("An error occurred when invoking service {} ", invocation.getServiceId(), e);
			if(!promise.isVoid()) {
				promise.tryFailure(e);
			}
		}
	}
}