/**
* Open a socket for the given connection, null if failed.
*
* @param connections connection array of size 1
*/
private Socket[] makeSock(ConnectionFactory.DiskConnection[] connections) {
if (connections.length != 1) {
throw new IllegalArgumentException(
"expected only one " + "connection, but array has lenght " + connections.length);
}
Socket[] sockets = new Socket[1];
try {
sockets[0] = ConnectionFactory.connect(connections[0]);
} catch (IOException ioe) {
// failed to connect to cache, wierd because this is local
if (sockets[0] != null) {
try {
sockets[0].close();
} catch (IOException ignored) {
}
}
return null;
}
return sockets;
}
/**
* **************************************
*
* <p>Query APIs
*
* <p>**************************************
*/
private MetadataClient.QueryResult queryPlus(
int[] mapIds,
String cacheId,
String query,
ArrayList attributes,
Cookie _cookie,
int maxResults,
int timeout,
boolean forceResults,
boolean abortOnFailure,
Object[] boundParameters,
MDOutputStream outputStream)
throws EMDException {
if (LOG.isLoggable(Level.FINE)) {
LOG.fine("EMDClient queryPlus called [" + query + "]");
}
// Sanity checks
if ((_cookie != null) && (maxResults == -1)) {
// Has to specify an offset AND a row count
throw new EMDException(
"Invalid argument : when "
+ "using cookies, you have to "
+ "specify the number of "
+ "entries to return");
}
ConnectionFactory.DiskConnection[] connections =
ConnectionFactory.getConnections(mapIds, cacheId);
if (LOG.isLoggable(Level.FINE)) {
StringBuffer buffer = new StringBuffer();
buffer.append("The query [" + query + "] is sent to : ");
for (int i = 0; i < connections.length; i++) {
connections[i].toString(buffer);
buffer.append(" <> ");
}
LOG.fine(buffer.toString());
}
Socket[] sockets = new Socket[connections.length];
ObjectBroker[] brokers = new ObjectBroker[connections.length];
StreamHead[] heads = new StreamHead[connections.length];
try {
EMDCookie cookie = (EMDCookie) _cookie;
NewObjectIdentifier oid = null;
int toBeSkipped = 0;
if (cookie != null) {
query = cookie.getQuery();
oid = cookie.getLastOid();
toBeSkipped = cookie.getToBeSkipped();
boundParameters = cookie.getBoundParameters();
}
// Construct the brokers and launch the queries
for (int i = 0; i < sockets.length; i++) {
sockets[i] = null;
// Connect to the node
try {
sockets[i] = ConnectionFactory.connect(connections[i]);
} catch (IOException e) {
LOG.warning(
"Failed to connect to node "
+ connections[i].getNodeAddress()
+ ". Skipping this node ["
+ e.getMessage()
+ "]");
String str = BundleAccess.getInstance().getBundle().getString("warn.emd.query.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_WARNING, MessageFormat.format(str, args));
try {
if (sockets[i] != null) {
sockets[i].close();
}
} catch (IOException ignored) {
}
sockets[i] = null;
throw new EMDException(
"Failed to connect to node "
+ connections[i].getNodeAddress()
+ ". Skipping this node.",
e);
}
// Launch the query
if (sockets[i] == null) {
brokers[i] = null;
heads[i] = null;
} else {
brokers[i] = new ObjectBroker(sockets[i]);
brokers[i].launchQueryClient(
cacheId,
connections[i].getDisks(),
query,
attributes,
cookie,
maxResults + toBeSkipped,
timeout,
forceResults,
boundParameters);
heads[i] = new StreamHead(brokers[i]);
}
}
// Merge the result and compute the output
ArrayList array = null;
MDHit lastHit = null;
if (outputStream == null) {
array = StreamHead.mergeStreams(heads, toBeSkipped, maxResults, abortOnFailure);
if (array != null && array.size() != 0) lastHit = (MDHit) array.get(array.size() - 1);
} else {
StreamHead.mergeStreams(heads, outputStream, toBeSkipped, maxResults);
lastHit = (MDHit) outputStream.getLastObject();
}
MetadataClient.QueryResult result = new MetadataClient.QueryResult();
if (lastHit != null) {
long atime = -1;
if (lastHit instanceof MDHitByATime) atime = ((MDHitByATime) lastHit).getATime();
result.cookie =
new EMDCookie(lastHit.constructOid(), query, 0, boundParameters, attributes, atime);
result.results = array;
} else {
result.cookie = null;
result.results = new ArrayList();
}
return (result);
} catch (IOException e) {
LOG.log(Level.SEVERE, "Failed to run the distributed query", e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.query.io");
Object[] args = {query};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
return (null);
} finally {
// Close the connections
for (int i = 0; i < sockets.length; i++) {
if (sockets[i] != null) {
try {
sockets[i].close();
sockets[i] = null;
} catch (IOException ignored) {
}
}
}
}
}
private int setMetadata(
String cacheId,
ConnectionFactory.DiskConnection[] connections,
NewObjectIdentifier oid,
Object argument) {
Socket[] sockets = new Socket[connections.length];
ObjectBroker[] brokers = new ObjectBroker[connections.length];
int inserts = 0;
for (int i = 0; i < connections.length; i++) {
try {
sockets[i] = ConnectionFactory.connect(connections[i]);
brokers[i] = new ObjectBroker(sockets[i]);
if (LOG.isLoggable(Level.FINE)) {
StringBuffer log = new StringBuffer();
log.append("RPC for setMetadata to ");
connections[i].toString(log);
LOG.fine(log.toString());
}
brokers[i].launchSetMetadata(cacheId, oid, argument, connections[i].getDisks());
} catch (ConnectException e) {
LOG.warning(
"Failed to update the metadata on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ cacheId
+ "]. Error is ["
+ e.getMessage()
+ "]");
String str = BundleAccess.getInstance().getBundle().getString("err.emd.setMetadata.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
} catch (IOException e) {
LOG.log(
Level.SEVERE,
"Failed to update the metadata on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ cacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.setMetadata.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
} catch (EMDException e) {
LOG.log(
Level.SEVERE,
"Failed to update the metadata on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ cacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.setMetadata.cache");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
}
}
for (int i = 0; i < connections.length; i++) {
try {
if (brokers[i] != null) {
brokers[i].waitForCompletion();
inserts++;
}
} catch (EMDException e) {
LOG.log(
Level.SEVERE,
"The setMetadata operation didn't return properly from ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ cacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.setMetadata.cache");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
} finally {
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
}
}
}
if (LOG.isLoggable(Level.FINE)) {
LOG.fine("Metadata insert succeeded with " + inserts + " caches successfully updated.");
}
return inserts;
}
/**
* **************************************
*
* <p>selectUnique APIs
*
* <p>**************************************
*/
public MetadataClient.SelectUniqueResult selectUnique(
String cacheId,
String query,
String attribute,
Cookie _cookie,
int maxResults,
int timeout,
boolean forceResults,
Object[] boundParameters,
MDOutputStream outputStream)
throws EMDException {
// Sanity checks
if ((_cookie != null) && (maxResults == -1)) {
// Has to specify an offset AND a row count
throw new EMDException(
"Invalid argument : when "
+ "using cookies, you have to "
+ "specify the number of "
+ "entries to return");
}
ConnectionFactory.DiskConnection[] connections = ConnectionFactory.getConnections();
Socket socket = null;
ArrayList sockets = new ArrayList();
ObjectBroker[] brokers = null;
StreamHead[] heads = null;
try {
EMDCookie cookie = (EMDCookie) _cookie;
String lastAttribute = null;
int toBeSkipped = 0;
if (cookie != null) {
query = cookie.getQuery();
attribute = cookie.getAttribute();
lastAttribute = cookie.getLastAttribute();
toBeSkipped = cookie.getToBeSkipped();
boundParameters = cookie.getBoundParameters();
}
// Construct the brokers and launch the queries
for (int i = 0; i < connections.length; i++) {
try {
socket = ConnectionFactory.connect(connections[i]);
sockets.add(socket);
} catch (IOException e) {
LOG.warning(
"Failed to connect to node "
+ connections[i].getNodeAddress()
+ ". Skipping this node ["
+ e.getMessage()
+ "]");
String str = BundleAccess.getInstance().getBundle().getString("warn.emd.selectUnique.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_WARNING, MessageFormat.format(str, args));
throw new EMDException(
"Failed to connect to node "
+ connections[i].getNodeAddress()
+ ". Skipping this node.",
e);
}
}
brokers = new ObjectBroker[sockets.size()];
heads = new StreamHead[sockets.size()];
Iterator iter = sockets.iterator();
for (int i = 0; i < brokers.length; i++) {
brokers[i] = new ObjectBroker((Socket) iter.next());
brokers[i].launchSelectUniqueClient(
cacheId,
query,
attribute,
lastAttribute,
maxResults + toBeSkipped,
timeout,
forceResults,
boundParameters);
heads[i] = new StreamHead(brokers[i]);
}
// Merge the result and compute the output
ArrayList array = null;
Object lastHit = null;
if (outputStream == null) {
array = StreamHead.mergeStreams(heads, toBeSkipped, maxResults, false);
if (array != null && array.size() != 0) lastHit = array.get(array.size() - 1);
} else {
StreamHead.mergeStreams(heads, outputStream, toBeSkipped, maxResults);
lastHit = outputStream.getLastObject();
}
MetadataClient.SelectUniqueResult result = new MetadataClient.SelectUniqueResult();
if (lastHit != null) {
if (array != null) {
ArrayList stringList = new ArrayList();
for (int i = 0; i < array.size(); i++) {
stringList.add(array.get(i).toString());
}
result.results = new StringList(stringList);
}
result.cookie = new EMDCookie(lastHit.toString(), query, attribute, 0, boundParameters);
} else {
result.cookie = null;
result.results = StringList.EMPTY_LIST;
}
return (result);
} catch (IOException e) {
EMDException newe =
new EMDException("Failed to run the distributed select unique [" + e.getMessage() + "]");
newe.initCause(e);
throw newe;
} finally {
// Close the connections
Iterator iter = sockets.iterator();
while (iter.hasNext()) {
try {
((Socket) iter.next()).close();
} catch (IOException ignored) {
}
}
}
}
private void removeLegalHold(
NewObjectIdentifier oid, String legalHold, ConnectionFactory.DiskConnection[] connections) {
if (LOG.isLoggable(Level.FINE)) {
LOG.fine("removeLegalHold has been called for oid " + oid);
}
String sysCacheId = CacheClientInterface.SYSTEM_CACHE;
// Precompute the connection
Socket socket = null;
for (int i = 0; i < connections.length; i++) {
try {
socket = ConnectionFactory.connect(connections[i]);
ObjectBroker broker = new ObjectBroker(socket);
broker.launchRemoveLegalHold(sysCacheId, oid, legalHold, connections[i].getDisks());
} catch (ConnectException e) {
LOG.warning(
"Couldn't remove the "
+ sysCacheId
+ " legal hold for ["
+ oid
+ "] from ["
+ connections[i].getNodeAddress()
+ "]. Error is ["
+ e.getMessage()
+ "]");
String str = BundleAccess.getInstance().getBundle().getString("err.emd.removeLegalHold.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
} catch (IOException e) {
LOG.log(
Level.SEVERE,
"Couldn't remove the "
+ sysCacheId
+ " legal hold for ["
+ oid
+ "] from ["
+ connections[i].getNodeAddress()
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.removeMetadata.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
} catch (EMDException e) {
LOG.log(
Level.SEVERE,
"Couldn't remove the "
+ sysCacheId
+ " legal hold for ["
+ oid
+ "] from ["
+ connections[i].getNodeAddress()
+ "]",
e);
String str =
BundleAccess.getInstance().getBundle().getString("err.emd.removeMetadata.cache");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
} finally {
if (socket != null) {
try {
socket.close();
} catch (IOException ignored) {
}
socket = null;
}
}
}
}
/** Private addLegalHold * */
private void addLegalHold(
ConnectionFactory.DiskConnection[] connections, NewObjectIdentifier oid, String legalHold) {
if (LOG.isLoggable(Level.FINE)) {
LOG.fine("Ading (" + oid + ", " + legalHold + ")");
}
if (connections == null) {
LOG.severe("connections is null");
return;
}
String sysCacheId = CacheClientInterface.SYSTEM_CACHE;
Socket[] sockets = new Socket[connections.length];
ObjectBroker[] brokers = new ObjectBroker[connections.length];
for (int i = 0; i < connections.length; i++) {
try {
sockets[i] = ConnectionFactory.connect(connections[i]);
brokers[i] = new ObjectBroker(sockets[i]);
if (LOG.isLoggable(Level.FINE)) {
StringBuffer log = new StringBuffer();
log.append("RPC for addLegalHold to ");
connections[i].toString(log);
LOG.fine(log.toString());
}
brokers[i].launchAddLegalHold(sysCacheId, oid, legalHold, connections[i].getDisks());
} catch (ConnectException e) {
LOG.warning(
"Failed to update the legal hold on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ sysCacheId
+ "]. Error is ["
+ e.getMessage()
+ "]");
String str = BundleAccess.getInstance().getBundle().getString("err.emd.addLegalHold.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
} catch (IOException e) {
LOG.log(
Level.SEVERE,
"Failed to add a legal hold on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ sysCacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.addLegalHold.io");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
} catch (EMDException e) {
LOG.log(
Level.SEVERE,
"Failed to update the metadata on ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ sysCacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.addLegalHold.cache");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
sockets[i] = null;
}
brokers[i] = null;
}
}
for (int i = 0; i < connections.length; i++) {
try {
if (brokers[i] != null) {
brokers[i].waitForCompletion();
}
} catch (EMDException e) {
LOG.log(
Level.SEVERE,
"The addLegalHold operation didn't return "
+ "properly from ["
+ connections[i]
+ "] for oid "
+ oid
+ " [cache "
+ sysCacheId
+ "]",
e);
String str = BundleAccess.getInstance().getBundle().getString("err.emd.addLegalHold.cache");
Object[] args = {connections[i].toString()};
LOG.log(ExtLevel.EXT_SEVERE, MessageFormat.format(str, args));
} finally {
if (sockets[i] != null) {
try {
sockets[i].close();
} catch (IOException ignored) {
}
}
}
}
}