require 'flydata/source_mysql/mysql_accessible'

module Flydata
  module SourceMysql
    class SyncDatabaseSizeCheck
      include MysqlAccessible

      SIZE_CHECK_QUERY = <<EOT
SELECT
  SUM(data_length) bytesize
FROM
  information_schema.tables
WHERE
  table_schema = '%s' AND table_name in (%s);
EOT

      def initialize(de_conf)
        super
        @query = SIZE_CHECK_QUERY % [
          de_conf['database'],
          tables = de_conf['tables'].collect{|t| "'#{t}'"}.join(',')]
      end

      def get_db_bytesize
        exec_mysql_query(@query).first['bytesize'].to_i
      end
    end
  end
end