Java Http代理

时间:2011-02-08 22:09:42

标签: java http proxy deflate

我正在用Java编写一个小代理,基本上可以选择2个特定文件并对它们进行一些额外的处理。一个URL只是在传递内容之前从内容中获取一些信息。另一个文件我想过滤响应内容,这只是xml deflate编码(我想删除一些子元素)。

现在,当我通过所有内容时,代理工作正常。但是,当我尝试过滤xml文件时,它实际上并没有将内容发送到客户端???

以下是一些代码:

在接受Socket连接时生成的Thread run()方法中,一旦我确定请求是针对我想要过滤的文件,我就调用:

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流中读取,因为它在内部使用mark和read,而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,但客户端只是没有收到它(在web调试器中显示内容长度为0,尽管在内容中没有Content-Length头原始回应或者)

关于发生了什么,或者我应该做什么的任何想法,我不是?

2 个答案:

答案 0 :(得分:0)

我没有看到任何错误,但如果您已经使用Document,那么您是否只能修改文档然后使用XML库来编写整个文档?这似乎比手动XML格式更强大。

答案 1 :(得分:-1)

你需要在deflate.write()之后添加对deflate.close()的调用,以便刷新输出并正确关闭流。