Class: Gapic::Schema::Loader

Inherits:
Object
  • Object
show all
Defined in:
lib/gapic/schema/loader.rb

Overview

Loader

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Loader

Initializes the loader



30
31
32
33
 
# File 'lib/gapic/schema/loader.rb', line 30

def initialize
  @prior_messages = []
  @prior_enums = []
end

Instance Method Details

#cached_enum(type_name) ⇒ Enum | nil

Retrieves an Enum if it has been seen before.

Parameters:

  • type_name (String) —

    the type name of the enum.

Returns:

  • (Enum | nil) —

    The enum if it has already been seen or nil if no enum can be found.



240
241
242
 
# File 'lib/gapic/schema/loader.rb', line 240

def cached_enum type_name
  cached_obj @prior_enums, type_name
end

#cached_message(type_name) ⇒ Enum | nil

Retrieves a Message if it has been seen before.

Parameters:

  • type_name (String) —

    the type name of the message.

Returns:

  • (Enum | nil) —

    The message if it has already been seen or nil if no message can be found.



249
250
251
 
# File 'lib/gapic/schema/loader.rb', line 249

def cached_message type_name
  cached_obj @prior_messages, type_name
end

#load_enum(registry, descriptor, address, docs, path) ⇒ Enum

Loads an enum.

Parameters:

  • descriptor (Google::Protobuf::EnumDescriptorProto) —

    the descriptor of this enum.

  • address (Enumerable<String>) —

    The address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:

  • (Enum) —

    The loaded enum.



96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/gapic/schema/loader.rb', line 96

def load_enum registry, descriptor, address, docs, path
  # Update Address.
  address = address.clone << descriptor.name

  # Load Enum Values
  values = descriptor.value.each_with_index.map do |value, i|
    load_enum_value registry, value, address, docs, path + [2, i]
  end

  # Construct, cache and return enum.
  enum = Enum.new descriptor, address, docs[path], values
  @prior_enums << enum
  add_to_registry registry, address, enum
end

#load_enum_value(registry, descriptor, address, docs, path) ⇒ EnumValue

Loads an enum value.

Parameters:

  • descriptor (Google::Protobuf::EnumValueDescriptorProto) —

    the descriptor of this enum value.

  • address (Enumerable<String>) —

    the address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:



122
123
124
125
126
127
128
129
 
# File 'lib/gapic/schema/loader.rb', line 122

def load_enum_value registry, descriptor, address, docs, path
  # Update Address.
  address = address.clone << descriptor.name

  # Construct and return value.
  enum_value = EnumValue.new descriptor, address, docs[path]
  add_to_registry registry, address, enum_value
end

#load_field(registry, descriptor, address, docs, path) ⇒ Field

Loads a field.

Parameters:

  • descriptor (Google::Protobuf::FieldDescriptorProto) —

    the descriptor of this field.

  • address (Enumerable<String>) —

    The address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:

  • (Field) —

    The loaded field.



179
180
181
182
183
184
185
186
187
 
# File 'lib/gapic/schema/loader.rb', line 179

def load_field registry, descriptor, address, docs, path
  # Update address.
  address = address.clone << descriptor.name

  # Construct and return the field.
  field = Field.new(descriptor, address, docs[path],
                    cached_message(descriptor.type_name), cached_enum(descriptor.type_name))
  add_to_registry registry, address, field
end

#load_file(file_descriptor, file_to_generate) ⇒ Object

Loads a file.

Parameters:



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'lib/gapic/schema/loader.rb', line 41

def load_file file_descriptor, file_to_generate
  # Setup.
  address = file_descriptor.package.split "."
  path = []
  registry = {}

  # Load the docs.
  location = file_descriptor.source_code_info.location
  docs = location.each_with_object({}) { |l, ans| ans[l.path] = l }

  # Load top-level enums.
  enums = file_descriptor.enum_type.each_with_index.map do |e, i|
    load_enum registry, e, address, docs, [5, i]
  end

  # Load top-level messages.
  messages = file_descriptor.message_type.each_with_index.map do |m, i|
    load_message registry, m, address, docs, [4, i]
  end
  messages.each(&method(:update_fields!))

  # Load services.
  services = file_descriptor.service.each_with_index.map do |s, i|
    load_service registry, s, address, docs, [6, i]
  end

  # Construct and return the file.
  File.new file_descriptor, address, docs[path], messages, enums,
           services, file_to_generate, registry
end

#load_message(registry, descriptor, address, docs, path) ⇒ Message

Loads a message. As a side effect, this alters @messages and @enums with the nested messages that are found.

Parameters:

  • descriptor (Google::Protobuf::DescriptorProto) —

    the descriptor of this message.

  • address (Enumerable<String>) —

    the address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:

  • (Message) —

    The loaded message.



143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
 
# File 'lib/gapic/schema/loader.rb', line 143

def load_message registry, descriptor, address, docs, path
  # Update Address.
  address = address.clone << descriptor.name

  # Load Children
  nested_messages = descriptor.nested_type.each_with_index.map do |m, i|
    load_message registry, m, address, docs, path + [3, i]
  end
  nested_enums = descriptor.enum_type.each_with_index.map do |e, i|
    load_enum registry, e, address, docs, path + [4, i]
  end
  fields = descriptor.field.each_with_index.map do |f, i|
    load_field registry, f, address, docs, path + [2, i]
  end
  extensions = descriptor.extension.each_with_index.map do |e, i|
    load_field registry, e, address, docs, path + [6, i]
  end

  # Construct, cache, and return.
  msg = Message.new(descriptor, address, docs[path], fields, extensions,
                    nested_messages, nested_enums)
  @prior_messages << msg
  add_to_registry registry, address, msg
end

#load_method(registry, descriptor, address, docs, path) ⇒ Method

Loads a method.

Parameters:

  • descriptor (Google::Protobuf::MethodDescriptorProto) —

    the descriptor of this service.

  • address (Enumerable<String>) —

    The address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:

  • (Method) —

    The loaded method.



225
226
227
228
229
230
231
232
233
 
# File 'lib/gapic/schema/loader.rb', line 225

def load_method registry, descriptor, address, docs, path
  # Update the address.
  address = address.clone << descriptor.name

  # Construct and return the method.
  method = Method.new(descriptor, address, docs[path],
                      cached_message(descriptor.input_type), cached_message(descriptor.output_type))
  add_to_registry registry, address, method
end

#load_service(registry, descriptor, address, docs, path) ⇒ Service

Loads a service.

Parameters:

  • descriptor (Google::Protobuf::ServiceDescriptorProto) —

    the descriptor of this service.

  • address (Enumerable<String>) —

    The address of the parent.

  • docs (Hash<Enumerable<Integer>, Google::Protobuf::SourceCodeInfo::Location>) —

    A mapping of a path to the docs. See Proto#docs for more info.

  • path (Enumerable<Integer>) —

    The current path. This is used to get the docs for a proto. See Proto#docs for more info.

Returns:

  • (Service) —

    The loaded service.



200
201
202
203
204
205
206
207
208
209
210
211
212
 
# File 'lib/gapic/schema/loader.rb', line 200

def load_service registry, descriptor, address, docs, path
  # Update the address.
  address = address.clone << descriptor.name

  # Load children
  methods = descriptor.method.each_with_index.map do |m, i|
    load_method registry, m, address, docs, path + [2, i]
  end

  # Construct and return the service.
  service = Service.new descriptor, address, docs[path], methods
  add_to_registry registry, address, service
end

#update_fields!(message) ⇒ Object

Updates the fields of a message and it's nested messages.

Parameters:

  • message (Message) —

    the message whose fields and nested messages to update.



76
77
78
79
80
81
82
83
 
# File 'lib/gapic/schema/loader.rb', line 76

def update_fields! message
  message.nested_messages.each(&method(:update_fields!))
  non_primitives = message.fields.reject { |f| f.type_name.empty? }
  non_primitives.each do |f|
    f.message ||= cached_message f.type_name
    f.enum ||= cached_enum f.type_name
  end
end