コード例 #1
0
ファイル: set_stmt.go プロジェクト: Alienero/Rambo
func (sei *session) handleSet(stmt *sqlparser.Set) error {
	// get first arg
	expr := stmt.Exprs[0]
	name := string(expr.Name.Name)
	switch name {
	case "dbnum":
		buf := sqlparser.NewTrackedBuffer(nil)
		expr.Expr.Format(buf)
		_, ok := expr.Expr.(sqlparser.NumVal)
		if ok {
			i, err := strconv.Atoi(buf.String())
			if err != nil {
				return sei.writeError(mysql.NewDefaultError(mysql.ER_WRONG_VALUE_FOR_VAR, buf.String(), name))
			}
			sei.dbnum = i
			glog.Infof("set dbnum=%v", sei.dbnum)
			return sei.writeOK(nil)
		}
		return sei.writeError(mysql.NewDefaultError(mysql.ER_WRONG_VALUE_FOR_VAR, buf.String(), name))
	case "partitionkey":
		buf := sqlparser.NewTrackedBuffer(nil)
		expr.Expr.Format(buf)
		_, ok := expr.Expr.(sqlparser.StrVal)
		if ok {
			sei.partitionKey = buf.String()
			glog.Infof("set partitionKey=%v", sei.partitionKey)
			return sei.writeOK(nil)
		}
		return sei.writeError(mysql.NewDefaultError(mysql.ER_WRONG_VALUE_FOR_VAR, buf.String(), name))
	default:
		return sei.writeError(mysql.NewDefaultError(mysql.ER_UNKNOWN_SYSTEM_VARIABLE, name))
	}
}
コード例 #2
0
ファイル: plan.go プロジェクト: Alienero/Rambo
func (p *Plan) generateOneSQL(stmt sqlparser.Statement, index int) {
	buf := sqlparser.NewTrackedBuffer(nil)
	stmt.Format(buf)
	sql := buf.String()
	glog.Infof("generate select SQL:%s", sql)
	p.SQLs = append(p.SQLs, &SQL{
		Backend: p.Table.Backends[index],
		SQL:     []string{sql},
	})
}
コード例 #3
0
ファイル: plan.go プロジェクト: Alienero/Rambo
func (p *Plan) generateSQLALL(stmt sqlparser.Statement) {
	buf := sqlparser.NewTrackedBuffer(nil)
	stmt.Format(buf)
	sql := buf.String()
	glog.Infof("generate select SQL:%s", sql)
	for _, backend := range p.Table.Backends {
		p.SQLs = append(p.SQLs, &SQL{
			Backend: backend,
			SQL:     []string{sql},
		})
	}
}