java - 需要阻塞直到接收到特定字符序列的 TCPIP 客户端

Question

我需要一个 java 7 TCP/IP 客户端，它将阻塞直到它接收到用户指定的字符序列(在我的例子中是消息终止符/分隔符——这会自动将数据“分 block ”为单独的消息以供进一步处理)。我预计这将是非常标准的代码，可以在网络上免费获得——但到目前为止运气不好。

使用标准行分隔符(例如 Oracle 的 KnockKnock 客户端中的 readLine())将接收到的数据“分 block ”，使事情复杂化是不可能的，因为这些字符是消息中的有效数据。消息格式是国际标准，无法更改。

在尝试了一些事情之后(见下文)，我想知道我是否采用了正确的方法。有没有我可以从中汲取灵感的免费软件示例？或者也许是满足我需要的类已经存在于“rt.jar”或其他地方的深处。 (顺便说一句，我使用 eclipse 查看了 rt.jar 的内容——大量的包/类(根据 http://www.findjar.com/jar/com.sun/jars/rt.jar.html?all=true JVM 6 包含 13200 多个类)使得手动搜索不切实际)。

---

我使用 Oracles 示例“KnockKnock”客户端作为起点。我的第一个想法是只需要修改一行:

while ( (fromServer = in.readLine()) != null )  

类似于:

while ( (fromServer = in.readLine( separator = UserSpecifiedRegExValue )) != null )  

不幸的是，Java 中不存在这种非常有用的 readLine() 重载/泛化。

Oracle 的示例之所以有效，是因为 readLine() 会阻塞，直到它在 TCP/IP 链接上接收到行分隔符值。我的想法是 readLine() 的通用版本也会阻塞，直到它收到用户指定的字符串(即消息终止符)，从而准确地给出我想要的。由于该方法不可用，我的下一个想法是将 readLine() 替换为 getNextMessage() 函数，该函数将阻塞，直到 TCP/IP 接收到用户指定的字符串。根据其他帖子，我想出了这个功能:

static String getNextMessage( java.io.BufferedReader MessageSource, 
                              String                 EndOfMessage_RegEx ) 
{           
    try ( java.util.Scanner s = new java.util.Scanner( MessageSource ) ) 
    { 
        return s.useDelimiter( EndOfMessage_RegEx ).hasNext() ? s.next() : ""; 
    }
}

并通过模拟 readLine() 对其进行测试，传递 O/S 特定的行分隔符，就像在这个变体中所做的那样:

final static String  LineSeparator     = System.getProperty( "line.separator" );  // LineSeparator = ODOA (<CR><LF>) on Win7
final static String  MessageSeparator  = Pattern.quote( LineSeparator );          // MessageSeparator = 5C510D0A5C45 (the RegEx string "\Q<CR><LF>\E")
final static Pattern EndOfMessageRegEx = Pattern.compile( MessageSeparator );

static String getNextMessage( java.io.BufferedReader MessageSource ) 

// This function needs to block until a complete message (terminated by 
// "EndOfMessageRegEx") is received by TCPIP from the other machine.

{           
    try ( java.util.Scanner s = new java.util.Scanner( MessageSource ).useDelimiter( EndOfMessageRegEx ) ) 
    { 
        if ( s.hasNext() ) 
        {
            return s.next();
        } 
        else 
        {
            return "";
        }
    }
}

不幸的是，这两个版本总是返回空字符串，立即终止我的客户端——如果 hasNext() 没有阻塞，这是有道理的。 ( hasNext() 文档说它“可能” - 即不保证 - 阻止。)我如何获得阻止效果？

我在这两个版本中看到的另一个问题是，每次调用该函数时，它们都毫无意义地重新创建了一个扫描器。

或者我是否被迫使用更原始的方法来创建缓冲区，使用 .read() 并改为搜索指定的字符串？

---

解决方案:移至接受的答案

Answer 1

考虑从 PushbackInputStream 派生的这个 InputStream:

public static class TerminatorInputString extends PushbackInputStream {

    private String terminator;

    public TerminatorInputString(InputStream inputStream, String string) {
        super(inputStream, 256);
        terminator = string;
    }

    public String nextMessage() throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] data = new byte[8];
        int len = read(data,  0,  data.length);
        while(len > 0) {
            baos.write(data, 0, len);
            byte[] all = baos.toByteArray();
            int idx = baos.toString().indexOf(terminator);
            if(idx >= 0) {                  
                String message = baos.toString().substring(0,  idx);
                byte[] unread = new byte[all.length-idx-terminator.length()];
                System.arraycopy(all, idx+terminator.length(), unread, 0, unread.length);
                super.unread(unread);
                return message;
            }
            len = read(data,  0,  data.length);
        }
        baos.flush();
        return new String(baos.toByteArray());
    }
}

它一直读取直到找到终止符，然后跳过终止符并继续执行。流结束将关闭最终消息。

测试框架:

public static void main(String[] args) {
    try {
        //System.in is a bad stream (if used in eclipse at least)
        // - it will only flush and make data available
        // on new line
        TerminatorInputString tis = new TerminatorInputString(System.in, "SCHWARZENEGGER");
        String message = tis.nextMessage();
        while(message != null) {
            System.out.println("MSG:>" + message + "<");
            message = tis.nextMessage();
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

有了这个输入

oneSCHWARZENEGGERtwoSCHWARZENEGGERthreeSCHWARZENEGGER

产生这个输出:

MSG:>one<
MSG:>two<
MSG:>three<