optimize

Author: tempacc21

extension.bzl

@@ -3,12 +3,13 @@
 load("@bazel_tools//tools/build_defs/repo:jvm.bzl", "jvm_maven_import_external")
 
 def get_wanted_module(module_ctx):
+    wanted_module = None
     for mod in module_ctx.modules:
         if mod.is_root and hasattr(mod.tags.client, "version"):
             return mod
-    for mod in module_ctx.modules:
-        for tags in mod.tags.client:
-            return mod
+        if wanted_module == None and mod.tags.client:
+            wanted_module = mod
+    return wanted_module
 
 def _openapi_generator_impl(module_ctx):
     wanted_module = get_wanted_module(module_ctx)