public void GenerateMethods(DbRowsetDeclaration d, CodeWriter.CodeWriter w)
{
using (w.B($"public async Task<Row> NextAsync()"))
{
w._($"if (await _reader.ReadAsync() == false) return null;",
$"var r = new Row();");
var fidx = 0;
foreach (var field in d.Fields)
{
w._($"var v{fidx} = _reader.GetValue({fidx});");
if (field.Nullable)
{
var ntype = field.Type + (DbTypeHelper.IsValueType(field.Type) ? "?" : "");
w._($"r.{field.Name} = (v{fidx} is DBNull) ? ({ntype})null : ({field.Type})v{fidx};");
}
else
{
var ivalue = DbTypeHelper.GetInitValue(field.Type);
w._($"r.{field.Name} = (v{fidx} is DBNull) ? {ivalue} : ({field.Type})v{fidx};");
}
fidx += 1;
}
w._($"return r;");
}
using (w.B($"public async Task<List<Row>> FetchAllRowsAndDisposeAsync()"))
{
w._($"var rows = new List<Row>();");
using (w.b($"while (true)"))
{
w._($"var row = await NextAsync();",
$"if (row == null) break;",
$"rows.Add(row);");
}
w._($"Dispose();",
$"return rows;");
}
using (w.B($"public async Task<List<T>> FetchAllRowsAndDisposeAsync<T>(Func<Row, T> selector)"))
{
w._($"var rows = new List<T>();");
using (w.b($"while (true)"))
{
w._($"var row = await NextAsync();",
$"if (row == null) break;",
$"rows.Add(selector(row));");
}
w._($"Dispose();",
$"return rows;");
}
using (w.B($"public void Dispose()"))
{
w._($"_reader.Dispose();");
}
}
public List <DbTableTypeDeclaration> Parse(string sql)
{
var decls = new List <DbTableTypeDeclaration>();
var text = sql.Replace('\n', ' ').Replace('\r', ' ');
var matches = Regex.Matches(text,
@"CREATE\s+TYPE\s+(\[dbo\]\.)?\[(\w+)\]\s?AS\s?TABLE\s?\((.+?)[^0-9]\)");
foreach (Match match in matches)
{
var decl = new DbTableTypeDeclaration();
decl.TypeName = match.Groups[2].Value;
decl.Fields = new List <DbField>();
var paramsText = match.Groups[3].Value.Trim();
if (paramsText.StartsWith("(") && paramsText.EndsWith(")"))
{
paramsText = paramsText.Substring(1, paramsText.Length - 2);
}
foreach (var p in paramsText.Split(','))
{
var fields = p.Split(new[] { ' ', '\t' }, StringSplitOptions.RemoveEmptyEntries)
.Where(w => w.ToLower() != "as").ToArray();
if (fields.Length == 0)
{
continue;
}
var fieldName = fields[0];
var fieldType = fields[1];
var type = DbTypeHelper.GetTypeFromSqlType(fieldType);
if (type == null)
{
throw new Exception("Cannot resolve type: " + fieldType);
}
decl.Fields.Add(new DbField
{
Name = fieldName.Replace("[", "").Replace("]", ""),
Type = type.Item1,
Len = type.Item2,
Dir = ""
});
}
decls.Add(decl);
}
return(decls);
}
public void Generate(DbRowsetDeclaration d, CodeWriter.CodeWriter w)
{
using (w.B($"public class {d.ClassName} : IDisposable"))
{
w._($"private DbDataReader _reader;");
w._();
using (w.B($"public {d.ClassName}(DbDataReader reader)"))
{
w._($"_reader = reader;");
}
using (w.B($"public class Row"))
{
foreach (var field in d.Fields)
{
w._($"public {DbTypeHelper.GetMemberDecl(field)};");
}
}
GenerateMethods(d, w);
}
}
public List <DbProcDeclaration> Parse(string sql)
{
var decls = new List <DbProcDeclaration>();
var text = sql.Replace('\n', ' ').Replace('\r', ' ');
var matches = Regex.Matches(text,
@"CREATE\s+PROCEDURE\s+(\[dbo\]\.)?\[(\w+)\](.+?)AS\s+BEGIN",
RegexOptions.IgnoreCase);
foreach (Match match in matches)
{
var decl = new DbProcDeclaration();
decl.ProcName = match.Groups[2].Value;
decl.Params = new List <DbField>();
var paramsText = match.Groups[3].Value.Trim();
if (paramsText.StartsWith("(") && paramsText.EndsWith(")"))
{
paramsText = paramsText.Substring(1, paramsText.Length - 2);
}
foreach (var p in paramsText.Split(','))
{
var parameters = p.Split(new[] { ' ', '\t' }, StringSplitOptions.RemoveEmptyEntries)
.Where(w => w.ToLower() != "as").ToArray();
if (parameters.Length == 0)
{
continue;
}
var paramName = parameters[0];
var paramType = parameters[1];
var paramOutput = (parameters.Length >= 3 && parameters[2].ToLower() == "output");
var paramReadonly = (parameters.Length >= 3 && parameters[2].ToLower() == "readonly");
var type = DbTypeHelper.GetTypeFromSqlType(paramType);
if (type == null)
{
if (paramReadonly)
{
type = Tuple.Create("DataTable", 0);
}
else
{
throw new Exception("Cannot resolve type: " + paramType);
}
}
decl.Params.Add(new DbField
{
Name = paramName.Substring(1),
Type = type.Item1,
Len = type.Item2,
Dir = paramOutput ? "out" : ""
});
}
decls.Add(decl);
}
return(decls);
}
