1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
# frozen_string_literal: true
module Keeps
module Helpers
class PostgresAi
Error = Class.new(StandardError)
def initialize
raise Error, "No credentials supplied" unless connection_string.present? && password.present?
end
def fetch_background_migration_status(job_class_name)
query = <<~SQL
SELECT id, created_at, updated_at, finished_at, started_at, status, job_class_name,
gitlab_schema, total_tuple_count
FROM batched_background_migrations
WHERE job_class_name = $1::text
SQL
pg_client.exec_params(query, [job_class_name])
end
def fetch_migrated_tuple_count(batched_background_migration_id)
query = <<~SQL
SELECT SUM("batched_background_migration_jobs"."batch_size")
FROM "batched_background_migration_jobs"
WHERE "batched_background_migration_jobs"."batched_background_migration_id" = #{batched_background_migration_id}
AND ("batched_background_migration_jobs"."status" IN (3))
SQL
pg_client.exec_params(query)
end
private
def connection_string
ENV["POSTGRES_AI_CONNECTION_STRING"]
end
def password
ENV["POSTGRES_AI_PASSWORD"]
end
def pg_client
@pg_client ||= PG.connect(connection_string, password: password)
end
end
end
end
|
