java/sql-dk/src/info/globalcode/sql/dk/CLIStarter.java
author František Kučera <franta-hg@frantovo.cz>
Sun, 15 Dec 2013 22:07:51 +0100
branchv_0
changeset 4 f5c3350f3d78
parent 3 efdf2b886feb
child 5 26223eb63851
permissions -rw-r--r--
data/types CLI options parsing
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
package info.globalcode.sql.dk;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     3
import java.sql.Types;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     4
import java.util.ArrayList;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     5
import java.util.Collections;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     6
import java.util.HashMap;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     7
import java.util.List;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     8
import java.util.Map;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     9
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
/**
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 *
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * @author Ing. František Kučera (frantovo.cz)
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 */
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    14
public class CLIStarter {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    16
	public static final String TYPE_NAME_SEPARATOR = ":";
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    17
	private static final Map<String, Integer> types;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    18
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    19
	static {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    20
		Map<String, Integer> m = new HashMap<>();
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    21
		m.put("int", Types.INTEGER);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    22
		m.put("string", Types.VARCHAR);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    23
		m.put("boolean", Types.BOOLEAN);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    24
		/**
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    25
		 * TODO: more types
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    26
		 */
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    27
		types = Collections.unmodifiableMap(m);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    28
	}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    29
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
	public static void main(String[] args) {
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    31
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    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"};
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    33
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
		CLIOptions options = parseOptions(args);
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    35
	}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
	private static CLIOptions parseOptions(String[] args) {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
		CLIOptions options = new CLIOptions();
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    40
		List<Integer> numberedTypes = new ArrayList<>();
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    41
		Map<String, Integer> namedTypes = new HashMap<>();
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    42
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    43
		for (int i = 0; i < args.length; i++) {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
			String arg = args[i];
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    45
			switch (arg) {
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    46
				case Tokens.TYPES:
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    47
					String typesString = args[++i];
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    48
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    49
					for (String oneType : typesString.split("\\s*,\\s*")) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    50
						int sepatratorIndex = oneType.indexOf(TYPE_NAME_SEPARATOR);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    51
						if (sepatratorIndex == -1) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    52
							numberedTypes.add(getType(oneType));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    53
						} else {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    54
							String namePart = oneType.substring(0, sepatratorIndex);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    55
							String typePart = oneType.substring(sepatratorIndex + TYPE_NAME_SEPARATOR.length(), oneType.length());
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    56
							namedTypes.put(namePart, getType(typePart));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    57
						}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    58
					}
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    59
					break;
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    60
				case Tokens.NAME_PREFIX:
3
efdf2b886feb name prefix
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    61
					options.setNamePrefix(args[++i]);
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    62
					break;
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
				case Tokens.DB:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    64
					options.setDatabaseName(args[++i]);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
				case Tokens.SQL:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    67
					options.setSql(args[++i]);
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    68
					options.setCommandType(CLIOptions.COMMAND_TYPE.QUERY);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
				case Tokens.SQL_UPDATE:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    71
				case Tokens.SQL_INSERT:
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    72
					options.setSql(args[++i]);
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    73
					options.setCommandType(CLIOptions.COMMAND_TYPE.UPDATE);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    75
				case Tokens.BATCH:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    76
					options.setBatch(true);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    77
					break;
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    78
				case Tokens.DATA: // --data is the last option
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    79
					for (i++; i < args.length; i++) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    80
						arg = args[i];
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    81
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    82
						if (arg.startsWith(options.getNamePrefix())) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    83
							String paramName = arg.substring(options.getNamePrefix().length());
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    84
							String paramValue = args[++i];
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    85
							options.addNamedParameter(new NamedParameter(paramName, paramValue, namedTypes.get(paramName)));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    86
						} else {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    87
							int paramIndex = options.getNumberedParameters().size();
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    88
							int paramType;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    89
							try {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    90
								paramType = numberedTypes.get(paramIndex);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    91
							} catch (IndexOutOfBoundsException e) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    92
								throw new IllegalArgumentException("Missing type for parameter #" + paramIndex, e);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    93
							} catch (NullPointerException e) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    94
								throw new IllegalArgumentException("Invalid type definition for parameter #" + paramIndex, e);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    95
							}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    96
							options.addNumberedParameter(new Parameter(arg, paramType));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    97
						}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    98
					}
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    99
					break;
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   100
				default:
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   101
					throw new IllegalArgumentException("Unknown option: " + arg);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   102
			}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   103
		}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   104
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   105
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   106
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   107
		return options;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   108
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   109
	}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   110
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   111
	public static class Tokens {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   112
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   113
		public static final String DB = "--db";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   114
		public static final String SQL = "--sql";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   115
		public static final String SQL_UPDATE = "--sql-update";
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   116
		public static final String SQL_INSERT = "--sql-insert";
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   117
		public static final String BATCH = "--batch";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   118
		public static final String DATA = "--data";
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   119
		public static final String NAME_PREFIX = "--name-prefix";
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   120
		public static final String TYPES = "--types";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   121
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   122
		private Tokens() {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   123
		}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   124
	}
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   125
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   126
	private static Integer getType(String typeString) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   127
		return types.get(typeString);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   128
	}
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   129
}