fly-ruby 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ea1ca64bcd11ed8a60774f0d8ec255ffb49bc6b8e21a8bd32e7410068bec35c0
-  data.tar.gz: fb55ad6c98cb631bf202aea206c53a9b2e9a7c738cd9835095dd647011cfc7f6
+  metadata.gz: bc19c3c6ff9a2dd0cfdc09640fa1bfd3a4eeeb60ff9be84774cdda50b8956faf
+  data.tar.gz: 44646945289997faed2261594766a55ae2c343f272c7aa85bedb116cb869037a
 SHA512:
-  metadata.gz: 7c3df04f8bb9f97bfb87903b0fffc14ef2b320802e1183752c07f8b769b629f6c66af21ae0e4bb5a23a2608cfdfdda2f3b1c387e3edcd1f64b5d2474a79f7a6f
-  data.tar.gz: ff4ca7c18d4b18ec237c8e1c59bce88dce849198934ed1d42c28ab2a081b577f8fa2cc38930d0478ec94a5bee580e2f12649b38d3ad46d176a774edaba60446c
+  metadata.gz: 17454e355c0f56c595fc3f01548b9bad679545d6d4da74ae20f813a441c878db6234ddee67cb67e2327449d18de6339accd9e14277103b036d1a7e9a767196ef
+  data.tar.gz: 065b6d3473c02437a1123751524a4812c8bc7bb5b1da0736f095abdf5c9b4437a13709b5e5a63b0fd8d241c52160cbe20a99e8117fc3e0a164f4cf665d3978d1

data/.github/workflows/test.yml

@@ -7,8 +7,6 @@ jobs:
     strategy:
       matrix:
         include:
-          - { os: ubuntu-latest, ruby_version: 2.5 }
-          - { os: ubuntu-latest, ruby_version: 2.6 }
           - { os: ubuntu-latest, ruby_version: 2.7 }
           - { os: ubuntu-latest, ruby_version: '3.0' }
     services:

data/Gemfile

@@ -6,6 +6,7 @@ gem 'rack-test'
 gem 'minitest'
 gem "rails"
 gem "pg"
+gem "sqlite3"
 gem "climate_control"
 gem "minitest-around"
 gem "m"

data/lib/fly-ruby/configuration.rb

@@ -62,12 +62,12 @@ module Fly
 
     def database_uri
       @database_uri ||= URI.parse(database_url)
-      @database_uri
     end
 
     def regional_database_url
       uri = database_uri.dup
       uri.host = regional_database_host
+      uri.port = regional_database_port
       uri.to_s
     end
 
@@ -75,13 +75,15 @@ module Fly
       "#{current_region}.#{database_uri.hostname}"
     end
 
-    # Rails-compatible database configuration
-    def regional_database_config
-      {
-        :host => regional_database_host,
-        :port => 5433,
-        :adapter => "postgresql"
-      }
+    def regional_database_port
+      port = if in_secondary_region?
+        case database_uri.scheme
+        when "postgres"
+          5433
+        end
+      end
+
+      port || database_uri.port
     end
 
     def redis_uri
@@ -103,6 +105,10 @@ module Fly
       database_url && primary_region && current_region && web?
     end
 
+    def hijack_database_connection?
+      in_secondary_region? && !database_uri.scheme.start_with?("sqlite")
+    end
+
     def in_secondary_region?
       primary_region && primary_region != current_region
     end

data/lib/fly-ruby/railtie.rb

@@ -1,19 +1,7 @@
 # frozen_string_literal: true
 
 class Fly::Railtie < Rails::Railtie
-  def hijack_database_connection
-    ActiveSupport::Reloader.to_prepare do
-      # If we already have a database connection when this initializer runs,
-      # we should reconnect to the region-local database. This may need some additional
-      # hooks for forking servers to work correctly.
-      if defined?(ActiveRecord)
-        config = ActiveRecord::Base.connection_db_config.configuration_hash
-        ActiveRecord::Base.establish_connection(config.symbolize_keys.merge(Fly.configuration.regional_database_config))
-      end
-    end
-  end
-
-  initializer("fly.regional_database") do |app|
+  initializer("fly.regional_database", before: "active_record.initialize_database") do |app|
     # Insert the request middleware high in the stack, but after static file delivery
     app.config.middleware.insert_after ActionDispatch::Executor, Fly::Headers if Fly.configuration.web?
 
@@ -22,8 +10,8 @@ class Fly::Railtie < Rails::Railtie
       # Insert the database exception handler at the bottom of the stack to take priority over other exception handlers
       app.config.middleware.use Fly::RegionalDatabase::DbExceptionHandlerMiddleware
 
-      if Fly.configuration.in_secondary_region?
-        hijack_database_connection
+      if Fly.configuration.hijack_database_connection?
+        ENV["DATABASE_URL"] = Fly.configuration.regional_database_url
       end
     elsif Fly.configuration.web?
       puts "Warning: DATABASE_URL, PRIMARY_REGION and FLY_REGION must be set to activate the fly-ruby middleware. Middleware not loaded."

data/lib/fly-ruby/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Fly
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

data/lib/fly-ruby.rb

@@ -22,16 +22,14 @@ module Fly
   end
 
   class Instance
-    attr_accessor :configuration
+    attr_writer :configuration
 
-    def initialize
-      self.configuration = Fly::Configuration.new
+    def configuration
+      @configuration ||= Fly::Configuration.new
     end
 
-    def configure
-      configuration = Fly::Configuration.new
-      yield(configuration) if block_given?
-      self.configuration = configuration
+    def configure(&block)
+      configuration.tap(&block)
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fly-ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Joshua Sierles