lib/prosopite.rb in prosopite-0.2.1 vs lib/prosopite.rb in prosopite-1.0.0

@@ -1,6 +1,5 @@
-require 'pg_query'
 
 module Prosopite
   class NPlusOneQueriesError < StandardError; end
   class << self
     attr_writer :raise,
@@ -69,9 +68,15 @@
 
     def fingerprint(query)
       if ActiveRecord::Base.connection.adapter_name.downcase.include?('mysql')
         mysql_fingerprint(query)
       else
+        begin
+          require 'pg_query'
+        rescue LoadError => e
+          msg = "Could not load the 'pg_query' gem. Add `gem 'pg_query'` to your Gemfile"
+          raise LoadError, msg, e.backtrace
+        end
         PgQuery.fingerprint(query)
       end
     end
 
     # Many thanks to https://github.com/genkami/fluent-plugin-query-fingerprint/