■ Parallel 클래스의 ForEach 정적 메소드를 사용해 디렉토리 크기를 구하는 방법을 보여준다.
▶ Program.cs
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
using System; using System.Collections.Generic; using System.IO; using System.Threading; using System.Threading.Tasks; namespace TestProject { /// <summary> /// 프로그램 /// </summary> class Program { //////////////////////////////////////////////////////////////////////////////////////////////////// Method ////////////////////////////////////////////////////////////////////////////////////////// Static //////////////////////////////////////////////////////////////////////////////// Private #region 프로그램 시작하기 - Main() /// <summary> /// 프로그램 시작하기 /// </summary> private static void Main() { string[] argumentArray = Environment.GetCommandLineArgs(); if(argumentArray.Length > 1) { List<Task<long>> taskList = new List<Task<long>>(); for(int i = 1; i < argumentArray.Length; i++) { taskList.Add(GetDirectorySizeAsync(argumentArray[i])); } try { Task.WaitAll(taskList.ToArray()); } catch(AggregateException) { } for(int i = 0 ; i < taskList.Count; i++) { if(taskList[i].Status == TaskStatus.Faulted) { Console.WriteLine($"{argumentArray[i + 1]} 디렉토리가 없습니다."); } else { Console.WriteLine($"{argumentArray[i + 1]} 디렉토리 크기 : {taskList[i].Result:N0} 바이트"); } } } else { Console.WriteLine("구문 에러 : 1개 이상의 디렉토리 경로를 지정해 주시기 바랍니다."); } } #endregion #region 디렉토리 크기 구하기 (비동기) - GetDirectorySizeAsync(directoryPath) /// <summary> /// 디렉토리 크기 구하기 (비동기) /// </summary> /// <param name="directoryPath">디렉토리 경로</param> /// <returns>디렉토리 크기</returns> private static Task<long> GetDirectorySizeAsync(string directoryPath) { if(!Directory.Exists(directoryPath)) { return Task.FromException<long>(new DirectoryNotFoundException("디렉토리가 없습니다.")); } else { string[] filePathArray = Directory.GetFiles(directoryPath, "*.*", SearchOption.AllDirectories); if(filePathArray.Length == 0) { return Task.FromResult(0L); } else { return Task.Run ( () => { long totalFileLength = 0; Parallel.ForEach ( filePathArray, (filePath) => { FileStream fileStream = new FileStream ( filePath, FileMode.Open, FileAccess.Read, FileShare.ReadWrite, 256, true ); long fileLength = fileStream.Length; Interlocked.Add(ref totalFileLength, fileLength); fileStream.Close(); } ); return totalFileLength; } ); } } } #endregion } } |
