/*
* Copyright (c) 2008-2017, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.client.spi;
import com.hazelcast.client.impl.HazelcastClientInstanceImpl;
import com.hazelcast.client.impl.protocol.ClientMessage;
import com.hazelcast.client.impl.protocol.codec.ClientDestroyProxyCodec;
import com.hazelcast.client.spi.impl.ClientInvocation;
import com.hazelcast.client.spi.impl.ListenerMessageCodec;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.partition.strategy.StringPartitioningStrategy;
import com.hazelcast.spi.serialization.SerializationService;
import com.hazelcast.util.ExceptionUtil;
import java.util.concurrent.Future;
import static com.hazelcast.util.ExceptionUtil.rethrow;
/**
* Base class for client proxies.
*
* Allows the client to proxy operations through member nodes.
*/
public abstract class ClientProxy implements DistributedObject {
protected final String name;
private final String serviceName;
private final ClientContext context;
private final SerializationService serializationService;
// these fields are for the legacy constructor without ClientContext
private volatile ClientContext lazyContext;
private volatile SerializationService lazySerializationService;
/**
* @deprecated since 3.9, please use {@link #ClientProxy(String, String, ClientContext)}
*/
@Deprecated
protected ClientProxy(String serviceName, String name) {
this(serviceName, name, null);
}
protected ClientProxy(String serviceName, String name, ClientContext context) {
this.serviceName = serviceName;
this.name = name;
this.context = context;
this.serializationService = context == null ? null : context.getSerializationService();
}
protected final String registerListener(ListenerMessageCodec codec, EventHandler handler) {
return getContext().getListenerService().registerListener(codec, handler);
}
protected final boolean deregisterListener(String registrationId) {
return getContext().getListenerService().deregisterListener(registrationId);
}
// public for testing
public final ClientContext getContext() {
return context != null ? context : lazyContext;
}
protected final ClientProxy setContext(ClientContext context) {
if (this.context != null) {
throw new IllegalStateException("The context has already been initialized!");
}
this.lazyContext = context;
this.lazySerializationService = context.getSerializationService();
return this;
}
protected SerializationService getSerializationService() {
return serializationService != null ? serializationService : lazySerializationService;
}
protected Data toData(Object o) {
return getSerializationService().toData(o);
}
protected <T> T toObject(Object data) {
return getSerializationService().toObject(data);
}
protected final HazelcastClientInstanceImpl getClient() {
return (HazelcastClientInstanceImpl) getContext().getHazelcastInstance();
}
@Deprecated
public final Object getId() {
return name;
}
@Override
public final String getName() {
return name;
}
@Override
public String getPartitionKey() {
return StringPartitioningStrategy.getPartitionKey(getDistributedObjectName());
}
@Override
public final String getServiceName() {
return serviceName;
}
protected String getDistributedObjectName() {
return name;
}
@Override
public final void destroy() {
if (preDestroy()) {
onDestroy();
ClientMessage clientMessage = ClientDestroyProxyCodec.encodeRequest(getDistributedObjectName(), getServiceName());
getContext().removeProxy(this);
try {
new ClientInvocation(getClient(), clientMessage).invoke().get();
postDestroy();
} catch (Exception e) {
throw rethrow(e);
}
}
}
/**
* Called when proxy is created.
* Overriding implementations can add initialization specific logic into this method
* like registering a listener, creating a cleanup task etc.
*/
protected void onInitialize() {
}
/**
* Called before proxy is destroyed.
* Overriding implementations should clean/release resources created during initialization.
*/
protected void onDestroy() {
}
/**
* Called before proxy is destroyed and determines whether destroy should be done.
*
* @return <code>true</code> if destroy should be done, otherwise <code>false</code>
*/
protected boolean preDestroy() {
return true;
}
/**
* Called after proxy is destroyed.
*/
protected void postDestroy() {
}
/**
* Called before client shutdown.
* Overriding implementations can add shutdown specific logic here.
*/
protected void onShutdown() {
}
protected <T> T invoke(ClientMessage clientMessage, Object key) {
final int partitionId = getContext().getPartitionService().getPartitionId(key);
return invokeOnPartition(clientMessage, partitionId);
}
protected <T> T invokeOnPartition(ClientMessage clientMessage, int partitionId) {
try {
final Future future = new ClientInvocation(getClient(), clientMessage, partitionId).invoke();
return (T) future.get();
} catch (Exception e) {
throw rethrow(e);
}
}
protected <T> T invokeOnPartitionInterruptibly(ClientMessage clientMessage, int partitionId) throws InterruptedException {
try {
final Future future = new ClientInvocation(getClient(), clientMessage, partitionId).invoke();
return (T) future.get();
} catch (Exception e) {
throw ExceptionUtil.rethrowAllowInterrupted(e);
}
}
protected <T> T invoke(ClientMessage clientMessage) {
try {
final Future future = new ClientInvocation(getClient(), clientMessage).invoke();
return (T) future.get();
} catch (Exception e) {
throw rethrow(e);
}
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
ClientProxy that = (ClientProxy) o;
if (!getInstanceName().equals(that.getInstanceName())) {
return false;
}
if (!getDistributedObjectName().equals(that.getDistributedObjectName())) {
return false;
}
if (!serviceName.equals(that.serviceName)) {
return false;
}
return true;
}
@Override
public int hashCode() {
int result = getInstanceName().hashCode();
result = 31 * result + serviceName.hashCode();
result = 31 * result + getDistributedObjectName().hashCode();
return result;
}
private String getInstanceName() {
ClientContext context = getContext();
return context != null ? context.getHazelcastInstance().getName() : "";
}
}