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))
}
}
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},
})
}
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},
})
}
}