C# Enum Generation
Ayende recently asked on the ALT.NET mailing list about the various methods developers use to provide lookup values, with the question framed as one between lookup tables and enums. My own preference is to use both, but keep it DRY with code generation.
To demonstrate the idea, I wrote a Ruby script that generates a C# enum file from some metadata. I much prefer Ruby to pure .NET solutions like CodeSmith—I find it easier and more powerful (I do think CodeSmith is excellent if there is no Ruby expertise on the team, however). The full source for this example can be grabbed here.
The idea is simple. I want a straightforward and extensible way to provide metadata for lookup values, following the Ruby Way of convention over configuration. XML is very popular in the .NET world, but the Ruby world views it as overly verbose, and prefers lighter markup languages like YAML. For my purposes, I decided not to mess with markup at all (although I’m still considering switching to YAML—the hash of hashes approach describes what I want well). Here’s some example metadata:
enums = {
'OrderType' => {},
'MethodOfPayment' => {:table => 'PaymentMethod',},
'StateProvince' => {:table => 'StateProvinces',
:name_column => 'Abbreviation',
:id_column => 'StateProvinceId',
:transformer => lambda {|value| value.upcase},
:filter => lambda {|value| !value.empty?}}
}
That list, which is valid Ruby code, describes three enums, which will be named OrderType
, MethodOfPayment
, and StateProvince
. The intention is that, where you followed your database standards, you should usually be able to get by without adding any extra metadata, as shown in the OrderType
example. The code generator will get the ids and enum names from the OrderType
table (expecting the columns to be named OrderTypeId
and Description
) and create the enum from those values. As StateProvince
shows, the table name and two column names can be overridden.
More interestingly, you can both transform and filter the enum names by passing lambdas (which are like anonymous delegates in C#). The ‘StateProvince’ example above will filter out any states that, after cleaning up any illegal characters, equal an empty string, and then it will upper case the name.
We use a pre-build event in our project to build the enum file. However, if you simply overwrite the file every time you build, you may slow down the build process considerably. MSBuild (used by Visual Studio) evidently sees that the timestamp has been updated, so it rebuilds the project, forcing a rebuild of all downstream dependent projects. A better solution is to only overwrite the file if there are changes:
require File.dirname(__FILE__) + '/enum_generator'
gen = EnumGenerator.new('localhost', ‘database-name’)
source = gen.generate_all(‘Namespace', enums)
filename = File.join(File.dirname(__FILE__), 'Enums.cs')
if Dir[filename].empty? || source != IO.read(filename)
File.open(filename, 'w') {|file| file << source}
end
I define the basic templates straight in the EnumGenerator
class, but allow them to be swapped out. In theory, the default name column and the default lambda for generating the id column name given the table name (or enum name) could be handled the same way. Below is the EnumGenerator
code:
class EnumGenerator
FILE_TEMPLATE = <<EOT
//------------------------------------------------------------------------------
// <auto-generated>
// This code was generated by a tool from <%= catalog %> on <%= server %>.
//
// Changes to this file may cause incorrect behavior and will be lost if
// the code is regenerated.
// </auto-generated>
//------------------------------------------------------------------------------
namespace <%= namespace %>
{
<%= enums %>
}
EOT
ENUM_TEMPLATE = <<EOT
public enum <%= enum_name %>
{
<% values.keys.sort.each_with_index do |id, i| -%>
<%= values[id] %> = <%= id %><%= ',' unless i == values.length - 1 %>
<% end -%>
}
EOT
# Change the templates by calling these setters
attr_accessor :enum_template, :file_template
attr_reader :server, :catalog
def initialize(server, catalog)
@server, @catalog = server, catalog
@enum_template, @file_template = ENUM_TEMPLATE, FILE_TEMPLATE
end
end
The code generation uses erb, the standard Ruby templating language:
def transform(template, template_binding)
erb = ERB.new(template, nil, '-')
erb.result template_binding
end
template_binding
describes the variables available to use in the template in much the same way that Castle Monorail’s PropertyBag
describes the variables available to the views. The difference is that, because Ruby is dynamic, you don’t have to explictly add values to the binding. The rest of the code is shown below:
def generate(enum_name, attributes)
table = attributes[:table] || enum_name
filter = attributes[:filter] || lambda {|value| true}
values = enum_values(table, attributes)
values.delete_if {|key, value| !filter.call(value)}
transform enum_template, binding
end
def generate_all(namespace, metadata)
enums = ''
metadata.keys.sort.each {|enum_name| enums << generate(enum_name, metadata[enum_name])}
enums = enums.gsub(/\n/m, "\n\t").strip
transform file_template, binding
end
private
def enum_values(table, attributes)
sql = get_sql table, attributes
@dbh ||= DBI.connect("DBI:ADO:Provider=SQLNCLI;server=#{server};database=#{catalog};Integrated Security=SSPI")
sth = @dbh.execute sql
values = {}
sth.each {|row| values[row['Id']] = clean(row['Name'], attributes[:transformer])}
sth.finish
values
end
def get_sql(table, attributes)
id_column = attributes[:id_column] || "#{table}Id"
name_column = attributes[:name_column] || "Description"
"SELECT #{id_column} AS Id, #{name_column} AS Name FROM #{table} ORDER BY Id"
end
def clean(enum_value, transformer=nil)
enum_value = '_' + enum_value if enum_value =~ /^\d/
enum_value = enum_value.gsub /[^\w]/, ''
transformer ||= lambda {|value| value}
transformer.call enum_value
end
Caveat Emptor: I wrote this code from scratch today; it is not the same code we currently use in production. I think it’s better, but if you find a problem with it please let me know.