rubyでzip圧縮といえばrubyzipというgemですが、githubに載っているサンプルスクリプトだと、日本語が文字化けしたり、なんと大容量ファイルを圧縮しようとした途端...その解決方法たちをご紹介します。
スポンサーリンク
rubyでディレクトリをzip圧縮する
rubyでzip圧縮をする必要があり、rubyzipというgemを使用することに決めました。また、ディレクトリを圧縮するスクリプトも、rubyzipのGitHubに載っていたので、簡単にできると思っていたのですが...
以下の2箇所でつまづく事になりました。。
- 日本語のディレクトリ名を使った場合に文字化けする
- ディレクトリの合計容量が大きかった場合に、処理が途中でコケる
これらを踏まえ、修正したスクリプトがこちらです。
require 'zip' # This is a simple example which uses rubyzip to # recursively generate a zip file from the contents of # a specified directory. The directory itself is not # included in the archive, rather just its contents. # # Usage: # directory_to_zip = "/tmp/input" # output_file = "/tmp/out.zip" # zf = ZipFileGenerator.new(directory_to_zip, output_file) # zf.write() class ZipFileGenerator # Initialize with the directory to zip and the location of the output archive. def initialize(input_dir, output_file) @input_dir = input_dir @output_file = output_file end # Zip the input directory. def write entries = Dir.entries(@input_dir) - %w(. ..) Zip.unicode_names = true ::Zip::File.open(@output_file, ::Zip::File::CREATE) do |zipfile| write_entries entries, '', zipfile end end private # A helper method to make the recursion work. def write_entries(entries, path, zipfile) entries.each do |e| zipfile_path = path == '' ? e : File.join(path, e) disk_file_path = File.join(@input_dir, zipfile_path) puts "Deflating #{disk_file_path}" if File.directory? disk_file_path recursively_deflate_directory(disk_file_path, zipfile, zipfile_path) else put_into_archive(disk_file_path, zipfile, zipfile_path) end end end def recursively_deflate_directory(disk_file_path, zipfile, zipfile_path) zipfile.mkdir zipfile_path.encode('cp932') subdir = Dir.entries(disk_file_path) - %w(. ..) write_entries subdir, zipfile_path, zipfile end def put_into_archive(disk_file_path, zipfile, zipfile_path) zipfile.get_output_stream(zipfile_path.encode('cp932')) do |f| #f.write(File.open(disk_file_path, 'rb').read) IO.copy_stream(File.open(disk_file_path, 'rb'), f) end end end

サンプルコードでは、大容量ファイルでメモリリークが発生していたみたいですね...!