Skip to content

Instantly share code, notes, and snippets.

@dcsobral
Forked from xeno-by/gist:2559714
Created September 22, 2012 21:50
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dcsobral/3767946 to your computer and use it in GitHub Desktop.
Save dcsobral/3767946 to your computer and use it in GitHub Desktop.
Mixing in a trait dynamically
Answers http://stackoverflow.com/questions/10373318/mixing-in-a-trait-dynamically.
Compile as follows:
scalac Common_1.scala Macros_2.scala
scalac Common_1.scala Test_3.scala -cp <path to the result of the previous compilation>
Tested in 2.10.0-M3, will most likely not compile by the time 2.10.0 final is released, because we're actively rehashing the API.
However the principles will remain the same in the final release, so the concept itself is okay.
upd. Code updated for 2.10.0-M7.
===Common_1.scala===
trait Persisted {
def id: Long
}
===Macros_2.scala===
import language.experimental.macros
import scala.reflect.macros.Context
object Macros {
def toPersisted[T](instance: T, id: Long): T with Persisted = macro impl[T]
def impl[T: c.AbsTypeTag](c: Context)(instance: c.Expr[T], id: c.Expr[Long]) = {
import c.universe._
import Flag._
val t = implicitly[c.AbsTypeTag[T]].tpe // in RC1 this will become c.absTypeOf[T]
val u = t.typeSymbol.asClass
if (!u.isCaseClass)
c.abort(c.enclosingPosition, "toPersisted only accepts case classes, you provided %s".format(t))
// use reflection API to get the list of all declared fields
// more info here: http://scalamacros.org/documentation.html
val accessors = t.members.sorted.collect{ case x: TermSymbol if x.isCaseAccessor && x.isMethod => x }
val fieldNames = accessors map (_.name)
// how did I know what trees to generate?
// read up the docs at http://scalamacros.org/documentation.html
val instanceParam = ValDef(NoMods, newTermName("instance"), TypeTree(u.toType), EmptyTree)
val idParam = ValDef(NoMods, newTermName("id"), Ident(definitions.LongClass), EmptyTree)
val superArgs = fieldNames map (fieldName => Select(Ident(instanceParam.name), fieldName))
val body = Apply(Select(Super(This(newTypeName("")), newTypeName("")), nme.CONSTRUCTOR), superArgs)
val ctor = DefDef(NoMods, nme.CONSTRUCTOR, Nil, List(List(instanceParam, idParam)), TypeTree(), Block(List(body), Literal(Constant(()))))
val idVal = idParam.duplicate
val tmpl = Template(List(Ident(u), Ident(c.mirror.staticClass("Persisted"))), emptyValDef, List(idVal, ctor))
val cdef = ClassDef(NoMods, newTypeName(c.fresh(u.name + "$Persisted")), Nil, tmpl)
val init = New(Ident(cdef.name), List(List(instance.tree, id.tree)))
c.Expr(Block(cdef, init))
}
}
===Test_3.scala===
object Test extends App {
import Macros._
case class Person(first: String, last: String)
val p = toPersisted(Person("hello", "world"), 42)
println(p.first)
println(p.last)
println(p.id)
}
===Output===
C:\Projects\Kepler\sandbox>scalac Common_1.scala Macros_2.scala
<scalac has exited with code 0>
C:\Projects\Kepler\sandbox>scalac -Ymacro-debug-lite Common_1.scala Test_3.scala -cp .
typechecking macro expansion Macros.toPersisted[Test.Person](Test.this.Person.apply("hello", "world"), 42L)
at source-C:\Projects\Kepler\sandbox\Test_3.scala,line-4,offset=114
{
class Person$Persisted1 extends Person with Persisted {
val id: Long = _;
def <init>(instance: Test.Person, id: Long) = {
super.<init>(instance.first, instance.last);
()
}
};
new Person$Persisted1(Test.this.Person.apply("hello", "world"), 42L)
}
*snip*
<scalac has exited with code 0>
C:\Projects\Kepler\sandbox>scala Test
hello
world
42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment