org.apache.curator.x.discovery.details.InstanceSerializer Java Examples

private void init() throws IOException {
  try {
    haContext.startCuratorFramework();
    InstanceSerializer<Void> instanceSerializer = new FixedJsonInstanceSerializer<Void>(Void.class);
    serviceDiscovery = ServiceDiscoveryBuilder.<Void>builder(Void.class)
              .basePath(HAContext.SENTRY_SERVICE_REGISTER_NAMESPACE)
              .serializer(instanceSerializer)
        .client(haContext.getCuratorFramework())
              .build();
    serviceDiscovery.start();
    serviceProvider = serviceDiscovery
            .serviceProviderBuilder()
            .serviceName(HAContext.SENTRY_SERVICE_REGISTER_NAMESPACE)
            .build();
    serviceProvider.start();
  } catch (Exception e) {
    throw new IOException(e);
  }
}
 

public void regService(String host, int port) throws Exception {

    haContext.startCuratorFramework();
    ServiceInstance<Void> serviceInstance = ServiceInstance.<Void>builder()
        .address(host)
        .port(port)
        .name(HAContext.SENTRY_SERVICE_REGISTER_NAMESPACE)
        .build();

    InstanceSerializer<Void> instanceSerializer = new FixedJsonInstanceSerializer<Void>(Void.class);
    ServiceDiscoveryBuilder.builder(Void.class)
        .basePath(HAContext.SENTRY_SERVICE_REGISTER_NAMESPACE)
        .client(haContext.getCuratorFramework())
        .serializer(instanceSerializer)
        .thisInstance(serviceInstance)
        .build()
        .start();
  }
 

@Bean
@ConditionalOnMissingBean(ServiceDiscoveryCustomizer.class)
public DefaultServiceDiscoveryCustomizer defaultServiceDiscoveryCustomizer(
		CuratorFramework curator, ZookeeperDiscoveryProperties properties,
		InstanceSerializer<ZookeeperInstance> serializer) {
	return new DefaultServiceDiscoveryCustomizer(curator, properties, serializer);
}
 

public DefaultServiceDiscoveryCustomizer(CuratorFramework curator,
		ZookeeperDiscoveryProperties properties,
		InstanceSerializer<ZookeeperInstance> instanceSerializer) {
	this.curator = curator;
	this.properties = properties;
	this.instanceSerializer = instanceSerializer;
}
 

public ZooKeeperAutoRegistrationCustomizer(CuratorFramework curator, ZookeeperDiscoveryProperties properties, InstanceSerializer<ZookeeperInstance> serializer) {
    super(curator, properties, serializer);
}
 

@Bean
public ZooKeeperAutoRegistrationCustomizer zooKeeperAutoRegistrationCustomizer(CuratorFramework curator,
                                                                               ZookeeperDiscoveryProperties properties,
                                                                               InstanceSerializer<ZookeeperInstance> serializer) {
    return new ZooKeeperAutoRegistrationCustomizer(curator, properties, serializer);
}
 

public ZooKeeperAutoRegistrationCustomizer(CuratorFramework curator, ZookeeperDiscoveryProperties properties, InstanceSerializer<ZookeeperInstance> serializer) {
    super(curator, properties, serializer);
}
 

@Bean
public ZooKeeperAutoRegistrationCustomizer zooKeeperAutoRegistrationCustomizer(CuratorFramework curator,
                                                                               ZookeeperDiscoveryProperties properties,
                                                                               InstanceSerializer<ZookeeperInstance> serializer) {
    return new ZooKeeperAutoRegistrationCustomizer(curator, properties, serializer);
}
 

/**
 * 这个bean启动后会独占线程，导致其他的bean无法执行。所以必须保证这个bean在最后才能够执行。
 * @return
 * @throws Exception
 */
@Bean(initMethod = "start", destroyMethod = "stop")
public ServerRunner serverRunner()
		throws Exception {
	String ip = this.ip;
	if (ip == null)
		ip = new IpPortResolver().getIpV4Address();

	String instanceId = this.ip + ":" + this.port;
	
	CuratorFramework curatorFramework =CuratorFrameworkFactory.builder()
			.connectString(this.connectString)
			.sessionTimeoutMs(this.sessionTimeoutMs)
			.connectionTimeoutMs(this.connectionTimeoutMs)
			.retryPolicy(this.retryPolicy())
			.aclProvider(this.aclProvider()).authorization(this.authInfo())
			.build();
	InstanceSerializer<RpcPayload> serializer = new JsonSerializer();

	TServerTransport transport = new TServerSocket(this.port);

	TThreadPoolServer.Args args = new TThreadPoolServer.Args(transport);
	args.transportFactory(new TTransportFactory());
	args.protocolFactory(new TBinaryProtocol.Factory());

	TProcessor processor= new TProtobufProcessor();		
	args.processor(processor);
	
	args.executorService(new ThreadPoolExecutor(this.minTheads,
			this.maxTheads, this.keepAliveTime, TimeUnit.SECONDS,
			new SynchronousQueue<Runnable>()));

	TServer server = new TThreadPoolServer(args);

	ServiceInstanceBuilder<RpcPayload> instanceBuilder = ServiceInstance
			.builder();
	instanceBuilder.name(this.serviceName)
			.uriSpec(new UriSpec(this.uriSpec)).payload(this.payload())
			.port(port).id(instanceId).address(ip);

	ServiceDiscoveryBuilder<RpcPayload> discoveryBuilder = ServiceDiscoveryBuilder
			.builder(RpcPayload.class);
	discoveryBuilder.client(curatorFramework).basePath(zkBasePath)
			.serializer(serializer).thisInstance(instanceBuilder.build())
			.build();
	return ServerRunner
			.newBuilder()
			.server(server)
			.curatorFramework(curatorFramework)
			.serviceDiscovery(discoveryBuilder.build())
			.zookeeperDeferRegisterPeriod(this.zookeeperDeferRegisterPeriod)
			.zookeeperUnregisterPeriod(this.zookeeperUnregisterPeriod).build();
}
 

public void setSerializer(InstanceSerializer<RpcPayload> serializer) {
	this.serializer = serializer;
}
 

public InstanceSerializer<RpcPayload> serializer() {
	return new JsonSerializer();
}
 

public <T> InstanceSerializer<T> getInstanceSerializer(
    TypeReference<ServiceInstance<T>> typeReference)
{
  return new JacksonInstanceSerializer<T>(objectReader, objectWriter, typeReference);
}
 

@Test
public void testSerialization() throws Exception
{
  ZKAssistedDiscovery discovery = new ZKAssistedDiscovery();
  discovery.setServiceName("ApexFlumeTest");
  discovery.setConnectionString("localhost:2181");
  discovery.setBasePath("/HelloApex");
  discovery.setup(null);
  ServiceInstance<byte[]> instance = discovery.getInstance(new Service<byte[]>()
  {
    @Override
    public String getHost()
    {
      return "localhost";
    }

    @Override
    public int getPort()
    {
      return 8080;
    }

    @Override
    public byte[] getPayload()
    {
      return null;
    }

    @Override
    public String getId()
    {
      return "localhost8080";
    }

  });
  InstanceSerializer<byte[]> instanceSerializer =
      discovery.getInstanceSerializerFactory().getInstanceSerializer(new TypeReference<ServiceInstance<byte[]>>()
      {
      });
  byte[] serialize = instanceSerializer.serialize(instance);
  logger.debug("serialized json = {}", new String(serialize));
  ServiceInstance<byte[]> deserialize = instanceSerializer.deserialize(serialize);
  assertArrayEquals("Metadata", instance.getPayload(), deserialize.getPayload());
}
 

@Bean
@ConditionalOnMissingBean
public InstanceSerializer<ZookeeperInstance> instanceSerializer() {
	return new JsonInstanceSerializer<>(ZookeeperInstance.class);
}
 

@Bean
@ConditionalOnMissingBean
public InstanceSerializer<ZookeeperInstance> deprecatedInstanceSerializer() {
	return new JsonInstanceSerializer<>(ZookeeperInstance.class);
}