GetQueryType parses a MongoDB shell statement and returns its query type.
(statement string)
| 10 | |
| 11 | // GetQueryType parses a MongoDB shell statement and returns its query type. |
| 12 | func GetQueryType(statement string) base.QueryType { |
| 13 | stmts, err := ParseMongoShell(statement) |
| 14 | if err != nil || len(stmts) == 0 { |
| 15 | return base.DML |
| 16 | } |
| 17 | |
| 18 | node, ok := GetOmniNode(stmts[0].AST) |
| 19 | if !ok || node == nil { |
| 20 | return base.DML |
| 21 | } |
| 22 | |
| 23 | a := analysis.Analyze(node) |
| 24 | if a == nil { |
| 25 | return base.DML |
| 26 | } |
| 27 | |
| 28 | return omniOperationToQueryType(a) |
| 29 | } |
| 30 | |
| 31 | // omniOperationToQueryType maps an omni StatementAnalysis to a base.QueryType. |
| 32 | func omniOperationToQueryType(a *analysis.StatementAnalysis) base.QueryType { |