io.snappydata.thrift.common.ThriftExceptionUtil Maven / Gradle / Ivy
/*
* Copyright (c) 2010-2015 Pivotal Software, 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. See accompanying
* LICENSE file.
*/
/*
* Changes for SnappyData data platform.
*
* Portions Copyright (c) 2018 SnappyData, 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. See accompanying
* LICENSE file.
*/
package io.snappydata.thrift.common;
import java.rmi.ServerException;
import java.sql.BatchUpdateException;
import java.sql.ClientInfoStatus;
import java.sql.SQLClientInfoException;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import com.pivotal.gemfirexd.internal.shared.common.error.ClientExceptionUtil;
import com.pivotal.gemfirexd.internal.shared.common.error.ExceptionSeverity;
import com.pivotal.gemfirexd.internal.shared.common.reference.SQLState;
import io.snappydata.thrift.SnappyException;
import io.snappydata.thrift.SnappyExceptionData;
import org.apache.thrift.TApplicationException;
import org.apache.thrift.protocol.TProtocolException;
import org.apache.thrift.transport.TTransportException;
/**
* This class provides utility routines for exceptions from Thrift layer.
*/
public abstract class ThriftExceptionUtil extends ClientExceptionUtil {
protected ThriftExceptionUtil() {
// no instance allowed
}
public static SQLException newSQLException(SnappyException se) {
SnappyExceptionData payload = se.getExceptionData();
List nextList = se.getNextExceptions();
// if SQLState is null for top-level exception then it could be for
// a different kind of exception like XAException, so skip
if (payload.getSqlState() == null || payload.getSqlState().isEmpty()) {
boolean foundValidPayload = false;
// search for a SQLException with non-null state in nextList
if (nextList != null) {
Iterator iter = nextList.iterator();
while (iter.hasNext()) {
SnappyExceptionData nextData = iter.next();
if (nextData.getSqlState() != null &&
nextData.getSqlState().length() > 0) {
payload = nextData;
iter.remove();
foundValidPayload = true;
break;
}
}
}
if (!foundValidPayload) {
// no valid exception state from server, so return a default one
payload.setSqlState(SQLState.DATA_UNEXPECTED_EXCEPTION.substring(0, 5));
payload.setErrorCode(ExceptionSeverity.STATEMENT_SEVERITY);
}
}
SQLException sqle = newSQLException(payload, se.getCause(),
se.getServerInfo());
// since SnappyException is always a wrapper, no need to record the stack
sqle.setStackTrace(se.getStackTrace());
// build next exceptions
SQLException current = sqle, next;
if (nextList != null) {
for (SnappyExceptionData nextData : nextList) {
// check for server stack indicator
if (SQLState.SNAPPY_SERVER_STACK_INDICATOR.equals(
nextData.getSqlState())) {
Throwable cause = sqle;
while (cause.getCause() != null) {
cause = cause.getCause();
}
try {
cause.initCause(new ServerException(nextData.getReason()));
continue;
} catch (IllegalStateException ignored) {
// continue to default
}
}
next = newSQLException(nextData, null, null);
current.setNextException(next);
current = next;
}
}
return sqle;
}
public static SQLException newSQLException(SnappyExceptionData payload,
Throwable cause, String serverInfo) {
String message = payload.getReason();
if (serverInfo != null) {
message = "(" + serverInfo + ") " + message;
}
List updateCounts = payload.getUpdateCounts();
if (updateCounts != null) {
final int numUpdates = updateCounts.size();
int[] updates = new int[numUpdates];
for (int i = 0; i < numUpdates; i++) {
updates[i] = updateCounts.get(i);
}
return new BatchUpdateException(message, payload.getSqlState(),
payload.getErrorCode(), updates, cause);
} else {
return factory.getSQLException(message, payload.getSqlState(),
payload.getErrorCode(), null, cause);
}
}
public static SQLClientInfoException newSQLClientInfoException(
String sqlState, Map failedProperties,
Throwable t, Object... args) {
return new SQLClientInfoException(getMessageUtil().getCompleteMessage(
sqlState, args), getSQLStateFromIdentifier(sqlState),
getSeverityFromIdentifier(sqlState), failedProperties, t);
}
public static SQLWarning newSQLWarning(SnappyExceptionData payload,
Throwable cause) {
return new SQLWarning(payload.getReason(), payload.getSqlState(),
payload.getErrorCode(), cause);
}
public static SnappyException newSnappyException(String sqlState, Throwable t,
String serverInfo, Object... args) {
SnappyExceptionData payload = new SnappyExceptionData(getMessageUtil()
.getCompleteMessage(sqlState, args),
getSeverityFromIdentifier(sqlState))
.setSqlState(getSQLStateFromIdentifier(sqlState));
SnappyException se = new SnappyException(payload, serverInfo);
if (t != null) {
if (t instanceof SnappyException) {
SnappyException next = (SnappyException)t;
se.addToNextExceptions(next.getExceptionData());
if (next.getNextExceptions() != null) {
se.getNextExceptions().addAll(next.getNextExceptions());
}
}
se.initCause(t);
}
return se;
}
public static String getExceptionString(Throwable t) {
String typeMessage = null;
if (t instanceof TTransportException) {
TTransportException tte = (TTransportException)t;
switch (tte.getType()) {
case TTransportException.ALREADY_OPEN:
typeMessage = "SOCKET ALREADY OPEN";
break;
case TTransportException.END_OF_FILE:
typeMessage = "SOCKET END OF TRANSMISSION";
break;
case TTransportException.NOT_OPEN:
typeMessage = "SOCKET NOT OPEN";
break;
case TTransportException.TIMED_OUT:
typeMessage = "SOCKET TIMED OUT";
break;
default:
typeMessage = "UNKNOWN SOCKET ERROR";
break;
}
} else if (t instanceof TProtocolException) {
TProtocolException tpe = (TProtocolException)t;
switch (tpe.getType()) {
case TProtocolException.BAD_VERSION:
typeMessage = "BAD PROTOCOL VERSION";
break;
case TProtocolException.INVALID_DATA:
typeMessage = "INVALID DATA IN PROTOCOL";
break;
case TProtocolException.NEGATIVE_SIZE:
typeMessage = "NEGATIVE SIZE IN PROTOCOL";
break;
case TProtocolException.NOT_IMPLEMENTED:
typeMessage = "NOT IMPLEMENTED";
break;
case TProtocolException.SIZE_LIMIT:
typeMessage = "PROTOCOL SIZE LIMIT HIT";
break;
default:
typeMessage = "UNKNOWN PROTOCOL ERROR";
break;
}
} else if (t instanceof TApplicationException) {
TApplicationException tae = (TApplicationException)t;
switch (tae.getType()) {
case TApplicationException.BAD_SEQUENCE_ID:
typeMessage = "BAD SEQUENCE ID";
break;
case TApplicationException.INTERNAL_ERROR:
typeMessage = "INTERNAL ERROR";
break;
case TApplicationException.INVALID_MESSAGE_TYPE:
typeMessage = "INVALID MESSAGE TYPE";
break;
case TApplicationException.MISSING_RESULT:
typeMessage = "MISSING RESULT";
break;
case TApplicationException.PROTOCOL_ERROR:
typeMessage = "PROTOCOL ERROR";
break;
case TApplicationException.UNKNOWN_METHOD:
typeMessage = "UNKNOWN METHOD";
break;
case TApplicationException.WRONG_METHOD_NAME:
typeMessage = "WRONG METHOD NAME";
break;
default:
typeMessage = "UNKNOWN APPLICATION ERROR";
break;
}
}
String message = t.getLocalizedMessage();
if (message == null || message.length() == 0) {
message = t.getClass().getName();
}
if (typeMessage != null) {
return message + " [" + typeMessage + ']';
} else {
return message;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy