Sha256: e923d3508fc620723b60b3712b07aa135b60a4290b02e5c8252987b735d182d5
Contents?: true
Size: 1.59 KB
Versions: 34
Compression:
Stored size: 1.59 KB
Contents
require 'flydata/table_meta'
require 'mysql2'
require 'flydata-core/mysql/config'
require 'flydata-core/table_def/mysql_table_def'
module Flydata
module SourceMysql
class TableMeta < Flydata::TableMeta
MANDATORY_OPTS = [
:host, :port, :username, :password,
:database, :tables,
]
OPTIONAL_OPTS = [
:ssl_ca
]
GET_TABLE_META_SQL = <<EOT
SELECT
T.table_name as table_name,
CCSA.character_set_name as character_set_name
FROM
information_schema.`TABLES` T,
information_schema.`COLLATION_CHARACTER_SET_APPLICABILITY` CCSA
WHERE
CCSA.collation_name = T.table_collation
AND T.table_schema = '%{database}'
AND T.table_name in (%{tables});
EOT
def initialize(opts)
missing_opts = MANDATORY_OPTS - opts.keys
raise "Mandatory option(s) are missing: #{missing_opts.join(', ')}" unless (missing_opts.empty?)
@db_opts = FlydataCore::Mysql::Config.build_mysql_db_opts(opts)
@database = opts[:database]
@tables = opts[:tables]
@table_meta = Hash.new{|h, k| h[k] = {}}
end
def reload
conn = Mysql2::Client.new(@db_opts)
sql = GET_TABLE_META_SQL % {
database: @database, tables: @tables.collect{|t| "'#{t}'"}.join(',') }
columns = conn.query(sql)
columns.collect do |col|
mysql_charset = col['character_set_name']
@table_meta[col['table_name'].to_sym][:encoding] = FlydataCore::TableDef::MysqlTableDef.ruby_encoding(mysql_charset)
@table_meta[col['table_name'].to_sym][:mysql_charset] = mysql_charset
end
ensure
conn.close rescue nil if conn
end
end
end
end
Version data entries
34 entries across 34 versions & 1 rubygems