| 12 | ) |
| 13 | |
| 14 | func GetStatementTypes(asts []base.AST) ([]storepb.StatementType, error) { |
| 15 | sqlTypeSet := make(map[storepb.StatementType]bool) |
| 16 | for _, ast := range asts { |
| 17 | node, ok := GetOmniNode(ast) |
| 18 | if ok { |
| 19 | sqlTypeSet[classifyOmniStatementType(node)] = true |
| 20 | continue |
| 21 | } |
| 22 | |
| 23 | return nil, errors.New("expected Oracle omni AST") |
| 24 | } |
| 25 | var sqlTypes []storepb.StatementType |
| 26 | for sqlType := range sqlTypeSet { |
| 27 | sqlTypes = append(sqlTypes, sqlType) |
| 28 | } |
| 29 | return sqlTypes, nil |
| 30 | } |
| 31 | |
| 32 | func classifyOmniStatementType(node oracleast.Node) storepb.StatementType { |
| 33 | switch n := node.(type) { |