koa-utils 0.0.8 → 0.0.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (3) hide show
  1. checksums.yaml +4 -4
  2. data/lib/koa-utils/health-check.rb +14 -6
  3. metadata +1 -1
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA1:
3
- metadata.gz: 5b6b0f306929583b66a794b3ab28a236b4033426
4
- data.tar.gz: 0e8c972e56b535beb8cd26b52a4983eea9b6f08f
3
+ metadata.gz: 44d367611792021c6ccb24e9481033a15c957fa3
4
+ data.tar.gz: b098b35bf9aadf96449c0f80c109e236865f7d98
5
5
  SHA512:
6
- metadata.gz: be1959550ee1e0cc8c5c71374eb8fc962ae6a3dc83d649ea9cf362f89dbcaa5ce6187f9b652de90c8ba78a98cac935bcd517998216afb71e8a0aa55526fd0770
7
- data.tar.gz: 42fb3008d33dc0e9af75354e913485254fb95173a55bc2a2035d381e258407c416fc1e6a1b598b8c0d8c0ecf157d76fd76bc1712d68a056625301932a51cb174
6
+ metadata.gz: 0224a865f0e7a23c517f4dea81ccb6c9e9a6e66b640164a15a842b044d2024dfd4bf42242e3b2acb643c456b349d908aab132be7413fe9b1f88c3c3dc1c7265b
7
+ data.tar.gz: afa0778b449961f71b50007278199b99e637ed374cb2ce64b30108f6f3dda047213a049b3d58e073344f8aa4d145b545e25dc31c8fe88e3f3f66d551a86eb665
@@ -1,19 +1,27 @@
1
1
  require 'json'
2
2
  require 'timeout'
3
3
 
4
- if !defined?(ActiveRecord)
5
- raise("KOAUtils::HealthCheck expects ActiveRecord to be defined.")
6
- end
7
-
8
4
  class KOAUtils::HealthCheck
9
5
  def self.call(env)
10
6
  begin
11
7
  Timeout::timeout(2) do
12
- ActiveRecord::Base.connection.execute("SELECT * from users limit 1")
8
+ db_check
13
9
  end
14
10
  rescue => e
15
- return [500, {"Content-Type" => "text/json"}, [JSON.dump({error: e.message})]]
11
+ return [500, {"Content-Type" => "text/json"},
12
+ [JSON.dump({error: e.message})]]
16
13
  end
17
14
  [200, {"Content-Type" => "text/json"}, [JSON.dump({info: Time.now})]]
18
15
  end
16
+
17
+ def self.db_check
18
+ query = "select * from information_schema.tables"
19
+ if defined?(ActiveRecord)
20
+ ActiveRecord::Base.connection.execute(query)
21
+ elsif defined?(Sequel)
22
+ c = Sequel.connect(ENV["DATABASE_URL"])
23
+ c.exec(query)
24
+ c.disconnect
25
+ end
26
+ end
19
27
  end
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: koa-utils
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.0.8
4
+ version: 0.0.9
5
5
  platform: ruby
6
6
  authors:
7
7
  - Eric Rykwalder