View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements. See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache license, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License. You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the license for the specific language governing permissions and
15   * limitations under the license.
16   */
17  package org.apache.logging.log4j.core.appender.rolling.action;
18  
19  import java.io.File;
20  import java.io.FileInputStream;
21  import java.io.FileOutputStream;
22  import java.io.IOException;
23  import java.nio.channels.FileChannel;
24  
25  /**
26   * File rename action.
27   */
28  public class FileRenameAction extends AbstractAction {
29  
30      /**
31       * Source.
32       */
33      private final File source;
34  
35      /**
36       * Destination.
37       */
38      private final File destination;
39  
40      /**
41       * If true, rename empty files, otherwise delete empty files.
42       */
43      private final boolean renameEmptyFiles;
44  
45      /**
46       * Creates an FileRenameAction.
47       *
48       * @param src              current file name.
49       * @param dst              new file name.
50       * @param renameEmptyFiles if true, rename file even if empty, otherwise delete empty files.
51       */
52      public FileRenameAction(final File src, final File dst, final boolean renameEmptyFiles) {
53          source = src;
54          destination = dst;
55          this.renameEmptyFiles = renameEmptyFiles;
56      }
57  
58      /**
59       * Rename file.
60       *
61       * @return true if successfully renamed.
62       */
63      @Override
64      public boolean execute() {
65          return execute(source, destination, renameEmptyFiles);
66      }
67  
68      /**
69       * Rename file.
70       *
71       * @param source           current file name.
72       * @param destination      new file name.
73       * @param renameEmptyFiles if true, rename file even if empty, otherwise delete empty files.
74       * @return true if successfully renamed.
75       */
76      public static boolean execute(final File source, final File destination, final boolean renameEmptyFiles) {
77          if (renameEmptyFiles || source.length() > 0) {
78              final File parent = destination.getParentFile();
79              if (parent != null && !parent.exists()) {
80                  if (!parent.mkdirs()) {
81                      LOGGER.error("Unable to create directory {}", parent.getAbsolutePath());
82                      return false;
83                  }
84              }
85              try {
86                  if (!source.renameTo(destination)) {
87                      try {
88                          copyFile(source, destination);
89                          return source.delete();
90                      } catch (final IOException iex) {
91                          LOGGER.error("Unable to rename file {} to {} - {}", source.getAbsolutePath(),
92                              destination.getAbsolutePath(), iex.getMessage());
93                      }
94                  }
95                  return true;
96              } catch (final Exception ex) {
97                  try {
98                      copyFile(source, destination);
99                      return source.delete();
100                 } catch (final IOException iex) {
101                     LOGGER.error("Unable to rename file {} to {} - {}", source.getAbsolutePath(),
102                         destination.getAbsolutePath(), iex.getMessage());
103                 }
104             }
105         } else {
106             try {
107                 source.delete();
108             } catch (final Exception ex) {
109                 LOGGER.error("Unable to delete empty file " + source.getAbsolutePath());
110             }
111         }
112 
113         return false;
114     }
115 
116     private static void copyFile(final File source, final File destination) throws IOException {
117         if (!destination.exists()) {
118             destination.createNewFile();
119         }
120 
121         FileChannel srcChannel = null;
122         FileChannel destChannel = null;
123         FileInputStream srcStream = null;
124         FileOutputStream destStream = null;
125         try {
126             srcStream = new FileInputStream(source);
127             destStream = new FileOutputStream(destination);
128             srcChannel = srcStream.getChannel();
129             destChannel = destStream.getChannel();
130             destChannel.transferFrom(srcChannel, 0, srcChannel.size());
131         } finally {
132             if (srcChannel != null) {
133                 srcChannel.close();
134             }
135             if (srcStream != null) {
136                 srcStream.close();
137             }
138             if (destChannel != null) {
139                 destChannel.close();
140             }
141             if (destStream != null) {
142                 destStream.close();
143             }
144         }
145     }
146 
147     @Override
148     public String toString() {
149         return FileRenameAction.class.getSimpleName() + '[' + source + " to " + destination //
150                 + ", renameEmptyFiles=" + renameEmptyFiles + ']';
151     }
152 }