From a68dec9749e95009b2168bf83e5a69ce2e123e71 Mon Sep 17 00:00:00 2001 From: Him188 Date: Tue, 28 Jan 2020 21:48:11 +0800 Subject: [PATCH] Add ` : ProtoBuf` --- .../src/jvmTest/kotlin/test/ProtoBufDataClassGenerator.kt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mirai-core-qqandroid/src/jvmTest/kotlin/test/ProtoBufDataClassGenerator.kt b/mirai-core-qqandroid/src/jvmTest/kotlin/test/ProtoBufDataClassGenerator.kt index b69d63923..e3a523b22 100644 --- a/mirai-core-qqandroid/src/jvmTest/kotlin/test/ProtoBufDataClassGenerator.kt +++ b/mirai-core-qqandroid/src/jvmTest/kotlin/test/ProtoBufDataClassGenerator.kt @@ -143,7 +143,7 @@ fun String.generateProtoBufDataClass(): GeneratedClass { val javaClassname = substringBetween("class", "{") val superclasses = javaClassname.split("$").map { it.trim().adjustClassName() }.toMutableList().apply { removeAt(this.lastIndex) } val className = substringBetween("class", "{").substringAfterLast("$").trim().adjustClassName() - return GeneratedClass(superclasses, className, "@Serializable\nclass $className") + return GeneratedClass(superclasses, className, "@Serializable\nclass $className : ProtoBuf") } val superclasses = substringBetween("class", "extends").split("$").map { it.trim().adjustClassName() }.toMutableList() @@ -154,7 +154,7 @@ fun String.generateProtoBufDataClass(): GeneratedClass { val ids = substringBetween("new int[]{", "}").split(",").map { it.trim() } if (ids.all { it.isBlank() }) { - return GeneratedClass(superclasses, className, "@Serializable\nclass $className") + return GeneratedClass(superclasses, className, "@Serializable\nclass $className : ProtoBuf") } val names = substringBetween("new String[]{", "}").split(",").map { it.trim() } @@ -326,7 +326,7 @@ fun String.generateProtoBufDataClass(): GeneratedClass { append("\n") } - append(")") + append(") : ProtoBuf") } return GeneratedClass(superclasses, className, source)