|
|
|
# Protocol Buffers - Google's data interchange format
|
|
|
|
# Copyright 2023 Google Inc. All rights reserved.
|
|
|
|
#
|
|
|
|
# Use of this source code is governed by a BSD-style
|
|
|
|
# license that can be found in the LICENSE file or at
|
|
|
|
# https://developers.google.com/open-source/licenses/bsd
|
|
|
|
|
|
|
|
require 'ffi-compiler/loader'
|
|
|
|
require 'google/protobuf/ffi/ffi'
|
|
|
|
require 'google/protobuf/ffi/internal/type_safety'
|
|
|
|
require 'google/protobuf/ffi/internal/pointer_helper'
|
|
|
|
require 'google/protobuf/ffi/internal/arena'
|
|
|
|
require 'google/protobuf/ffi/internal/convert'
|
|
|
|
require 'google/protobuf/ffi/descriptor'
|
|
|
|
require 'google/protobuf/ffi/enum_descriptor'
|
|
|
|
require 'google/protobuf/ffi/field_descriptor'
|
|
|
|
require 'google/protobuf/ffi/oneof_descriptor'
|
Implement service & method descriptor lookup in Ruby (#15817)
This PR implements lookup of service descriptor and method descriptor objects in Ruby as described in issue https://github.com/protocolbuffers/protobuf/issues/14891.
It contains three implementations - one for the CRuby extension API, one for JRuby, and one for FFI.
With this patch,
* `DescriptorPool#lookup('fully.qualified.service.name')` works and returns a `Google::Protobuf::ServiceDescriptor` object
* You can call `#options` on that to get the service options
* You can call `#methods` on that to get the services' methods as `Google::Protobuf::MethodDescriptor` objects,
* You can call `MethodDescriptor#options` to get method options
* You can also get the streaming flags & input/output types of the method with `#input_type`, `#output_type`, `#client_streaming`, and `#server_streaming`.
In order to make the FFI implementation work, I had to mark some more methods in the UPB header as exported - I guess that's something which will have to be done on the UPB side, like this https://github.com/protocolbuffers/upb/commit/01fed1cc1ba255bf22b49393ba054b8d270e6ba3
CC @dazuma & @haberman from the original issue, and @JasonLunn (since you work on protobuf it seems - small world!)
I apologies for the large volume of copy-pasta'd code from the existing descriptor class implementations into the new ones - I felt this was probably better than designing new abstractions to reduce it off the bat though; this feels like it "fits in" with the existing implementation.
Closes #15817
COPYBARA_INTEGRATE_REVIEW=https://github.com/protocolbuffers/protobuf/pull/15817 from KJTsanaktsidis:ktsanaktsidis/add_service_method_descriptors 54d72184310d7d9bfc06d4a58956e3871170c43d
PiperOrigin-RevId: 618221016
9 months ago
|
|
|
require 'google/protobuf/ffi/method_descriptor'
|
|
|
|
require 'google/protobuf/ffi/service_descriptor'
|
|
|
|
require 'google/protobuf/ffi/descriptor_pool'
|
|
|
|
require 'google/protobuf/ffi/file_descriptor'
|
|
|
|
require 'google/protobuf/ffi/map'
|
|
|
|
require 'google/protobuf/ffi/object_cache'
|
|
|
|
require 'google/protobuf/ffi/repeated_field'
|
|
|
|
require 'google/protobuf/ffi/message'
|
|
|
|
|
|
|
|
module Google
|
|
|
|
module Protobuf
|
|
|
|
def self.deep_copy(object)
|
|
|
|
case object
|
|
|
|
when RepeatedField
|
|
|
|
RepeatedField.send(:deep_copy, object)
|
|
|
|
when Google::Protobuf::Map
|
|
|
|
Google::Protobuf::Map.deep_copy(object)
|
|
|
|
when Google::Protobuf::MessageExts
|
|
|
|
object.class.send(:deep_copy, object.instance_variable_get(:@msg))
|
|
|
|
else
|
|
|
|
raise NotImplementedError
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.discard_unknown(message)
|
|
|
|
raise FrozenError if message.frozen?
|
|
|
|
raise ArgumentError.new "Expected message, got #{message.class} instead." if message.instance_variable_get(:@msg).nil?
|
|
|
|
unless Google::Protobuf::FFI.message_discard_unknown(message.instance_variable_get(:@msg), message.class.descriptor, 128)
|
|
|
|
raise RuntimeError.new "Messages nested too deeply."
|
|
|
|
end
|
|
|
|
nil
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|