From 1eba8264adbe42115caac4db655d450a99791ceb Mon Sep 17 00:00:00 2001 From: Endi Sukma Dewata Date: Thu, 7 Jun 2012 03:16:37 -0500 Subject: Added REST error handler. A new getEntity() method has been added to obtain the entity from a Response object and also map HTTP errors into exceptions. Ticket #161 --- .../netscape/cms/client/group/GroupRestClient.java | 6 +++--- .../netscape/cms/client/user/UserRestClient.java | 6 +++--- .../cms/servlet/csadmin/CMSRestClient.java | 22 ++++++++++++++++++++++ .../netscape/cms/servlet/test/DRMRestClient.java | 2 +- 4 files changed, 29 insertions(+), 7 deletions(-) diff --git a/base/common/src/com/netscape/cms/client/group/GroupRestClient.java b/base/common/src/com/netscape/cms/client/group/GroupRestClient.java index c5070a447..54b31fef6 100644 --- a/base/common/src/com/netscape/cms/client/group/GroupRestClient.java +++ b/base/common/src/com/netscape/cms/client/group/GroupRestClient.java @@ -59,13 +59,13 @@ public class GroupRestClient extends CMSRestClient { public GroupData addGroup(GroupData groupData) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)groupClient.addGroup(groupData); - return response.getEntity(); + return getEntity(response); } public GroupData modifyGroup(String groupID, GroupData groupData) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)groupClient.modifyGroup(groupID, groupData); - return response.getEntity(); + return getEntity(response); } public void removeGroup(String groupID) { @@ -83,7 +83,7 @@ public class GroupRestClient extends CMSRestClient { public GroupMemberData addGroupMember(String groupID, String memberID) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)groupMemberClient.addGroupMember(groupID, memberID); - return response.getEntity(); + return getEntity(response); } public void removeGroupMember(String groupID, String memberID) { diff --git a/base/common/src/com/netscape/cms/client/user/UserRestClient.java b/base/common/src/com/netscape/cms/client/user/UserRestClient.java index a214b261e..8bbdefe26 100644 --- a/base/common/src/com/netscape/cms/client/user/UserRestClient.java +++ b/base/common/src/com/netscape/cms/client/user/UserRestClient.java @@ -59,13 +59,13 @@ public class UserRestClient extends CMSRestClient { public UserData addUser(UserData userData) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)userClient.addUser(userData); - return response.getEntity(); + return getEntity(response); } public UserData modifyUser(String userID, UserData userData) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)userClient.modifyUser(userID, userData); - return response.getEntity(); + return getEntity(response); } public void removeUser(String userID) { @@ -83,7 +83,7 @@ public class UserRestClient extends CMSRestClient { public UserCertData addUserCert(String userID, UserCertData userCertData) { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse)userCertClient.addUserCert(userID, userCertData); - return response.getEntity(); + return getEntity(response); } public void removeUserCert(String userID, String certID) { diff --git a/base/common/src/com/netscape/cms/servlet/csadmin/CMSRestClient.java b/base/common/src/com/netscape/cms/servlet/csadmin/CMSRestClient.java index ff028660a..7f2c75007 100644 --- a/base/common/src/com/netscape/cms/servlet/csadmin/CMSRestClient.java +++ b/base/common/src/com/netscape/cms/servlet/csadmin/CMSRestClient.java @@ -17,8 +17,12 @@ import org.apache.http.conn.scheme.SchemeSocketFactory; import org.apache.http.impl.client.DefaultHttpClient; import org.apache.http.params.HttpParams; import org.jboss.resteasy.client.ClientExecutor; +import org.jboss.resteasy.client.ClientResponse; +import org.jboss.resteasy.client.ClientResponseFailure; import org.jboss.resteasy.client.ProxyFactory; +import org.jboss.resteasy.client.core.BaseClientResponse; import org.jboss.resteasy.client.core.executors.ApacheHttpClient4Executor; +import org.jboss.resteasy.client.core.extractors.ClientErrorHandler; import org.jboss.resteasy.spi.ResteasyProviderFactory; import org.mozilla.jss.ssl.SSLCertificateApprovalCallback; import org.mozilla.jss.ssl.SSLSocket; @@ -27,6 +31,7 @@ public abstract class CMSRestClient { protected String clientCertNickname; protected ResteasyProviderFactory providerFactory; + protected ClientErrorHandler errorHandler; protected ClientExecutor executor; protected URI uri; @@ -59,6 +64,7 @@ public abstract class CMSRestClient { executor = new ApacheHttpClient4Executor(httpclient); providerFactory = ResteasyProviderFactory.getInstance(); providerFactory.addClientErrorInterceptor(new CMSErrorInterceptor()); + errorHandler = new ClientErrorHandler(providerFactory.getClientErrorInterceptors()); } private class ServerCertApprovalCB implements SSLCertificateApprovalCallback { @@ -168,4 +174,20 @@ public abstract class CMSRestClient { public T createProxy(Class clazz) { return ProxyFactory.create(clazz, uri, executor, providerFactory); } + + @SuppressWarnings("unchecked") + public T getEntity(ClientResponse response) { + BaseClientResponse clientResponse = (BaseClientResponse)response; + try { + clientResponse.checkFailureStatus(); + + } catch (ClientResponseFailure e) { + errorHandler.clientErrorHandling((BaseClientResponse) e.getResponse(), e); + + } catch (RuntimeException e) { + errorHandler.clientErrorHandling(clientResponse, e); + } + + return response.getEntity(); + } } diff --git a/base/kra/functional/src/com/netscape/cms/servlet/test/DRMRestClient.java b/base/kra/functional/src/com/netscape/cms/servlet/test/DRMRestClient.java index 772612bf4..82d246e04 100644 --- a/base/kra/functional/src/com/netscape/cms/servlet/test/DRMRestClient.java +++ b/base/kra/functional/src/com/netscape/cms/servlet/test/DRMRestClient.java @@ -48,7 +48,7 @@ public class DRMRestClient extends CMSRestClient { @SuppressWarnings("unchecked") ClientResponse response = (ClientResponse) systemCertClient .getTransportCert(); - CertificateData certData = response.getEntity(); + CertificateData certData = getEntity(response); String transportCert = certData.getB64(); return transportCert; } -- cgit