public void CanShowHelp()
{
// Arrange
var opts = new GetOpt("desc",
new[]
{
new CommandLineOption('a', null, "a param", ParameterType.None, null),
new CommandLineOption('b', "btext", "a param", ParameterType.None, null),
new CommandLineOption('c', null, "a param", ParameterType.None, null),
});
// Act
opts.ShowUsage(false);
// Assert
var output = this.StdOutLines;
Assert.AreEqual(8, output.Length);
Assert.AreEqual("desc", output[0]);
Assert.IsTrue(output[1].StartsWith("Usage: "));
Assert.IsTrue(output[1].EndsWith(" -abch"));
}
private static void Main(string[] args)
{
string Server = "localhost";
string DatabaseName = null;
string ConnectionString = null;
string Table = null;
string Query = null;
string Format = null;
bool Follow = false;
int SleepInterval = 200;
bool Retry = false;
var opts = new GetOpt("Tail a database query just like you tail a file",
new[]
{
new CommandLineOption('S', "server", "Database Server [localhost]", ParameterType.String, o => Server = (string)o),
new CommandLineOption('d', "database", "Database Name", ParameterType.String, o => DatabaseName = (string)o),
new CommandLineOption('c', "connectionstring", "Connection String (Overrides --server and --database)", ParameterType.String, o => ConnectionString = (string)o),
new CommandLineOption('t', "table", "Table Name to Query", ParameterType.String, o => Table = (string)o),
new CommandLineOption('q', "query", "Custom SQL Query (Overrides --table)", ParameterType.String, o => Query = (string)o),
new CommandLineOption('F', "format", "Custom String.Format output", ParameterType.String, o => Format = (string)o),
new CommandLineOption('f', "follow", "Follow output as it gets added", ParameterType.None, o => Follow = true),
new CommandLineOption('s', "sleep-interval", "Sleep interval in ms [200]", ParameterType.Integer, o => SleepInterval = (int)o),
new CommandLineOption('r', "retry", "keep trying to query if it is unsuccessful", ParameterType.None, o => Retry = true)
});
opts.ParseOptions(args);
var valid = true;
if (String.IsNullOrEmpty(ConnectionString) && String.IsNullOrEmpty(DatabaseName))
{
Console.Error.WriteLine("Database connection information required.");
valid = false;
}
if (String.IsNullOrEmpty(Table) && String.IsNullOrEmpty(Query))
{
Console.Error.WriteLine("Table Name or Custom SQL Query required.");
valid = false;
}
if (!valid)
{
opts.ShowUsage();
return;
}
bool ThrewException = false;
do
{
ThrewException = false;
try
{
using (SqlConnection conn = new SqlConnection())
{
conn.ConnectionString = ConnectionString ?? String.Format("Server={0};Database={1};Trusted_Connection=true", Server, DatabaseName);
conn.Open();
var cmd = new SqlCommand(Query ?? String.Format("SELECT * FROM {0}", Table), conn);
string lastRow = null;
var triggered = true;
do
{
using (SqlDataReader reader = cmd.ExecuteReader())
{
while (reader.Read())
{
var values = new object[reader.FieldCount];
reader.GetValues(values);
var currentRow = String.IsNullOrEmpty(Format) ? String.Join(" ", values) : String.Format(Format, values);
if (!triggered)
{
if (lastRow == currentRow)
{
triggered = true;
}
continue;
}
Console.WriteLine(currentRow);
lastRow = currentRow;
}
triggered = false;
}
if(Follow) Thread.Sleep(TimeSpan.FromMilliseconds(SleepInterval));
} while (Follow);
}
}
catch (Exception e)
{
Console.Error.WriteLine(e.Message);
ThrewException = true;
}
if (Retry && ThrewException) Thread.Sleep(TimeSpan.FromMilliseconds(SleepInterval));
} while (Retry && ThrewException);
}
