public ProjectRunner(
string projectName,
IModelContext context,
ITimeService timeService,
IReporter reporter,
string dataDbConnectionString,
IDataDbCreatorFactory dataDbCreatorFactory,
IUsersRunnerFactory usersRunnerFactory,
IServiceRunnerFactory serviceRunnerFactory,
ICyclicRunnerFactory cyclicRunnerFactory,
IIoDeviceRunTargetFactory ioDeviceRunTargetFactory,
IAlarmsRunTargetFactory alarmsRunTargetFactory,
ITrendsRunTargetFactory trendsRunTargetFactory,
ILogRunTargetFactory logMaintainerFactory)
{
_projectName = projectName;
_context = context;
_timeService = timeService;
_reporter = reporter;
_dataDbConnectionString = dataDbConnectionString;
_dataDbCreator = dataDbCreatorFactory.Create(_dataDbConnectionString, _reporter);
_usersRunnerFactory = usersRunnerFactory;
_serviceRunnerFactory = serviceRunnerFactory;
_cyclicRunnerFactory = cyclicRunnerFactory;
_ioDeviceRunTargetFactory = ioDeviceRunTargetFactory;
_alarmsRunTargetFactory = alarmsRunTargetFactory;
_trendsRunTargetFactory = trendsRunTargetFactory;
_logMaintainerFactory = logMaintainerFactory;
}
public Runner(ISemanticVersioning versioning, INuGet nuget, NuGetConfiguration config, IReporter reporter)
{
_versioning = versioning;
_nuget = nuget;
_config = config;
_reporter = reporter;
}
public Simulator(IReporter reporter, ILogger logger)
{
_reporter = reporter;
_logger = logger;
robot = new Robot();
instruction = new Instruction();
}
public void Start(IReporter report)
{
Tests = 0;
Errors = 0;
Failures = 0;
suiteChrono.Start();
this.report = report ?? new JUnitReport("Unamed");
}
public ProviderServiceValidator(
IHttpRequestSender httpRequestSender,
IReporter reporter) : this(
new ProviderServiceResponseComparer(reporter),
httpRequestSender,
reporter)
{
}
public static void BeforeRunAssembly(TestContext context)
{
log = LogManager.GetLogger(typeof(MbUnitHelperSuper));
killDriverExecutablesProcesses();
Settings.ProjectData.Default.CURRENT_TEST_SUITE = "";
reporter = new Reporter();
reporter.addNewTestExecution();
}
public ProviderServiceValidator(HttpClient httpClient, IReporter reporter)
: this(new ProviderServiceResponseComparer(reporter),
httpClient,
new HttpRequestMessageMapper(),
new ProviderServiceResponseMapper(),
reporter)
{
}
private IProviderServiceValidator GetSubject()
{
_mockResponseComparer = Substitute.For<IProviderServiceResponseComparer>();
_mockHttpRequestSender = Substitute.For<IHttpRequestSender>();
_mockReporter = Substitute.For<IReporter>();
return new ProviderServiceValidator(_mockResponseComparer, _mockHttpRequestSender, _mockReporter);
}
public static IReporter getReporterInstance()
{
if (reporter == null)
{
reporter = new Reporter();
}
return reporter;
}
/// <summary>
/// construct an Automation Engine
/// </summary>
/// <param name="parser">the file parser</param>
/// <param name="workingDir">the working directory</param>
public Automation(IFileParser parser, IReporter reporter, string workingDir = "")
: this(workingDir)
{
Parser = parser;
Parser.WorkingDir = WorkingDir;
Reporter = reporter;
Reporter.WorkingDir = WorkingDir;
}
public MockProviderAdminRequestHandler(
IMockProviderRepository mockProviderRepository,
IReporter reporter,
IProviderServiceRequestComparer requestComparer)
{
_mockProviderRepository = mockProviderRepository;
_reporter = reporter;
_requestComparer = requestComparer;
}
public ProviderServiceRequestComparer(IReporter reporter)
{
_reporter = reporter;
_httpMethodComparer = new HttpMethodComparer(MessagePrefix, _reporter);
_httpPathComparer = new HttpPathComparer(MessagePrefix, _reporter);
_httpQueryStringComparer = new HttpQueryStringComparer(MessagePrefix, _reporter);
_httpHeaderComparer = new HttpHeaderComparer(MessagePrefix, _reporter);
_httpBodyComparer = new HttpBodyComparer(MessagePrefix, _reporter);
}
internal ProviderServiceValidator(
IProviderServiceResponseComparer providerServiceResponseComparer,
IHttpRequestSender httpRequestSender,
IReporter reporter)
{
_providerServiceResponseComparer = providerServiceResponseComparer;
_httpRequestSender = httpRequestSender;
_reporter = reporter;
}
protected void Attach(IReporter Reporter)
{
if (_reporterList == null)
{
_reporterList = new List<IReporter>();
}
_reporterList.Add(Reporter);
Reporter.Start(this);
}
public void SetReporter(IReporter value)
{
if (Reporter != null)
Handlers -= Handlers;
Reporter = value;
if (Reporter != null)
Handlers += value.WriteLine;
}
public ProviderServiceValidator(
IHttpRequestSender httpRequestSender,
IReporter reporter,
PactVerifierConfig config)
: this(new ProviderServiceResponseComparer(),
httpRequestSender,
reporter,
config)
{
}
public void SendMetrics(IReporter reporter, List<IMetric> metrics)
{
Debug.Log("Sending metrics to New Relic");
try {
reporter.Post(metrics);
} catch (System.Exception ex) {
Debug.LogException(ex);
}
}
// Copies all files from one directory to another.
public static void CopyAllFiles(string sourceDirectory, string targetDirectory, bool recursive, IReporter reporter)
{
if (sourceDirectory == null)
throw new ArgumentNullException("sourceDirectory");
if (targetDirectory == null)
throw new ArgumentNullException("targetDirectory");
_currentPath = sourceDirectory;
// Call the recursive method.
CopyAllFiles(new DirectoryInfo(sourceDirectory), new DirectoryInfo(targetDirectory), recursive,reporter);
}
public bool StartJob(XmlDocument xmlSettings, IReporter reporter)
{
this.reporterMaster = reporter;
logger.Info(logger.Name);
this.reporterMaster.WriteMessage(0, "test me", true);
return true;
}
public ProjectRunnerFactory(
ITimeService timeService,
IReporter reporter,
IModelContextFactory contextFactory,
INpgHelper npgHelper)
{
_timeService = timeService;
_reporter = reporter;
_contextFactory = contextFactory;
_npgHelper = npgHelper;
}
internal ProviderServiceValidator(
IProviderServiceResponseComparer providerServiceResponseComparer,
IHttpRequestSender httpRequestSender,
IReporter reporter,
PactVerifierConfig config)
{
_providerServiceResponseComparer = providerServiceResponseComparer;
_httpRequestSender = httpRequestSender;
_reporter = reporter;
_config = config;
}
public Service()
{
InitializeComponent();
eventLog1.Source = PHmiConstants.PHmiServiceName;
_reporter = new EventLogReporter(eventLog1);
_runnerFactory = new ProjectRunnerFactory(
new TimeService(),
_reporter,
new PHmiModelContextFactory(),
new NpgHelper());
}
private IMockProviderAdminRequestHandler GetSubject()
{
_mockProviderRepository = Substitute.For<IMockProviderRepository>();
_mockReporter = Substitute.For<IReporter>();
_mockRequestComparer = Substitute.For<IProviderServiceRequestComparer>();
return new MockProviderAdminRequestHandler(
_mockProviderRepository,
_mockReporter,
_mockRequestComparer);
}
public ProviderServiceValidator(
IProviderServiceResponseComparer providerServiceResponseComparer,
HttpClient httpClient,
IHttpRequestMessageMapper httpRequestMessageMapper,
IProviderServiceResponseMapper providerServiceResponseMapper,
IReporter reporter)
{
_providerServiceResponseComparer = providerServiceResponseComparer;
_httpClient = httpClient;
_httpRequestMessageMapper = httpRequestMessageMapper;
_providerServiceResponseMapper = providerServiceResponseMapper;
_reporter = reporter;
}
public Record(IReporter source, IEvent e, int aStamp)
{
if (null == source)
{
throw new ArgumentNullException("reporter");
}
if (null == e)
{
throw new ArgumentNullException("e");
}
this._source = source;
this._event = e;
this._stamp = aStamp;
}
public void Run(Microsoft.VisualStudio.TestTools.UnitTesting.TestContext testContext, TempReader temp)
{
_reporter = new Reporter(testContext);
string dllFullname = DllLoader.GetDllFullNameContainTCInProject(temp.ProjectName, temp.TcFolder);
Type t;
if (DllLoader.HasFoundTypeByTcid(dllFullname, temp.TcID.ToString(), out t))
{
DataRow globalData, testData;
this.LoadData(temp.PID.ToString(), temp.TcID.ToString(), out globalData, out testData);
Execute(new DataUse(globalData), new DataUse(testData), t, temp.PID.ToString(), temp.TcID.ToString());
}
else
throw new NotImplementedException(string.Format("Test case#{0} in Project {1} does not exist!", temp.TcID, t, temp.PID));
}
internal static PacketHandler Create(WebSocket sender, Packet packet, IReporter reporter)
{
var type = Type.GetType(String.Format("Client.PacketHandlers.{0}", packet.Name));
if (type == null)
return null;
var handler = Activator.CreateInstance(type) as PacketHandler;
if (handler == null)
return null;
handler.BasePacket = packet;
handler.Logger = new Core.Logger(packet.Name + "(PacketHandler)", reporter);
handler.Sender = sender;
return handler;
}
public Cook(string publicIPv6, string name, OvenUri oven, IServer server)
{
if(string.IsNullOrEmpty(publicIPv6))
throw new ArgumentNullException("clientPublicIPv6");
if(string.IsNullOrEmpty(name))
throw new ArgumentNullException("name");
if(null == oven)
throw new ArgumentNullException("oven");
_name = name;
_oven = new Autogenerated.ControlledOvenProxy(oven, server);
UriBuilder builder = new UriBuilder("http", publicIPv6);
builder.Fragment = name + "/";
_uri = builder.Uri;
_alarm = new AlarmBell(_uri);
_reporter = new Reporter(_uri, new Autogenerated.ReportingOvenProxy(oven, server));
}
internal static PacketHandler Create(WebSocket sender, long userId, Packet packet, string tag, IReporter reporter)
{
var type = Type.GetType(String.Format("Server.PacketHandlers.{0}", packet.Name));
if (type == null)
return null;
var handler = Activator.CreateInstance(type) as PacketHandler;
if (handler == null)
return null;
handler.BasePacket = packet;
handler.Logger = new Logger(tag + " " + packet.Name, reporter);
handler.Sender = sender;
handler.UserId = userId;
return handler;
}
public static int Main(string[] args)
{
System.Console.WriteLine("Keys : \n R : Rerun\n B : Run in browser\n Any other key : Exit\n\n");
var currentConfigurationFile = Path.Combine(Directory.GetCurrentDirectory(), "forseti.yaml");
_executor = Executor.WithForsetiConfigurationFile(currentConfigurationFile);
_executor.ReportWith<Reporter>();
_executor.RegisterWatcher<Windows.Growl.GrowlHarnessWatcher>();
_executor.RegisterWatcher<ConsoleHarnessWatcher>();
_reporter = new Reporter(_executor.GetReportingOptions());
_reporter.ReportSummary(_executor.ExecuteTests());
for( ;; )
{
var key = System.Console.ReadKey();
if( key.KeyChar != 0x0 )
{
switch (key.Key)
{
case ConsoleKey.R:
_reporter.ReportOn(_executor.ExecuteTests());
break;
case ConsoleKey.B:
var target = Path.GetTempPath() + @"Forseti/runner.html";
System.Diagnostics.Process.Start(target);
break;
default:
System.Diagnostics.Process.GetCurrentProcess().Kill ();
break;
}
}
Thread.Sleep(20);
}
return 0;
}
void IPlugin.PerformAction(IPluginContext context, IReporter reporter)
{
_pluginContext = context;
_reporter = reporter;
string line;
var lineNumber = 0;
while ((line = context.FileContents.ReadLine()) != null)
{
lineNumber++;
#region for store procedure
var SpItems = ParseString.GetNames(Pattern.sp, line, ErrorNameType.sp_);
if (SpItems.schema.Length > 0 ||
SpItems.objName.Length > 0)
{
TwoPartNameCheck(SpItems.schema, line, lineNumber);
if (!SpItems.objName.StartsWith("usp_", StringComparison.Ordinal))
{
var column = line.IndexOf(SpItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-usp_ perfix and lowercase",
"Should use usp_ stored procedure prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for user function
var fuItems = ParseString.GetNames(Pattern.fn, line, ErrorNameType.uf_);
if (fuItems.schema.Length > 0 ||
fuItems.objName.Length > 0)
{
TwoPartNameCheck(fuItems.schema, line, lineNumber);
if (!fuItems.objName.StartsWith("ufn_", StringComparison.Ordinal))
{
var column = line.IndexOf(fuItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-ufn_ perfix and lowercase",
"Should use ufn_ user function prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for user type
var utItems = ParseString.GetNames(Pattern.ut, line, ErrorNameType.ubt_);
if (utItems.schema.Length > 0 ||
utItems.objName.Length > 0)
{
TwoPartNameCheck(utItems.schema, line, lineNumber);
if (!utItems.objName.StartsWith("ut_", StringComparison.Ordinal))
{
var column = line.IndexOf(utItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-ut_ perfix and lowercase",
"Should use ut_ user Type prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for View
var uvItems = ParseString.GetNames(Pattern.uv, line, ErrorNameType.uvv_);
if (uvItems.schema.Length > 0 ||
uvItems.objName.Length > 0)
{
TwoPartNameCheck(uvItems.schema, line, lineNumber);
if (!uvItems.objName.StartsWith("uv_", StringComparison.Ordinal))
{
var column = line.IndexOf(uvItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-uv_ perfix and lowercase",
"Should use uv_ view prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for Index
var ixItems = ParseString.GetNames(Pattern.ix, line, ErrorNameType.idx_);
if (ixItems.schema.Length > 0 ||
ixItems.objName.Length > 0)
{
if (!ixItems.objName.StartsWith("ix_", StringComparison.Ordinal))
{
var column = line.IndexOf(ixItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-ix_ perfix and lowercase",
"Should use ix_ index prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for Trigger
var triItems = ParseString.GetNames(Pattern.tri, line, ErrorNameType.trr_);
if (triItems.schema.Length > 0 ||
triItems.objName.Length > 0)
{
TwoPartNameCheck(triItems.schema, line, lineNumber);
if (!triItems.objName.StartsWith("tri_", StringComparison.Ordinal))
{
var column = line.IndexOf(triItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-tri_ perfix and lowercase",
"Should use tri_ trigger prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
continue;
}
#endregion
#region for sequence
var seqItems = ParseString.GetNames(Pattern.seq, line, ErrorNameType.seqq_);
if (seqItems.schema.Length > 0 ||
seqItems.objName.Length > 0)
{
TwoPartNameCheck(seqItems.schema, line, lineNumber);
if (!seqItems.objName.StartsWith("seq_", StringComparison.Ordinal))
{
var column = line.IndexOf(seqItems.objName);
_reporter.ReportViolation(new CustomeRuleViolation(
_pluginContext.FilePath,
"prefer-seq_ perfix and lowercase",
"Should use seq_ sequence prefix and lowercase",
lineNumber,
column,
RuleViolationSeverity.Error));
}
}
#endregion
}
}
public BlockyDecompression([NotNull] Stream reader, [NotNull] IReporter <OfcNumber> target) : this(reader)
{
_writer = target;
_numberWriter = this;
}
public TrailingWhiteSpace(IPluginContext context, IReporter reporter)
{
this.context = context;
this.reporter = reporter;
}
public SlackReportGenerator(IDatabaseContextProvider databaseProvider, IReporter reporter)
{
_databaseProvider = databaseProvider;
_reporter = reporter;
}
public Program(IConsole console)
{
_console = console;
_reporter = new ConsoleReporter(console);
}
public MsBuildFileSetFactoryTest(ITestOutputHelper output)
{
_reporter = new TestReporter(output);
_testAssets = new TestAssetsManager(output);
}
public RounderReporter([NotNull] IReporter <OfcNumber> nextReporter, [NotNull] IConfiguaration config)
{
_nextReporter = nextReporter;
Configuaration = config;
_routeMode = true;
}
public void reportResults(String category, String message, LogLevel logLevel)
{
reporter = MbUnitHelperSuper.getReporterInstance();
reporter.reportStepResults(true, category, message, logLevel.ToString(), "");
}
public void AddReporter(IReporter reporter)
{
CurrentReporter().AddReporter(reporter);
}
public void RemoveReporter(IReporter reporter)
{
CurrentReporter().RemoveReporter(reporter);
}
/// <summary>
/// Removes the reporter to report for the current active thread.
/// </summary>
public static void RemoveReporter(IReporter reporter)
{
s_reporter.RemoveReporter(reporter);
}
/// <summary>
/// Adds the reporter to report for the current active thread.
/// </summary>
public static void AddReporter(IReporter reporter)
{
s_reporter.AddReporter(reporter);
}
public HotReload(ProcessRunner processRunner, IReporter reporter)
{
_staticFileHandler = new StaticFileHandler(reporter);
_scopedCssFileHandler = new ScopedCssFileHandler(reporter);
_compilationHandler = new CompilationHandler(reporter);
}
private static (JwtCreatorOptions, bool, string) ValidateArguments(
IReporter reporter,
CommandOption projectOption,
CommandOption schemeNameOption,
CommandOption nameOption,
CommandOption audienceOption,
CommandOption issuerOption,
CommandOption notBeforeOption,
CommandOption expiresOnOption,
CommandOption validForOption,
CommandOption rolesOption,
CommandOption scopesOption,
CommandOption claimsOption)
{
var isValid = true;
var project = DevJwtCliHelpers.GetProject(projectOption.Value());
if (project == null)
{
reporter.Error(Resources.ProjectOption_ProjectNotFound);
isValid = false;
// Break out early if we haven't been able to resolve a project
// since we depend on it for the managing of JWT tokens
return(
null,
isValid,
string.Empty
);
}
var scheme = schemeNameOption.HasValue() ? schemeNameOption.Value() : "Bearer";
var optionsString = schemeNameOption.HasValue() ? $"{Resources.JwtPrint_Scheme}: {scheme}{Environment.NewLine}" : string.Empty;
var name = nameOption.HasValue() ? nameOption.Value() : Environment.UserName;
optionsString += $"{Resources.JwtPrint_Name}: {name}{Environment.NewLine}";
var audience = audienceOption.HasValue() ? audienceOption.Values : DevJwtCliHelpers.GetAudienceCandidatesFromLaunchSettings(project);
optionsString += audienceOption.HasValue() ? $"{Resources.JwtPrint_Audiences}: {string.Join(", ", audience)}{Environment.NewLine}" : string.Empty;
if (audience is null || audience.Count == 0)
{
reporter.Error(Resources.CreateCommand_NoAudience_Error);
isValid = false;
}
var issuer = issuerOption.HasValue() ? issuerOption.Value() : DevJwtsDefaults.Issuer;
optionsString += issuerOption.HasValue() ? $"{Resources.JwtPrint_Issuer}: {issuer}{Environment.NewLine}" : string.Empty;
var notBefore = DateTime.UtcNow;
if (notBeforeOption.HasValue())
{
if (!ParseDate(notBeforeOption.Value(), out notBefore))
{
reporter.Error(Resources.FormatCreateCommand_InvalidDate_Error("--not-before"));
isValid = false;
}
optionsString += $"{Resources.JwtPrint_NotBefore}: {notBefore:O}{Environment.NewLine}";
}
var expiresOn = notBefore.AddMonths(3);
if (expiresOnOption.HasValue())
{
if (!ParseDate(expiresOnOption.Value(), out expiresOn))
{
reporter.Error(Resources.FormatCreateCommand_InvalidDate_Error("--expires-on"));
isValid = false;
}
if (validForOption.HasValue())
{
reporter.Error(Resources.CreateCommand_InvalidExpiresOn_Error);
isValid = false;
}
else
{
optionsString += $"{Resources.JwtPrint_ExpiresOn}: {expiresOn:O}{Environment.NewLine}";
}
}
if (validForOption.HasValue())
{
if (!TimeSpan.TryParseExact(validForOption.Value(), _timeSpanFormats, CultureInfo.InvariantCulture, out var validForValue))
{
reporter.Error(Resources.FormatCreateCommand_InvalidPeriod_Error("--valid-for"));
}
expiresOn = notBefore.Add(validForValue);
if (expiresOnOption.HasValue())
{
reporter.Error(Resources.CreateCommand_InvalidExpiresOn_Error);
isValid = false;
}
else
{
optionsString += $"{Resources.JwtPrint_ExpiresOn}: {expiresOn:O}{Environment.NewLine}";
}
}
var roles = rolesOption.HasValue() ? rolesOption.Values : new List <string>();
optionsString += rolesOption.HasValue() ? $"{Resources.JwtPrint_Roles}: [{string.Join(", ", roles)}]{Environment.NewLine}" : string.Empty;
var scopes = scopesOption.HasValue() ? scopesOption.Values : new List <string>();
optionsString += scopesOption.HasValue() ? $"{Resources.JwtPrint_Scopes}: {string.Join(", ", scopes)}{Environment.NewLine}" : string.Empty;
var claims = new Dictionary <string, string>();
if (claimsOption.HasValue())
{
if (!DevJwtCliHelpers.TryParseClaims(claimsOption.Values, out claims))
{
reporter.Error(Resources.CreateCommand_InvalidClaims_Error);
isValid = false;
}
optionsString += $"{Resources.JwtPrint_CustomClaims}: [{string.Join(", ", claims.Select(kvp => $"{kvp.Key}={kvp.Value}"))}]{Environment.NewLine}";
}
return(
new JwtCreatorOptions(scheme, name, audience, issuer, notBefore, expiresOn, roles, scopes, claims),
isValid,
optionsString);
public Builder WithReporter(IReporter reporter)
{
this._reporter = reporter;
return(this);
}
protected void Detach(IReporter Reporter)
{
Reporter.Stop();
_reporterList.Remove(Reporter);
}
public HttpPathComparer(string messagePrefix, IReporter reporter)
{
_messagePrefix = messagePrefix;
_reporter = reporter;
}
public ConfigReader(IReporter reporter)
: this(reporter, new FileSystem())
{
}
public StrykerRunner(IInitialisationProcess initialisationProcess = null, IMutationTestProcess mutationTestProcess = null, IReporter reporter = null)
{
_initialisationProcess = initialisationProcess;
_mutationTestProcess = mutationTestProcess;
_reporter = reporter;
}
public ProgramBase(IConsole console)
{
_console = console ?? throw new ArgumentNullException(nameof(console));
_reporter = new ConsoleReporter(_console, verbose: false, quiet: false);
}
public static WorkingBeatmap GetBeatmap(int beatmapId, bool verbose = false, bool forceDownload = true, IReporter reporter = null)
{
string fileLocation = Path.Combine(AppSettings.BEATMAPS_PATH, beatmapId.ToString()) + ".osu";
if ((forceDownload || !File.Exists(fileLocation)) && AppSettings.ALLOW_DOWNLOAD)
{
Stream stream;
if (verbose)
{
reporter?.Verbose($"Downloading {beatmapId}.");
}
stream = GetBeatmapByBid(beatmapId);
if (stream == null)
{
var req = new WebRequest(string.Format(AppSettings.DOWNLOAD_PATH, beatmapId))
{
AllowInsecureRequests = true,
};
req.Failed += _ =>
{
if (verbose)
{
reporter?.Error($"Failed to download {beatmapId}.");
}
};
req.Finished += () =>
{
if (verbose)
{
reporter?.Verbose($"{beatmapId} successfully downloaded.");
}
};
req.Perform();
RedisHelper.HSet("beatmap", beatmapId.ToString(), req.GetResponseData());
stream = req.ResponseStream;
}
if (AppSettings.SAVE_DOWNLOADED)
{
using (var fileStream = File.Create(fileLocation))
{
stream.CopyTo(fileStream);
stream.Seek(0, SeekOrigin.Begin);
}
}
return(new LoaderWorkingBeatmap(stream));
}
return(!File.Exists(fileLocation) ? null : new LoaderWorkingBeatmap(fileLocation));
}
private HttpQueryStringComparer GetSubject()
{
_mockReporter = Substitute.For <IReporter>();
return(new HttpQueryStringComparer("query", _mockReporter));
}
/// <summary>
/// Initializes a new instance of the <see cref="CloseBrowser" /> class.
/// </summary>
/// <param name="step">The step.</param>
/// <param name="reporter">The reporter.</param>
/// <param name="driver">The driver.</param>
public CloseBrowser(ITestStep step, IReporter reporter, IAutomationDriver driver)
: base(step, reporter, driver)
{
}
public AspNetCoreDeltaApplier(IReporter reporter)
{
_reporter = reporter;
}
public RuleVisitorBuilder(IConfigReader configReader, IReporter reporter, IPluginHandler pluginHandler) : this(configReader, reporter)
{
this.pluginHandler = pluginHandler;
}
public RuleVisitorBuilder(IConfigReader configReader, IReporter reporter)
{
this.reporter = reporter;
this.configReader = configReader;
}
public ProjectIdResolver(IReporter reporter, string workingDirectory)
{
_workingDirectory = workingDirectory;
_reporter = reporter;
_targetsFile = FindTargetsFile();
}
public HttpSecuityCheckCommand(ISecurityCheckPipeline securityCheckPipline, IReporter reporter, UriHelper uriHelper)
{
SecurityCheckPipeline = securityCheckPipline ?? throw new ArgumentNullException(nameof(securityCheckPipline));
Reporter = reporter ?? throw new ArgumentNullException(nameof(reporter));
UriHelper = uriHelper ?? throw new ArgumentNullException(nameof(uriHelper));
}
public void OnExecute(CommandLineApplication app, IConsole console)
{
reporter = new Reporter(console, LogFile)
{
IsQuiet = Quiet,
IsVerbose = Verbose
};
if (Concurrency < 1)
{
reporter.Error("Concurrency level must be above 1.");
return;
}
threadBeatmapIds = new int[Concurrency];
if (AppSettings.RUN_AS_SANDBOX_DOCKER)
{
reporter.Output("Waiting for database...");
while (true)
{
try
{
bool initialised = false;
using (var conn = Database.GetConnection())
{
if (conn.QuerySingle <int>("SELECT `count` FROM `osu_counts` WHERE `name` = 'docker_db_step'") >= 1)
{
initialised = true;
}
}
if (initialised)
{
break;
}
}
catch
{
}
Thread.Sleep(1000);
}
}
var beatmaps = new ConcurrentQueue <int>(GetBeatmaps() ?? Enumerable.Empty <int>());
totalBeatmaps = beatmaps.Count;
var tasks = new Task[Concurrency];
for (int i = 0; i < Concurrency; i++)
{
int tmp = i;
tasks[i] = Task.Factory.StartNew(() =>
{
var calc = new ServerDifficultyCalculator(Rulesets, Converts, DryRun);
while (beatmaps.TryDequeue(out int beatmapId))
{
threadBeatmapIds[tmp] = beatmapId;
reporter.Verbose($"Processing difficulty for beatmap {beatmapId}.");
try
{
var beatmap = BeatmapLoader.GetBeatmap(beatmapId, Verbose, ForceDownload, reporter);
// ensure the correct online id is set
beatmap.BeatmapInfo.OnlineBeatmapID = beatmapId;
calc.ProcessBeatmap(beatmap);
reporter.Verbose($"Difficulty updated for beatmap {beatmapId}.");
}
catch (Exception e)
{
reporter.Error($"{beatmapId} failed with {e}");
}
Interlocked.Increment(ref processedBeatmaps);
}
});
}
reporter.Output($"Processing {totalBeatmaps} beatmaps.");
using (new Timer(_ => outputProgress(), null, 1000, 1000))
using (new Timer(_ => outputHealth(), null, 5000, 5000))
Task.WaitAll(tasks);
if (AppSettings.RUN_AS_SANDBOX_DOCKER)
{
using (var conn = Database.GetConnection())
{
conn.Execute("INSERT INTO `osu_counts` (`name`, `count`) VALUES (@Name, @Count) ON DUPLICATE KEY UPDATE `count` = @Count", new
{
Name = "docker_db_step",
Count = 2
});
}
}
outputProgress();
reporter.Output("Done.");
}
public static void Init(ILogger logger, IReporter reporter)
{
Logger = logger;
Reporter = reporter;
}
