func (c *Session) handleComStmtPrepare(sqlstmt string) error {
stmt, err := parser.Parse(sqlstmt)
if err != nil {
log.Warningf(`parse sql "%s" error "%s"`, sqlstmt, err.Error())
return c.handleMySQLError(
mysql.NewDefaultError(mysql.ER_SYNTAX_ERROR, err.Error()))
}
// Only a few statements supported by prepare statements
// http://dev.mysql.com/worklog/task/?id=2871
switch v := stmt.(type) {
case parser.ISelect, *parser.Insert, *parser.Update, *parser.Delete,
*parser.Replace,
parser.IDDLStatement,
*parser.ShowTables,
*parser.ShowColumns,
*parser.ShowVariables,
*parser.ShowIndex,
*parser.Set,
*parser.DescribeTable,
*parser.Do:
return c.prepare(v, sqlstmt)
default:
log.Warnf("statement %T[%s] not support prepare ops", stmt, sqlstmt)
return c.handleMySQLError(
mysql.NewDefaultError(mysql.ER_UNSUPPORTED_PS))
}
}
func (c *Session) comQuery(sqlstmt string) error {
stmt, err := parser.Parse(sqlstmt)
if err != nil {
log.Warningf(`parse sql "%s" error "%s"`, sqlstmt, err.Error())
return c.handleMySQLError(
NewDefaultError(ER_SYNTAX_ERROR, err.Error()))
}
switch v := stmt.(type) {
case parser.ISelect:
return c.handleQuery(v, sqlstmt)
case *parser.Insert, *parser.Update, *parser.Delete, *parser.Replace:
return c.handleExec(stmt, sqlstmt, false)
case *parser.Set:
return c.handleSet(v, sqlstmt)
case *parser.Begin, *parser.StartTrans:
return c.handleBegin()
case *parser.Commit:
return c.handleCommit()
case *parser.Rollback:
return c.handleRollback()
case parser.IShow:
return c.handleShow(sqlstmt, v)
case parser.IDDLStatement:
return c.handleDDL(v, sqlstmt)
case *parser.Do, *parser.Call, *parser.FlushTables:
return c.handleExec(stmt, sqlstmt, false)
case *parser.Use:
if err := c.useDB(hack.String(stmt.(*parser.Use).DB)); err != nil {
return c.handleMySQLError(err)
} else {
return c.fc.WriteOK(nil)
}
default:
log.Warnf("statement %T[%s] not support now", stmt, sqlstmt)
return nil
}
return nil
}
func (c *Session) comQuery(sqlstmt string) error {
//TODO accerlate the flow control module and the figerprint module
// err := c.intercept(sqlstmt)
// if err != nil {
// return err
// }
// c.updatefp(sqlstmt)
log.Infof("session %d: %s", c.sessionId, sqlstmt)
stmt, err := parser.Parse(sqlstmt)
if err != nil {
log.Warningf(`parse sql "%s" error "%s"`, sqlstmt, err.Error())
return c.handleMySQLError(
NewDefaultError(ER_SYNTAX_ERROR, err.Error()))
}
switch v := stmt.(type) {
case parser.ISelect:
return c.handleQuery(v, sqlstmt)
case *parser.Insert, *parser.Update, *parser.Delete, *parser.Replace:
return c.handleExec(stmt, sqlstmt, false)
case *parser.Set:
return c.handleSet(v, sqlstmt)
case *parser.Begin, *parser.StartTrans:
return c.handleBegin()
case *parser.Commit:
return c.handleCommit()
case *parser.Rollback:
// log.Debug(hack.String(stmt.(*parser.Rollback).Point))
if len(stmt.(*parser.Rollback).Point) > 0 {
return c.handleExec(stmt, sqlstmt, false)
}
return c.handleRollback()
case parser.IShow:
return c.handleShow(sqlstmt, v)
case parser.IDDLStatement:
return c.handleDDL(v, sqlstmt)
case *parser.Do, *parser.Call, *parser.FlushTables:
return c.handleExec(stmt, sqlstmt, false)
//add the describe table module
case *parser.DescribeTable, *parser.DescribeStmt:
return c.handleQuery(v, sqlstmt)
case *parser.Use:
if err := c.useDB(hack.String(stmt.(*parser.Use).DB)); err != nil {
return c.handleMySQLError(err)
} else {
return c.fc.WriteOK(nil)
}
case *parser.SavePoint:
return c.handleExec(stmt, sqlstmt, false)
// return c.handleQuery(v, sqlstmt)
case *parser.SetTrans:
// log.Warnf("set tx iso level ")
t_sl := hack.Slice(sqlstmt)
tmp := make([]byte, len(t_sl))
copy(tmp, t_sl)
// log.Debug(sqlstmt, t_sl, tmp, len(t_sl))
c.txIsolationInDef = false
sql := hack.String(tmp)
// log.Debug(sql, len(sql))
c.txIsolationStmt = sql
// log.Warnf("set tx iso level finish ")
if c.isInTransaction() {
return c.handleExec(stmt, sqlstmt, false)
}
return c.fc.WriteOK(nil)
default:
log.Warnf("session %d : statement %T[%s] not support now", c.sessionId, stmt, sqlstmt)
err := errors.New("statement not support now")
return c.handleMySQLError(
NewDefaultError(ER_SYNTAX_ERROR, err.Error()))
}
return nil
}