gosora/query_gen/micro_builders.go

package qgen

type dateCutoff struct {
    Column   string
    Quantity int
    Unit     string
    Type     int
}

type prebuilder struct {
    adapter Adapter
}

func (b *prebuilder) Select(nlist ...string) *selectPrebuilder {
    name := optString(nlist, "")
    return &selectPrebuilder{name, "", "", "", "", "", nil, nil, "", b.adapter}
}

func (b *prebuilder) Count(nlist ...string) *selectPrebuilder {
    name := optString(nlist, "")
    return &selectPrebuilder{name, "", "COUNT(*)", "", "", "", nil, nil, "", b.adapter}
}

func (b *prebuilder) Insert(nlist ...string) *insertPrebuilder {
    name := optString(nlist, "")
    return &insertPrebuilder{name, "", "", "", b.adapter}
}

func (b *prebuilder) Update(nlist ...string) *updatePrebuilder {
    name := optString(nlist, "")
    return &updatePrebuilder{name, "", "", "", nil, nil, b.adapter}
}

func (b *prebuilder) Delete(nlist ...string) *deletePrebuilder {
    name := optString(nlist, "")
    return &deletePrebuilder{name, "", "", nil, b.adapter}
}

type deletePrebuilder struct {
    name       string
    table      string
    where      string
    dateCutoff *dateCutoff

    build Adapter
}

func (b *deletePrebuilder) Table(table string) *deletePrebuilder {
    b.table = table
    return b
}

func (b *deletePrebuilder) Where(where string) *deletePrebuilder {
    if b.where != "" {
        b.where += " AND "
    }
    b.where += where
    return b
}

// TODO: We probably want to avoid the double allocation of two builders somehow
func (b *deletePrebuilder) FromAcc(acc *accDeleteBuilder) *deletePrebuilder {
    b.table = acc.table
    b.where = acc.where
    b.dateCutoff = acc.dateCutoff
    return b
}

func (b *deletePrebuilder) Text() (string, error) {
    return b.build.SimpleDelete(b.name, b.table, b.where)
}

func (b *deletePrebuilder) Parse() {
    b.build.SimpleDelete(b.name, b.table, b.where)
}

type updatePrebuilder struct {
    name          string
    table         string
    set           string
    where         string
    dateCutoff    *dateCutoff // We might want to do this in a slightly less hacky way
    whereSubQuery *selectPrebuilder

    build Adapter
}

func qUpdate(table string, set string, where string) *updatePrebuilder {
    return &updatePrebuilder{table: table, set: set, where: where}
}

func (b *updatePrebuilder) Table(table string) *updatePrebuilder {
    b.table = table
    return b
}

func (b *updatePrebuilder) Set(set string) *updatePrebuilder {
    b.set = set
    return b
}

func (b *updatePrebuilder) Where(where string) *updatePrebuilder {
    if b.where != "" {
        b.where += " AND "
    }
    b.where += where
    return b
}

func (b *updatePrebuilder) WhereQ(sel *selectPrebuilder) *updatePrebuilder {
    b.whereSubQuery = sel
    return b
}

func (b *updatePrebuilder) Text() (string, error) {
    return b.build.SimpleUpdate(b)
}

func (b *updatePrebuilder) Parse() {
    b.build.SimpleUpdate(b)
}

type selectPrebuilder struct {
    name       string
    table      string
    columns    string
    where      string
    orderby    string
    limit      string
    dateCutoff *dateCutoff
    inChain    *selectPrebuilder
    inColumn   string // for inChain

    build Adapter
}

func (b *selectPrebuilder) Table(table string) *selectPrebuilder {
    b.table = table
    return b
}

func (b *selectPrebuilder) Columns(columns string) *selectPrebuilder {
    b.columns = columns
    return b
}

func (b *selectPrebuilder) Where(where string) *selectPrebuilder {
    if b.where != "" {
        b.where += " AND "
    }
    b.where += where
    return b
}

func (b *selectPrebuilder) InQ(subBuilder *selectPrebuilder) *selectPrebuilder {
    b.inChain = subBuilder
    return b
}

func (b *selectPrebuilder) Orderby(orderby string) *selectPrebuilder {
    b.orderby = orderby
    return b
}

func (b *selectPrebuilder) Limit(limit string) *selectPrebuilder {
    b.limit = limit
    return b
}

// TODO: We probably want to avoid the double allocation of two builders somehow
func (b *selectPrebuilder) FromAcc(acc *AccSelectBuilder) *selectPrebuilder {
    b.table = acc.table
    if acc.columns != "" {
        b.columns = acc.columns
    }
    b.where = acc.where
    b.orderby = acc.orderby
    b.limit = acc.limit

    b.dateCutoff = acc.dateCutoff
    if acc.inChain != nil {
        b.inChain = &selectPrebuilder{"", acc.inChain.table, acc.inChain.columns, acc.inChain.where, acc.inChain.orderby, acc.inChain.limit, acc.inChain.dateCutoff, nil, "", b.build}
        b.inColumn = acc.inColumn
    }
    return b
}

func (b *selectPrebuilder) FromCountAcc(acc *accCountBuilder) *selectPrebuilder {
    b.table = acc.table
    b.where = acc.where
    b.limit = acc.limit

    b.dateCutoff = acc.dateCutoff
    if acc.inChain != nil {
        b.inChain = &selectPrebuilder{"", acc.inChain.table, acc.inChain.columns, acc.inChain.where, acc.inChain.orderby, acc.inChain.limit, acc.inChain.dateCutoff, nil, "", b.build}
        b.inColumn = acc.inColumn
    }
    return b
}

// TODO: Add support for dateCutoff
func (b *selectPrebuilder) Text() (string, error) {
    return b.build.SimpleSelect(b.name, b.table, b.columns, b.where, b.orderby, b.limit)
}

// TODO: Add support for dateCutoff
func (b *selectPrebuilder) Parse() {
    b.build.SimpleSelect(b.name, b.table, b.columns, b.where, b.orderby, b.limit)
}

type insertPrebuilder struct {
    name    string
    table   string
    columns string
    fields  string

    build Adapter
}

func (b *insertPrebuilder) Table(table string) *insertPrebuilder {
    b.table = table
    return b
}

func (b *insertPrebuilder) Columns(columns string) *insertPrebuilder {
    b.columns = columns
    return b
}

func (b *insertPrebuilder) Fields(fields string) *insertPrebuilder {
    b.fields = fields
    return b
}

func (b *insertPrebuilder) Text() (string, error) {
    return b.build.SimpleInsert(b.name, b.table, b.columns, b.fields)
}

func (b *insertPrebuilder) Parse() {
    b.build.SimpleInsert(b.name, b.table, b.columns, b.fields)
}

/*type countPrebuilder struct {
    name  string
    table string
    where string
    limit string

    build Adapter
}

func (b *countPrebuilder) Table(table string) *countPrebuilder {
    b.table = table
    return b
}

func b *countPrebuilder) Where(where string) *countPrebuilder {
    if b.where != "" {
        b.where += " AND "
    }
    b.where += where
    return b
}

func (b *countPrebuilder) Limit(limit string) *countPrebuilder {
    b.limit = limit
    return b
}

func (b *countPrebuilder) Text() (string, error) {
    return b.build.SimpleCount(b.name, b.table, b.where, b.limit)
}

func (b *countPrebuilder) Parse() {
    b.build.SimpleCount(b.name, b.table, b.where, b.limit)
}*/

func optString(nlist []string, defaultStr string) string {
    if len(nlist) == 0 {
        return defaultStr
    }
    return nlist[0]
}