xorm/statement.go

package xorm

import (
	"fmt"
)

type Statement struct {
	TableName  string
	Table      *Table
	Session    *Session
	LimitStr   int
	OffsetStr  int
	WhereStr   string
	ParamStr   []interface{}
	OrderStr   string
	JoinStr    string
	GroupByStr string
	HavingStr  string
}

func (statement *Statement) Limit(start int, size ...int) *Statement {
	statement.LimitStr = start
	if len(size) > 0 {
		statement.OffsetStr = size[0]
	}
	return statement
}

func (statement *Statement) Offset(offset int) *Statement {
	statement.OffsetStr = offset
	return statement
}

func (statement *Statement) OrderBy(order string) *Statement {
	statement.OrderStr = order
	return statement
}

func (statement *Statement) Select(colums string) *Statement {
	//statement.ColumnStr = colums
	return statement
}

func (statement Statement) generateSql() string {
	columnStr := statement.Table.ColumnStr()
	return statement.genSelectSql(columnStr)
}

func (statement Statement) genCountSql() string {
	return statement.genSelectSql("count(*) as total")
}

func (statement Statement) genSelectSql(columnStr string) (a string) {
	session := statement.Session
	if session.Engine.Protocol == "mssql" {
		if statement.OffsetStr > 0 {
			a = fmt.Sprintf("select ROW_NUMBER() OVER(order by %v )as rownum,%v from %v",
				statement.Table.PKColumn().Name,
				columnStr,
				statement.Table.Name)
			if statement.WhereStr != "" {
				a = fmt.Sprintf("%v WHERE %v", a, statement.WhereStr)
			}
			a = fmt.Sprintf("select %v from (%v) "+
				"as a where rownum between %v and %v",
				columnStr,
				a,
				statement.OffsetStr,
				statement.LimitStr)
		} else if statement.LimitStr > 0 {
			a = fmt.Sprintf("SELECT top %v %v FROM %v", statement.LimitStr, columnStr, statement.Table.Name)
			if statement.WhereStr != "" {
				a = fmt.Sprintf("%v WHERE %v", a, statement.WhereStr)
			}
			if statement.GroupByStr != "" {
				a = fmt.Sprintf("%v %v", a, statement.GroupByStr)
			}
			if statement.HavingStr != "" {
				a = fmt.Sprintf("%v %v", a, statement.HavingStr)
			}
			if statement.OrderStr != "" {
				a = fmt.Sprintf("%v ORDER BY %v", a, statement.OrderStr)
			}
		} else {
			a = fmt.Sprintf("SELECT %v FROM %v", columnStr, statement.Table.Name)
			if statement.WhereStr != "" {
				a = fmt.Sprintf("%v WHERE %v", a, statement.WhereStr)
			}
			if statement.GroupByStr != "" {
				a = fmt.Sprintf("%v %v", a, statement.GroupByStr)
			}
			if statement.HavingStr != "" {
				a = fmt.Sprintf("%v %v", a, statement.HavingStr)
			}
			if statement.OrderStr != "" {
				a = fmt.Sprintf("%v ORDER BY %v", a, statement.OrderStr)
			}
		}
	} else {
		a = fmt.Sprintf("SELECT %v FROM %v", columnStr, statement.Table.Name)
		if statement.JoinStr != "" {
			a = fmt.Sprintf("%v %v", a, statement.JoinStr)
		}
		if statement.WhereStr != "" {
			a = fmt.Sprintf("%v WHERE %v", a, statement.WhereStr)
		}
		if statement.GroupByStr != "" {
			a = fmt.Sprintf("%v %v", a, statement.GroupByStr)
		}
		if statement.HavingStr != "" {
			a = fmt.Sprintf("%v %v", a, statement.HavingStr)
		}
		if statement.OrderStr != "" {
			a = fmt.Sprintf("%v ORDER BY %v", a, statement.OrderStr)
		}
		if statement.OffsetStr > 0 {
			a = fmt.Sprintf("%v LIMIT %v, %v", a, statement.OffsetStr, statement.LimitStr)
		} else if statement.LimitStr > 0 {
			a = fmt.Sprintf("%v LIMIT %v", a, statement.LimitStr)
		}
	}
	return
}

/*func (statement *Statement) genInsertSQL() string {
	table = statement.Table
	colNames := make([]string, len(table.Columns))
	for idx, col := range table.Columns {
		if col.Name == "" {
			continue
		}
		colNames[idx] = col.Name
	}
	return strings.Join(colNames, ", ")

	colNames := make([]string, len(table.Columns))
	for idx, col := range table.Columns {
		if col.Name == "" {
			continue
		}
		colNames[idx] = "?"
	}
	strings.Join(colNames, ", ")
}*/