Java Http Proxy - PullRequest
       16

Java Http Proxy

2 голосов
/ 09 февраля 2011

Я пишу небольшой прокси в Java, который в основном выбирает 2 конкретных файла и выполняет с ними дополнительную обработку. Один URL-адрес просто извлекает некоторую информацию из содержимого перед его передачей. Другой файл, который я хочу отфильтровать по содержимому ответа, который просто закодирован в xml deflate (хочу удалить некоторые дочерние элементы).

Теперь прокси работает нормально, когда я просто передаю все содержимое. Тем не менее, когда я пытаюсь отфильтровать XML-файл, он фактически не отправляет содержимое клиенту ???

Вот код:

В методе Thread run (), который создается при приеме соединения Socket, как только я определяю, что запрос относится к файлу, который я хочу отфильтровать, я вызываю:

filterRaceFile(serverIn, clientOut);     // This won't send content     
//streamHTTPData(serverIn, clientOut, true); // This passes through fine (but all content of course).

а вот и сам метод фильтрации:

private void filterRaceFile(InputStream is, OutputStream os) {
    // Pass through headers before using deflate and filtering xml
    processHeader(is, os, new StringBuilder(), new StringBuilder());        

    // Seems to be 1 line left, inflater doesn't like it if we don't do this anyway...?
    try {
        os.write(readLine(is, false).getBytes());
    } catch (IOException e) {
        e.printStackTrace();
    }       

    InflaterInputStream inflate = new InflaterInputStream(is);
    DeflaterOutputStream deflate = new DeflaterOutputStream(os);
    int c = 0;
    try {
        DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();
        Document xdoc = db.parse(inflate);
        Node id = xdoc.getElementsByTagName("id").item(0);
        Node msg = xdoc.getElementsByTagName("message").item(0);
        StringBuilder xml_buf = new StringBuilder("<race>\n");

        xml_buf.append("<id>").append(id.getTextContent()).append("</id>\n");
        xml_buf.append("<message>").append(msg.getTextContent()).append("</message>\n");
        xml_buf.append("<boats>\n");

        NodeList allBoats = xdoc.getElementsByTagName("boat");
        int N = allBoats.getLength();

        for (int i = 0; i < N; ++i)
        {
            Node boat = allBoats.item(i);
            Element boat_el = (Element)boat;                
            double lat = Double.parseDouble(boat_el.getElementsByTagName("lat").item(0).getTextContent());
            double lon = Double.parseDouble(boat_el.getElementsByTagName("lon").item(0).getTextContent());
            double dist = Geodesic.vincenty_earth_dist(proxy.userLat, proxy.userLon, lat, lon)[0];
            if (dist <= LOS_DIST)
            {
                String boatXML = xmlToString(boat);
                //<?xml version="1.0" encoding="UTF-8"?> is prepended to the xml
                int pos = boatXML.indexOf("?>")+2;
                boatXML = boatXML.substring(pos);
                xml_buf.append(boatXML);
                ++c;
            }
        }
        System.out.printf("%d boats within LOS distance\n", c);

        xml_buf.append("</boats>\n");
        xml_buf.append("</race>");

        byte[] xml_bytes = xml_buf.toString().getBytes("UTF-8");
        deflate.write(xml_bytes);

    } catch (Exception e) {
        e.printStackTrace();
    }

    // flush the OutputStream and return
    try {
        os.flush();
    } catch (Exception e) {
        e.printStackTrace();
    }
    }

У меня также есть простой метод сквозной передачи, который просто записывает InputStream сервера в OutputStream клиента, также использует readLine и работает нормально - то есть любой другой URL-адрес отображается в браузере без проблем, поэтому readLine в порядке. Логический параметр должен сообщить ему, что он читает из потока deflate, поскольку он использует метку и внутреннее чтение, что не поддерживается в потоках deflate.

XML очень прост:

<race> 
  {some data to always pass thru}
    <boats>
       <boat>
          <id>1234</id>
          ....
          <lat>-23.3456</lat>
          <lon>17.2345</lon>
       </boat>
       {more boat elements}
     </boats>
   </race>

И он производит xml, который я хочу, чтобы он нормально отправлялся клиенту, но клиент просто его не получает (показывает content-length 0 в web-отладчике, хотя в Content-Length нет заголовка в оригинальный ответ либо)

Есть какие-либо идеи относительно того, что происходит, или что я должен делать, что я не ?? 1020 *

Ответы [ 2 ]

0 голосов
/ 05 декабря 2014

Вам нужно добавить вызов deflate.close () после deflate.write (), чтобы очистить вывод и правильно закрыть поток.

0 голосов
/ 09 февраля 2011

Я не вижу никакой ошибки, но если вы уже используете Document, вы не можете просто изменить документ, а затем использовать свою библиотеку XML для записи всего документа?Это кажется более надежным, чем ручное форматирование XML.

...