Add warning when field is unresolvable

valaphee · Apr 6, 2022 · 09c33f6 · 09c33f6
1 parent d9adeb7
commit 09c33f6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/kotlin/com/valaphee/protod/ProtoWriter.kt b/src/main/kotlin/com/valaphee/protod/ProtoWriter.kt
@@ -82,9 +82,9 @@ class ProtoWriter(
                 }
             }"
         }
-        val messageExtensions = messageExtensions[".${options.descriptorForType.fullName}"] ?: emptyMap()
+        val messageExtensions = messageExtensions[".${options.descriptorForType.fullName}"]
         options.unknownFields.asMap().forEach { unknownField ->
-            messageExtensions[unknownField.key]?.let { messageExtension ->
+            messageExtensions?.get(unknownField.key)?.let { messageExtension ->
                 val messageExtensionFields = checkNotNull(messages[messageExtension.typeName]).fieldList
                 unknownField.value.lengthDelimitedList.forEach {
                     val codedInputStream = it.newCodedInput()
@@ -119,7 +119,7 @@ class ProtoWriter(
                         }"
                     }
                 }
-            }
+            } ?: println("Unresolvable field .${options.descriptorForType.fullName} ${unknownField.key}")
         }
         return generatedOptions
     }