chiark / gitweb /
fetch yarrg timestamp asynchronously
[jarrg-ian.git] / src / com / tedpearson / ypp / market / MarketUploader.java
index 4254f002e20f234f1357d267f23b9588db80adf4..bac55220cf9658d6538a907bc754f30214a5d6c2 100644 (file)
@@ -65,6 +65,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
 
        private AccessibleContext sidePanel;
        private HashMap<String,Integer> commodMap;
+        public PrintStream dtxt = null;
 
        private PropertyChangeListener changeListener = new PropertyChangeListener() {
                public void propertyChange(PropertyChangeEvent e) {
@@ -75,7 +76,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                                int index = text.indexOf(":");
                                String name = text.substring(0,index);
                                islandName = name;
-                               //System.out.println(islandName);
+                               // if (dtxt!=null) dtxt.println(islandName);
                                sidePanel.removePropertyChangeListener(this);
                                latch.countDown();
                        }
@@ -91,8 +92,9 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        }
 
         private void progresslog(String s) {
+           if (dtxt == null) return;
            long now = new Date().getTime();
-           System.out.println("progress "+(now - startTime)+"ms "+s);
+           dtxt.println("progress "+(now - startTime)+"ms "+s);
        }
 
         private void progressNote(ProgressMonitor pm, String s) {
@@ -236,6 +238,14 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                // check if we've been turned off in the control panel
                Preferences prefs = Preferences.userNodeForPackage(getClass());
 
+               if (prefs.getBoolean("writeDebugFiles", false)) {
+                   try {
+                       dtxt = new PrintStream(new File("jarrg-debug-log.txt"));
+                   } catch (java.io.FileNotFoundException e) {
+                       System.err.println("JARRG: Error opening debug log: "+e);
+                   }
+               }
+
                if (prefs.getBoolean("useLiveServers", false)) {
                        YARRG_URL = YARRG_LIVE_URL;
                        PCTB_HOST_URL = PCTB_LIVE_HOST_URL;
@@ -386,7 +396,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                        html = m.replaceAll(" ");
                }
                String whole_msg = "<html><h1>Error</h1>"+msg+"<h1>PCTB Server said:</h1><blockquote>"+html+"</blockquote>";
-               System.err.println("###" + whole_msg + "###");
+               if (dtxt!=null) dtxt.println("###" + whole_msg + "###");
 
                JOptionPane.showMessageDialog(frame,whole_msg,"Error",JOptionPane.ERROR_MESSAGE);
        }
@@ -399,19 +409,31 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        *
        *       @exception Exception if an error we didn't expect occured
        */
+        private class YarrgTimestampFetcher extends Thread {
+           public String ts = null;
+           public void run() {
+               try {
+                   ts = getYarrgTimestamp();
+                   progresslog("(async) yarrg timestamp ready.");
+               } catch(Exception e) {
+                   error("Error getting YARRG timestamp: "+e);
+               }
+           }
+       };
+
        private void runUpload() throws Exception {
                progresslog("starting");
 
-               String yarrgts = "";
                ProgressMonitor pm = new ProgressMonitor(frame,"Processing Market Data","Getting table data",0,100);
                pm.setMillisToDecideToPopup(0);
                pm.setMillisToPopup(0);
                boolean doneyarrg = false, donepctb = false;
+               YarrgTimestampFetcher yarrgts_thread = null;
 
                if (uploadToYarrg) {
-                       progresslog("yarrg timestamp...");
-                       yarrgts = getYarrgTimestamp();
-                       progresslog("yarrg timestamp done.");
+                       progresslog("(async) yarrg timestamp...");
+                       yarrgts_thread = new YarrgTimestampFetcher();
+                       yarrgts_thread.start();
                }
 
                AccessibleTable accesstable = findMarketTable();
@@ -428,7 +450,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                        return;
                }
 
-               progresslog("getisland...");
+               progresslog("(async) getisland...");
                getIsland();
                progresslog("getocean...");
                getOcean();
@@ -437,6 +459,20 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                if (latch != null) {
                    latch.await(2, java.util.concurrent.TimeUnit.SECONDS);
                }
+               progresslog("(async) getisland done");
+
+               String yarrgts = null;
+               if (yarrgts_thread != null) {
+                   progresslog("(async) yarrg timestamp join...");
+                   yarrgts_thread.join();
+                   progresslog("(async) yarrg timestamp joined.");
+                   yarrgts = yarrgts_thread.ts;
+               }
+
+               if (islandName == null) {
+                   error("Could not find island name in YPP user interface.");
+                   return;
+               }
 
                progresslog("table check...");
 
@@ -497,11 +533,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                        progressNote(pm, "Yarrg: Uploading");
                        progresslog("yarrg upload...");
 
-                       if (islandName != null) {
-                               doneyarrg = runYarrg(yarrgts, oceanName, islandName, yarrgdata);
-                       } else {
-                               System.out.println("Couldn't upload to Yarrg - no island name found");
-                       }
+                       doneyarrg = runYarrg(yarrgts, oceanName, islandName, yarrgdata);
                        progresslog("yarrg done.");
                }
 
@@ -522,14 +554,16 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                    }
                    // get commod map
                
+                   progresslog("pctb commodmap...");
                    HashMap<String,Integer> commodMap = getCommodMap();
                    if(commodMap == null) {
                        return;
                    }
+                   progresslog("pctb commodmap done.");
                    int[] offerCount = getBuySellMaps(data,buys,sells,stallMap,commodMap);
                    //println(buys.toString());
-                   //System.out.println(sells);
-                   //System.out.println("\n\n\n"+buys);
+                   // if (dtxt!=null) dtxt.println(sells);
+                   // if (dtxt!=null) dtxt.println("\n\n\n"+buys);
 
                    ByteArrayOutputStream outStream = new ByteArrayOutputStream();
                    pm.setProgress(60);
@@ -546,7 +580,15 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                    writeBuySellOffers(buys,sells,offerCount,out);
                    out.finish();
                    progresslog("pctb send...");
-                   InputStream in = sendInitialData(new ByteArrayInputStream(outStream.toByteArray()));
+
+                   byte[] ba = outStream.toByteArray();
+                   if (dtxt!=null) {
+                       FileOutputStream dbgdata = new FileOutputStream(new File("jarrg-debug-pctb-marketdata.gz"));
+                       dbgdata.write(ba);
+                       dbgdata.close();
+                   }
+
+                   InputStream in = sendInitialData(new ByteArrayInputStream(ba));
                    progresslog("pctb sent.");
                    if (in == null) return;
                    pm.setProgress(80);
@@ -596,13 +638,13 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                Accessible node1 = window;
                Accessible node = descendNodes(node1,new int[] {0,1,0,0,0,0,1,0,0,1,0,0}); // commod market
                // commod market: {0,1,0,0,0,0,1,0,0,1,0}  {0,1,0,0,0,0,1,0,1,0,0,1,0,0})
-               //System.out.println(node);
+               // if (dtxt!=null) dtxt.println(node);
                if (!(node instanceof JTable)) {
                        node = descendNodes(node1,new int[] {0,1,0,0,0,0,1,0,1,0,0,1,0,0}); // commod market
                }
                if (!(node instanceof JTable)) return null;
                AccessibleTable table = node.getAccessibleContext().getAccessibleTable();
-               //System.out.println(table);
+               // if (dtxt!=null) dtxt.println(table);
                return table;
        }
        
@@ -619,7 +661,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        private Accessible descendNodes(Accessible parent, int[] path) {
                for(int i=0;i<path.length;i++) {
                        if (null == (parent = descend(parent, path[i]))) return null;
-                       // System.out.println(parent.getClass());
+                       // if (dtxt!=null) dtxt.println(parent.getClass());
                }
                return parent;
        }
@@ -676,13 +718,6 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        
        /**
        *       Gets the list of commodities and their associated commodity ids.
-       *       On the first run, the data is downloaded from the PCTB server. 
-       *       After the first run, the data is cached using <code>Preferences</code>.
-       *       <p>
-       *       Potential issues: When more commodities are added to the server, this
-       *       program will currently break unless the user deletes the preferences
-       *       file or we give them a new release with a slighly different storage
-       *       location for the data.
        *
        *       @return a map where the key is the commodity and the value is the commodity id.
        */
@@ -691,7 +726,6 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                        return commodMap;
                }
                HashMap<String,Integer> map = new HashMap<String,Integer>();
-               Preferences prefs = Preferences.userNodeForPackage(getClass());
                String xml;
                try {
                        URL host = new URL(PCTB_HOST_URL + "commodmap.php");
@@ -704,7 +738,14 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                        int first = sb.indexOf("<pre>") + 5;
                        int last = sb.indexOf("</body>");
                        xml = sb.substring(first,last);
-                       //System.out.println(xml);
+                       //if (dtxt!=null) {
+                       //        dtxt.println(">>");
+                       //        dtxt.println(sb);;
+                       //        dtxt.println("||");
+                       //        dtxt.println(xml);
+                       //        dtxt.println("<<");
+                       //}
+                       // if (dtxt!=null) dtxt.println(xml);
                        Reader reader = new CharArrayReader(xml.toCharArray());
                        Document d = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(new InputSource(reader));
                        NodeList maps = d.getElementsByTagName("CommodMap");
@@ -774,7 +815,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                                        buySellCount[1]++;
                                }
                        } catch(IllegalArgumentException e) {
-                               System.err.println("Error: Unsupported Commodity \"" + offer.get(0) + "\"");
+                               if (dtxt!=null) dtxt.println("Error: Unsupported Commodity \"" + offer.get(0) + "\"");
                        }
                }
                if (buySellCount[0]==0 && buySellCount[1]==0) {
@@ -944,7 +985,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        */
        private boolean finishUpload(InputStream in) throws IOException {
                String html = readstreamstring(in);
-               //System.out.println(html);
+               // if (dtxt!=null) dtxt.println(html);
                Matcher m;
 
                Pattern params = Pattern.compile("(?s)<input type=\"hidden\" name=\"action\" value=\"setisland\" />.+?<input type=\"hidden\" name=\"forcereload\" value=\"([^\"]+)\" />.+?<input type=\"hidden\" name=\"filename\" value=\"([^\"]+)\" />");
@@ -979,7 +1020,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                Pattern done = Pattern.compile("Your data has been integrated into the database. Thank you!");
                m = done.matcher(complete);
                if(m.find()) {
-                       System.out.println("FILE upload successful!!!");
+                       if (dtxt!=null) dtxt.println("FILE upload successful!!!");
                        return true;
                } else {
                        error_html("Something was wrong with the final upload parameters!", complete);
@@ -1029,13 +1070,13 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
        BufferedReader br = new BufferedReader(new InputStreamReader(in));
        String yarrgresult; 
        while((yarrgresult = br.readLine()) != null) {
-           System.out.println(yarrgresult);
+           if (dtxt!=null) dtxt.println(yarrgresult);
        }
        return true;
     }
 
     private int calculateArbitrageCommodity(ArrayList<SortedSet<int[]>> arb_bs) {
-       //System.out.println("ARBITRAGE?");
+       // if (dtxt!=null) dtxt.println("ARBITRAGE?");
        int profit = 0;
        SortedSet<int[]> buys = arb_bs.get(0);
        SortedSet<int[]> sells = arb_bs.get(1);
@@ -1051,7 +1092,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
 
            int unitprofit = buy[0] - sell[0];
            int count = buy[1] < sell[1] ? buy[1] : sell[1];
-           //System.out.println(" sell @"+sell[0]+" x"+sell[1]+" buy @"+buy[0]+" x"+buy[1]
+           // if (dtxt!=null) dtxt.println(" sell @"+sell[0]+" x"+sell[1]+" buy @"+buy[0]+" x"+buy[1]
            //                 +" => x"+count+" @"+unitprofit);
 
            if (unitprofit <= 0)
@@ -1063,7 +1104,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
            if (buy[1]==0) buys.remove(buy);
            if (sell[1]==0) sells.remove(sell);
        }
-       //System.out.println(" PROFIT "+profit);
+       // if (dtxt!=null) dtxt.println(" PROFIT "+profit);
        return profit;
     }
 
@@ -1084,15 +1125,15 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
 
        for (ArrayList<String> row : data) {
            String thiscommod = row.get(0);
-           //System.out.println("ROW "+row.toString());
+           // if (dtxt!=null) dtxt.println("ROW "+row.toString());
            if (lastcommod == null || !thiscommod.equals(lastcommod)) {
                if (lastcommod != null)
                    arbitrage += calculateArbitrageCommodity(arb_bs);
-               //System.out.println("ROW rdy");
+               // if (dtxt!=null) dtxt.println("ROW rdy");
                arb_bs = new ArrayList<SortedSet<int[]>>(2);
                arb_bs.add(0, new TreeSet<int[]>(compar));
                arb_bs.add(1, new TreeSet<int[]>(compar));
-               //System.out.println("ROW init");
+               // if (dtxt!=null) dtxt.println("ROW init");
                lastcommod = thiscommod;
            }
            for (int bs = 0; bs < 2; bs++) {
@@ -1100,7 +1141,7 @@ public class MarketUploader implements TopLevelWindowListener, GUIInitializedLis
                if (pricestr == null)
                    continue;
                int[] entry = new int[2];
-               //System.out.println("ROW BS "+bs);
+               // if (dtxt!=null) dtxt.println("ROW BS "+bs);
                entry[0] = parseQty(pricestr);
                entry[1] = parseQty(row.get(bs*2 + 3));
                arb_bs.get(bs).add(entry);