Class: Baza::Driver::Mysql::Sql::Column

Inherits:
Object
  • Object
show all
Defined in:
lib/baza/driver/mysql/sql/column.rb

Constant Summary collapse

DATA_SQL_ALLOWED_KEYS = 
[:type, :maxlength, :name, :primarykey, :autoincr, :default, :comment, :after, :first, :storage, :null, :renames].freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(data) ⇒ Column

Returns a new instance of Column.



6
7
8
 
# File 'lib/baza/driver/mysql/sql/column.rb', line 6

def initialize(data)
  @data = data
end

Instance Attribute Details

#data ⇒ Object (readonly)

Returns the value of attribute data.



4
5
6
 
# File 'lib/baza/driver/mysql/sql/column.rb', line 4

def data
  @data
end

Instance Method Details

#sql ⇒ Object 



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
 
# File 'lib/baza/driver/mysql/sql/column.rb', line 10

def sql
  data.each_key do |key|
    raise "Invalid key: '#{key}' (#{key.class.name})." unless DATA_SQL_ALLOWED_KEYS.include?(key)
  end

  raise "No type given." unless data[:type]
  type = data[:type].to_sym

  data[:maxlength] = 255 if type == :varchar && data[:maxlength].to_s.strip.empty?

  sql = "#{Baza::Driver::Mysql::SEPARATOR_COLUMN}#{Baza::Driver::Mysql.escape_column(data.fetch(:name))}#{Baza::Driver::Mysql::SEPARATOR_COLUMN} #{type}"
  sql << "(#{data[:maxlength]})" if data[:maxlength]
  sql << " PRIMARY KEY" if data[:primarykey]
  sql << " AUTO_INCREMENT" if data[:autoincr]
  sql << " NOT NULL" if data.key?(:null) && !data[:null]

  if data.key?(:default_func)
    sql << " DEFAULT #{data[:default_func]}"
  elsif data.key?(:default) && !data[:default].nil?
    sql << " DEFAULT #{Baza::Driver::Mysql.sqlval(data.fetch(:default))}"
  end

  sql << " COMMENT '#{Baza::Driver::Mysql.escape(data.fetch(:comment))}'" if data.key?(:comment)
  sql << " AFTER #{Baza::Driver::Mysql::SEPARATOR_COLUMN}#{Baza::Driver::Mysql.escape_column(data.fetch(:after))}#{Baza::Driver::Mysql::SEPARATOR_COLUMN}" if data[:after] && !data[:first]
  sql << " FIRST" if data[:first]
  sql << " STORAGE #{data[:storage].to_s.upcase}" if data[:storage]

  [sql]
end