X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Ffts%2Fservice%2Fthreedbeacons%2FTDBeaconsFTSRestClient.java;h=08edcdf5e485025395e2a2aa46bbc7dbc131d2c7;hb=dfa42fba9774fe4659d234661c1e0d2c17980505;hp=e9564615252928532bf4b35fc2817b07c2acc2f6;hpb=bc4b17d7c6b2e7df9262371e644abcdfb8ac3e91;p=jalview.git diff --git a/src/jalview/fts/service/threedbeacons/TDBeaconsFTSRestClient.java b/src/jalview/fts/service/threedbeacons/TDBeaconsFTSRestClient.java index e956461..08edcdf 100644 --- a/src/jalview/fts/service/threedbeacons/TDBeaconsFTSRestClient.java +++ b/src/jalview/fts/service/threedbeacons/TDBeaconsFTSRestClient.java @@ -21,25 +21,34 @@ import jalview.datamodel.SequenceI; import jalview.fts.api.FTSData; import jalview.fts.api.FTSDataColumnI; import jalview.fts.api.FTSRestClientI; +import jalview.fts.api.StructureFTSRestClientI; import jalview.fts.core.FTSRestClient; import jalview.fts.core.FTSRestRequest; import jalview.fts.core.FTSRestResponse; +import jalview.fts.core.FTSDataColumnPreferences.PreferenceSource; import jalview.fts.service.pdb.PDBFTSRestClient; import jalview.util.JSONUtils; import jalview.util.MessageManager; import jalview.util.Platform; public class TDBeaconsFTSRestClient extends FTSRestClient + implements StructureFTSRestClientI { - private static final String DEFAULT_THREEDBEACONS_DOMAIN = - "https://wwwdev.ebi.ac.uk/pdbe/pdbe-kb/3dbeacons/api/uniprot/summary/"; - - private static FTSRestClientI instance = null; - + /** + * production server URI + */ + private static String TDB_PROD_API="https://www.ebi.ac.uk/pdbe/pdbe-kb/3dbeacons/api/uniprot/summary/"; + /** + * dev server URI + */ + private static String TDB_DEV_API="https://wwwdev.ebi.ac.uk/pdbe/pdbe-kb/3dbeacons/api/uniprot/summary/"; + private static String DEFAULT_THREEDBEACONS_DOMAIN = TDB_PROD_API; + + public static FTSRestClientI instance = null; + protected TDBeaconsFTSRestClient() - { + { } - @SuppressWarnings("unchecked") @Override public FTSRestResponse executeRequest(FTSRestRequest tdbRestRequest) @@ -59,29 +68,38 @@ public class TDBeaconsFTSRestClient extends FTSRestClient else /** * Java only + * * @j2sIgnore */ { client = Client.create(new DefaultClientConfig()); clientResponseClass = ClientResponse.class; } + WebResource webResource; - webResource = client.resource(DEFAULT_THREEDBEACONS_DOMAIN) - .path(query); + webResource = client.resource(DEFAULT_THREEDBEACONS_DOMAIN+query); + URI uri = webResource.getURI(); System.out.println(uri.toString()); - + // Execute the REST request - ClientResponse clientResponse = webResource - .accept(MediaType.APPLICATION_JSON).get(clientResponseClass); - + ClientResponse clientResponse; + if (isMocked()) { + clientResponse = null; + } + else + { + clientResponse = webResource.accept(MediaType.APPLICATION_JSON) + .get(clientResponseClass); + } + // Get the JSON string from the response object or directly from the // client (JavaScript) Map jsonObj = null; String responseString = null; - + // Check the response status and report exception if one occurs - int responseStatus = clientResponse.getStatus(); + int responseStatus = isMocked() ? (mockQueries.containsKey(query) ? 200 : 404) : clientResponse.getStatus(); switch (responseStatus) { // if success @@ -92,17 +110,20 @@ public class TDBeaconsFTSRestClient extends FTSRestClient } else { - responseString = clientResponse.getEntity(String.class); + responseString = isMocked() ? mockQueries.get(query): clientResponse.getEntity(String.class); } break; case 400: throw new Exception(parseJsonExceptionString(responseString)); + case 404: + return emptyTDBeaconsJsonResponse(); default: throw new Exception( getMessageByHTTPStatusCode(responseStatus, "3DBeacons")); } // Process the response and return the result to the caller. - return parseTDBeaconsJsonResponse(responseString, jsonObj, tdbRestRequest); + return parseTDBeaconsJsonResponse(responseString, jsonObj, + tdbRestRequest); } catch (Exception e) { String exceptionMsg = e.getMessage(); @@ -123,52 +144,70 @@ public class TDBeaconsFTSRestClient extends FTSRestClient throw e; } } - + + } + + /** + * returns response for when the 3D-Beacons service doesn't have a record for + * the given query - in 2.11.2 this triggers a failover to the PDBe FTS + * + * @return null + */ + private FTSRestResponse emptyTDBeaconsJsonResponse() + { + return null; } - - public String setSearchTerm(String term) { + + public String setSearchTerm(String term) + { return term; } - + public static FTSRestResponse parseTDBeaconsJsonResponse( String tdbJsonResponseString, FTSRestRequest tdbRestRequest) { return parseTDBeaconsJsonResponse(tdbJsonResponseString, (Map) null, tdbRestRequest); } - + @SuppressWarnings("unchecked") - public static FTSRestResponse parseTDBeaconsJsonResponse(String tdbJsonResponseString, - Map jsonObj, FTSRestRequest tdbRestRequest) + public static FTSRestResponse parseTDBeaconsJsonResponse( + String tdbJsonResponseString, Map jsonObj, + FTSRestRequest tdbRestRequest) { FTSRestResponse searchResult = new FTSRestResponse(); List result = null; - + try { if (jsonObj == null) { - jsonObj = (Map) JSONUtils.parse(tdbJsonResponseString); + jsonObj = (Map) JSONUtils + .parse(tdbJsonResponseString); } - - Object uniprot_entry = jsonObj.get("uniprot_entry"); - // TODO: decide if anything from uniprot_entry needs to be reported via the FTSRestResponse object - // Arnaud added seqLength = (Long) ((Map) jsonObj.get("uniprot_entry")).get("sequence_length"); + + Object uniprot_entry = jsonObj.get("uniprot_entry"); + // TODO: decide if anything from uniprot_entry needs to be reported via + // the FTSRestResponse object + // Arnaud added seqLength = (Long) ((Map) + // jsonObj.get("uniprot_entry")).get("sequence_length"); List structures = (List) jsonObj.get("structures"); result = new ArrayList<>(); - + int numFound = 0; - for (Iterator strucIter = structures.iterator(); strucIter.hasNext();) + for (Iterator strucIter = structures.iterator(); strucIter + .hasNext();) { - Map structure = (Map) strucIter.next(); + Map structure = (Map) strucIter + .next(); result.add(getFTSData(structure, tdbRestRequest)); numFound++; } - - searchResult.setNumberOfItemsFound(numFound); + + searchResult.setNumberOfItemsFound(numFound); searchResult.setSearchSummary(result); - + } catch (ParseException e) { e.printStackTrace(); @@ -176,20 +215,35 @@ public class TDBeaconsFTSRestClient extends FTSRestClient return searchResult; } -private static FTSData getFTSData(Map tdbJsonStructure, + private static FTSData getFTSData(Map tdbJsonStructure, FTSRestRequest tdbRequest) { - // TODO Auto-generated method stub + // TODO: consider reusing PDBFTSRestClient.getFTSData ? + String primaryKey = null; Object[] summaryRowData; + + SequenceI associatedSequence; + Collection displayFields = tdbRequest.getWantedFields(); + SequenceI associatedSeq = tdbRequest.getAssociatedSequence(); int colCounter = 0; - summaryRowData = new Object[displayFields.size()]; - - for (FTSDataColumnI field : displayFields) { - String fieldData = (tdbJsonStructure.get(field.getCode()) == null) ? " " + summaryRowData = new Object[(associatedSeq != null) + ? displayFields.size() + 1 + : displayFields.size()]; + if (associatedSeq != null) + { + associatedSequence = associatedSeq; + summaryRowData[0] = associatedSequence; + colCounter = 1; + } + + for (FTSDataColumnI field : displayFields) + { + String fieldData = (tdbJsonStructure.get(field.getCode()) == null) + ? " " : tdbJsonStructure.get(field.getCode()).toString(); - // System.out.println("Field : " + field + " Data : " + fieldData); + // System.out.println("Field : " + field + " Data : " + fieldData); if (field.isPrimaryKeyColumn()) { primaryKey = fieldData; @@ -199,7 +253,7 @@ private static FTSData getFTSData(Map tdbJsonStructure, { summaryRowData[colCounter++] = null; } - else + else { try { @@ -212,14 +266,14 @@ private static FTSData getFTSData(Map tdbJsonStructure, : fieldData; } catch (Exception e) { - //e.printStackTrace(); + // e.printStackTrace(); System.out.println("offending value:" + fieldData + fieldData); } } } final String primaryKey1 = primaryKey; final Object[] summaryRowData1 = summaryRowData; - + return new FTSData() { @@ -234,7 +288,7 @@ private static FTSData getFTSData(Map tdbJsonStructure, { return primaryKey1; } - + /** * Returns a string representation of this object; */ @@ -250,7 +304,7 @@ private static FTSData getFTSData(Map tdbJsonStructure, } return summaryFieldValues.toString(); } - + /** * Returns hash code value for this object */ @@ -268,19 +322,20 @@ private static FTSData getFTSData(Map tdbJsonStructure, }; } -// private static FTSData getFTSData(Map doc, -// FTSRestRequest tdbRestRequest) -// { -// String primaryKey = null; -// -// Object[] summaryRowData; -// -// Collection displayFields = tdbRestRequest.getWantedFields(); -// int colCounter = 0; -// summaryRowData = new Object[displayFields.size() + 1]; -// -// return null; -// } + // private static FTSData getFTSData(Map doc, + // FTSRestRequest tdbRestRequest) + // { + // String primaryKey = null; + // + // Object[] summaryRowData; + // + // Collection displayFields = + // tdbRestRequest.getWantedFields(); + // int colCounter = 0; + // summaryRowData = new Object[displayFields.size() + 1]; + // + // return null; + // } private String parseJsonExceptionString(String jsonErrorString) { @@ -293,7 +348,7 @@ private static FTSData getFTSData(Map tdbJsonStructure, { return "/fts/tdbeacons_data_columns.txt"; } - + public static FTSRestClientI getInstance() { if (instance == null) @@ -302,7 +357,7 @@ private static FTSData getFTSData(Map tdbJsonStructure, } return instance; } - + private Collection allDefaultDisplayedStructureDataColumns; public Collection getAllDefaultDisplayedStructureDataColumns() @@ -316,5 +371,26 @@ private static FTSData getFTSData(Map tdbJsonStructure, } return allDefaultDisplayedStructureDataColumns; } - + + @Override + public String[] getPreferencesColumnsFor(PreferenceSource source) + { + String[] columnNames = null; + switch (source) + { + case SEARCH_SUMMARY: + columnNames = new String[] { "", "Display", "Group" }; + break; + case STRUCTURE_CHOOSER: + columnNames = new String[] { "", "Display", "Group" }; + break; + case PREFERENCES: + columnNames = new String[] { "3DB Beacons Field", "Show in search summary", + "Show in structure summary" }; + break; + default: + break; + } + return columnNames; + } }