private string SomeAssoc(StandardDB db, Dictionary <string, object> cache, string target, IEnumerable <uint> uids)
{
var zanfcache = cache["from-zan"] as ILookup <uint, int>;
var zans = uids.SelectMany(uid => zanfcache[uid]);
LOG.LogInformation($"we get {zans.Count()} zans");
var anss = zans.Select(idx => db.zans[idx].to).GroupBy(x => x);
var sb = new StringBuilder("[");
foreach (var ans in anss)
{
sb.AppendFormat("{{\"key\":{0},\"count\":{1}}},", ans.Key, ans.Count());
}
sb.Remove(sb.Length - 1, 1);
sb.Append("]");
return(sb.ToString());
}
private User[] GetUsers(StandardDB db, Dictionary <string, object> cache, string[] uids)
{
if (uids[0] == "#ALL_USER")
{
return(db.users.ToArray());
}
var uidcache = cache["uid-user"] as Dictionary <uint, int>;
if (uids[0] == "#BAN_UID")
{
var banuid = cache["banuid"] as HashSet <uint>;
return(banuid.Select(uid => uidcache.TryGetValue(uid, out var idx) ? idx : -1).Where(x => x >= 0)
.Select(idx => db.users[idx]).ToArray());
}
return(uids.Select(uid => UIDPool.Get(uid)).Where(uid => uid != uint.MaxValue)
.Select(uid => uidcache.TryGetValue(uid, out var idx) ? idx : -1).Where(x => x >= 0)
.Select(idx => db.users[idx]).ToArray());
}
private Dictionary <string, object> BuildCache(StandardDB db)
{
Console.WriteLine($"building cache for {ObjName}");
var caches = new Dictionary <string, object>();
{
Dictionary <uint, int> uidcache = new Dictionary <uint, int>();
HashSet <uint> banuid = new HashSet <uint>();
int idx = 0;
foreach (var user in db.users)
{
uidcache[user.id_] = idx++;
if (user.status_ == UserStatus.ban || user.status_ == UserStatus.sban)
{
banuid.Add(user.id_);
}
}
caches["uid-user"] = uidcache;
caches["banuid"] = banuid;
LOG.LogInformation($"user's index cache built");
}
{
caches["from-zan"] = db.zans.Zip(Enumerable.Range(0, db.zans.Count), (zan, idx) => (zan.from_, idx)).ToLookup(x => x.from_, x => x.idx);
caches["to-zan"] = db.zans.Zip(Enumerable.Range(0, db.zans.Count), (zan, idx) => (zan.to, idx)).ToLookup(x => x.to, x => x.idx);
LOG.LogInformation($"zanart's index cache built");
caches["from-zanart"] = db.zanarts.Zip(Enumerable.Range(0, db.zanarts.Count), (zan, idx) => (zan.from_, idx)).ToLookup(x => x.from_, x => x.idx);
caches["to-zanart"] = db.zanarts.Zip(Enumerable.Range(0, db.zanarts.Count), (zan, idx) => (zan.to, idx)).ToLookup(x => x.to, x => x.idx);
LOG.LogInformation($"zanart's index cache built");
}
{
var dict1 = db.answers.ToDictionary(ans => ans.id, ans => ans.author_);
caches["ans-author"] = dict1;
caches["author-ans"] = dict1.ToLookup(kv => kv.Value, kv => kv.Key);
caches["ans-qst"] = db.answers.ToDictionary(ans => ans.id, ans => ans.question);
LOG.LogInformation($"answer's index cache built");
var dict2 = db.articles.ToDictionary(art => art.id, art => art.author_);
caches["art-author"] = dict2;
caches["author-art"] = dict2.ToLookup(kv => kv.Value, kv => kv.Key);
LOG.LogInformation($"article's index cache built");
}
GC.Collect(2, GCCollectionMode.Optimized, false, true);
return(caches);
}
private string[][] GetZanLinks(StandardDB db, Dictionary <string, object> cache, string[] uids, string target)
{
Console.WriteLine($"here get {uids.Length} uids");
if (target == "to")
{
var zanfcache = cache["from-zan"] as ILookup <uint, int>;
var zanartfcache = cache["from-zanart"] as ILookup <uint, int>;
var ansathcache = cache["ans-author"] as Dictionary <uint, uint>;
var artathcache = cache["art-author"] as Dictionary <uint, uint>;
var zanans = uids.SelectMany(uid =>
{
var id = UIDPool.Get(uid);
if (id == uint.MaxValue)
{
return(Enumerable.Empty <string[]>());
}
return(zanfcache[id].Select(idx =>
{
ansathcache.TryGetValue(db.zans[idx].to, out var ath);
return new string[] { uid, UIDPool.GetString(ath) };
}));
}).Where(pair => pair[1] != null);
var zanart = uids.SelectMany(uid =>
{
var id = UIDPool.Get(uid);
if (id == uint.MaxValue)
{
return(Enumerable.Empty <string[]>());
}
return(zanartfcache[id].Select(idx =>
{
artathcache.TryGetValue(db.zanarts[idx].to, out var ath);
return new string[] { uid, UIDPool.GetString(ath) };
}));
}).Where(pair => pair[1] != null);
return(zanans.Concat(zanart).ToArray());
}
else if (target == "from")
{
var athanscache = cache["author-ans"] as ILookup <uint, uint>;
var athartcache = cache["author-art"] as ILookup <uint, uint>;
var zantcache = cache["to-zan"] as ILookup <uint, int>;
var zanarttcache = cache["to-zanart"] as ILookup <uint, int>;
var zanans = uids.SelectMany(uid =>
{
var id = UIDPool.Get(uid);
if (id == uint.MaxValue)
{
return(Enumerable.Empty <string[]>());
}
return(athanscache[id].SelectMany(ansid => zantcache[ansid])
.Select(idx => new string[] { db.zans[idx].from, uid }));
});
var zanart = uids.SelectMany(uid =>
{
var id = UIDPool.Get(uid);
if (id == uint.MaxValue)
{
return(Enumerable.Empty <string[]>());
}
return(athanscache[id].SelectMany(artid => zanarttcache[artid])
.Select(idx => new string[] { db.zanarts[idx].from, uid }));
});
return(zanans.Concat(zanart).ToArray());
}
else
{
return(null);
}
}
