lutaml 0.10.2 → 0.10.3

data/lib/lutaml/xmi/parsers/xmi_base.rb

@@ -485,9 +485,7 @@ module Lutaml
         # @return [Array<Hash>]
         # @note xpath %(//diagrams/diagram/model[@package="#{node['xmi:id']}"])
         def serialize_model_diagrams(node_id, with_package: false) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
-          diagrams = @xmi_root_model.extension.diagrams.diagram.select do |d|
-            d.model.package == node_id
-          end
+          diagrams = diagram_lookup[node_id]
 
           diagrams.map do |diagram|
             h = {
@@ -506,6 +504,17 @@ module Lutaml
           end
         end
 
+        # Lazy-built hash index for O(1) diagram lookups by package
+        # @return [Hash] Mapping of package_id => [diagrams]
+        def diagram_lookup
+          @diagram_lookup ||= begin
+            idx = Hash.new { |h, k| h[k] = [] }
+            diagrams = @xmi_root_model.extension&.diagrams&.diagram || []
+            diagrams.each { |d| idx[d.model.package] << d if d.model&.package }
+            idx
+          end
+        end
+
         # @param xmi_id [String]
         # @return [Array<Hash>]
         # @note xpath %(//element[@xmi:idref="#{xmi_id}"]/links/*)
@@ -932,8 +941,22 @@ module Lutaml
         # @param source_or_target [String]
         # @return [String]
         def connector_node_by_id(xmi_id, source_or_target)
-          @xmi_root_model.extension.connectors.connector.find do |con|
-            con.send(source_or_target.to_sym).idref == xmi_id
+          connector_lookup[[source_or_target.to_sym, xmi_id]]
+        end
+
+        # Lazy-built hash index for O(1) connector lookups
+        # @return [Hash] Mapping of [direction, idref] => connector
+        def connector_lookup
+          @connector_lookup ||= begin
+            lookup = {}
+            connectors = @xmi_root_model.extension&.connectors&.connector || []
+            connectors.each do |con|
+              %i[source target].each do |dir|
+                idref = con.send(dir)&.idref
+                lookup[[dir, idref]] = con if idref
+              end
+            end
+            lookup
           end
         end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lutaml
 version: !ruby/object:Gem::Version
-  version: 0.10.2
+  version: 0.10.3
 platform: ruby
 authors:
 - Ribose Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-04-23 00:00:00.000000000 Z
+date: 2026-04-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: expressir