From 831003bb7dc0a7896f8112d28430bd5d9b657385 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=96mer=20Sinan=20A=C4=9Facan?= <omersa@google.com>
Date: Tue, 13 Jun 2023 12:24:02 +0200
Subject: [PATCH] Minor refactoring in ExtensionRegistry.getExtension (#841)

---
 protobuf/lib/src/protobuf/extension_registry.dart | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/protobuf/lib/src/protobuf/extension_registry.dart b/protobuf/lib/src/protobuf/extension_registry.dart
index 8d925e0e..81b658ec 100644
--- a/protobuf/lib/src/protobuf/extension_registry.dart
+++ b/protobuf/lib/src/protobuf/extension_registry.dart
@@ -27,13 +27,8 @@ class ExtensionRegistry {
 
   /// Retrieves an extension from the registry that adds tag number [tagNumber]
   /// to the [messageName] message type.
-  Extension? getExtension(String messageName, int tagNumber) {
-    final map = _extensions[messageName];
-    if (map != null) {
-      return map[tagNumber];
-    }
-    return null;
-  }
+  Extension? getExtension(String messageName, int tagNumber) =>
+      _extensions[messageName]?[tagNumber];
 
   /// Creates a shallow copy of [message], with all extensions in `this` parsed
   /// from the unknown fields of [message] and of every nested submessage.