编解码器 - 读入固定长度的字符串

问题描述

我正在编写一个文件解析器,用于读取包含固定长度、0 填充字符串的现有文件格式。

因此,例如,我需要解析的文件中有两个用于二进制结构的 case 类。第一个包含一个 4 个字符的字符串,可以是两个值之一,后者包含一个 8 个字符的字符串(其中值

case class WadHeader( magic : String,items : Int,dirOffset : Int)
case class LumpIndex( offset : Int,size : Int,lumpName : String)

我尝试编写一个简单的编解码器来解析第一个

  implicit val headerCodec : Codec[WadHeader] = {
    ("magic" | bytes(4)) ::
      ("items" | uint32) ::
      ("dirOffset" | uint32)
  }.as[WadHeader]

但是,我发现它无法成功将其转换为 WadHeader(大概是因为魔法值与案例类定义不完全匹配。我希望能够摄取固定的- size 字节串并将其解码为 String 对象。

不幸的是,搜索文档只会出现“贪婪”字符串或大小前缀字符串选项。

解决方法

好的 - 所以我想出了一个可以正常工作的解决方案。可能有一种更简单/更干净的方法来做到这一点,但效果很好。

首先,当我提前知道长度时,我定义了一个新的fixedString编解码器用于读取字符串:

  def fixedString(size: Int): Codec[String] = new Codec[String] {
    private val codec = fixedSizeBytes(size,ascii)
    def sizeBound: SizeBound = SizeBound.exact(size * 8L)
    def encode(b: String): Attempt[BitVector] = codec.encode(b)
    def decode(b: BitVector): Attempt[DecodeResult[String]] = {
      codec.decode(b) match {
        case Successful(DecodeResult(value,remainder)) =>
          val decoded = value.toSeq.takeWhile(_>0).mkString

          Attempt.successful(DecodeResult(decoded,remainder))
        case fail : scodec.Attempt.Failure => fail
      }
    }
    override def toString = s"fixedString($size)"
  }

这适用于字符串。第二个只是我的一个愚蠢的错误(uint32 解码为 Long,而不是 Int),这需要我相应地更新我的案例类定义:

case class WadHeader( magic : String,items : Long,dirOffset : Long)

object WadHeader {
  implicit val codec : Codec[WadHeader] = {
    ("magic" | fixedString(4)) ::
      ("items" | uint32) ::
      ("dirOffset" | uint32)
  }.as[WadHeader]
}

编辑:5/7 - 发现我可以包装 fixedSizeCodec(size,ascii) 而不是 bytes 并且它完成了我想要的大部分内容并相应地更新了解决方案。根据要求,fixedSizeCodec(size,cstring) 也可能是一个非常好的解决方案 - 但是对于我的用例,对于使用完整字段长度的字符串失败,因为没有空间用于终止 nul。