Adds check for SP only commands.
[mkjsf-moved-to-github.git] / src / main / java / com / indexdata / mkjsf / pazpar2 / Pz2Bean.java
index 30ac455..5b45551 100644 (file)
@@ -26,6 +26,7 @@ import com.indexdata.mkjsf.errors.ErrorHelper;
 import com.indexdata.mkjsf.pazpar2.commands.CommandParameter;\r
 import com.indexdata.mkjsf.pazpar2.commands.Pazpar2Command;\r
 import com.indexdata.mkjsf.pazpar2.commands.Pazpar2Commands;\r
+import com.indexdata.mkjsf.pazpar2.commands.sp.ServiceProxyCommand;\r
 import com.indexdata.mkjsf.pazpar2.data.RecordResponse;\r
 import com.indexdata.mkjsf.pazpar2.data.ResponseDataObject;\r
 import com.indexdata.mkjsf.pazpar2.data.ResponseParser;\r
@@ -141,17 +142,7 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
       return "";\r
     } else {\r
       logger.debug("Executing record command");\r
-      ResponseDataObject responseObject = doCommand("record");\r
-      if ((pzreq.getRecord().hasParameterValue("offset") ||\r
-            pzreq.getRecord().hasParameterValue("checksum")) &&\r
-            !responseObject.getType().equals("record")) {\r
-        logger.debug("Storing record offset response as 'record'");\r
-        RecordResponse recordResponse = new RecordResponse();\r
-        recordResponse.setType("record");\r
-        recordResponse.setXml(responseObject.getXml());\r
-        recordResponse.setAttribute("activeclients", "0");\r
-        pzresp.put("record", recordResponse);\r
-      }\r
+      doCommand("record");\r
       return pzresp.getRecord().getActiveClients();\r
     }\r
   }\r
@@ -170,7 +161,7 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
       logger.error("Ignoring show,stat,termlist,bytarget commands due to problem with most recent search.");\r
       return "";\r
     } else if (!hasQuery()) {\r
-      logger.error("Ignoring show,stat,termlist,bytarget commands because there is not yet a query.");\r
+      logger.debug("Ignoring show,stat,termlist,bytarget commands because there is not yet a query.");\r
       return "";\r
     } else {\r
       return update("show,stat,termlist,bytarget");\r
@@ -184,7 +175,7 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
    * @return Number of activeclients at the time of the 'show' command\r
    */\r
   public String update (String commands) {\r
-    logger.info("Request to update: " + commands);\r
+    logger.debug("Request to update: " + commands);\r
     try {\r
       if (commands.equals("search")) {\r
         doSearch();\r
@@ -205,7 +196,6 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
           return "0";\r
         } else {\r
           logger.debug("Processing request for " + commands); \r
-          \r
           List<CommandThread> threadList = new ArrayList<CommandThread>();\r
           StringTokenizer tokens = new StringTokenizer(commands,",");\r
           while (tokens.hasMoreElements()) {          \r
@@ -321,29 +311,53 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
   }\r
   \r
   /**\r
-   * Validates the request then executes the command and parses the response.\r
+   * Executes the command and parses the response to create data objects.\r
    * If the parsed response is of a known type it will be cached in 'pzresp'\r
    * \r
    * @param commandName The command to be executed\r
    * @return An XML response parsed to form a response data object\r
    */\r
   protected ResponseDataObject doCommand(String commandName) {\r
-    ResponseDataObject responseObject = null;     \r
-    logger.info("Request "+commandName + ": "+ pzreq.getCommand("search").toString());\r
-    Pazpar2Command command = pzreq.getCommand(commandName);\r
-    long start = System.currentTimeMillis();\r
-    HttpResponseWrapper commandResponse = searchClient.executeCommand(command);\r
-    long end = System.currentTimeMillis();\r
-    logger.debug("Executed " + command.getCommandName() + " in " + (end-start) + " ms." );\r
-    responseLogger.debug("Response was: " + commandResponse.getResponseString());\r
-    responseObject = ResponseParser.getParser().getDataObject((ClientCommandResponse)commandResponse);\r
-    if (ResponseParser.docTypes.contains(responseObject.getType())) {\r
-      logger.debug("Storing " + responseObject.getType() + " in pzresp. ");\r
-      pzresp.put(commandName, responseObject);\r
+    if (pzreq.getCommand(commandName).spOnly() && isPazpar2Service()) {\r
+      logger.warn("Skipping " + commandName + " - SP-only command, un-supported by Pazpar2");\r
+      return new ResponseDataObject();\r
     } else {\r
-      logger.info("Unrecognized response object type not cached in pzresp: " + responseObject.getType());\r
+      ResponseDataObject responseObject = null;     \r
+      logger.info("Request "+commandName + ": "+ pzreq.getCommand("search").toString());\r
+      Pazpar2Command command = pzreq.getCommand(commandName);\r
+      long start = System.currentTimeMillis();\r
+      HttpResponseWrapper commandResponse = searchClient.executeCommand(command);\r
+      long end = System.currentTimeMillis();\r
+      logger.debug("Executed " + command.getCommandName() + " in " + (end-start) + " ms." );\r
+      responseLogger.debug("Response was: " + commandResponse.getResponseString());\r
+      if (commandResponse.getContentType().contains("xml")) {\r
+        responseObject = ResponseParser.getParser().getDataObject((ClientCommandResponse)commandResponse);\r
+        if (ResponseParser.docTypes.contains(responseObject.getType())) {\r
+          logger.debug("Storing " + responseObject.getType() + " in pzresp. ");\r
+          pzresp.put(commandName, responseObject);\r
+        } else {\r
+          if (commandName.equals("record")) {\r
+            logger.debug("Command was 'record' but response not '<record>' - assuming raw record response.");\r
+            ResponseDataObject recordResponse = new RecordResponse(); \r
+            recordResponse.setType("record");\r
+            recordResponse.setXml(responseObject.getXml());          \r
+            recordResponse.setAttribute("activeclients", "0");\r
+            pzresp.put("record", recordResponse); \r
+          }        \r
+        }\r
+      } else if (commandResponse.isBinary()) {\r
+        responseObject = new RecordResponse(); \r
+        responseObject.setType(commandName);\r
+        logger.info("Binary response");\r
+        responseObject.setAttribute("activeclients", "0");\r
+        responseObject.setXml("<record>binary response</record>");\r
+        responseObject.setBinary(commandResponse.getBytes());\r
+        pzresp.put("record", responseObject);\r
+      } else {\r
+        logger.error("Response was not found to be XML or binary. The response was not handled.");\r
+      }\r
+      return responseObject;\r
     }\r
-    return responseObject;\r
   }\r
     \r
   @Override\r
@@ -399,6 +413,17 @@ public class Pz2Bean implements Pz2Interface, StateListener, Configurable, Seria
     return (searchClient!=null ? searchClient.getServiceUrl() : "");\r
   }\r
   \r
+  public void setServiceId () {\r
+    pzreq.getRecord().removeParametersInState();\r
+    pzreq.getSearch().removeParametersInState();\r
+    pzresp.resetSearchResponses();\r
+    pz2Client.setServiceId(pzreq.getInit().getService());\r
+  }\r
+  \r
+  public String getServiceId () {\r
+    return pzreq.getInit().getService();\r
+  }\r
+  \r
   public boolean getServiceUrlIsDefined() {\r
     return (searchClient != null && searchClient.hasServiceUrl());\r
   }\r