1 /* 2 * Copyright 2020 The Netty Project 3 * 4 * The Netty Project licenses this file to you under the Apache License, 5 * version 2.0 (the "License"); you may not use this file except in compliance 6 * with the License. You may obtain a copy of the License at: 7 * 8 * https://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 * License for the specific language governing permissions and limitations 14 * under the License. 15 */ 16 package io.netty.handler.codec.http.multipart; 17 18 import java.io.File; 19 import java.util.Collections; 20 import java.util.Set; 21 import java.util.concurrent.ConcurrentHashMap; 22 23 /** 24 * DeleteFileOnExitHook. 25 */ 26 final class DeleteFileOnExitHook { 27 private static final Set<String> FILES = Collections.newSetFromMap(new ConcurrentHashMap<String, Boolean>()); 28 29 private DeleteFileOnExitHook() { 30 } 31 32 static { 33 // DeleteOnExitHook must be the last shutdown hook to be invoked. 34 // Application shutdown hooks may add the first file to the 35 // delete on exit list and cause the DeleteOnExitHook to be 36 // registered during shutdown in progress. 37 Runtime.getRuntime().addShutdownHook(new Thread() { 38 39 @Override 40 public void run() { 41 runHook(); 42 } 43 }); 44 } 45 46 /** 47 * Remove from the pool to reduce space footprint. 48 * 49 * @param file tmp file path 50 */ 51 public static void remove(String file) { 52 FILES.remove(file); 53 } 54 55 /** 56 * Add to the hook and clean up when the program exits. 57 * 58 * @param file tmp file path 59 */ 60 public static void add(String file) { 61 FILES.add(file); 62 } 63 64 /** 65 * Check in the hook files. 66 * 67 * @param file target file 68 * @return true or false 69 */ 70 public static boolean checkFileExist(String file) { 71 return FILES.contains(file); 72 } 73 74 /** 75 * Clean up all the files. 76 */ 77 static void runHook() { 78 for (String filename : FILES) { 79 new File(filename).delete(); 80 } 81 } 82 }