RailsでRakeタスクの追加 #Rails

Railsで使っているDBに対するバッチ処理を行うのにrails runner 使ったりもしていたけど結局Rake使うのが一番捗ると今更気づいた今日このごろ。

サンプル

name,commentを持つMobというモデルを作成
それに100件のデータを追加
奇数IDのみ表示
全削除

作成単なるScaffold

rails new SampleRake
cd SampleRake
rails g scaffold mob name comment
rake db:migrate

Rakeタスクの作成

rakeファイルの作成

rails g task mob_task add_100 show_odd remove_all

結果
create lib/tasks/mob_task.rake

lib/tasks/mob_task.rake

namespace :mob_task do
  desc "TODO"
  task :add_100 => :environment do
  end

  desc "TODO"
  task :show_odd => :environment do
  end

  desc "TODO"
  task :remove_all => :environment do
  end

end

100件追加タスクの作成と実行

add100

  task :add_100 => :environment do
+   100.times do |i|
+     mob = Mob.new
+     mob.name = "user_#{i}"
+     mob.comment = "sug" + "e" * i
+     mob.save
+   end
  end

実行
rake mob_task:add100

結果(localhost:3000/mobs)

Name Comment

user_0 sug Show Edit Destroy
user_1 suge Show Edit Destroy
user_2 sugee Show Edit Destroy
user_3 sugeee Show Edit Destroy
・・・・

奇数IDのみ出力

  task :show_odd => :environment do
+   Mob.all.each do |m|
+     p "id:#{m.id}, name:#{m.name}, comment:#{m.comment}" if m.id % 2 == 1
+   end
  end

実行
rake mob_task:show_odd

結果

"id:1, name:user_0, comment:sug"
"id:3, name:user_2, comment:sugee"
"id:5, name:user_4, comment:sugeeee"
"id:7, name:user_6, comment:sugeeeeee"
"id:9, name:user_8, comment:sugeeeeeeee"

全削除

  task :remove_all => :environment do
+   Mob.delete_all
  end

実行
rake mob_task:remove_all

結果
消えてます。

他にもいろいろできそうなので活用してみます。

RailsでRakeタスクの追加