com.paremus.dosgi.net.server.ServerResponseSerializer 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.server;

import static com.paremus.dosgi.net.server.ServerMessageType.FAILURE_SERIALIZATION_ERROR;
import static com.paremus.dosgi.net.server.ServerMessageType.RETURN_SERIALIZATION_ERROR;
import static com.paremus.dosgi.net.server.ServerMessageType.UNKNOWN_ERROR;

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

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

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

@Sharable
public class ServerResponseSerializer extends ChannelOutboundHandlerAdapter {
	
	private static final Logger LOG = LoggerFactory.getLogger(ServerResponseSerializer.class);
	
	@Override
	public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
		
		@SuppressWarnings("unchecked")
		AbstractRSAMessage response = (AbstractRSAMessage) msg;
		try {
			/* See Protocol_V1 and Protocol_V2 for header structure */
			ByteBuf buf = ctx.alloc().ioBuffer();
			try {
				response.write(buf, promise);
			} catch (Exception e) {
				buf.clear();
				getErrorResponse(response, e).write(buf, promise);
			}
		
			ctx.writeAndFlush(buf, promise);
		} catch (Exception e) {
			LOG.error("An error occurred when invoking service {} ", response.getServiceId(), e);
		}
	}

	private AbstractRSAMessage getErrorResponse(
			AbstractRSAMessage response, Exception e) {
		AbstractRSAMessage toReturn;
		switch(response.getType()) {
			case SUCCESS:
				toReturn = new ServerErrorMessageResponse(RETURN_SERIALIZATION_ERROR, 
						response.getServiceId(), response.getCallId(), e.getMessage());
				break;
			case FAILURE:
				toReturn = new ServerErrorMessageResponse(FAILURE_SERIALIZATION_ERROR, 
						response.getServiceId(), response.getCallId(), e.getMessage());
				break;
			default:
				toReturn = new ServerErrorMessageResponse(UNKNOWN_ERROR, 
						response.getServiceId(), response.getCallId(), e.getMessage());
		}
		return toReturn;
	}
}