@Override public Pair<THBaseService.Client, TTransport> getClient() throws IOException { TSocket sock = new TSocket(connection.getHost(), connection.getPort()); sock.setSocketTimeout(connection.getOperationTimeout()); sock.setConnectTimeout(connection.getConnectTimeout()); TTransport tTransport = sock; if (connection.isFramed()) { tTransport = new TFramedTransport(tTransport); } try { sock.open(); } catch (TTransportException e) { throw new IOException(e); } TProtocol prot; if (connection.isCompact()) { prot = new TCompactProtocol(tTransport); } else { prot = new TBinaryProtocol(tTransport); } THBaseService.Client client = new THBaseService.Client(prot); return new Pair<>(client, tTransport); }
protected void talkToThriftServer() throws Exception { TSocket sock = new TSocket(InetAddress.getLocalHost().getHostName(), port); TTransport transport = sock; if (specifyFramed || implType.isAlwaysFramed) { transport = new TFramedTransport(transport); } sock.open(); try { TProtocol prot; if (specifyCompact) { prot = new TCompactProtocol(transport); } else { prot = new TBinaryProtocol(transport); } Hbase.Client client = new Hbase.Client(prot); if (!tableCreated){ TestThriftServer.createTestTables(client); tableCreated = true; } TestThriftServer.checkTableList(client); } finally { sock.close(); } }
sock.open(); try { TProtocol tProtocol;
private void reloadClient() { transport = new TSocket(readerHost, readerPort); protocol = new TBinaryProtocol(transport); client = new LogReader.Client(protocol); try { transport.open(); } catch (TTransportException e) { } }
public void sendBatch(LogBatch batch) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); LogWriter.Client client = new LogWriter.Client(protocol); try { transport.open(); client.send_batch(batch); transport.close(); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
public void saveInsight(String indexCode, String insightCode, String jsonValue) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); FrontendManager.Client client = new FrontendManager.Client(protocol); try { transport.open(); client.save_insight(indexCode, insightCode, jsonValue); transport.close(); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void enqueueRemoveDocument(String indexId, String sourceDocId) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Storage.Client client = new Storage.Client(protocol); try { transport.open(); client.enqueueRemoveStore(indexId, sourceDocId); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void enqueueUpdateCategories(String indexId, String sourceDocId, Map<String, String> categories) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Storage.Client client = new Storage.Client(protocol); try { transport.open(); client.enqueueUpdateCategories(indexId, sourceDocId, categories); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void updateBoosts(String docId, Map<Integer, Double> updatedBoosts) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.updateBoost(docId, updatedBoosts); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void updateTimestamp(String docId, int timestampBoost) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.updateTimestampBoost(docId, timestampBoost); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
public List<IndexInfo> listIndexes() { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); FrontendManager.Client client = new FrontendManager.Client(protocol); try { transport.open(); List<IndexInfo> indexes = client.list_indexes(); transport.close(); return indexes; } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void dump() { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.dump(); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void enqueueUpdateTimestamp(String indexId, String sourceDocId, int timestamp) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Storage.Client client = new Storage.Client(protocol); try { transport.open(); client.enqueueUpdateTimestamp(indexId, sourceDocId, timestamp); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void promoteResult(String docid, String query) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.promoteResult(docid, query); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void del(String docid) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.delDoc(docid); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void removeScoreFunction(int functionIndex) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.removeScoreFunction(functionIndex); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public Map<String, String> getStats() { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); Map<String,String> retValue = client.get_stats(); transport.close(); return retValue; } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
@Override public void add(String docId, com.flaptor.indextank.index.Document document, int timestampBoost, Map<Integer, Double> dynamicBoosts) { TSocket transport = new TSocket(host, port); TProtocol protocol = new TBinaryProtocol(transport); Indexer.Client client = new Indexer.Client(protocol); try { transport.open(); client.addDoc(docId, new Document(document.asMap()), timestampBoost, dynamicBoosts); transport.close(); } catch (IndextankException e) { throw new RuntimeException(e); } catch (TTransportException e) { throw new RuntimeException(e); } catch (TException e) { throw new RuntimeException(e); } }
/** * Inits the client. * * @throws TTransportException * the t transport exception */ public static void initClient() throws TTransportException { TSocket socket = new TSocket("127.0.0.1", 9160); TTransport transport = new TFramedTransport(socket); TProtocol protocol = new TBinaryProtocol(transport, true, true); client = new Cassandra.Client(protocol); socket.open(); }
/** * Inits the client. * * @param port * @param ip * * @throws TTransportException * the t transport exception */ public static void initClient(String ip, int port) throws TTransportException { TSocket socket = new TSocket(ip, port); TTransport transport = new TFramedTransport(socket); TProtocol protocol = new TBinaryProtocol(transport, true, true); client = new Cassandra.Client(protocol); socket.open(); }