java directory copy / progress bar

 

😒 test용으로 구현 java 1.8 사용 

 

😒 에이전트의 FS 기능 개발 중 필요해서 구현.

 

 

특정 디렉토리 경로 및 모든 하위 파일을 file output stream으로 쓰면서 소스와 타겟의 바이트단위 크기를 비교한다.

재귀호출을 통해 디렉토리안의 디렉토리도 모두 확인하도록 구현.

 

1. static 변수를 통해 진행률 등 기록하는 방식. 

2. static 변수 없이, 파라미터로 처리하는 방식.

     멀티  스레드 환경에서 전역변수를 참조하면 전역변수의 값이 꼬이므로, 2번의 방식이 적합함.

 

 

FileDownloaderTest.java

package test;

import com.gemiso.ariel.common.util.FileUtil;

import java.io.*;

public class FsCopyDirTest {

    static long totalBytesCopied;
    static long expectedBytes;

    public static void main(String[] args) {
        String srcDirPath = "C:\\Users\\user\\Desktop\\테스트용 영상 및 이미지\\디렉토리 카피 테스트용 1";
        //String srcDirPath = "C:\\Users\\admin\\Desktop\\다운로드 소스";
        String tgtDirPath = "C:\\Users\\user\\Desktop\\테스트용 영상 및 이미지\\디렉토리 카피 타겟";
        //String tgtDirPath = "C:\\Users\\admin\\Desktop\\다운로드 타겟";

        File srcDir = new File(srcDirPath);
        File tgtDir = new File(tgtDirPath);

        expectedBytes = FileUtil.getDirSize(srcDir);
        
        // 전역변수 없이 실행하게된다면 0값으로 실행.
        // expectedBytes = Fileutil.getDirSize(srcDir,0);
        
        copy(srcDir, tgtDir);
    }


    public static void copy(File srcDir, File tgtDir) {

        boolean isDirectory = srcDir.isDirectory();

        String[] srcList = srcDir.list();

        boolean is_cancel = false;

        try {
            int len = 0;
            byte[] buf = new byte[1024];
            int rates = 0;

            System.out.println(" 카피 시작! ");

            File finalTgtDir = tgtDir;
            File[] target_file = srcDir.listFiles();

            // 폴더 사이즈를 구할 경우 하위파일을 모두 탐색하고 파일 크기 합을 구하지 않는다면
            // directory의 사이즈를 4096으로 고정해서 뱉어낸다.

            System.out.println(" 소스 파일 사이즈 : " + expectedBytes);

            for (File file : target_file) {
                File temp = new File(tgtDir.getAbsolutePath() + File.separator + file.getName());
                if (file.isDirectory()) {
                    temp.mkdir();
                    copy(file, temp);
                } else {

                    FileInputStream fis = null;
                    FileOutputStream fos = null;
                    try {
                        fis = new FileInputStream(file);
                        fos = new FileOutputStream(temp);
                        byte[] b = new byte[1024];
                        while ((len = fis.read(b)) != -1) {
                            totalBytesCopied += len;

                            fos.write(b, 0, len);

                            rates = (int) Math.floor(( totalBytesCopied * 1.0 / expectedBytes) * 1000) / 10;

                            System.out.println(" 파일 별 사이즈 : " + totalBytesCopied + " / 원본 dir 사이즈 : " + expectedBytes);
                            System.out.println(" 진행률 : " + rates);
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    } finally {
                        try {
                            fis.close();
                            fos.close();
                        } catch (IOException e) {
                            // TODO Auto-generated catch block
                            e.printStackTrace();
                        }
                    }

                }
            }

        } catch (Exception e) {
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            PrintStream pinrtStream = new PrintStream(out);
            e.printStackTrace(pinrtStream);
            System.out.println(out.toString());
        }

    }
    
    
    // 멀티스레드 환경에서 스태틱 변수를 참조하고 싶지 않은 경우 아래 로직 수행 : 
     public static void copy(File srcDir, File tgtDir, long totalBytesCopied, long expectedBytes) {

        boolean isDirectory = srcDir.isDirectory();

        String[] srcList = srcDir.list();

        boolean is_cancel = false;

        try {
            int len = 0;
            byte[] buf = new byte[1024];

            System.out.println(" 카피 시작! ");

            File finalTgtDir = tgtDir;
            File[] target_file = srcDir.listFiles();

            for (File file : target_file) {
                File temp = new File(tgtDir.getAbsolutePath() + File.separator + file.getName());
                if (file.isDirectory()) {
                    temp.mkdir();
                    copy(file, temp, totalBytesCopied, expectedBytes);
                } else {

                    FileInputStream fis = null;
                    FileOutputStream fos = null;
                    DataOutputStream dos = null;
                    try {
                        fis = new FileInputStream(file);
                        fos = new FileOutputStream(temp);
                        dos = new DataOutputStream(fos);


                        byte[] b = new byte[1024];
                        while ((len = fis.read(b)) != -1) {
                            totalBytesCopied += len;
                            dos.write(b, 0, len);

                            FileUtil.rates = (int) Math.floor(( totalBytesCopied * 1.0 / expectedBytes) * 1000) / 10;

                            System.out.println(" 파일 별 사이즈 : " + totalBytesCopied + " / 원본 dir 사이즈 : " + expectedBytes + " / rate : " + FileUtil.rates);
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    } finally {
                        try {
                            fis.close();
                            fos.close();
                        } catch (IOException e) {
                            // TODO Auto-generated catch block
                            e.printStackTrace();
                        }
                    }

                }
            }

        } catch (Exception e) {
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            PrintStream pinrtStream = new PrintStream(out);
            e.printStackTrace(pinrtStream);
            System.out.println(out.toString());
        }

    }

    
    
    
    
    
}

 

 

FileUtil.java

package com.org.walk.util;

import java.io.File;
import java.io.IOException;

public class Utils {

 	public static long getDirSize(File folder) {
		File[] files = folder.listFiles();

		long length=0;

		int count = files.length;
		for (int i = 0; i < count; i++) {
			if (files[i].isFile()) {
				// System.out.println(" 파일 네임 : " + files[i].getName() + " / size : " + files[i].length());
				length += files[i].length();
			} else {
				// System.out.println(" DIR 네임 : " + files[i].getName());
				length += getDirSize(files[i]);
			}
		}
		return length;
	}
    
    
    // 멀티 스레드 환경에서 static 변수를 동시에 참조하고 싶지 않은 경우라면 아래 로직 수행 : 
    public static long getDirSize(File folder, long length) {
		File[] files = folder.listFiles();

		long _length;

		if (length!=0) {
			_length = length;
		} else {
			_length = 0;
		}

		int count = files.length;
		for (int i = 0; i < count; i++) {
			if (files[i].isFile()) {
				
				_length += files[i].length();
				
			} else {
				
				_length += getDirSize(files[i],0);
				
			}
		}

		return _length;
	}
    
    
    
    
}