提供几个快捷方法

fuqiuluo · Sep 3, 2023 · af08214 · af08214
1 parent 7c98547
commit af08214
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/main/kotlin/moe/fuqiuluo/proto2json/Proto.kt b/src/main/kotlin/moe/fuqiuluo/proto2json/Proto.kt
@@ -175,6 +175,12 @@ val Proto.asString: ByteString
 val Proto.asNumber: Number
  get() = (this as ProtoNumber).value
 
+val Proto.asInt: Int
+ get() = (this as ProtoNumber).value.toInt()
+
+val Proto.asLong: Long
+ get() = (this as ProtoNumber).value.toLong()
+
 val Proto.asMap: ProtoMap
  get() = (this as ProtoMap)
 

diff --git a/src/test/kotlin/net/oicq/parse_pb_3.kt b/src/test/kotlin/net/oicq/parse_pb_3.kt
@@ -12,8 +12,8 @@ fun main() {
 
 
  val face = obj[2].asList.first { elem ->
- elem.asMap.let { 53 in it && it[53, 1].asNumber.toInt() == 33 }
+ elem.asMap.let { 53 in it && it[53, 1].asInt == 33 }
  }
- println("表情ID: " + face.asMap[53, 2, 1].asNumber)
- println("表情简介: " + face.asMap[53, 2, 2].asUtf8String)
+ println("表情ID: " + face[53, 2, 1].asInt)
+ println("表情简介: " + face[53, 2, 2].asUtf8String)
 }