package ch.usi.da.smr.transport;
/*
* Copyright (c) 2013 Università della Svizzera italiana (USI)
*
* This file is part of URingPaxos.
*
* URingPaxos is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* URingPaxos is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with URingPaxos. If not, see <http://www.gnu.org/licenses/>.
*/
import java.nio.ByteBuffer;
import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TFramedTransport;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException;
import ch.usi.da.paxos.message.Control;
import ch.usi.da.smr.message.Message;
import ch.usi.da.smr.thrift.gen.PaxosProposerService;
import ch.usi.da.smr.thrift.gen.Value;
/**
* Name: ThriftABSender<br>
* Description: <br>
*
* Creation date: Mar 12, 2013<br>
* $Id$
*
* @author Samuel Benz benz@geoid.ch
*/
public class ThriftABSender implements ABSender {
private final TTransport transport;
private final TProtocol protocol;
private final PaxosProposerService.Client proposer;
public ThriftABSender(String host, int port) throws TTransportException {
transport = new TFramedTransport(new TSocket(host,port));
protocol = new TBinaryProtocol(transport);
proposer = new PaxosProposerService.Client(protocol);
transport.open();
}
@Override
public long abroadcast(Message m){
Value value = null;
if(m.isSetControl()){
Control c = m.getControl();
ch.usi.da.smr.thrift.gen.ControlType type = null;
switch(c.getType()){
case Prepare:
type = ch.usi.da.smr.thrift.gen.ControlType.PREPARE; break;
case Subscribe:
type = ch.usi.da.smr.thrift.gen.ControlType.SUBSCRIBE; break;
case Unsubscribe:
type = ch.usi.da.smr.thrift.gen.ControlType.UNSUBSCRIBE; break;
}
ch.usi.da.smr.thrift.gen.Control control = new ch.usi.da.smr.thrift.gen.Control(c.getID(),type,c.getGroupID(),c.getRingID());
value = new Value();
value.setControl(control);
}else{
value = new Value(ByteBuffer.wrap(Message.toByteArray(m)));
}
try {
//long start = System.nanoTime();
long ret = 1; //proposer.propose(value);
proposer.nb_propose(value);
//System.err.println(System.nanoTime() - start);
return ret;
} catch (TException e) {
e.printStackTrace();
return -1;
}
}
public PaxosProposerService.Client getProposer(){
return proposer;
}
@Override
public void close(){
transport.close();
}
}