feature/multi-threading #2
@ -1,13 +1,19 @@
|
|||||||
using System.Reflection;
|
using System.Reflection;
|
||||||
using System.Security.Cryptography;
|
using System.Security.Cryptography;
|
||||||
using System.Threading.Tasks;
|
|
||||||
using Microsoft.Data.Sqlite;
|
using Microsoft.Data.Sqlite;
|
||||||
namespace Chksum.Utils;
|
namespace Chksum.Utils;
|
||||||
public class ChksumUtils {
|
public class ChksumUtils {
|
||||||
|
|
||||||
private int getFileCount() {
|
private int getTotalFileCount() {
|
||||||
int fileCount = Directory.GetFiles(Directory.GetCurrentDirectory()).Length; // Get file count in current directory
|
int totalFileCount = Directory.GetFiles(Directory.GetCurrentDirectory(), "*", SearchOption.AllDirectories).Length;
|
||||||
return fileCount;
|
return totalFileCount - 3; // Remove the program, datbase and library from the totalFileCount
|
||||||
|
}
|
||||||
|
|
||||||
|
private string[] indexFiles() {
|
||||||
|
string[] indexedFiles = Directory.GetFiles(Directory.GetCurrentDirectory(), "*", SearchOption.AllDirectories);
|
||||||
|
string[] filesToExclude = { "Chksum", "chksum.db", "libe_sqlite3.so" };
|
||||||
|
indexedFiles = indexedFiles.Where(file => !filesToExclude.Contains(Path.GetFileName(file))).ToArray();
|
||||||
|
return indexedFiles;
|
||||||
}
|
}
|
||||||
|
|
||||||
public string DatabaseRoot { get; set; } = string.Empty;
|
public string DatabaseRoot { get; set; } = string.Empty;
|
||||||
@ -64,21 +70,6 @@ public class ChksumUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private string[] CalculateMD5(string[] filenames) {
|
|
||||||
string[] checksums = new string[filenames.Length];
|
|
||||||
|
|
||||||
Parallel.ForEach(filenames, (filename, state, index) => {
|
|
||||||
using (var md5 = MD5.Create()) {
|
|
||||||
using (var stream = File.OpenRead(filename)) {
|
|
||||||
var hash = md5.ComputeHash(stream);
|
|
||||||
checksums[index] = BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
return checksums;
|
|
||||||
}
|
|
||||||
|
|
||||||
private Dictionary<string, string> CalculateChecksums(string[] filenames) {
|
private Dictionary<string, string> CalculateChecksums(string[] filenames) {
|
||||||
Dictionary<string, string> checksums = new Dictionary<string, string>();
|
Dictionary<string, string> checksums = new Dictionary<string, string>();
|
||||||
|
|
||||||
@ -99,15 +90,9 @@ public class ChksumUtils {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public void doTheThing() {
|
public void doTheThing() {
|
||||||
foreach (var directory in Directory.GetDirectories(Directory.GetCurrentDirectory()))
|
|
||||||
using (var connection = new SqliteConnection("Data Source=" + DatabaseRoot + "chksum.db;Mode=ReadWrite")) {
|
using (var connection = new SqliteConnection("Data Source=" + DatabaseRoot + "chksum.db;Mode=ReadWrite")) {
|
||||||
Directory.SetCurrentDirectory(directory); // Set new root
|
if (getTotalFileCount() >= 1) {
|
||||||
if (getFileCount() >= 1) {
|
Dictionary<string, string> fileHashes = CalculateChecksums(indexFiles());
|
||||||
string[] filenames = Directory.GetFiles(directory);
|
|
||||||
Dictionary<string, string> fileHashes = CalculateChecksums(filenames);
|
|
||||||
//DirectoryInfo dir = new DirectoryInfo(Directory.GetCurrentDirectory());
|
|
||||||
//FileInfo[] files = dir.GetFiles();
|
|
||||||
int index = 0;
|
|
||||||
foreach (var file in fileHashes) {
|
foreach (var file in fileHashes) {
|
||||||
string absolutePathToFile = file.Key;
|
string absolutePathToFile = file.Key;
|
||||||
string fileName = Path.GetFileName(absolutePathToFile);
|
string fileName = Path.GetFileName(absolutePathToFile);
|
||||||
@ -130,7 +115,6 @@ public class ChksumUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
doTheThing();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user