Generate Scala code from your database to use with the incredible library quill. Only tested with postgresql, but could in theory work with any jdbc compliant database.
Say you have some database with this schema
create table test_user(
id integer not null,
name varchar(255),
primary key (id)
);
create table article(
id integer not null,
author_id integer,
is_published boolean
);
ALTER TABLE article
ADD CONSTRAINT author_id_fk
FOREIGN KEY (author_id)
REFERENCES test_user (id);scala-db-codegen will then generate "type all the things!"
code like this
package com.geirsson.codegen
import java.util.Date
import io.getquill.WrappedValue
object Tables {
/////////////////////////////////////////////////////
// Article
/////////////////////////////////////////////////////
case class Article(id: Article.Id, authorId: Option[TestUser.Id], isPublished: Option[Article.IsPublished])
object Article {
def create(id: Int, authorId: Option[Int], isPublished: Option[Boolean]): Article = {
Article(Id(id), authorId.map(TestUser.Id.apply), isPublished.map(IsPublished.apply))
}
case class Id(value: Int) extends AnyVal with WrappedValue[Int]
case class IsPublished(value: Boolean) extends AnyVal with WrappedValue[Boolean]
}
/////////////////////////////////////////////////////
// TestUser
/////////////////////////////////////////////////////
case class TestUser(id: TestUser.Id, name: Option[TestUser.Name])
object TestUser {
def create(id: Int, name: Option[String]): TestUser = {
TestUser(Id(id), name.map(Name.apply))
}
case class Id(value: Int) extends AnyVal with WrappedValue[Int]
case class Name(value: String) extends AnyVal with WrappedValue[String]
}
}
It could in theory also generate the code differently.
Download 13kb bootstrap script
scala-db-codegen
and execute it.
The script will download all dependencies on first execution.
# print to stdout, works with running postgres instance on
# localhost:5432 with user "postgres", password "postgres" and database "postgres"
$ scala-db-codegen
# Override any default settings with flags.
$ scala-db-codegen --user myuser --password mypassword --url jdbc:postgresql://myhost:8888/postgres --file Tables.scala --type-map "bool,Boolean;int4,Int;int8,Long"
...For more details:
$ scala-db-codegen --help
Usage: scala-db-codegen [options]
--usage
Print usage and exit
--help | -h
Print help message and exit
--user <value>
user on database server
--password <value>
password for user on database server
--url <value>
jdbc url
--schema <value>
schema on database
--jdbc-driver <value>
only tested with postgresql
--imports <value>
top level imports of generated file
--package <value>
package name for generated classes
--type-map <value>
Which types should write to which types? Format is: numeric,BigDecimal;int8,Long;...
--excluded-tables <value>
Do not generate classes for these tables.
--file <value>
Write generated code to this filename. Prints to stdout if not set.// 2.11 only
libraryDependencies += "com.geirsson" %% "scala-db-codegen" % "<version>"Consult the source code, at least for now ;)
Clone this repo into a subdirectory of your project. In your build.sbt:
import sbt.Project.projectToRef
lazy val `scala-db-codegen` = ProjectRef(file("scala-db-codegen"), "scala-db-codegen")
lazy val codegen = project.dependsOn(`scala-db-codegen`)Run from sbt:
codegen/runMain com.geirsson.codegen.Codegen --package tables --file myfile.scalaHack on the code to further customize to your needs.
The Slick code generator is excellent and please use that if you are using Slick. Really, the slick codegenerator is extremely customizable and can probably even do stuff that this library does.
I created this library because I struggled to get the slick code generator to do exactly what I wanted. Instead of learning more about slick models and which methods to override on the slick code generator, I decided to roll my own code generator and hopefully learn more about jdbc along the way :)
- No longer abort on missing key in
--type-map, see #3. Thanks @nightscape!
- Map nullable columns to
Optiontypes. - Rename maven artifact name to
scala-db-codegenfor consistency.
- Basic code generation
- Command line interface
