cgroup.go

package sqlbuilder

import "bytes"

//
// Author: 陈永佳 chenyongjia@parkingwang.com, yoojiachen@gmail.com
//

type ConditionGroup struct {
	buffer     *bytes.Buffer
	conditions SQLStatement
}

func Group(conditions SQLStatement) *ConditionGroup {
	return newGroup(conditions)
}

func (slf *ConditionGroup) Group(conditions SQLStatement) *ConditionGroup {
	slf.conditions = conditions
	return slf
}

func (slf *ConditionGroup) And() *ConditionGroup {
	return newGroupWith(slf, " AND ")
}

func (slf *ConditionGroup) Or() *ConditionGroup {
	return newGroupWith(slf, " OR ")
}

func (slf *ConditionGroup) Compile() string {
	slf.buffer.WriteString(brackets(slf.conditions.Compile()))
	return slf.buffer.String()
}

//

func newGroup(conditions SQLStatement) *ConditionGroup {
	return &ConditionGroup{
		buffer:     new(bytes.Buffer),
		conditions: conditions,
	}
}

func newGroupWith(preStatement SQLStatement, op string) *ConditionGroup {
	group := newGroup(nil)
	group.buffer.WriteString(preStatement.Compile())
	group.buffer.WriteString(op)
	return group
}