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

40
sqlbuilder/select_statement_table.go
View file

@ -1,45 +1,53 @@
package sqlbuilder
type SelectStatementTable struct {
statement SelectStatement
type ExpressionTable interface {
ReadableTable
RefIntColumnName(name string) *IntegerColumn
RefIntColumn(column Column) *IntegerColumn
RefStringColumn(column Column) *StringColumn
}
type expressionTableImpl struct {
statement Expression
columns []Column
alias string
}
// Returns the tableName's name in the database
func (t *SelectStatementTable) SchemaName() string {
func (t *expressionTableImpl) SchemaName() string {
return ""
}
func (s *SelectStatementTable) TableName() string {
func (s *expressionTableImpl) TableName() string {
return s.alias
}
func (s *SelectStatementTable) Columns() []Column {
func (s *expressionTableImpl) Columns() []Column {
return s.columns
}
func (s *SelectStatementTable) RefIntColumnName(name string) Column {
func (s *expressionTableImpl) RefIntColumnName(name string) *IntegerColumn {
intColumn := NewIntegerColumn(name, NotNullable)
intColumn.setTableName(s.alias)
return intColumn
}
func (s *SelectStatementTable) RefIntColumn(column Column) *IntegerColumn {
func (s *expressionTableImpl) RefIntColumn(column Column) *IntegerColumn {
intColumn := NewIntegerColumn(column.TableName()+"."+column.Name(), NotNullable)
intColumn.setTableName(s.alias)
return intColumn
}
func (s *SelectStatementTable) RefStringColumn(column Column) *StringColumn {
func (s *expressionTableImpl) RefStringColumn(column Column) *StringColumn {
strColumn := NewStringColumn(column.Name(), NotNullable)
strColumn.setTableName(column.TableName())
return strColumn
}
func (s *SelectStatementTable) SerializeSql(out *queryData) error {
func (s *expressionTableImpl) SerializeSql(out *queryData) error {
out.WriteString("( ")
err := s.statement.Serialize(out)
@ -54,33 +62,33 @@ func (s *SelectStatementTable) SerializeSql(out *queryData) error {
}
// Generates a select query on the current tableName.
func (s *SelectStatementTable) SELECT(projections ...Projection) SelectStatement {
func (s *expressionTableImpl) SELECT(projections ...Projection) SelectStatement {
return newSelectStatement(s, projections)
}
// Creates a inner join tableName expression using onCondition.
func (s *SelectStatementTable) INNER_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
func (s *expressionTableImpl) INNER_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
return InnerJoinOn(s, table, onCondition)
}
//func (s *SelectStatementTable) InnerJoinUsing(table ReadableTable, col1 Column, col2 Column) ReadableTable {
//func (s *expressionTableImpl) InnerJoinUsing(table ReadableTable, col1 Column, col2 Column) ReadableTable {
// return INNER_JOIN(s, table, col1.Eq(col2))
//}
// Creates a left join tableName expression using onCondition.
func (s *SelectStatementTable) LeftJoinOn(table ReadableTable, onCondition BoolExpression) ReadableTable {
func (s *expressionTableImpl) LEFT_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
return LeftJoinOn(s, table, onCondition)
}
// Creates a right join tableName expression using onCondition.
func (s *SelectStatementTable) RightJoinOn(table ReadableTable, onCondition BoolExpression) ReadableTable {
func (s *expressionTableImpl) RIGHT_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
return RightJoinOn(s, table, onCondition)
}
func (s *SelectStatementTable) FULL_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
func (s *expressionTableImpl) FULL_JOIN(table ReadableTable, onCondition BoolExpression) ReadableTable {
return FullJoin(s, table, onCondition)
}
func (s *SelectStatementTable) CrossJoin(table ReadableTable) ReadableTable {
func (s *expressionTableImpl) CROSS_JOIN(table ReadableTable) ReadableTable {
return CrossJoin(s, table)
}