Rails 5.2使用ActiveStorage和Creek导入XLSX

时间:2018-08-13 04:14:11

标签: ruby-on-rails ruby import sidekiq rails-activestorage

我有一个名为ImportTemp的模型,该模型用于将导入的XLSX文件存储到数据库。我正在使用ActiveStorage来存储文件。

这是模型代码:

class ImportTemp < ApplicationRecord
  belongs_to :user
  has_one_attached :file
  has_one_attached :log_result
end

这是我的导入控制器代码:

def import
  # Check filetype
  case File.extname(params[:file].original_filename)
  when ".xlsx"
    # Add File to ImportFile model
    import = ImportTemp.new(import_type: 'UnitsUpload', user: current_user)
    import.file.attach(params[:file])
    import.save

    # Import unit via sidekiq with background jobs
    ImportUnitWorker.perform_async(import.id)

    # Notice
    flash.now[:notice] = "We are processing your xlsx, we will inform you after it's done via notifications."
    # Unit.import_file(xlsx)
  else flash.now[:error] = t('shared.info.unknown')+": #{params[:file].original_filename}"
  end
end

上传xlsx文件后,将在sidekiq中处理导入。这是工作程序代码(实际上仍然无法导入):

class ImportUnitWorker
  include Sidekiq::Worker
  sidekiq_options retry: false

def perform(file_id)
  import_unit = ImportTemp.find(file_id)

  # Open the uploaded xlsx to Creek
  creek = Creek::Book.new(Rails.application.routes.url_helpers.rails_blob_path(import_unit.file, only_path: true))
  sheet = creek.sheets[0]

  puts "Opening Sheet #{sheet.name}"
  sheet.rows.each do |row|
    puts row
  end

  units = []

  # Unit.import(units)
end

但是在我尝试之后,它给了我错误:

Zip::Error (File /rails/active_storage/blobs/eyJfcmFpbHMiOnsibWVzc2FnZSI6IkJBaHBCZz09IiwiZXhwIjpudWxsLCJwdXIiOiJibG9iX2lkIn19--3960b6ba5b55f7004e09967d16dfabe63f09f0a9/2018-08-10_10_39_audit_gt.xlsx not found)

但是如果我尝试使用浏览器打开它,则链接看起来像这样:

http://localhost:3000/rails/active_storage/blobs/eyJfcmFpbHMiOnsibWVzc2FnZSI6IkJBaHBCZz09IiwiZXhwIjpudWxsLCJwdXIiOiJibG9iX2lkIn19--3960b6ba5b55f7004e09967d16dfabe63f09f0a9/2018-08-10_10_39_audit_gt.xlsx

正在运行,并且已下载xlsx。我的问题是怎么了?为什么在sidekiq中找不到文件?

3 个答案:

答案 0 :(得分:1)

Rails.application.routes.url_helpers.rails_blob_path不会返回磁盘上文件的路径。而是返回一个路径,该路径可以与主机名结合使用以生成用于下载文件的URL,以供在链接中使用。

您有两个选择:

  • 如果您希望让ImportUnitWorker对正在使用的存储服务无动于衷,请将该文件“下载”到磁盘上的临时文件中。切换到Rails master并使用ActiveStorage::Blob#open

    def perform(import_id)
      import = ImportTemp.find(import_id)
      units  = []
    
      import.file.open do |file|
        book  = Creek::Book.new(file.path)
        sheet = creek.sheets[0]
    
        # ...
      end
    
      Unit.import(units)
    end
    
  • 如果您不介意ImportWorker知道使用磁盘服务,请向服务询问磁盘上文件的路径。 ActiveStorage::Service::DiskService#path_for(key)在Rails 5.2中是私有的,因此可以用send强制调用它,也可以升级到公开的Rails master:

    def perform(import_id)
      import = ImportTemp.find(import_id)
      units  = []
    
      path   = ActiveStorage::Blob.service.send(:path_for, import.file.key)
      book   = Creek::Book.new(path)
      sheet  = creek.sheets[0]
    
      # ...
    
      Unit.import(units)
    end
    

答案 1 :(得分:0)

我最终使用了乔治·克拉格霍恩(George Claghorn)建议的Tempfile。我不知道这是最佳解决方案还是最佳做法,但现在对我有用。我将在等待Rails 6稳定版推出ActiveStorage::Blob#open功能时使用此解决方案。

def perform(file_id)
  import    = ImportTemp.find(file_id)
  temp_unit = Tempfile.new([ 'unit_import_temp', '.xlsx' ], :encoding => 'ascii-8bit')
  units     = []

  begin
    # Write xlsx from ImportTemp to Tempfile
    temp_unit.write(import.file.download)

    # Open the temp xlsx to Creek
    book   = Creek::Book.new(temp_unit.path)
    sheet  = book.sheets[0]

    sheet.rows.each do |row|
      # Skip the header
      next if row.values[0] == 'Name' || row.values[1] == 'Abbreviation'
      cells = row.values

      # Add cells to new Unit
      unit = Unit.new(name: cells[0], abbrev: cells[1], desc: cells[2])
      units << unit
    end

    # Import the unit
    Unit.import(units)
  ensure
    temp_unit.close
    temp_unit.unlink   # deletes the temp file
  end
end

答案 2 :(得分:0)

现在的答案似乎是(除非我遗漏了什么):

Creek::Book.new file.service_url, check_file_extension: false, remote: true