92 lines
3.0 KiB
C#
92 lines
3.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using MongoDB.Bson;
|
|
using MongoDB.Driver;
|
|
|
|
namespace BuildFeed.Model
|
|
{
|
|
public partial class BuildRepository
|
|
{
|
|
public async Task<string[]> SelectAllLabs(int limit = -1, int skip = 0)
|
|
{
|
|
var query = _buildCollection.Aggregate()
|
|
.Group(new BsonDocument("_id", $"${nameof(Build.Lab)}"))
|
|
.Sort(new BsonDocument("_id", 1))
|
|
.Skip(skip);
|
|
|
|
if (limit > 0)
|
|
{
|
|
query = query.Limit(limit);
|
|
}
|
|
|
|
var grouping = await query.ToListAsync();
|
|
|
|
return (from g in grouping
|
|
where !g["_id"].IsBsonNull
|
|
select g["_id"].AsString).ToArray();
|
|
}
|
|
|
|
public async Task<string[]> SelectLabsForVersion(int major, int minor)
|
|
{
|
|
var query = _buildCollection.Aggregate()
|
|
.Match(new BsonDocument
|
|
{
|
|
new BsonElement(nameof(Build.MajorVersion), major),
|
|
new BsonElement(nameof(Build.MinorVersion), minor)
|
|
})
|
|
.Group(new BsonDocument("_id", $"${nameof(Build.Lab)}"))
|
|
.Sort(new BsonDocument("_id", 1));
|
|
|
|
var grouping = await query.ToListAsync();
|
|
|
|
return (from g in grouping
|
|
where !g["_id"].IsBsonNull
|
|
select g["_id"].AsString).ToArray();
|
|
}
|
|
|
|
public async Task<List<string>> SearchLabs(string search)
|
|
{
|
|
var result = await _buildCollection.Aggregate()
|
|
.Match(b => b.Lab != null)
|
|
.Match(b => b.Lab != "")
|
|
.Match(b => b.Lab.ToLower().Contains(search.ToLower()))
|
|
.Group(b => b.Lab.ToLower(),
|
|
// incoming bullshit hack
|
|
bg => new Tuple<string>(bg.Key))
|
|
.ToListAsync();
|
|
|
|
// work ourselves out of aforementioned bullshit hack
|
|
return result.Select(b => b.Item1).ToList();
|
|
}
|
|
|
|
public async Task<long> SelectAllLabsCount()
|
|
{
|
|
var query = _buildCollection.Aggregate()
|
|
.Group(new BsonDocument("_id", new BsonDocument(nameof(Build.Lab), $"${nameof(Build.Lab)}")))
|
|
.Sort(new BsonDocument("_id", 1));
|
|
|
|
var grouping = await query.ToListAsync();
|
|
|
|
return grouping.Count;
|
|
}
|
|
|
|
public async Task<List<Build>> SelectLab(string lab, int limit = -1, int skip = 0)
|
|
{
|
|
var query = _buildCollection.Find(new BsonDocument(nameof(Build.LabUrl), lab))
|
|
.Sort(sortByCompileDate)
|
|
.Skip(skip);
|
|
|
|
if (limit > 0)
|
|
{
|
|
query = query.Limit(limit);
|
|
}
|
|
|
|
return await query.ToListAsync();
|
|
}
|
|
|
|
public async Task<long> SelectLabCount(string lab)
|
|
=> await _buildCollection.CountDocumentsAsync(new BsonDocument(nameof(Build.LabUrl), lab));
|
|
}
|
|
} |