holm/GetMoarFediverse
1
1
Fork 0
forked from extern/GetMoarFediverse
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do up to 2 * tags concurrent requests (#20)

Browse Source
This commit is contained in:
Gervasio Marchand 2022-12-22 04:37:29 -08:00 committed by GitHub
parent 44b47acbfe
commit 83ca6bb63b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/Program.cs
View File

@ -35,28 +35,30 @@ if (!File.Exists(importedPath))
File.WriteAllText(importedPath, ""); File.WriteAllText(importedPath, "");
} }
ParallelOptions parallelOptions = new()
{
MaxDegreeOfParallelism = 8
};
var robotsFileParser = new RobotsFileParser(); var robotsFileParser = new RobotsFileParser();
var sitesRobotFile = new ConcurrentDictionary<string, RobotsFile>(); var sitesRobotFile = new ConcurrentDictionary<string, RobotsFile>();
await Parallel.ForEachAsync(Config.Instance.Sites, parallelOptions, async (site, _) => await Parallel.ForEachAsync(Config.Instance.Sites,
{ new ParallelOptions { MaxDegreeOfParallelism = Config.Instance.Sites.Length },
sitesRobotFile[site.Host] = await robotsFileParser.FromUriAsync(new Uri($"http://{site.Host}/robots.txt")); async (site, _) =>
}); {
sitesRobotFile[site.Host] = await robotsFileParser.FromUriAsync(new Uri($"http://{site.Host}/robots.txt"));
}
);
List<(string host, string tag)> sitesTags; List<(string host, string tag)> sitesTags;
int numberOfTags;
if (Config.Instance.MastodonPostgresConnectionString.HasValue()) if (Config.Instance.MastodonPostgresConnectionString.HasValue())
{ {
var tags = await MastodonConnectionHelper.GetFollowedTagsAsync(); var tags = await MastodonConnectionHelper.GetFollowedTagsAsync();
numberOfTags = tags.Count;
sitesTags = Config.Instance.Sites sitesTags = Config.Instance.Sites
.SelectMany(s => tags.Select(t => (s.Host, t))) .SelectMany(s => tags.Select(t => (s.Host, t)))
.OrderBy(e => e.t)
.ToList(); .ToList();
} }
else else
{ {
numberOfTags = Config.Instance.Tags.Length;
sitesTags = Config.Instance.Sites sitesTags = Config.Instance.Sites
.SelectMany(s => Config.Instance.Tags.Select(tag => (s.Host, tag))) .SelectMany(s => Config.Instance.Tags.Select(tag => (s.Host, tag)))
.Concat(Config.Instance.Sites.SelectMany(s => s.SiteSpecificTags.Select(tag => (s.Host, tag)))) .Concat(Config.Instance.Sites.SelectMany(s => s.SiteSpecificTags.Select(tag => (s.Host, tag))))
@ -67,7 +69,7 @@ else
var importedList = File.ReadAllLines(importedPath).ToList(); var importedList = File.ReadAllLines(importedPath).ToList();
var imported = importedList.ToHashSet(); var imported = importedList.ToHashSet();
var statusesToLoadBag = new ConcurrentBag<string>(); var statusesToLoadBag = new ConcurrentBag<string>();
await Parallel.ForEachAsync(sitesTags, parallelOptions, async (st, _) => await Parallel.ForEachAsync(sitesTags, new ParallelOptions{MaxDegreeOfParallelism = numberOfTags * 2}, async (st, _) =>
{ {
var (site, tag) = st; var (site, tag) = st;
Console.WriteLine($"Fetching tag #{tag} from {site}"); Console.WriteLine($"Fetching tag #{tag} from {site}");