Standardise on double quoted strings
[rails.git] / db / migrate / 020_populate_node_tags_and_remove.rb
1 require "migrate"
2
3 class PopulateNodeTagsAndRemove < ActiveRecord::Migration
4   def self.up
5     have_nodes = select_value("SELECT count(*) FROM current_nodes").to_i != 0
6
7     if have_nodes
8       prefix = File.join Dir.tmpdir, "020_populate_node_tags_and_remove.#{$PROCESS_ID}."
9
10       cmd = "db/migrate/020_populate_node_tags_and_remove_helper"
11       src = "#{cmd}.c"
12       if !File.exist?(cmd) || File.mtime(cmd) < File.mtime(src)
13         system("cc -O3 -Wall `mysql_config --cflags --libs` " +
14           "#{src} -o #{cmd}") || fail
15       end
16
17       conn_opts = ActiveRecord::Base.connection.instance_eval { @connection_options }
18       args = conn_opts.map(&:to_s) + [prefix]
19       fail "#{cmd} failed" unless system cmd, *args
20
21       tempfiles = %w(nodes node_tags current_nodes current_node_tags)
22                   .map { |base| prefix + base }
23       nodes, node_tags, current_nodes, current_node_tags = tempfiles
24     end
25
26     execute "TRUNCATE nodes"
27     remove_column :nodes, :tags
28     remove_column :current_nodes, :tags
29
30     add_column :nodes, :version, :bigint, :null => false
31
32     create_table :current_node_tags, :id => false do |t|
33       t.column :id,          :bigint, :null => false
34       t.column :k,           :string, :default => "", :null => false
35       t.column :v,           :string, :default => "", :null => false
36     end
37
38     create_table :node_tags, :id => false do |t|
39       t.column :id,          :bigint, :null => false
40       t.column :version,     :bigint, :null => false
41       t.column :k,           :string, :default => "", :null => false
42       t.column :v,           :string, :default => "", :null => false
43     end
44
45     # now get the data back
46     csvopts = "FIELDS TERMINATED BY ',' ENCLOSED BY '\"' ESCAPED BY '\"' LINES TERMINATED BY '\\n'"
47
48     if have_nodes
49       execute "LOAD DATA INFILE '#{nodes}' INTO TABLE nodes #{csvopts} (id, latitude, longitude, user_id, visible, timestamp, tile, version)"
50       execute "LOAD DATA INFILE '#{node_tags}' INTO TABLE node_tags #{csvopts} (id, version, k, v)"
51       execute "LOAD DATA INFILE '#{current_nodes}' INTO TABLE current_nodes #{csvopts} (id, latitude, longitude, user_id, visible, timestamp, tile)"
52       execute "LOAD DATA INFILE '#{current_node_tags}' INTO TABLE current_node_tags #{csvopts} (id, k, v)"
53     end
54
55     tempfiles.each { |fn| File.unlink fn } if have_nodes
56   end
57
58   def self.down
59     fail ActiveRecord::IrreversibleMigration
60     #    add_column :nodes, "tags", :text, :default => "", :null => false
61     #    add_column :current_nodes, "tags", :text, :default => "", :null => false
62   end
63 end