Class: JunglePath::DBModel::Schema

Inherits:
Object
  • Object
show all
Defined in:
lib/jungle_path/db_model/schema.rb

Instance Method Summary collapse

Constructor Details

#initialize(db) ⇒ Schema

Returns a new instance of Schema.



4
5
6
7
 
# File 'lib/jungle_path/db_model/schema.rb', line 4

def initialize db
	# db should be a "Sequel" db object that is connected to some database.
	@db = db
end

Instance Method Details

#create_table(table_class) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/jungle_path/db_model/schema.rb', line 39

def create_table table_class
	# translate column type keys into Sequel methods used by Sequel (DB.create_table).

	# call the Sequel DB.create_table method with a block:
	#puts "\ttable_name: #{table_class.table_name}"
	schema = self
	@db.create_table? table_class.table_name do
		made_pk = false
		table_class.columns.each_value do |column|
			puts "\tcolumn: #{column.name} #{schema.get_sequel_method(column.type)} #{column.foreign_key_table_name}."
			method = schema.get_sequel_method(column.type, (table_class.primary_key_columns.count > 1))
			args_hash = {}
			if column.foreign_key? and column.not_null?
				#args = [column.name, column.foreign_key_table_name, :null=>false]
				args = [column.name, column.foreign_key_table_name]
				args_hash[:null] = false
			elsif column.foreign_key?
				args = [column.name, column.foreign_key_table_name]
			elsif column.primary_key?
				args = [column.name]
				#made_pk = true
			else
				args = [column.name]
			end
			args_hash[:unique] = true if column.unique?
			args_hash[:null] = false if column.not_null?
			args_hash[:type] = schema.get_postgresql_override_type(column.override_type) unless column.override_type == nil
			args_hash[:default] = column.default_value unless column.default_value == nil
			args << args_hash
			puts "\t\tmethod: #{method}. args:#{args}.\n"
			made_pk = true if column.type == :primary_key and table_class.primary_key_columns.count == 1 # should only happen when pk is single column.
			puts "column: #{column.name}, override_type: #{column.override_type}, override_type.class: #{column.override_type.class}, method: #{method}, args: #{args}."
			send(method, *args)

			#index:
			if column.index
				send('index', column.index)
			end

			#unique index:
			if column.unique_index
				send('unique', column.unique_index)
			end
		end

		if not made_pk
			#puts "creating primary_key for table #{table_class.name}. keys: #{table_class.primary_key_columns.keys}."
			send('primary_key', table_class.primary_key_columns.keys, :name=>"#{table_class.table_name}_pk".to_sym)
		end
		#puts "end of block"
	end
end

#create_table!(table_class) ⇒ Object 



92
93
94
 
# File 'lib/jungle_path/db_model/schema.rb', line 92

def create_table! table_class
	nil
end

#drop_table(table_class) ⇒ Object 



96
97
98
 
# File 'lib/jungle_path/db_model/schema.rb', line 96

def drop_table table_class
	@db.drop_table? table_class.table_name
end

#get_postgresql_override_type(type) ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/jungle_path/db_model/schema.rb', line 25

def get_postgresql_override_type type
	return 'text' if type == :string
	return 'integer' if type == :integer
	return 'bigint' if type == :bigint
	return 'timestamp' if type == :timestamp
	return 'timestamp' if type == :timestamp_local
	return 'boolean' if type == :boolean
	return 'date' if type == :date
	return 'json' if type == :json
	return 'jsonb' if type == :jsonb
	return 'double' if type == :float
	return type.to_s
end

#get_sequel_method(type, multi_column_pk = false) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/jungle_path/db_model/schema.rb', line 9

def get_sequel_method type, multi_column_pk=false
	return 'foreign_key' if type == :foreign_key
	return 'primary_key' if type == :primary_key and multi_column_pk == false
	return 'Integer' if type == :primary_key and multi_column_pk == true
	return 'String' if type == :string
	return 'Integer' if type == :integer
	return 'DateTime' if type == :timestamp
	return 'DateTime' if type == :timestamp_local
	return 'TrueClass' if type == :boolean
	return 'Date' if type == :date
	return 'json' if type == :json
	return 'jsonb' if type == :jsonb
	return 'Float' if type == :float
	return type
end