Auto/eager load constants for convenience

Shopify · Jan 15, 2025 · cafa5bd · cafa5bd
1 parent b520393
commit cafa5bd
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/lib/protoboeuf.rb b/lib/protoboeuf.rb
@@ -1,4 +1,6 @@
 # frozen_string_literal: true
 
-module Protoboeuf
+module ProtoBoeuf
+  autoload :CodeGen, "protoboeuf/codegen"
+  autoload :Google, "protoboeuf/google"
 end
diff --git a/lib/protoboeuf/google.rb b/lib/protoboeuf/google.rb
@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+# There isn't a clean 1:1 mapping between constants and *.rb files, so eager load instead of autoload.
+
+Dir[File.expand_path("**/*.rb", __dir__)].each { |file| require file }
diff --git a/test/gem_test.rb b/test/gem_test.rb
@@ -14,9 +14,10 @@ def test_can_be_required
 
         ::ProtoBoeuf
 
-        require "protoboeuf/codegen"
-
+        # The following should auto/eagerload
         ::ProtoBoeuf::CodeGen
+        ::ProtoBoeuf::Google::Api::FieldBehavior
+        ::ProtoBoeuf::Google::Protobuf::Any
 
         exit 0
       RUBY