lib/wyrm/pump.rb in wyrm-0.3.0 vs lib/wyrm/pump.rb in wyrm-0.3.1

@@ -47,14 +47,15 @@
 
     # add extensions
     @db.extension :pagination
 
     # turn on postgres streaming if available
-    if defined?( Sequel::Postgres ) && defined?(Sequel::Postgres.supports_streaming?) && Sequel::Postgres.supports_streaming?
-      logger.debug "Streaming for postgres"
+    # also gets called for non-postgres dbs, but that seems to be fine.
+    if defined?( Sequel::Postgres ) && @db.is_a?(Sequel::Postgres::Database) && defined?(Sequel::Postgres.supports_streaming?) && Sequel::Postgres.supports_streaming?
       @db.extension :pg_streaming
+      logger.info "Streaming for #{@db.uri}"
     else
-      logger.info "No streaming for postgres"
+      logger.info "No streaming for #{@db.uri}"
     end
   end
 
   # return an object that responds to ===
   # which returns true if ==='s parameter