| Index: third_party/protobuf/ruby/src/main/java/com/google/protobuf/jruby/RubyBuilder.java
|
| diff --git a/third_party/protobuf/ruby/src/main/java/com/google/protobuf/jruby/RubyBuilder.java b/third_party/protobuf/ruby/src/main/java/com/google/protobuf/jruby/RubyBuilder.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..5addae58dad7fd6336e30e7c37c84f6a020e7797
|
| --- /dev/null
|
| +++ b/third_party/protobuf/ruby/src/main/java/com/google/protobuf/jruby/RubyBuilder.java
|
| @@ -0,0 +1,167 @@
|
| +/*
|
| + * Protocol Buffers - Google's data interchange format
|
| + * Copyright 2014 Google Inc. All rights reserved.
|
| + * https://developers.google.com/protocol-buffers/
|
| + *
|
| + * Redistribution and use in source and binary forms, with or without
|
| + * modification, are permitted provided that the following conditions are
|
| + * met:
|
| + *
|
| + * * Redistributions of source code must retain the above copyright
|
| + * notice, this list of conditions and the following disclaimer.
|
| + * * Redistributions in binary form must reproduce the above
|
| + * copyright notice, this list of conditions and the following disclaimer
|
| + * in the documentation and/or other materials provided with the
|
| + * distribution.
|
| + * * Neither the name of Google Inc. nor the names of its
|
| + * contributors may be used to endorse or promote products derived from
|
| + * this software without specific prior written permission.
|
| + *
|
| + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| + */
|
| +
|
| +package com.google.protobuf.jruby;
|
| +
|
| +import org.jruby.*;
|
| +import org.jruby.anno.JRubyClass;
|
| +import org.jruby.anno.JRubyMethod;
|
| +import org.jruby.runtime.*;
|
| +import org.jruby.runtime.builtin.IRubyObject;
|
| +
|
| +@JRubyClass(name = "Builder")
|
| +public class RubyBuilder extends RubyObject {
|
| + public static void createRubyBuilder(Ruby runtime) {
|
| + RubyModule protobuf = runtime.getClassFromPath("Google::Protobuf");
|
| + RubyClass cBuilder = protobuf.defineClassUnder("Builder", runtime.getObject(), new ObjectAllocator() {
|
| + @Override
|
| + public IRubyObject allocate(Ruby runtime, RubyClass klazz) {
|
| + return new RubyBuilder(runtime, klazz);
|
| + }
|
| + });
|
| + cBuilder.defineAnnotatedMethods(RubyBuilder.class);
|
| + }
|
| +
|
| + public RubyBuilder(Ruby runtime, RubyClass metaClass) {
|
| + super(runtime, metaClass);
|
| + this.cDescriptor = (RubyClass) runtime.getClassFromPath("Google::Protobuf::Descriptor");
|
| + this.cEnumDescriptor = (RubyClass) runtime.getClassFromPath("Google::Protobuf::EnumDescriptor");
|
| + this.cMessageBuilderContext = (RubyClass) runtime.getClassFromPath("Google::Protobuf::MessageBuilderContext");
|
| + this.cEnumBuilderContext = (RubyClass) runtime.getClassFromPath("Google::Protobuf::EnumBuilderContext");
|
| + }
|
| +
|
| + /*
|
| + * call-seq:
|
| + * Builder.new => builder
|
| + *
|
| + * Creates a new Builder. A Builder can accumulate a set of new message and enum
|
| + * descriptors and atomically register them into a pool in a way that allows for
|
| + * (co)recursive type references.
|
| + */
|
| + @JRubyMethod
|
| + public IRubyObject initialize(ThreadContext context) {
|
| + Ruby runtime = context.runtime;
|
| + this.pendingList = runtime.newArray();
|
| + return this;
|
| + }
|
| +
|
| + /*
|
| + * call-seq:
|
| + * Builder.add_message(name, &block)
|
| + *
|
| + * Creates a new, empty descriptor with the given name, and invokes the block in
|
| + * the context of a MessageBuilderContext on that descriptor. The block can then
|
| + * call, e.g., MessageBuilderContext#optional and MessageBuilderContext#repeated
|
| + * methods to define the message fields.
|
| + *
|
| + * This is the recommended, idiomatic way to build message definitions.
|
| + */
|
| + @JRubyMethod(name = "add_message")
|
| + public IRubyObject addMessage(ThreadContext context, IRubyObject name, Block block) {
|
| + RubyDescriptor msgdef = (RubyDescriptor) cDescriptor.newInstance(context, Block.NULL_BLOCK);
|
| + IRubyObject ctx = cMessageBuilderContext.newInstance(context, msgdef, this, Block.NULL_BLOCK);
|
| + msgdef.setName(context, name);
|
| + if (block.isGiven()) {
|
| + if (block.arity() == Arity.ONE_ARGUMENT) {
|
| + block.yield(context, ctx);
|
| + } else {
|
| + Binding binding = block.getBinding();
|
| + binding.setSelf(ctx);
|
| + block.yieldSpecific(context);
|
| + }
|
| + }
|
| + this.pendingList.add(msgdef);
|
| + return context.runtime.getNil();
|
| + }
|
| +
|
| + /*
|
| + * call-seq:
|
| + * Builder.add_enum(name, &block)
|
| + *
|
| + * Creates a new, empty enum descriptor with the given name, and invokes the block in
|
| + * the context of an EnumBuilderContext on that descriptor. The block can then
|
| + * call EnumBuilderContext#add_value to define the enum values.
|
| + *
|
| + * This is the recommended, idiomatic way to build enum definitions.
|
| + */
|
| + @JRubyMethod(name = "add_enum")
|
| + public IRubyObject addEnum(ThreadContext context, IRubyObject name, Block block) {
|
| + RubyEnumDescriptor enumDef = (RubyEnumDescriptor) cEnumDescriptor.newInstance(context, Block.NULL_BLOCK);
|
| + IRubyObject ctx = cEnumBuilderContext.newInstance(context, enumDef, Block.NULL_BLOCK);
|
| + enumDef.setName(context, name);
|
| +
|
| + if (block.isGiven()) {
|
| + if (block.arity() == Arity.ONE_ARGUMENT) {
|
| + block.yield(context, ctx);
|
| + } else {
|
| + Binding binding = block.getBinding();
|
| + binding.setSelf(ctx);
|
| + block.yieldSpecific(context);
|
| + }
|
| + }
|
| +
|
| + this.pendingList.add(enumDef);
|
| + return context.runtime.getNil();
|
| + }
|
| +
|
| + /*
|
| + * call-seq:
|
| + * Builder.finalize_to_pool(pool)
|
| + *
|
| + * Adds all accumulated message and enum descriptors created in this builder
|
| + * context to the given pool. The operation occurs atomically, and all
|
| + * descriptors can refer to each other (including in cycles). This is the only
|
| + * way to build (co)recursive message definitions.
|
| + *
|
| + * This method is usually called automatically by DescriptorPool#build after it
|
| + * invokes the given user block in the context of the builder. The user should
|
| + * not normally need to call this manually because a Builder is not normally
|
| + * created manually.
|
| + */
|
| + @JRubyMethod(name = "finalize_to_pool")
|
| + public IRubyObject finalizeToPool(ThreadContext context, IRubyObject rbPool) {
|
| + RubyDescriptorPool pool = (RubyDescriptorPool) rbPool;
|
| + for (int i = 0; i < this.pendingList.size(); i++) {
|
| + IRubyObject defRb = this.pendingList.entry(i);
|
| + if (defRb instanceof RubyDescriptor) {
|
| + pool.addToSymtab(context, (RubyDescriptor) defRb);
|
| + } else {
|
| + pool.addToSymtab(context, (RubyEnumDescriptor) defRb);
|
| + }
|
| + }
|
| + this.pendingList = context.runtime.newArray();
|
| + return context.runtime.getNil();
|
| + }
|
| +
|
| + protected RubyArray pendingList;
|
| + private RubyClass cDescriptor, cEnumDescriptor, cMessageBuilderContext, cEnumBuilderContext;
|
| +}
|
|
|