3 class PopulateNodeTagsAndRemove < ActiveRecord::Migration[4.2]
5 have_nodes = select_value("SELECT count(*) FROM current_nodes").to_i.nonzero?
8 prefix = File.join Dir.tmpdir, "020_populate_node_tags_and_remove.#{$PROCESS_ID}."
10 cmd = "db/migrate/020_populate_node_tags_and_remove_helper"
12 if !File.exist?(cmd) || File.mtime(cmd) < File.mtime(src)
13 system("cc -O3 -Wall `mysql_config --cflags --libs` " \
14 "#{src} -o #{cmd}") || raise
17 conn_opts = ActiveRecord::Base.connection.instance_eval { @connection_options }
18 args = conn_opts.map(&:to_s) + [prefix]
19 raise "#{cmd} failed" unless system cmd, *args
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
26 execute "TRUNCATE nodes"
27 remove_column :nodes, :tags
28 remove_column :current_nodes, :tags
30 add_column :nodes, :version, :bigint, :null => false
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
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
45 # now get the data back
46 csvopts = "FIELDS TERMINATED BY ',' ENCLOSED BY '\"' ESCAPED BY '\"' LINES TERMINATED BY '\\n'"
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)"
55 tempfiles.each { |fn| File.unlink fn } if have_nodes
59 raise ActiveRecord::IrreversibleMigration
60 # add_column :nodes, "tags", :text, :default => "", :null => false
61 # add_column :current_nodes, "tags", :text, :default => "", :null => false