mirror of https://github.com/aldy505/bob.git
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package bob
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/lann/builder"
|
|
)
|
|
|
|
var ErrEmptyTable = errors.New("sql: no rows in result set")
|
|
var ErrEmptyTablePgx = errors.New("no rows in result set")
|
|
|
|
// BobBuilderType is the type for BobBuilder
|
|
type BobBuilderType builder.Builder
|
|
|
|
// BobBuilder interface wraps the ToSql method
|
|
type BobBuilder interface {
|
|
ToSql() (string, []interface{}, error)
|
|
}
|
|
|
|
// CreateTable creates a table with CreateBuilder interface
|
|
func (b BobBuilderType) CreateTable(table string) CreateBuilder {
|
|
return CreateBuilder(b).Name(table)
|
|
}
|
|
|
|
func (b BobBuilderType) CreateTableIfNotExists(table string) CreateBuilder {
|
|
return CreateBuilder(b).Name(table).IfNotExists()
|
|
}
|
|
|
|
// HasTable checks if a table exists with HasBuilder interface
|
|
func (b BobBuilderType) HasTable(table string) HasBuilder {
|
|
return HasBuilder(b).HasTable(table)
|
|
}
|
|
|
|
// HasColumn checks if a column exists with HasBuilder interface
|
|
func (b BobBuilderType) HasColumn(column string) HasBuilder {
|
|
return HasBuilder(b).HasColumn(column)
|
|
}
|
|
|
|
func (b BobBuilderType) DropTable(table string) DropBuilder {
|
|
return DropBuilder(b).DropTable(table)
|
|
}
|
|
|
|
func (b BobBuilderType) DropTableIfExists(table string) DropBuilder {
|
|
return DropBuilder(b).DropTable(table).IfExists()
|
|
}
|
|
|
|
func (b BobBuilderType) RenameTable(from, to string) RenameBuilder {
|
|
return RenameBuilder(b).From(from).To(to)
|
|
}
|
|
|
|
func (b BobBuilderType) Truncate(table string) TruncateBuilder {
|
|
return TruncateBuilder(b).Truncate(table)
|
|
}
|
|
|
|
// BobStmtBuilder is the parent builder for BobBuilderType
|
|
var BobStmtBuilder = BobBuilderType(builder.EmptyBuilder)
|
|
|
|
// CreateTable creates a table with CreateBuilder interface
|
|
func CreateTable(table string) CreateBuilder {
|
|
return BobStmtBuilder.CreateTable(table)
|
|
}
|
|
|
|
// CreateTableIfNotExists creates a table with CreateBuilder interface, if the table doesn't exists
|
|
func CreateTableIfNotExists(table string) CreateBuilder {
|
|
return BobStmtBuilder.CreateTableIfNotExists(table)
|
|
}
|
|
|
|
// HasTable checks if a table exists with HasBuilder interface
|
|
func HasTable(table string) HasBuilder {
|
|
return BobStmtBuilder.HasTable(table)
|
|
}
|
|
|
|
// HasColumn checks if a column exists with HasBuilder interface
|
|
func HasColumn(col string) HasBuilder {
|
|
return BobStmtBuilder.HasColumn(col)
|
|
}
|
|
|
|
func DropTable(table string) DropBuilder {
|
|
return BobStmtBuilder.DropTable(table)
|
|
}
|
|
|
|
func DropTableIfExists(table string) DropBuilder {
|
|
return BobStmtBuilder.DropTableIfExists(table)
|
|
}
|
|
|
|
func RenameTable(from, to string) RenameBuilder {
|
|
return BobStmtBuilder.RenameTable(from, to)
|
|
}
|
|
|
|
func Truncate(table string) TruncateBuilder {
|
|
return BobStmtBuilder.Truncate(table)
|
|
} |