T4Templates/DataRow.ttinclude at master · essenbee/T4Templates · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
using System;
using System.Text;
using System.Data;
using System.Collections.Generic;

namespace DataRows
{

public static partial class Reader
{
<#
    foreach (var dataRow in model)
    {
#>
    public static IEnumerable<<#=dataRow.Name#>> Read<#=dataRow.Name#>(this IDbConnection dbConnection, string query)
    {

<#
    foreach (var field in dataRow.Fields)
    {
#>
            <#=field.Type#> default_<#=field.Name#> = default;
<#
    }
#>

            using (var cmd = dbConnection.CreateCommand())
            {
                cmd.CommandType = CommandType.Text;
                cmd.CommandText = query;
                using (var reader = cmd.ExecuteReader())
                {
                    var fc = reader.FieldCount;
                    var row = new object[fc];

<#
    foreach (var field in dataRow.Fields)
    {
#>
                    var ord_<#=field.Name#> = reader.GetOrdinal("<#=field.Name#>");
<#
    }
#>

                    while (reader.Read())
                    {
                        // TODO: Unnecessary boxing?
                        // TODO: Two strategies, one for wide rows and one for narrow rows?
                        var afc = reader.GetValues(row);
                        if (afc != fc)
                        {
                            throw new Exception($"Expected {fc} fields but got {afc} fields");
                        }

<#
    foreach (var field in dataRow.Fields)
    {
#>
                        var val_<#=field.Name#> = (row[ord_<#=field.Name#>] is DBNull) ? default_<#=field.Name#> : ((<#=field.Type#>)row[ord_<#=field.Name#>]);
<#
    }
#>
                        yield return new <#=dataRow.Name#>
                        {
<#
    foreach (var field in dataRow.Fields)
    {
#>
                            <#=field.Name#> = val_<#=field.Name#>,
<#
    }
#>
                        };

                    }
                }
            }
        }
<#
    }
#>
}


<#
    foreach (var dataRow in model)
    {
#>
    public class <#=dataRow.Name#>
    {
        public override string ToString ()
        {
            var sb = new StringBuilder ();

            sb.Append ("{ <#=dataRow.Name#> ");
<#
        foreach (var field in dataRow.Fields)
    {
#>
            sb.Append(", <#=field.Name#> = ");
            sb.Append(<#=field.Name#>);
 <#
    }
#>
            sb.Append (" }");

            return sb.ToString ();
        }
<#
    foreach (var field in dataRow.Fields)
    {
#>
    public <#=field.Type#> <#=field.Name#> { get; set;}
<#
    }
#>
    }
<#
    }
#>
}

<#+
    class DataRow
    {
        public string   Name    ;
        public Field[]  Fields  ;
    }

    class Field
    {
        public string Name;
        public string Type;
    }
#>