query_builder.go

package query

import (
	"fmt"
	"strings"
)

type Builder struct {
	table string
	query string
}

func Table(name string) *Builder {
	return &Builder{table: name}
}

func (b *Builder) Select(q ...string) *Builder {
	b.query += fmt.Sprintf("SELECT %s FROM %s WHERE 1 = 1", strings.Join(q, ","), b.table)
	return b
}

func (b *Builder) Limit(page, limit int32) *Builder {
	if page != 0 && limit != 0 {
		b.query += " LIMIT ? OFFSET ?"
	}

	return b
}

func (b *Builder) Query(sql string, args []interface{}) *Builder {
	b.query += sql
	return b
}

func (b *Builder) String() string {
	return b.query
}