问题描述
我正在编写一个文件解析器,用于读取包含固定长度、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。