feat(6178): support enums as integers in jruby

protocolbuffers · Feb 2, 2023 · 1163142 · 1163142
1 parent 9d81de2
commit 1163142
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ruby/src/main/java/com/google/protobuf/jruby/RubyMessage.java b/ruby/src/main/java/com/google/protobuf/jruby/RubyMessage.java
@@ -677,6 +677,7 @@ public static IRubyObject decode(ThreadContext context, IRubyObject recv, IRubyO
    * @param options [Hash] options for the decoder
    *  preserve_proto_fieldnames: set true to use original fieldnames (default is to camelCase)
    *  emit_defaults: set true to emit 0/false values (default is to omit them)
+   *  enums_as_integers: set true to enmit enum values as integer (default is string)
    */
   @JRubyMethod(name = "encode_json", required = 1, optional = 1, meta = true)
   public static IRubyObject encodeJson(
@@ -698,6 +699,7 @@ public static IRubyObject encodeJson(
 
       IRubyObject emitDefaults = options.fastARef(runtime.newSymbol("emit_defaults"));
       IRubyObject preserveNames = options.fastARef(runtime.newSymbol("preserve_proto_fieldnames"));
+      IRubyObject enumAsIntegers = options.fastARef(runtime.newSymbol("enums_as_integers"));
 
       if (emitDefaults != null && emitDefaults.isTrue()) {
         printer = printer.includingDefaultValueFields();
@@ -706,6 +708,10 @@ public static IRubyObject encodeJson(
       if (preserveNames != null && preserveNames.isTrue()) {
         printer = printer.preservingProtoFieldNames();
       }
+
+      if (enumAsIntegers != null && printingEnumsAsInts.isTrue()) {
+        printer = printer.printingEnumsAsInts();
+      }
     }
     printer =
         printer.usingTypeRegistry(