Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Allow set statements to be used as tables and expressions.

Browse source
This commit is contained in:
zer0sub 2019-05-05 12:37:23 +02:00
parent 5ad213885f
commit 5de001d7e0
8 changed files with 134 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

23
sqlbuilder/select_statement.go
View file

@ -11,6 +11,7 @@ type SelectStatement interface {
Expression
DISTINCT() SelectStatement
FROM(table ReadableTable) SelectStatement
WHERE(expression BoolExpression) SelectStatement
GROUP_BY(expressions ...Clause) SelectStatement
HAVING(boolExpression BoolExpression) SelectStatement
@ -21,7 +22,11 @@ type SelectStatement interface {
FOR_UPDATE() SelectStatement
AsTable(alias string) *SelectStatementTable
AsTable(alias string) ExpressionTable
}
var SELECT = func(projection ...Projection) SelectStatement {
return newSelectStatement(nil, projection)
}
// NOTE: SelectStatement purposely does not implement the Table interface since
@ -59,8 +64,7 @@ func defaultProjectionAliasing(projections []Projection) []Projection {
}
func newSelectStatement(table ReadableTable, projections []Projection) SelectStatement {
return &selectStatementImpl{
newSelect := &selectStatementImpl{
table: table,
projections: defaultProjectionAliasing(projections),
limit: -1,
@ -68,6 +72,15 @@ func newSelectStatement(table ReadableTable, projections []Projection) SelectSta
forUpdate: false,
distinct: false,
}
newSelect.expressionInterfaceImpl.parent = newSelect
return newSelect
}
func (s *selectStatementImpl) FROM(table ReadableTable) SelectStatement {
s.table = table
return s
}
func (s *selectStatementImpl) Serialize(out *queryData, options ...serializeOption) error {
@ -176,8 +189,8 @@ func (q *selectStatementImpl) Sql() (query string, args []interface{}, err error
return queryData.buff.String(), queryData.args, nil
}
func (s *selectStatementImpl) AsTable(alias string) *SelectStatementTable {
return &SelectStatementTable{
func (s *selectStatementImpl) AsTable(alias string) ExpressionTable {
return &expressionTableImpl{
statement: s,
alias: alias,
}