3

我需要编写执行以下操作的代码：

  1. 连接到一个 tcp 套接字
  2. 读取以 "\r\n" 结尾的包含数字 N 的行
  3. 读取 N 个字节
  4. 使用那些 N 个字节

我目前正在使用以下代码：

val socket = new Socket(InetAddress.getByName(host), port)
val in = socket.getInputStream;
val out = new PrintStream(socket.getOutputStream)

val reader = new DataInputStream(in)
val baos = new ByteArrayOutputStream
val buffer = new Array[Byte](1024)

out.print(cmd + "\r\n")
out.flush

val firstLine = reader.readLine.split("\\s")

if(firstLine(0) == "OK") {
  def read(written: Int, max: Int, baos: ByteArrayOutputStream): Array[Byte] = {
    if(written >= max) baos.toByteArray
    else {
      val count = reader.read(buffer, 0, buffer.length)
      baos.write(buffer, 0, count)
      read(written + count, max, baos)
    }
  }

  read(0, firstLine(1).toInt, baos)
} else {
  // RAISE something
}

baos.toByteArray()

这段代码的问题是使用DataInputStream#readLine会引发弃用警告，但我找不到同时实现read(...)和的类readLine(...)。BufferedReader例如，实现read但它读取字符而不是字节。我可以将这些字符转换为字节，但我认为这不安全。

在scala中写这样的东西还有其他方法吗？

谢谢

4

3 回答 3

3

请注意，在 JVM 上，char 有 2 个字节，因此“\r\n”是 4 个字节。对于存储在 JVM 之外的字符串，这通常不是真的。

我认为最安全的方法是以原始字节读取文件，直到达到“\r\n”的二进制表示，现在您可以在第一个字节上创建一个读取器（将字节转换为 JVM 兼容字符），您可以在其中请确保只有文本，对其进行解析，然后安全地继续处理其余的二进制数据。

于 2013-05-16T04:13:13.627 回答
2

您可以实现在一个类中使用 read(...) 和 readLine(...) 的目标。这个想法是使用 BufferedReader.read():Int。BufferedReader 类缓冲了内容，因此您可以一次读取一个字节而不会降低性能。

更改可以是：（没有 scala 样式优化）

import java.io.BufferedInputStream
import java.io.BufferedReader
import java.io.ByteArrayOutputStream
import java.io.PrintStream
import java.net.InetAddress
import java.net.Socket
import java.io.InputStreamReader


object ReadLines extends App {
  val host = "127.0.0.1"
  val port = 9090
  val socket = new Socket(InetAddress.getByName(host), port)
  val in = socket.getInputStream;
  val out = new PrintStream(socket.getOutputStream)

//  val reader = new DataInputStream(in)
  val bufIns = new BufferedInputStream(in)
  val reader = new BufferedReader(new InputStreamReader(bufIns, "utf8"));

  val baos = new ByteArrayOutputStream
  val buffer = new Array[Byte](1024)

  val cmd = "get:"
  out.print(cmd + "\r\n")
  out.flush

  val firstLine = reader.readLine.split("\\s")

  if (firstLine(0) == "OK") {
    def read(written: Int, max: Int, baos: ByteArrayOutputStream): Array[Byte] = {
      if (written >= max) {
        println("get: " + new String(baos.toByteArray))
        baos.toByteArray()
      } else {
//         val count = reader.read(buffer, 0, buffer.length)
        var count = 0
        var b = reader.read()
        while(b != -1){
          buffer(count) = b.toByte
          count += 1
          if (count < max){
              b = reader.read()
          }else{
            b = -1
          }
        }
        baos.write(buffer, 0, count)
        read(written + count, max, baos)
      }
    }

    read(0, firstLine(1).toInt, baos)
  } else {
    // RAISE something
  }

  baos.toByteArray()
}

为了测试，下面是一个服务器代码：

object ReadLinesServer extends App {
  val serverSocket = new ServerSocket(9090)
  while(true){
    println("accepted a connection.")
    val socket = serverSocket.accept()
    val ops = socket.getOutputStream()
    val printStream = new PrintStream(ops, true, "utf8")
    printStream.print("OK 2\r\n")   // 1 byte for alpha-number char
    printStream.print("ab")
  }
}
于 2013-05-20T06:46:44.317 回答
0

似乎这是我能找到的最佳解决方案：

val reader = new BufferedReader(new InputStreamReader(in))
val buffer = new Array[Char](1024)

out.print(cmd + "\r\n")
out.flush

val firstLine = reader.readLine.split("\\s")

if(firstLine(0) == "OK") {
  def read(readCount: Int, acc: List[Byte]): Array[Byte] = {
    if(readCount <= 0) acc.toArray
    else {
      val count = reader.read(buffer, 0, buffer.length)
      val asBytes = buffer.slice(0, count).map(_.toByte)

      read(readCount - count, acc ++ asBytes)
    }
  }

  read(firstLine(1).toInt, List[Byte]())
} else {
  // RAISE
}

即buffer.map(_.toByte).toArray用于将 char Array 转换为 Byte Array 而无需关心编码。

于 2013-05-21T08:46:47.907 回答