com.hazelcast.client.spi.ClientProxy Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2015, 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.client.BaseClientAddListenerRequest;
import com.hazelcast.client.impl.client.BaseClientRemoveListenerRequest;
import com.hazelcast.client.impl.client.ClientDestroyRequest;
import com.hazelcast.client.impl.client.ClientRequest;
import com.hazelcast.client.spi.impl.ClientInvocation;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.nio.Address;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.partition.strategy.StringPartitioningStrategy;
import com.hazelcast.util.ExceptionUtil;
import java.util.concurrent.Future;
public abstract class ClientProxy implements DistributedObject {
private final String serviceName;
private final String objectName;
private volatile ClientContext context;
protected ClientProxy(String serviceName, String objectName) {
this.serviceName = serviceName;
this.objectName = objectName;
}
protected final String registerListener(BaseClientAddListenerRequest addRequest,
BaseClientRemoveListenerRequest removeRequest, EventHandler handler) {
return context.getListenerService().registerListener(addRequest, removeRequest, handler);
}
protected final boolean deregisterListener(String registrationId) {
return context.getListenerService().deregisterListener(registrationId);
}
protected final ClientContext getContext() {
return context;
}
protected final HazelcastClientInstanceImpl getClient() {
return (HazelcastClientInstanceImpl) context.getHazelcastInstance();
}
protected final void setContext(ClientContext context) {
this.context = context;
}
@Deprecated
public final Object getId() {
return objectName;
}
@Override
public final String getName() {
return objectName;
}
@Override
public String getPartitionKey() {
return StringPartitioningStrategy.getPartitionKey(getName());
}
@Override
public final String getServiceName() {
return serviceName;
}
@Override
public final void destroy() {
onDestroy();
ClientDestroyRequest request = new ClientDestroyRequest(objectName, getServiceName());
context.removeProxy(this);
try {
new ClientInvocation(getClient(), request).invoke().get();
} catch (Exception e) {
throw ExceptionUtil.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 client shutdown.
* Overriding implementations can add shutdown specific logic here.
*/
protected void onShutdown() {
}
protected T invoke(ClientRequest req, Object key) {
try {
final int partitionId = context.getPartitionService().getPartitionId(key);
final Future future = new ClientInvocation(getClient(), req, partitionId).invoke();
Object result = future.get();
return toObject(result);
} catch (Exception e) {
throw ExceptionUtil.rethrow(e);
}
}
protected T invokeOnPartition(ClientRequest req, int partitionId) {
try {
final Future future = new ClientInvocation(getClient(), req, partitionId).invoke();
Object result = future.get();
return toObject(result);
} catch (Exception e) {
throw ExceptionUtil.rethrow(e);
}
}
protected T invokeInterruptibly(ClientRequest req, Object key) throws InterruptedException {
try {
final int partitionId = context.getPartitionService().getPartitionId(key);
final Future future = new ClientInvocation(getClient(), req, partitionId).invoke();
Object result = future.get();
return toObject(result);
} catch (Exception e) {
throw ExceptionUtil.rethrowAllowInterrupted(e);
}
}
protected T invoke(ClientRequest req) {
try {
final Future future = new ClientInvocation(getClient(), req).invoke();
Object result = future.get();
return toObject(result);
} catch (Exception e) {
throw ExceptionUtil.rethrow(e);
}
}
protected T invoke(ClientRequest req, Address address) {
try {
final Future future = new ClientInvocation(getClient(), req, address).invoke();
Object result = future.get();
return toObject(result);
} catch (Exception e) {
throw ExceptionUtil.rethrow(e);
}
}
protected Data toData(Object o) {
return getContext().getSerializationService().toData(o);
}
protected T toObject(Object data) {
return getContext().getSerializationService().toObject(data);
}
protected void throwExceptionIfNull(Object o) {
if (o == null) {
throw new NullPointerException("Object is null");
}
}
private String getInstanceName() {
ClientContext ctx = context;
if (ctx != null) {
HazelcastInstance instance = ctx.getHazelcastInstance();
return instance.getName();
}
return "";
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
ClientProxy that = (ClientProxy) o;
String instanceName = getInstanceName();
if (!instanceName.equals(that.getInstanceName())) {
return false;
}
if (!objectName.equals(that.objectName)) {
return false;
}
if (!serviceName.equals(that.serviceName)) {
return false;
}
return true;
}
@Override
public int hashCode() {
String instanceName = getInstanceName();
int result = instanceName.hashCode();
result = 31 * result + serviceName.hashCode();
result = 31 * result + objectName.hashCode();
return result;
}
}