(wr *pipe.Writer, sqlProgram string, db *database.DB, modelDir string, session *pb.Session)
| 132 | } |
| 133 | |
| 134 | func runSQLProgram(wr *pipe.Writer, sqlProgram string, db *database.DB, modelDir string, session *pb.Session) error { |
| 135 | sqlProgram, err := parser.RemoveCommentInSQLStatement(sqlProgram) |
| 136 | if err != nil { |
| 137 | return err |
| 138 | } |
| 139 | |
| 140 | stmts, err := parser.Parse(db.DriverName, sqlProgram) |
| 141 | if err != nil { |
| 142 | return err |
| 143 | } |
| 144 | // NOTE(tony): We generate IR and execute its translated program one-by-one since IR generation may depend on the execution |
| 145 | // of the previous statement. For example, consider a SQL program |
| 146 | // |
| 147 | // create table some_table as (select ...); |
| 148 | // select * from some_table to train ... |
| 149 | // |
| 150 | // The IR generation on the second statement would fail since it requires inspection the schema of some_table, |
| 151 | // which depends on the execution of create table some_table as (select ...);. |
| 152 | sqls := RewriteStatementsWithHints(stmts, db.DriverName) |
| 153 | for _, sql := range sqls { |
| 154 | if err := runSingleSQLFlowStatement(wr, sql, db, modelDir, session); err != nil { |
| 155 | return err |
| 156 | } |
| 157 | } |
| 158 | return nil |
| 159 | } |
| 160 | |
| 161 | func runSingleSQLFlowStatement(wr *pipe.Writer, sql *parser.SQLFlowStmt, db *database.DB, modelDir string, session *pb.Session) (e error) { |
| 162 | defer func(startTime int64) { |
no test coverage detected