bufio 扫描仪和处理新行

今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《bufio 扫描仪和处理新行》，主要内容是讲解等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

我有 2 个进程通过 tcp 套接字进行通信。 a 方向 b 方发送字符串，有时使用标准加密/密码包对其进行加密。生成的字符串可能包含换行符，但 b 方的 bufio 扫描器将其解释为请求的结束。我希望 b 方继续接受行、追加它们并等待已知的命令结束字符，然后再进一步处理它。 b 端将向 a 端返回响应，因此连接保持打开状态，因此不能使用关闭连接事件作为命令分隔符。

单行命令一切正常，但加密输出中的这些新行字符会导致问题（大约 10% 的时间）。

a 方将以以下格式发送（第三个是我尝试正确处理的问题字符串的合法示例）：

callcommand()

callcommand("一","二","三")

callcommand("string","要关注的加密数据","[7b��cr��l��g���bh�@x��������� �(z �$�0�带5y7+�u�qt�ΐl�k�(�n�u��j���qk�bx�+�l\8h��-g�y.�. �1�f�i�c�ş㳿��o�xz�8?��c�e �tb�?4�hd w��� �<��Е�gc����n�v����p8����o3")

我们可以相当可靠地说命令结束键是一个右括号“)”和一个换行符。

a方发送给b方的功能：

func writer(text string) string {
    conn, err := net.dial("tcp", tcpdest)
    t := time.now()
    if err != nil {
        if _, t := err.(*net.operror); t {
            fmt.println("some problem connecting.\r\n")
        } else {
            fmt.println("unknown error: " + err.error()+"\r\n")
        }
    } else {
        conn.setwritedeadline(time.now().add(1 * time.second))
        _, err = conn.write([]byte(text+"\r\n"))
        if err != nil {
            fmt.println("error writing to stream.\r\n")
        } else {
            timenow := time.now()           
            if timenow.sub(t.add(time.duration(5*time.second))).seconds() > 5 {
                return "timeout"
            }
            scanner := bufio.newscanner(conn)
            for {
                ok := scanner.scan()
                if !ok {
                    break
                }
                if strings.hasprefix(scanner.text(), "callcommand(") && strings.hassuffix(scanner.text(), ")") {
                    conn.close()
                    return scanner.text()
                }
            }
        }
    }
    return "unspecified error"
}

b方对传入连接的处理：

src := "192.168.68.100:9000"
listener, _ := net.Listen("tcp", src)

defer listener.Close()

for {
    conn, err := listener.Accept()
    if err != nil {
        fmt.Println("Some connection error: %s\r\n", err)
    }
    go handleConnection(conn)
}   

func handleConnection(conn net.Conn) {
    remoteAddr := conn.RemoteAddr().String()
    fmt.Println("Client connected from " + remoteAddr + "\r\n")

    scanner := bufio.NewScanner(conn)
    wholeString := ""
    for {
        ok := scanner.Scan()

        if !ok {
            break
        }

        //Trying to find the index of a new-line character, to help me understand how it's being processed
        fmt.Println(strings.Index(scanner.Text(), "\n"))
        fmt.Println(strings.Index(wholeString, "\n"))

        //for the first line received, add it to wholeString
        if len(wholeString) == 0 {
            wholeString = scanner.Text()
        }

        re := regexp.MustCompile(`[a-zA-Z]+\(.*\)\r?\n?`)

        if re.Match([]byte(wholeString)) {
            fmt.Println("Matched command format")
            handleRequest(wholeString, conn)
        } else if len(wholeString) > 0 && !re.Match([]byte(wholeString)) {
            //Since we didn't match regex, we can assume there's a new-line mid string, so append to wholeString
            wholeString += "\n"+scanner.Text()
        }

    }
    conn.Close()
    fmt.Println("Client at " + remoteAddr + " disconnected.\r\n")
}

func handleRequest(request string, conn net.Conn) {
    fmt.Println("Received: "+request)
}

我不太确定 b 侧的这种方法是否正确，但在上面包含了我的代码。我见过一些实现，但很多似乎依赖于关闭连接来开始处理请求，这不适合我的场景。

感谢任何指点，谢谢。

解决方案

您的通信“协议”（一行是一条消息，不完全是协议）显然无法处理二进制数据。如果您想在协议中发送文本数据，您可以将二进制数据转换为文本，例如使用 base64 编码。您还需要一些语义来指示某些文本是从二进制转换而来的。

或者您可以更改协议以本地处理二进制数据。您可以在后面添加二进制数据的长度，这样您就知道必须将此数据读取为二进制数据，而不是将换行符解释为消息的结尾。

有许多协议在这方面做得很好，也许您不需要想出您的自定义协议。如果你想发送短信，http 使用起来非常简单，你可以将数据格式化为 json，使用 base64 将二进制数据转换为文本：

{
    "command": "string",
    "args": [
        "binaryDataAsBase64"
    ]
}

今天关于《bufio 扫描仪和处理新行》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！