33 lines
735 B
Ruby
33 lines
735 B
Ruby
require "big_query"
|
|
|
|
module BigQuery
|
|
class Base
|
|
def query(q)
|
|
client.query(q)
|
|
end
|
|
|
|
def escape(s)
|
|
Regexp.escape(s).gsub(/\\/, '\0\0').gsub(/['"]/, '\\\\\0')
|
|
end
|
|
|
|
def client
|
|
@_client ||= BigQuery::Client.new(
|
|
"json_key" => client_options[:google_key_path],
|
|
"project_id" => google_config["project_id"],
|
|
"dataset" => client_options[:google_data_set]
|
|
)
|
|
end
|
|
|
|
def client_options
|
|
@_client_options ||= {
|
|
google_key_path: "/var/www/danbooru/shared/google-key.json",
|
|
google_data_set: "danbooru_#{Rails.env}"
|
|
}
|
|
end
|
|
|
|
def google_config
|
|
@_google_config ||= JSON.parse(File.read(client_options[:google_key_path]))
|
|
end
|
|
end
|
|
end
|