lsm/PdfPig
1
0
Fork 0
mirror of https://github.com/UglyToad/PdfPig.git synced 2025-06-28 15:30:17 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

support specifying number of documents for console runner

Browse Source
This commit is contained in:
Eliot Jones 2021-06-02 13:45:12 -04:00
parent 14276d5d16
commit ea4889778d
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tools/ConsoleRunner/Program.cs
View File

@ -9,9 +9,9 @@ namespace UglyToad.PdfPig.ConsoleRunner
{ {
public static int Main(string[] args) public static int Main(string[] args)
{ {
if (args.Length != 1) if (args.Length == 0)
{ {
Console.WriteLine("Only 1 argument, path to test file directory, may be provided."); Console.WriteLine("At least 1 argument, path to test file directory, may be provided.");
return 7; return 7;
} }
@ -23,12 +23,25 @@ namespace UglyToad.PdfPig.ConsoleRunner
return 7; return 7;
} }
var maxCount = default(int?);
if (args.Length > 1 && int.TryParse(args[1], out var countIn))
{
maxCount = countIn;
}
var hasError = false; var hasError = false;
var errorBuilder = new StringBuilder(); var errorBuilder = new StringBuilder();
var fileList = Directory.GetFiles(path, "*.pdf"); var fileList = Directory.GetFiles(path, "*.pdf");
var runningCount = 0;
foreach (var file in fileList) foreach (var file in fileList)
{ {
if (maxCount.HasValue && runningCount >= maxCount)
{
break;
}
try try
{ {
var numWords = 0; var numWords = 0;
@ -57,6 +70,8 @@ namespace UglyToad.PdfPig.ConsoleRunner
.Append(ex) .Append(ex)
.AppendLine(); .AppendLine();
} }
runningCount++;
} }
if (hasError) if (hasError)