2014-09-29 47 views
0

我正在处理的应用程序每晚都会运行导入任务以从外部数据库和进程获取信息并将其写入本地PostgreSQL数据库。这是可以在我们的应用程序中查看的。如何将这些多个rake任务合并为一个任务

我们做到以下几点:

  • 创建到外部数据库
  • 的连接进行一些货币转换
  • 请在基于平均
  • 检查数据的一些计算,如果已经在项目存在并更新(如果已更改)
  • 同时检查它是如何分类的,并根据我们的分类更改分类
  • 创建项目,如果不存在

所有这些数据是在外部数据库不同的表,所以我会为每个表(约30)任务,但我过于重复自己所以寻找一些关于如何干这个坏男孩的指导。

是否可以在外部数据库表名的集合上创建某种循环?我已经尝试了几次,但没有取得任何成功。请在下面找到两个不同导入脚本示例的代码。

desc 'Imports ABC Data' 
task :abcdeal_import => :environment do 

    # Connect to legacy db 

    class OldDb < ActiveRecord::Base 
     establish_connection :cleardb 
     self.table_name = 'abcdeal' 
    end 

    class Import < OldDb 
     self.table_name = 'abcdeal' 
    end 

    # Currency Conversion 

    require 'money' 
    require 'money/bank/google_currency' 

    # set the seconds after than the current rates are automatically expired 
    Money::Bank::GoogleCurrency.ttl_in_seconds = 86400 
    # set default bank to instance of GoogleCurrency 
    bank = Money::Bank::GoogleCurrency.new 

     @gbp = bank.get_rate(:USD, :GBP).to_f 
     @eur = bank.get_rate(:USD, :EUR).to_f 
     @aud = bank.get_rate(:USD, :AUD).to_f  
     @cad = bank.get_rate(:USD, :CAD).to_f  
     @nzd = bank.get_rate(:USD, :NZD).to_f  

    Import.all.find_each do |u| 

     # Define Variables 

     @price = u.UnitPrice 
     @interest = u.UnitSold 
     #@interest_changes = u.unitsold_changes 
     @company = u.Company 
     @legacy_id = u.id 
     @end_date = u.EndDate 
     @category = u.Category 
     @revenue = u.Revenue 

     # Calculate average 

     @company_average = company_average(@company)   

     if @company_average == 0 
      @company_average = 1 
     end 

     # Does item exist? 
     if Dailydeal.exists?(:legacy_id => @legacy_id, :company => @company) 
     # It exists. 
      @historical_interest = Product.where(:legacy_id => @legacy_id).pluck(:interest)     
      # Another statement to check that the Interest != UnitSold 
      if @interest != @historical_interest 
      Product.where(:legacy_id => @legacy_id, :company => @company).update_all(:interest => @interest, :end_date => @end_date, :interest_changes => @interest_changes, :status => average_change_interest, :revenue => @revenue, :revenue_eur => eur_revenue_currency_conversion, :revenue_aud => aud_revenue_currency_conversion, :revenue_gbp => gbp_revenue_currency_conversion, :revenue_cad => cad_revenue_currency_conversion, :revenue_nzd => nzd_revenue_currency_conversion) 
      end 
     else 
     # Doesn't exist 

     # Set Category    

     case @category 
      when 'Apparel' 
       @category = 'Apparel & Footwear' 

      when 'Gadgets' 
       @category = 'Electronics & Accessories' 

      when 'Gifts' 
       @category = 'Miscellaneous' 

      when 'Health & Cosmetics' 
       @category = 'Health & Beauty' 

      when 'Home' 
       @category = 'Home & Furniture' 

      when 'Other' 
       @category = 'Miscellaneous' 

      when 'Toys' 
       @category = 'Toys & Kids' 

      else 

       @category 

      end 

     Product.create(
       :name => u.ProductName, 
       :link => u.ProductLink, 
       :image_url => u.ImageUrl, 
       :price => u.UnitPrice, 
       :interest => u.UnitSold, 
       :start_date => u.StartDate, 
       :end_date => u.EndDate, 
       :revenue => u.Revenue, 
       :company => u.Company, 
       :category => @category, 
       :country => u.Country, 
       :price_eur => eur_currency_conversion, 
       :price_aud => aud_currency_conversion, 
       :price_gbp => gbp_currency_conversion, 
       :price_cad => cad_currency_conversion, 
       :price_nzd => nzd_currency_conversion, 
       :revenue_eur => eur_revenue_currency_conversion, 
       :revenue_aud => aud_revenue_currency_conversion, 
       :revenue_gbp => gbp_revenue_currency_conversion, 
       :revenue_cad => cad_revenue_currency_conversion, 
       :revenue_nzd => nzd_revenue_currency_conversion, 
       :status => average_change_interest, 
       :legacy_id => u.id) 
     end 
    end 

    puts "Successfully imported data from abc" 

end 

desc 'Imports def Data' 
task :def_import => :environment do 

    # Connect to legacy db 

    class OldDb < ActiveRecord::Base 
     establish_connection :cleardb 
     self.table_name = 'def' 
    end 

    class Import < OldDb 
     self.table_name = 'def' 
    end 

    # Currency Conversion 

    require 'money' 
    require 'money/bank/google_currency' 

    # set the seconds after than the current rates are automatically expired 
    Money::Bank::GoogleCurrency.ttl_in_seconds = 86400 
    # set default bank to instance of GoogleCurrency 
    bank = Money::Bank::GoogleCurrency.new 

     @gbp = bank.get_rate(:USD, :GBP).to_f 
     @eur = bank.get_rate(:USD, :EUR).to_f 
     @aud = bank.get_rate(:USD, :AUD).to_f  
     @cad = bank.get_rate(:USD, :CAD).to_f  
     @nzd = bank.get_rate(:USD, :NZD).to_f  

    Import.all.find_each do |u| 

     # Define Variables 

     @price = u.UnitPrice 
     @interest = u.UnitSold 
     #@interest_changes = u.unitsold_changes 
     @company = u.Company 
     @legacy_id = u.id 
     @end_date = u.EndDate 
     @category = u.Category 
     @revenue = u.Revenue 

     # Calculate average 

     @company_average = company_average(@company) 

     if @company_average == 0 
      @company_average = 1 
     end 

     # Does item exist? 
     if Product.exists?(:legacy_id => @legacy_id, :company => @company) 
     # It exists. 
      @historical_interest = Product.where(:legacy_id => @legacy_id).pluck(:interest)     
      # Another statement to check that the Interest != UnitSold 
      if @interest != @historical_interest 
       Product.where(:legacy_id => @legacy_id, :company => @company).update_all(:interest => @interest, :end_date => @end_date, :interest_changes => @interest_changes, :status => average_change_interest, :revenue => @revenue, :revenue_eur => eur_revenue_currency_conversion, :revenue_aud => aud_revenue_currency_conversion, :revenue_gbp => gbp_revenue_currency_conversion, :revenue_cad => cad_revenue_currency_conversion, :revenue_nzd => nzd_revenue_currency_conversion) 
      end 
     else 
     # Doesn't exist 

     # Set Categories  

     case @category 

      when 'Adult Products' 
       @category 

      when 'Arts & Crafts', 'Furniture', 'Garden', 'Large Appliances', 'Lighting', 'Outdoor & Patio', 'Stationary', 'Storage & Organization' 
       @category = 'Home & Furniture' 

      when 'Audio & Audio Accessories', 'Electronics Accessories', 'Gadgets', 'GPS & Car Accessories', 'Tools & Hardware', 'Tablets' 
       @category = 'Electronics & Accessories' 

      when 'Automotive Services' 
       @category = 'Automotive' 

      when 'Bath', 'Bedding' 
       @category = 'Bed & Bath' 

      when 'Books & Media', 'Canada', 'Gifts', 'Miscellaneous', 'Recreational' 
       @category = 'Miscellaneous' 

      when 'Cafe, Bakery & Treats', 'Kitchen' 
       @category = 'Kitchen' 

      when 'Casual Restaurants', 'Cleaning Services', 'Dance Classes', 'Fast Food', 'Fine Dining', 'Home Services', 'Local Exhibits & Shows', 'Other Services', 'Other Workshops & Classes', 'Outdoor Adventures', 'Other Activities', 'Tours & Sightseeing', 'Fitness Classes', 'Massage', 'Med Spa', 'Salon & Hair Care Services', 'Spa Services', 'Yoga, Pilates & Aerobics' 
       @category = 'Experiences' 

      when 'Cats', 'Dogs', 'Pets', 'Pets Accessories' 
       @category = 'Pets Accessories' 

      when 'Clothing', 'Clothing, Fashion & Accessories', 'Fashion & Accessories', 'Footwear', "Men's", 'Unisex', "Women's" 
       @category = 'Apparel & Footwear' 

      when 'Earrings', 'Fashion Accessories' 
       @category = 'Jewellery & Accessories' 

      when 'Fitness', 'Hair Care', 'Hair Removal', 'Health Care', 'Manicure & Pedicure', 'Other Beauty & Spa', 'Other Health & Fitness', 'Personal Care', 'Teeth Whitening', 'Wellness & Nutrition', 'Facial' 
      @category = 'Health & Beauty' 

      when 'Golf', 'Sports', 'Football', 'Baseball', 'Hockey', 'Ice Hockey' 
       @category = 'Sporting Accessories' 

     else 

      @category 

     end    

     Product.create(
       :name => u.ProductName, 
       :link => u.ProductLink, 
       :image_url => u.ImageUrl, 
       :price => u.UnitPrice, 
       :interest => u.UnitSold, 
       :start_date => u.StartDate, 
       :end_date => u.EndDate, 
       :revenue => u.Revenue, 
       :company => u.Company, 
       :category => @category, 
       :country => u.Country, 
       :price_eur => eur_currency_conversion, 
       :price_aud => aud_currency_conversion, 
       :price_gbp => gbp_currency_conversion, 
       :price_cad => cad_currency_conversion, 
       :price_nzd => nzd_currency_conversion, 
       :revenue_eur => eur_revenue_currency_conversion, 
       :revenue_aud => aud_revenue_currency_conversion, 
       :revenue_gbp => gbp_revenue_currency_conversion, 
       :revenue_cad => cad_revenue_currency_conversion, 
       :revenue_nzd => nzd_revenue_currency_conversion, 
       :status => average_change_interest, 
       :legacy_id => u.id) 
     end 
    end 

    puts "Successfully imported data from def" 

end 

回答

0

这的确看起来重复的,你为什么不能拿出这样的结构,而不是把所有的东西在下面的格式

. 
├── Rakefile 
└── models 
    ├── import.rb 
    └── old_db.rb 

我做了类似的事情,我的项目之一在这里我从数据库中提取数据,你也可以在这里参考它https://github.com/ankit8898/sql-server-mapper

+0

嘿非常感谢你的回复。我正在使用Heroku Scheduler来运行导入任务,因此它们都在一个名为scheduler.rake的文件中。所以您认为我应该专门创建这些模型?然后从耙子任务中调用它们? – bnussey 2014-09-29 20:27:17

+0

@bnussey,是的,我更喜欢把它放在不同的空间里。因此,您可以创建模型并在.rake中要求它们。你可以在上面的网址 – AnkitG 2014-09-29 20:28:20

+0

中查看回购请注意。我是在寻找某种方式来干掉整个任务,就像它正在做的所有事情,任何想法一样?也许一次写脚本然后以某种方式循环通过不同的模型 – bnussey 2014-09-29 20:31:23