I'm just getting started with Anorm and parser combinators. It seems like there is an awful lot of boilerplate code. For example, I have
case class Model(
id:Int,
field1:String,
field2:Int,
// a bunch of fields omitted
)
val ModelParser:RowParser[RegdataStudentClass] = {
int("id") ~
str("field1") ~
int("field2") ~
// a bunch of fields omitted
map {
case id ~ field1 ~ field2 //more omissions
=> Model(id, field1, field2, // still more omissions
)
}
}
Each database field is repeated four (!) times before the whole thing is defined. It seems like the parser should be able to be deduced semi-automatically from the case class. Any tools or other techniques to suggest to reduce the work involved here?
Thanks for any pointers.
Here's the solution I eventually developed. I currently have this as a class in my Play project; it could (should!) be turned into a stand-alone tool. To use it, change the tableName
val to the name of your table. Then run it using the main
at the bottom of the class. It will print a skeleton of the case class and the parser combinator. Most of the time these skeletons require very little tweaking.
Byron
package tools
import scala.sys.process._
import anorm._
/**
* Generate a parser combinator for a specified table in the database.
* Right now it's just specified with the val "tableName" a few lines
* down.
*
* 20121024 bwbecker
*/
object ParserGenerator {
val tableName = "uwdata.uwdir_person_by_student_id"
/**
* Convert the sql type to an equivalent Scala type.
*/
def fieldType(field:MetaDataItem):String = {
val t = field.clazz match {
case "java.lang.String" => "String"
case "java.lang.Boolean" => "Boolean"
case "java.lang.Integer" => "Int"
case "java.math.BigDecimal" => "BigDecimal"
case other => other
}
if (field.nullable) "Option[%s]" format (t)
else t
}
/**
* Drop the schema name from a string (tablename or fieldname)
*/
def dropSchemaName(str:String):String =
str.dropWhile(c => c != '.').drop(1)
def formatField(field:MetaDataItem):String = {
"\t" + dropSchemaName(field.column) + " : " + fieldType(field)
}
/**
* Derive the class name from the table name: drop the schema,
* remove the underscores, and capitalize the leading letter of each word.
*/
def deriveClassName(tableName:String) =
dropSchemaName(tableName).split("_").map(w => w.head.toUpper + w.tail).mkString
/**
* Query the database to get the metadata for the given table.
*/
def getFieldList(tableName:String):List[MetaDataItem] = {
val sql = SQL("""select * from %s limit 1""" format (tableName))
val results:Stream[SqlRow] = util.Util.DB.withConnection { implicit connection => sql() }
results.head.metaData.ms
}
/**
* Generate a case class definition with one data member for each field in
* the database table.
*/
def genClassDef(className:String, fields:List[MetaDataItem]):String = {
val fieldList = fields.map(formatField(_)).mkString(",\n")
""" case class %s (
%s
)
""" format (className, fieldList )
}
/**
* Generate a parser for the table.
*/
def genParser(className:String, fields:List[MetaDataItem]):String = {
val header:String = "val " + className.take(1).toLowerCase() + className.drop(1) +
"Parser:RowParser[" + className + "] = {\n"
val getters = fields.map(f =>
"\tget[" + fieldType(f) + "](\"" + dropSchemaName(f.column) + "\")"
).mkString(" ~ \n")
val mapper = " map {\n case " + fields.map(f => dropSchemaName(f.column)).mkString(" ~ ") +
" =>\n\t" + className + "(" + fields.map(f => dropSchemaName(f.column)).mkString(", ") + ")\n\t}\n}"
header + getters + mapper
}
def main(args:Array[String]) = {
val className = deriveClassName(tableName)
val fields = getFieldList(tableName)
println( genClassDef(className, fields) )
println( genParser(className, fields))
}
}
Well, you actually don't have to repeat anything at all. You can use flatten
to make a tuple and then create your model instance out of that tuple:
(int("id") ~ str("field1") ~ int("field2"))
.map(flatten)
.map { tuple => (Model apply _).tupled(tuple) }
However, if you need to do some further transformations, you will need to modify the tuple somehow:
(int("id") ~ str("field1") ~ int("field2"))
.map(flatten)
.map { tuple => (Model apply _).tupled(tuple.copy(_1=..., _2=....) }
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With