public void GetCompanyByMarketCapTest()
{
var compRepo = new MongoDBRepository <Company>(nameof(Company));
CompanyCollection companyCollection = new CompanyCollection();
var statRepo = new MongoDBRepository <Stat>("IEXStat");
var finRepo = new MongoDBRepository <Financial>(nameof(Financial));
long minCap = 1_000_000;
long maxCap = 2_000_000_000;
var filter = Builders <Stat> .Filter.Lte(x => x.Marketcap, maxCap) &
Builders <Stat> .Filter.Gte(x => x.Marketcap, minCap);
var results = statRepo.DbCollection.FindAsync(filter).Result.ToList();
Assert.NotNull(results);
results.ForEach((x) =>
{
Console.WriteLine(x.Marketcap);
Assert.True(x.Marketcap <= maxCap, $"Actual={x.Marketcap} <= Expected={maxCap}");
});
var companies = companyCollection.GetCompanyId(results.Select(x => x.Symbol).ToList());
Assert.AreEqual(companies.Count, results.Count);
}
internal static void InsertBatch(IEnumerable <CIKInfo> cikInfos, List <IEXDataType> listIexDataTypes, ChartOption chartOption, int itemCount, Period period)
{
int total = cikInfos.Count();
int count = 0;
int batchCount = 50;
CompanyCollection companyRepo = new CompanyCollection();
do
{
var symbols = cikInfos.Select(x => x.Ticker).Skip(count).Take(batchCount).ToArray();
count += symbols.Count();
var dictSymbolComp = companyRepo.GetCompanyId(symbols.ToList());
//get data
IEXClientHandler response = new IEXClientHandler();
var results = response.GetBatchReport(symbols, listIexDataTypes, ChartOption._1d, itemCount, period);
//insert to db
var comp = new MongoDBRepository <Company>(nameof(Company));
var stat = new MongoDBRepository <Stat>("IEXStat");
var fin = new MongoDBRepository <Financial>(nameof(Financial));
var quote = new MongoDBRepository <Quote>("Quote");
foreach (string symbol in results.Keys)
{
ObjectId compId = ObjectId.GenerateNewId(); //ObjectId.GenerateNewId().ToString();
results[symbol].Company._id = compId;
if (results[symbol].Company != null)
{
comp.MongoInsert(results[symbol].Company).Wait();
}
if (results[symbol].Stat != null)
{
results[symbol].Stat.companyId = compId;
stat.MongoInsert(results[symbol].Stat).Wait();
}
if (results[symbol].Financials.financials != null)
{
Parallel.For(0, results[symbol].Financials.financials.Count, (i) =>
{
results[symbol].Financials.financials[i].companyId = compId;
});
fin.MongoInsertMany(results[symbol].Financials.financials).Wait();
}
if (results[symbol].Quote != null)
{
results[symbol].Quote.companyId = compId;
quote.MongoInsert(results[symbol].Quote).Wait();
}
}
} while (count < total);
}
internal static void InsertChartData(IEnumerable <CIKInfo> cikInfos)
{
IEXClientHandler response = new IEXClientHandler();
CompanyCollection compColl = new CompanyCollection();
//symbols from list
var symbols = cikInfos.Select(x => x.Ticker).ToList();
//checking for company exists in db
var dictComp = compColl.GetCompanyId(symbols);
//insert to db
var chartCollection = new MongoDBRepository <Chart>("Charts");
foreach (string symbol in dictComp.Keys)
{
try
{
var results = response.GetBatchReport(symbol, new List <IEXDataType> {
IEXDataType.chart
}, ChartOption._5y, 100);
if (!dictComp.ContainsKey(symbol))
{
Console.WriteLine($"Not data for symbol {symbol}.");
continue;
}
//update charts with companyId
Parallel.For(0, results.Charts.Count, (i) =>
{
results.Charts[i].companyId = dictComp[symbol];
});
chartCollection.MongoInsertMany(results.Charts).Wait();
}
catch (Exception ex)
{
Console.WriteLine(ex);
}
}
}