1 package info.globalcode.sql.dk; |
1 package info.globalcode.sql.dk; |
|
2 |
|
3 import java.sql.Types; |
|
4 import java.util.ArrayList; |
|
5 import java.util.Collections; |
|
6 import java.util.HashMap; |
|
7 import java.util.List; |
|
8 import java.util.Map; |
2 |
9 |
3 /** |
10 /** |
4 * |
11 * |
5 * @author Ing. František Kučera (frantovo.cz) |
12 * @author Ing. František Kučera (frantovo.cz) |
6 */ |
13 */ |
7 public class CLIStarter { |
14 public class CLIStarter { |
8 |
15 |
|
16 public static final String TYPE_NAME_SEPARATOR = ":"; |
|
17 private static final Map<String, Integer> types; |
|
18 |
|
19 static { |
|
20 Map<String, Integer> m = new HashMap<>(); |
|
21 m.put("int", Types.INTEGER); |
|
22 m.put("string", Types.VARCHAR); |
|
23 m.put("boolean", Types.BOOLEAN); |
|
24 /** |
|
25 * TODO: more types |
|
26 */ |
|
27 types = Collections.unmodifiableMap(m); |
|
28 } |
|
29 |
9 public static void main(String[] args) { |
30 public static void main(String[] args) { |
10 |
31 |
11 args = new String[]{"--sql", "SELECT * FROM tabulka;", "--db", "databáze_1"}; |
32 args = new String[]{"--sql", "SELECT * FROM tabulka;", "--db", "databáze_1", "--types", "int,bbb,omfg:int,omg:boolean", "--data", "xxx", ":omfg", "hodnota omfg", ":omg", "true"}; |
12 |
33 |
13 CLIOptions options = parseOptions(args); |
34 CLIOptions options = parseOptions(args); |
14 } |
35 } |
15 |
36 |
16 private static CLIOptions parseOptions(String[] args) { |
37 private static CLIOptions parseOptions(String[] args) { |
17 CLIOptions options = new CLIOptions(); |
38 CLIOptions options = new CLIOptions(); |
18 |
39 |
19 String typesString = null; |
40 List<Integer> numberedTypes = new ArrayList<>(); |
|
41 Map<String, Integer> namedTypes = new HashMap<>(); |
20 |
42 |
21 for (int i = 0; i < args.length; i++) { |
43 for (int i = 0; i < args.length; i++) { |
22 String arg = args[i]; |
44 String arg = args[i]; |
23 switch (arg) { |
45 switch (arg) { |
24 case Tokens.TYPES: |
46 case Tokens.TYPES: |
25 typesString = args[++i]; |
47 String typesString = args[++i]; |
|
48 |
|
49 for (String oneType : typesString.split("\\s*,\\s*")) { |
|
50 int sepatratorIndex = oneType.indexOf(TYPE_NAME_SEPARATOR); |
|
51 if (sepatratorIndex == -1) { |
|
52 numberedTypes.add(getType(oneType)); |
|
53 } else { |
|
54 String namePart = oneType.substring(0, sepatratorIndex); |
|
55 String typePart = oneType.substring(sepatratorIndex + TYPE_NAME_SEPARATOR.length(), oneType.length()); |
|
56 namedTypes.put(namePart, getType(typePart)); |
|
57 } |
|
58 } |
26 break; |
59 break; |
27 case Tokens.NAME_PREFIX: |
60 case Tokens.NAME_PREFIX: |
28 options.setNamePrefix(args[++i]); |
61 options.setNamePrefix(args[++i]); |
29 break; |
|
30 } |
|
31 } |
|
32 |
|
33 |
|
34 for (int i = 0; i < args.length; i++) { |
|
35 String arg = args[i]; |
|
36 switch (arg) { |
|
37 case Tokens.TYPES: |
|
38 case Tokens.NAME_PREFIX: |
|
39 i++; |
|
40 break; |
62 break; |
41 case Tokens.DB: |
63 case Tokens.DB: |
42 options.setDatabaseName(args[++i]); |
64 options.setDatabaseName(args[++i]); |
43 break; |
65 break; |
44 case Tokens.SQL: |
66 case Tokens.SQL: |
51 options.setCommandType(CLIOptions.COMMAND_TYPE.UPDATE); |
73 options.setCommandType(CLIOptions.COMMAND_TYPE.UPDATE); |
52 break; |
74 break; |
53 case Tokens.BATCH: |
75 case Tokens.BATCH: |
54 options.setBatch(true); |
76 options.setBatch(true); |
55 break; |
77 break; |
56 case Tokens.DATA: |
78 case Tokens.DATA: // --data is the last option |
|
79 for (i++; i < args.length; i++) { |
|
80 arg = args[i]; |
|
81 |
|
82 if (arg.startsWith(options.getNamePrefix())) { |
|
83 String paramName = arg.substring(options.getNamePrefix().length()); |
|
84 String paramValue = args[++i]; |
|
85 options.addNamedParameter(new NamedParameter(paramName, paramValue, namedTypes.get(paramName))); |
|
86 } else { |
|
87 int paramIndex = options.getNumberedParameters().size(); |
|
88 int paramType; |
|
89 try { |
|
90 paramType = numberedTypes.get(paramIndex); |
|
91 } catch (IndexOutOfBoundsException e) { |
|
92 throw new IllegalArgumentException("Missing type for parameter #" + paramIndex, e); |
|
93 } catch (NullPointerException e) { |
|
94 throw new IllegalArgumentException("Invalid type definition for parameter #" + paramIndex, e); |
|
95 } |
|
96 options.addNumberedParameter(new Parameter(arg, paramType)); |
|
97 } |
|
98 } |
57 break; |
99 break; |
|
100 default: |
|
101 throw new IllegalArgumentException("Unknown option: " + arg); |
58 } |
102 } |
59 } |
103 } |
|
104 |
|
105 |
60 |
106 |
61 return options; |
107 return options; |
62 |
108 |
63 } |
109 } |
64 |
110 |