lib/rb: Add SimpleJSONProtocol

Author: AlexisMontagne

lib/rb/lib/thrift/protocol/json_protocol.rb

@@ -761,9 +761,73 @@ def read_binary
     end
   end
 
+  class SimpleJsonProtocol < JsonProtocol
+    READ_EXCEPTION = ProtocolException.new(
+      ProtocolException::NOT_IMPLEMENTED,
+      'op not implemented'
+    )
+
+    def write_message_begin(name, _type, _seqid)
+      write_json_object_start
+      write_json_string(name)
+    end
+
+    def read_message_begin
+      raise READ_EXCEPTION
+    end
+
+    def write_message_end
+      write_json_object_end
+    end
+
+    def write_field_begin(name, _type, _id)
+      write_json_string(name)
+    end
+
+    def read_field_begin
+      raise READ_EXCEPTION
+    end
+
+    def write_field_end; end
+
+    def write_map_begin(_ktype, _vtype, _size)
+      write_json_object_start
+    end
+
+    def read_map_begin
+      raise READ_EXCEPTION
+    end
+
+    def write_map_end
+      write_json_object_end
+    end
+
+    def write_list_begin(_etype, _size)
+      write_json_array_start
+    end
+
+    def read_list_begin
+      raise READ_EXCEPTION
+    end
+
+    def write_set_begin(_etype, _size)
+      write_json_array_start
+    end
+
+    def read_set_begin
+      raise READ_EXCEPTION
+    end
+  end
+
+  class SimpleJsonProtocolFactory < BaseProtocolFactory
+    def get_protocol(trans)
+      Thrift::SimpleJsonProtocol.new(trans)
+    end
+  end
+
   class JsonProtocolFactory < BaseProtocolFactory
     def get_protocol(trans)
-      return Thrift::JsonProtocol.new(trans)
+      Thrift::JsonProtocol.new(trans)
     end
   end
 end

lib/rb/spec/json_protocol_spec.rb

@@ -21,6 +21,23 @@
 require 'spec_helper'
 
 describe 'JsonProtocol' do
+  describe Thrift::SimpleJsonProtocol do
+    before(:each) do
+      @trans = Thrift::MemoryBufferTransport.new
+      @prot = Thrift::SimpleJsonProtocol.new(@trans)
+    end
+
+    it 'should pretty print object' do
+      SpecNamespace::Hello.new.write(@prot)
+
+      @trans.read(@trans.available).should == '{"greeting":"hello world"}'
+    end
+
+    it 'shound not be able to read message' do
+      @trans.write('{"greeting":"hello world"}')
+      expect {@prot.read_message_begin}.to raise_error(Thrift::ProtocolException)
+    end
+  end
 
   describe Thrift::JsonProtocol do
     before(:each) do