本文整理匯總了Golang中github.com/pingcap/tidb/ast.SelectStmt.SetResultFields方法的典型用法代碼示例。如果您正苦於以下問題：Golang SelectStmt.SetResultFields方法的具體用法？Golang SelectStmt.SetResultFields怎麽用？Golang SelectStmt.SetResultFields使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類github.com/pingcap/tidb/ast.SelectStmt的用法示例。
在下文中一共展示了SelectStmt.SetResultFields方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Golang代碼示例。
示例1: extractSelectAgg
// extractSelectAgg extracts aggregate functions and converts ColumnNameExpr to aggregate function.
func (b *planBuilder) extractSelectAgg(sel *ast.SelectStmt) []*ast.AggregateFuncExpr {
extractor := &ast.AggregateFuncExtractor{AggFuncs: make([]*ast.AggregateFuncExpr, 0)}
res := make([]*ast.ResultField, 0, len(sel.GetResultFields()))
for _, f := range sel.GetResultFields() {
n, ok := f.Expr.Accept(extractor)
if !ok {
b.err = errors.New("Failed to extract agg expr!")
return nil
}
ve, ok := f.Expr.(*ast.ValueExpr)
if ok && len(f.Column.Name.O) > 0 {
agg := &ast.AggregateFuncExpr{
F: ast.AggFuncFirstRow,
Args: []ast.ExprNode{ve},
}
extractor.AggFuncs = append(extractor.AggFuncs, agg)
n = agg
}
// Clone the ResultField.
// For "select c from t group by c;" both "c"s in select field and grouby clause refer to the same ResultField.
// So we should not affact the "c" in groupby clause.
nf := &ast.ResultField{
ColumnAsName: f.ColumnAsName,
Column: f.Column,
Table: f.Table,
DBName: f.DBName,
Expr: n.(ast.ExprNode),
}
res = append(res, nf)
}
sel.SetResultFields(res)
// Extract agg funcs from having clause.
if sel.Having != nil {
n, ok := sel.Having.Expr.Accept(extractor)
if !ok {
b.err = errors.New("Failed to extract agg expr from having clause")
return nil
}
sel.Having.Expr = n.(ast.ExprNode)
}
// Extract agg funcs from orderby clause.
if sel.OrderBy != nil {
for _, item := range sel.OrderBy.Items {
n, ok := item.Expr.Accept(extractor)
if !ok {
b.err = errors.New("Failed to extract agg expr from orderby clause")
return nil
}
item.Expr = n.(ast.ExprNode)
// If item is PositionExpr, we need to rebind it.
// For PositionExpr will refer to a ResultField in fieldlist.
// After extract AggExpr from fieldlist, it may be changed (See the code above).
if pe, ok := item.Expr.(*ast.PositionExpr); ok {
pe.Refer = sel.GetResultFields()[pe.N-1]
}
}
}
return extractor.AggFuncs
}示例2: extractSelectAgg
// extractSelectAgg extracts aggregate functions and converts ColumnNameExpr to aggregate function.
func (b *planBuilder) extractSelectAgg(sel *ast.SelectStmt) []*ast.AggregateFuncExpr {
extractor := &ast.AggregateFuncExtractor{AggFuncs: make([]*ast.AggregateFuncExpr, 0)}
res := make([]*ast.ResultField, 0, len(sel.GetResultFields()))
for _, f := range sel.GetResultFields() {
n, ok := f.Expr.Accept(extractor)
if !ok {
b.err = errors.New("Failed to extract agg expr!")
return nil
}
ve, ok := f.Expr.(*ast.ValueExpr)
if ok && len(f.Column.Name.O) > 0 {
agg := &ast.AggregateFuncExpr{
F: ast.AggFuncFirstRow,
Args: []ast.ExprNode{ve},
}
extractor.AggFuncs = append(extractor.AggFuncs, agg)
n = agg
}
// Clone the ResultField.
// For "select c from t group by c;" both "c"s in select field and grouby clause refer to the same ResultField.
// So we should not affact the "c" in groupby clause.
nf := &ast.ResultField{
ColumnAsName: f.ColumnAsName,
Column: f.Column,
Table: f.Table,
DBName: f.DBName,
Expr: n.(ast.ExprNode),
}
res = append(res, nf)
}
sel.SetResultFields(res)
// Extract agg funcs from orderby clause.
if sel.OrderBy != nil {
for _, item := range sel.OrderBy.Items {
n, ok := item.Expr.Accept(extractor)
if !ok {
b.err = errors.New("Failed to extract agg expr from orderby clause")
return nil
}
item.Expr = n.(ast.ExprNode)
}
}
// TODO: extract aggfuncs from having clause.
return extractor.AggFuncs
}