From 9a25e98cc41f20571fb3e6f63663278915f5b6c3 Mon Sep 17 00:00:00 2001
From: tony castiglione <tony@kennasecurity.com>
Date: Thu, 21 Jun 2018 21:26:31 +0000
Subject: [PATCH] Update how Dataset gets set for the Sequel gem, it changed
 recently

---
 lib/sequel/adapters/redshift.rb | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/lib/sequel/adapters/redshift.rb b/lib/sequel/adapters/redshift.rb
index bb4b0ee..2bd5fd1 100644
--- a/lib/sequel/adapters/redshift.rb
+++ b/lib/sequel/adapters/redshift.rb
@@ -41,6 +41,10 @@ def supports_index_parsing?
 
       private
 
+      def dataset_class_default
+        Dataset
+      end
+
       def type_literal_generic_string(column)
         super(column.merge(text: false))
       end
@@ -128,8 +132,6 @@ def schema_parse_table(table_name, opts)
     end
 
     class Dataset < Postgres::Dataset
-      Database::DatasetClass = self
-
       # Redshift doesn't support RETURNING statement
       def insert_returning_sql(sql)
         # do nothing here