dbtogo

dbtogo

dbtogo is a tool to connect to a database and output Go structs mirroring the tables and fields

The generated code is configurable through modifying the template.

Support

Right now only supports mysql, sqlite3, mssql (MS SQL Sever), and postgresql.

Why?

I don't like using ORMs that do too much magic for me. This allows me to generate structs for my massive databases and use database/sql or some other light wrapper (sqlx).

I use this for our production backend. Pull requests are welcome.

Usage

dbtogo --help will tell you what you want to know.

dbtogo mssql "server=localhost;database=db1;user id=user1;password=password1"
dbtogo mysql mysqluser:pass@tcp(host:port)/db
dbtogo postgres postgres://user:pass@localhost:5432/db?sslmode=disable
dbtogo sqlite3 ./foo.db

Example

Assume we have the following tables in a sqlite3 database:

CREATE TABLE Foo (
id integer not null primary key,
name text);

CREATE TABLE Bar(
message varchar(255),
date datetime,
count int,
awesome bit);

dbtogo with the default template would output:

// GENERATED BY dbtogo (github.com/kdar/dbtogo); DO NOT EDIT
// ---args:

package model

import "database/sql"

var InsertStmts = map[string]string{
    "Foo": "INSERT INTO Foo (Id,Name) VALUES (:Id,:Name)",
    "Bar": "INSERT INTO Bar (Message,Date,Count,Awesome) VALUES (:Message,:Date,:Count,:Awesome)",
}

type Foo struct {
    Id   sql.NullInt64
    Name sql.NullString
}

type Bar struct {
    Message sql.NullString
    Date    *time.Time
    Count   sql.NullInt64
    Awesome sql.NullBool
}

Templating

The example template should get you started.

// GENERATED BY dbtogo (github.com/kdar/dbtogo); DO NOT EDIT
// ---args: {{join .SafeArgs " "}}

package {{.Package}}

{{define "columns"}}{{$len:=len .}}{{range $i, $v := .}}{{$v.Name|capitalize}}{{if lt $i (sub $len 1)}},{{end}}{{end}}{{end}}
{{define "values"}}{{$len:=len .}}{{range $i, $v := .}}:{{$v.Name|capitalize}}{{if lt $i (sub $len 1)}},{{end}}{{end}}{{end}}

import "database/sql"

var InsertStmts = map[string]string{
{{range $_, $table := .Tables}}  "{{$table.Name}}": "INSERT INTO {{$table.Name}} ({{template "columns" $table.Fields}}) VALUES ({{template "values" $table.Fields}})",
{{end}}}

{{range $_, $table := .Tables}}
type {{$table.Name}} struct {
{{range $_, $field := $table.Fields}}  {{$field.Name|capitalize}} {{$field|typenull}}
{{end}}}
{{end}}

Available template functions

String functions

Modify inflection functions

Math functions

Field functions (these functions operate on dbtogo.Field)

Inflection modification

Sometimes, the inflection library doesn't do what you want. For me, I ran into problems with words like "Address" and "Diagnosis" and turning them singular. It would return "Addres" and "Diagnosi" respectively. To fix this, put the following in your template before you do any inflection calls:

{{ addsingular "sis" "sis" }}
{{ addsingular "ess" "ess" }}