Contents

require 'oj'
require 'active_support/concern'
require 'pg_serializable/errors'
require 'pg_serializable/visitable'
require 'pg_serializable/nodes'
require 'pg_serializable/trait_manager'
require 'pg_serializable/trait'
require 'pg_serializable/visitors'

Oj.mimic_JSON()

module ActiveRecord
  class Relation
    include PgSerializable::Visitable

    def json(trait: :default)
      to_pg_json accept(PgSerializable::Visitors::Json.new, trait: trait)
    end
  end
end

module PgSerializable
  extend ActiveSupport::Concern

  included do
    include Visitable

    def json(trait: :default)
      self.class.to_pg_json accept(PgSerializable::Visitors::Json.new, trait: trait)
    end
  end

  class_methods do
    def json(trait: :default)
      to_pg_json accept(PgSerializable::Visitors::Json.new, trait: trait)
    end

    def serializable(&blk)
      trait_manager.instance_eval &blk
      begin
        validate_traits!
      rescue ActiveRecord::NoDatabaseError
      end
    end

    def trait_manager
      @trait_manager ||= TraitManager.new(self)
    end

    def accept visitor, **kwargs
      visitor.visit self, **kwargs
    end

    def to_pg_json(scope)
      res = scope.as_json.first
      ::Oj.dump(res['coalesce'] || res['json_build_object'])
    end

    private

    delegate :validate_traits!, to: :trait_manager
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
pg_serializable-1.3.0	lib/pg_serializable.rb
pg_serializable-1.2.2	lib/pg_serializable.rb