58 lines
1.5 KiB
Ruby
58 lines
1.5 KiB
Ruby
module JamRuby
|
|
|
|
def strip_quotes str
|
|
return nil if str.nil?
|
|
|
|
if str.start_with? '"'
|
|
str = str[1..-1]
|
|
end
|
|
|
|
if str.end_with? '"'
|
|
str = str.chop
|
|
end
|
|
|
|
return str
|
|
end
|
|
|
|
# creates messages (implementation: protocol buffer) objects cleanly
|
|
class Database
|
|
|
|
@@log = Logging.logger[Database]
|
|
|
|
#def self.db_timezone
|
|
# @@db_timezone ||= TZInfo::Timezone.get(fetch_db_timezone)
|
|
#end
|
|
|
|
def self.db_timezone
|
|
@@db_timezone ||= ActiveSupport::TimeZone.new(fetch_db_timezone)
|
|
end
|
|
|
|
def self.fetch_db_timezone
|
|
result = ActiveRecord::Base.connection.execute('show timezone')
|
|
tz = result.getvalue(0, 0)
|
|
result.clear
|
|
tz
|
|
end
|
|
|
|
def self.copy(table_name, file)
|
|
@@log.debug("issuing COPY to #{table_name} from #{file}")
|
|
|
|
raw = GeoIpBlocks.connection.raw_connection
|
|
|
|
result = raw.copy_data "COPY #{table_name} FROM STDIN" do
|
|
File.open(file, 'r').each do |line|
|
|
raw.put_copy_data line
|
|
end
|
|
end
|
|
|
|
count = GeoIpBlocks.connection.select_value("select count(*) from #{table_name}").to_i
|
|
ActiveRecord::Base.logger.debug "loaded #{count} records into #{table_name}"
|
|
end
|
|
|
|
def self.copy_table(table_name)
|
|
copied_name = "#{table_name}_copied"
|
|
GeoIpBlocks.connection.execute "CREATE TABLE #{copied_name} (LIKE #{table_name} INCLUDING DEFAULTS INCLUDING CONSTRAINTS INCLUDING COMMENTS INCLUDING STORAGE)"
|
|
copied_name
|
|
end
|
|
end
|
|
end |