org.apache.thrift.server.TServerEventHandler Java Examples

The following examples show how to use org.apache.thrift.server.TServerEventHandler. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source File: GfxdThriftServerSelector.java    From gemfirexd-oss with Apache License 2.0 6 votes vote down vote up 
protected ClientProcessData(GfxdTSocket socket, int connectionNumber,
    TProcessor proc, TTransport in, TTransport out, TProtocol inp,
    TProtocol outp, TServerEventHandler eventHandler) {
  this.clientSocket = socket;
  this.connectionNumber = connectionNumber;
  this.processor = proc;
  this.inputTransport = in;
  this.outputTransport = out;
  this.inputProtocol = inp;
  this.outputProtocol = outp;
  this.eventHandler = eventHandler;
  if (eventHandler != null) {
    this.connectionContext = eventHandler.createContext(inp, outp);
  }
  else {
    this.connectionContext = null;
  }
  this.idle = true;
}
Example #2
Source File: GfxdThriftServerSelector.java    From gemfirexd-oss with Apache License 2.0 6 votes vote down vote up 
protected ClientProcessData(GfxdTSocket socket, int connectionNumber,
    TProcessor proc, TTransport in, TTransport out, TProtocol inp,
    TProtocol outp, TServerEventHandler eventHandler) {
  this.clientSocket = socket;
  this.connectionNumber = connectionNumber;
  this.processor = proc;
  this.inputTransport = in;
  this.outputTransport = out;
  this.inputProtocol = inp;
  this.outputProtocol = outp;
  this.eventHandler = eventHandler;
  if (eventHandler != null) {
    this.connectionContext = eventHandler.createContext(inp, outp);
  }
  else {
    this.connectionContext = null;
  }
  this.idle = true;
}
Example #3
Source File: ConfigurationTest.java    From ikasoa with MIT License 5 votes vote down vote up 
@Test
public void testThriftServerConfiguration() {
	ThriftServerConfiguration thriftServerConfiguration = new ThriftServerConfiguration();
	// 是否有默认值
	assertNotNull(thriftServerConfiguration.getTransportFactory());
	assertNotNull(thriftServerConfiguration.getProtocolFactory());
	// 测试事件处理器配置
	TServerEventHandler testServerEventHandler = new TestServerEventHandler();
	thriftServerConfiguration.setServerEventHandler(testServerEventHandler);
	assertEquals(thriftServerConfiguration.getServerEventHandler(), testServerEventHandler);
	// 测试线程池配置
	ExecutorService executorService = Executors.newSingleThreadExecutor();
	thriftServerConfiguration.setExecutorService(executorService);
	assertEquals(thriftServerConfiguration.getExecutorService(), executorService);
}
Example #4
Source File: SentryService.java    From incubator-sentry with Apache License 2.0 5 votes vote down vote up 
/**
 * Add Thrift event handler to underlying thrift threadpool server
 * @param eventHandler
 */
public void setThriftEventHandler(TServerEventHandler eventHandler) throws IllegalStateException {
  if (thriftServer == null) {
    throw new IllegalStateException("Server is not initialized or stopped");
  }
  thriftServer.setServerEventHandler(eventHandler);
}
Example #5
Source File: CatalogThriftService.java    From metacat with Apache License 2.0 4 votes vote down vote up 
/**
 * {@inheritDoc}
 */
@Override
public TServerEventHandler getServerEventHandler() {
    return new CatalogThriftEventHandler();
}
Example #6
Source File: GfxdThriftServerThreadPool.java    From gemfirexd-oss with Apache License 2.0 4 votes vote down vote up 
/**
 * Loops on processing a client forever
 */
@Override
public void run() {
  TProcessor processor = null;
  TTransport inputTransport = null;
  TTransport outputTransport = null;
  TProtocol inputProtocol = null;
  TProtocol outputProtocol = null;

  final TServerEventHandler eventHandler = getEventHandler();
  ServerContext connectionContext = null;

  final ConnectionListener listener = connListener;
  final TTransport client = this.client;
  Socket clientSocket = null;

  try {
    processor = processorFactory_.getProcessor(client);
    inputTransport = inputTransportFactory_.getTransport(client);
    outputTransport = outputTransportFactory_.getTransport(client);
    inputProtocol = inputProtocolFactory_.getProtocol(inputTransport);
    outputProtocol = outputProtocolFactory_.getProtocol(outputTransport);

    if (eventHandler != null) {
      connectionContext = eventHandler.createContext(inputProtocol,
          outputProtocol);
    }
    // register with ConnectionListener
    if (listener != null) {
      if (client instanceof GfxdTSocket) {
        clientSocket = ((GfxdTSocket)client).getSocket();
      }
      else if (client instanceof TSocket) {
        clientSocket = ((TSocket)client).getSocket();
      }
      listener.connectionOpened(clientSocket, this.connectionNumber);
    }
    // we check stopped_ first to make sure we're not supposed to be
    // shutting down. this is necessary for graceful shutdown.
    while (true) {

      if (eventHandler != null) {
        eventHandler.processContext(connectionContext, inputTransport,
            outputTransport);
      }

      if (stopped || !processor.process(inputProtocol, outputProtocol)) {
        break;
      }
    }
  } catch (TTransportException tte) {
    // Assume the client died and continue silently
  } catch (TException te) {
    LOGGER.error("Thrift error occurred during processing of message.", te);
  } catch (Exception e) {
    LOGGER.error("Error occurred during processing of message.", e);
  }

  if (eventHandler != null) {
    eventHandler.deleteContext(connectionContext, inputProtocol,
        outputProtocol);
  }

  if (inputTransport != null) {
    inputTransport.close();
  }

  if (outputTransport != null) {
    outputTransport.close();
  }

  // deregister with ConnectionListener
  if (listener != null) {
    listener.connectionClosed(clientSocket, this.connectionNumber);
  }
}
Example #7
Source File: ServiceProviderFactoryBean.java    From ourea with Apache License 2.0 4 votes vote down vote up 
private void buildServerConfig(Properties properties) {

        ConfigUtils.assertEmpty("port", properties.getProperty("port"));
        serverConfig = new ThriftServerConfig(Integer.parseInt(properties.getProperty("port")));

        if (StringUtils.isNoneEmpty(properties.getProperty("group"))) {
            serverConfig.setGroup(properties.getProperty("group"));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("version"))) {
            serverConfig.setVersion(properties.getProperty("version"));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("MinWorkerThreads"))) {
            serverConfig.setMinWorkerThreads(Integer.parseInt(properties.getProperty("MinWorkerThreads")));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("MaxWorkerThreads"))) {
            serverConfig.setMaxWorkerThreads(Integer.parseInt(properties.getProperty("MaxWorkerThreads")));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("weight"))) {
            serverConfig.setWeight(Integer.parseInt(properties.getProperty("weight")));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("status"))) {
            serverConfig.setStatus(Boolean.parseBoolean(properties.getProperty("status")));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("directInvoke"))) {
            serverConfig.setDirectInvoke(Boolean.parseBoolean(properties.getProperty("directInvoke")));
        }
        if (StringUtils.isNoneEmpty(properties.getProperty("daemonRun"))) {
            serverConfig.setDaemonRun(Boolean.parseBoolean(properties.getProperty("daemonRun")));
        }

        if (StringUtils.isNoneEmpty(properties.getProperty("serverEventHandler"))) {

            try {
                Class<TServerEventHandler> handlerClass = (Class<TServerEventHandler>) Class
                        .forName(properties.getProperty("serverEventHandler"));
                serverConfig.setServerEventHandler(handlerClass.newInstance());
            } catch (Exception e) {
                LOGGER.warn(
                        "config full loadBalanceStrategy class not find.use default RoundRobinLoadBalanceStrategy.");
            }
        }
    }
Example #8
Source File: ThriftServerConfig.java    From ourea with Apache License 2.0 4 votes vote down vote up 
public TServerEventHandler getServerEventHandler() {
    return serverEventHandler;
}
Example #9
Source File: ThriftServerConfig.java    From ourea with Apache License 2.0 4 votes vote down vote up 
public void setServerEventHandler(TServerEventHandler serverEventHandler) {
    this.serverEventHandler = serverEventHandler;
}
Example #10
Source File: GfxdThriftServerThreadPool.java    From gemfirexd-oss with Apache License 2.0 4 votes vote down vote up 
/**
 * Loops on processing a client forever
 */
@Override
public void run() {
  TProcessor processor = null;
  TTransport inputTransport = null;
  TTransport outputTransport = null;
  TProtocol inputProtocol = null;
  TProtocol outputProtocol = null;

  final TServerEventHandler eventHandler = getEventHandler();
  ServerContext connectionContext = null;

  final ConnectionListener listener = connListener;
  final TTransport client = this.client;
  Socket clientSocket = null;

  try {
    processor = processorFactory_.getProcessor(client);
    inputTransport = inputTransportFactory_.getTransport(client);
    outputTransport = outputTransportFactory_.getTransport(client);
    inputProtocol = inputProtocolFactory_.getProtocol(inputTransport);
    outputProtocol = outputProtocolFactory_.getProtocol(outputTransport);

    if (eventHandler != null) {
      connectionContext = eventHandler.createContext(inputProtocol,
          outputProtocol);
    }
    // register with ConnectionListener
    if (listener != null) {
      if (client instanceof GfxdTSocket) {
        clientSocket = ((GfxdTSocket)client).getSocket();
      }
      else if (client instanceof TSocket) {
        clientSocket = ((TSocket)client).getSocket();
      }
      listener.connectionOpened(clientSocket, this.connectionNumber);
    }
    // we check stopped_ first to make sure we're not supposed to be
    // shutting down. this is necessary for graceful shutdown.
    while (true) {

      if (eventHandler != null) {
        eventHandler.processContext(connectionContext, inputTransport,
            outputTransport);
      }

      if (stopped || !processor.process(inputProtocol, outputProtocol)) {
        break;
      }
    }
  } catch (TTransportException tte) {
    // Assume the client died and continue silently
  } catch (TException te) {
    LOGGER.error("Thrift error occurred during processing of message.", te);
  } catch (Exception e) {
    LOGGER.error("Error occurred during processing of message.", e);
  }

  if (eventHandler != null) {
    eventHandler.deleteContext(connectionContext, inputProtocol,
        outputProtocol);
  }

  if (inputTransport != null) {
    inputTransport.close();
  }

  if (outputTransport != null) {
    outputTransport.close();
  }

  // deregister with ConnectionListener
  if (listener != null) {
    listener.connectionClosed(clientSocket, this.connectionNumber);
  }
}
Example #11
Source File: SentryService.java    From incubator-sentry with Apache License 2.0 4 votes vote down vote up 
public TServerEventHandler getThriftEventHandler() throws IllegalStateException {
  if (thriftServer == null) {
    throw new IllegalStateException("Server is not initialized or stopped");
  }
  return thriftServer.getEventHandler();
}
Example #12
Source File: AbstractThriftServer.java    From metacat with Apache License 2.0 2 votes vote down vote up 
/**
 * Returns the server event handler.
 *
 * @return server event handler
 */
public abstract TServerEventHandler getServerEventHandler();