diff --git a/README.md b/README.md index 4121dfe2..bb976b28 100644 --- a/README.md +++ b/README.md @@ -94,6 +94,22 @@ member when the resource names are included in the `include` option. render @posts, include: 'authors,comments' ``` +### Specify a serializer + +If you wish to use a serializer other than the default, you can explicitly pass it to the renderer. + +#### 1. For a resource: + +```ruby + render json: @post, serializer: PostPreviewSerializer +``` + +#### 2. For an array resource: + +```ruby + render json: @posts, serializer: PaginatedSerializer, each_serializer: PostPreviewSerializer +``` + ## Installation Add this line to your application's Gemfile: diff --git a/lib/action_controller/serialization.rb b/lib/action_controller/serialization.rb index 2b460cb4..150b9aa2 100644 --- a/lib/action_controller/serialization.rb +++ b/lib/action_controller/serialization.rb @@ -8,16 +8,25 @@ module ActionController ADAPTER_OPTION_KEYS = [:include, :root] + def get_serializer(resource, options) + @_serializer ||= if (serializer = options.delete :serializer) + options[:serializer] = options.delete :each_serializer + serializer + else + ActiveModel::Serializer.serializer_for(resource) + end + end + [:_render_option_json, :_render_with_renderer_json].each do |renderer_method| define_method renderer_method do |resource, options| - serializer = ActiveModel::Serializer.serializer_for(resource) - if serializer - adapter_opts, serializer_opts = - options.partition { |k, _| ADAPTER_OPTION_KEYS.include? k } + adapter_opts, serializer_opts = + options.partition { |k, _| ADAPTER_OPTION_KEYS.include? k }.map { |h| Hash[h] } + + if (serializer = get_serializer(resource, serializer_opts)) # omg hax - object = serializer.new(resource, Hash[serializer_opts]) - adapter = ActiveModel::Serializer.adapter.new(object, Hash[adapter_opts]) + object = serializer.new(resource, serializer_opts) + adapter = ActiveModel::Serializer.adapter.new(object, adapter_opts) super(adapter, options) else super(resource, options) diff --git a/lib/active_model/serializer/array_serializer.rb b/lib/active_model/serializer/array_serializer.rb index b52a4f51..83693c97 100644 --- a/lib/active_model/serializer/array_serializer.rb +++ b/lib/active_model/serializer/array_serializer.rb @@ -6,7 +6,10 @@ module ActiveModel def initialize(objects, options = {}) @objects = objects.map do |object| - serializer_class = ActiveModel::Serializer.serializer_for(object) + serializer_class = options.fetch( + :serializer, + ActiveModel::Serializer.serializer_for(object) + ) serializer_class.new(object) end end diff --git a/test/action_controller/explicit_serializer_test.rb b/test/action_controller/explicit_serializer_test.rb new file mode 100644 index 00000000..e4bddd04 --- /dev/null +++ b/test/action_controller/explicit_serializer_test.rb @@ -0,0 +1,53 @@ +require 'test_helper' + +module ActionController + module Serialization + class ExplicitSerializerTest < ActionController::TestCase + class MyController < ActionController::Base + def render_using_explicit_serializer + @profile = Profile.new(name: 'Name 1', + description: 'Description 1', + comments: 'Comments 1') + render json: @profile, serializer: ProfilePreviewSerializer + end + + def render_array_using_explicit_serializer + array = [ + Profile.new(name: 'Name 1', + description: 'Description 1', + comments: 'Comments 1'), + Profile.new(name: 'Name 2', + description: 'Description 2', + comments: 'Comments 2') + ] + render json: array, + serializer: PaginatedSerializer, + each_serializer: ProfilePreviewSerializer + end + end + + tests MyController + + def test_render_using_explicit_serializer + get :render_using_explicit_serializer + + assert_equal 'application/json', @response.content_type + assert_equal '{"name":"Name 1"}', @response.body + end + + def test_render_array_using_explicit_serializer + get :render_array_using_explicit_serializer + assert_equal 'application/json', @response.content_type + + expected = { + 'paginated' => [ + { 'name' => 'Name 1' }, + { 'name' => 'Name 2' } + ] + } + + assert_equal expected.to_json, @response.body + end + end + end +end diff --git a/test/fixtures/poro.rb b/test/fixtures/poro.rb index 7d8d57b4..9b52c870 100644 --- a/test/fixtures/poro.rb +++ b/test/fixtures/poro.rb @@ -35,6 +35,12 @@ class ProfileSerializer < ActiveModel::Serializer urls :posts, :comments end +class ProfilePreviewSerializer < ActiveModel::Serializer + attributes :name + + urls :posts, :comments +end + Post = Class.new(Model) Comment = Class.new(Model) Author = Class.new(Model) @@ -67,3 +73,9 @@ BlogSerializer = Class.new(ActiveModel::Serializer) do belongs_to :writer has_many :articles end + +PaginatedSerializer = Class.new(ActiveModel::Serializer::ArraySerializer) do + def json_key + 'paginated' + end +end