src/librustc_incremental/persist/work_product.rs

   1 // Copyright 2012-2015 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 //! This module contains files for saving intermediate work-products.
  12
  13 use persist::fs::*;
  14 use rustc::dep_graph::{WorkProduct, WorkProductId};
  15 use rustc::session::Session;
  16 use rustc::session::config::OutputType;
  17 use rustc::util::fs::link_or_copy;
  18 use std::path::PathBuf;
  19 use std::fs as std_fs;
  20
  21 pub fn save_trans_partition(sess: &Session,
  22                             cgu_name: &str,
  23                             partition_hash: u64,
  24                             files: &[(OutputType, PathBuf)]) {
  25     debug!("save_trans_partition({:?},{},{:?})",
  26            cgu_name,
  27            partition_hash,
  28            files);
  29     if sess.opts.incremental.is_none() {
  30         return;
  31     }
  32     let work_product_id = WorkProductId::from_cgu_name(cgu_name);
  33
  34     let saved_files: Option<Vec<_>> =
  35         files.iter()
  36              .map(|&(kind, ref path)| {
  37                  let file_name = format!("cgu-{}.{}", cgu_name, kind.extension());
  38                  let path_in_incr_dir = in_incr_comp_dir_sess(sess, &file_name);
  39                  match link_or_copy(path, &path_in_incr_dir) {
  40                      Ok(_) => Some((kind, file_name)),
  41                      Err(err) => {
  42                          sess.warn(&format!("error copying object file `{}` \
  43                                              to incremental directory as `{}`: {}",
  44                                             path.display(),
  45                                             path_in_incr_dir.display(),
  46                                             err));
  47                          None
  48                      }
  49                  }
  50              })
  51              .collect();
  52     let saved_files = match saved_files {
  53         Some(v) => v,
  54         None => return,
  55     };
  56
  57     let work_product = WorkProduct {
  58         cgu_name: cgu_name.to_string(),
  59         input_hash: partition_hash,
  60         saved_files,
  61     };
  62
  63     sess.dep_graph.insert_work_product(&work_product_id, work_product);
  64 }
  65
  66 pub fn delete_workproduct_files(sess: &Session, work_product: &WorkProduct) {
  67     for &(_, ref file_name) in &work_product.saved_files {
  68         let path = in_incr_comp_dir_sess(sess, file_name);
  69         match std_fs::remove_file(&path) {
  70             Ok(()) => { }
  71             Err(err) => {
  72                 sess.warn(
  73                     &format!("file-system error deleting outdated file `{}`: {}",
  74                              path.display(), err));
  75             }
  76         }
  77     }
  78 }