Mercurial Hosting > nabble
diff src/jdbcpgbackup/JdbcPgBackup.java @ 0:7ecd1a4ef557
add content
author | Franklin Schmidt <fschmidt@gmail.com> |
---|---|
date | Thu, 21 Mar 2019 19:15:52 -0600 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/jdbcpgbackup/JdbcPgBackup.java Thu Mar 21 19:15:52 2019 -0600 @@ -0,0 +1,129 @@ +/* Copyright (c) 2012 Tomislav Gountchev <tomi@gountchev.net> */ + +package jdbcpgbackup; + +import java.util.Arrays; +import java.util.HashMap; +import java.util.Map; + +public final class JdbcPgBackup { + + public static final String USAGE = + "Usage: JdbcPgBackup -m dump|restore [-h hostname] [-p port] [-t (timing)] " + + "[-d database] [-U user] [-P password] [-f filename] [-o (schema only)] " + + "[-s schema[,schema...]] [-n schema[,schema...]] [-b batchsize]"; + + private static Map<String,String> parseArgs(String[] args) { + Map<String,String> params = new HashMap<String,String>(); + for (int i = 0; i<args.length; i++) { + if (args[i].startsWith("-") && args[i].length() == 2) { + char option = args[i].charAt(1); + switch(option) { + case 'm': + params.put("mode", args[++i]); + break; + case 'h': + params.put("hostname", args[++i]); + break; + case 'p': + params.put("port", args[++i]); + break; + case 'd': + params.put("database", args[++i]); + break; + case 'U': + params.put("user", args[++i]); + break; + case 'P': + params.put("password", args[++i]); + break; + case 'f': + params.put("filename", args[++i]); + break; + case 's': + params.put("schemas", args[++i]); + break; + case 'n': + params.put("toschemas", args[++i]); + break; + case 'b': + params.put("batch", args[++i]); + break; + case 't': + params.put("debug", "true"); + break; + case 'o': + params.put("nodata", "true"); + break; + default: + throw new RuntimeException("invalid parameter: " + args[i]); + } + } else throw new RuntimeException("invalid parameters"); + } + return params; + } + + public static void main(String[] args) { + if (args.length == 0) { + System.err.println(USAGE); + System.exit(1); + } + Map<String, String> params = null; + try { + params = parseArgs(args); + } catch (RuntimeException e) { + System.err.println(USAGE); + System.err.println(e.getMessage()); + System.exit(1); + } + ZipBackup backup = new ZipBackup(params); + try { + String[] schemas = null; + String[] toSchemas = null; + String schemasParam = params.get("schemas"); + if (schemasParam != null) { + schemas = schemasParam.split(","); + } + String toSchemasParam = params.get("toschemas"); + if (toSchemasParam != null) { + toSchemas = toSchemasParam.split(","); + if (schemas == null || schemas.length == 0 || schemas.length != toSchemas.length) + throw new RuntimeException("non-matching source schema (-s) and destination schema (-n) parameters"); + } + String mode = params.get("mode"); + + if ("true".equals(params.get("debug"))) + ZipBackup.setTimingOutput(System.err); + + if ("dump".equals(mode)) { + boolean nodata = "true".equals(params.get("nodata")); + DataFilter dataFilter = nodata ? DataFilter.NO_DATA : DataFilter.ALL_DATA; + String batchS = params.get("batch"); + int batch = batchS == null ? ZipBackup.DEFAULT_BATCH_SIZE : Integer.parseInt(batchS); + if (schemas == null) { + backup.dumpAll(dataFilter, batch); + } else { + backup.dump(Arrays.asList(schemas), dataFilter); + } + } else if ("restore".equals(mode)) { + if (schemas == null) { + backup.restoreAll(); + } else if (toSchemas == null) { + for (String schema : schemas) { + backup.restoreSchema(schema); + } + } else { + for (int i=0; i<schemas.length; i++) { + backup.restoreSchemaTo(schemas[i], toSchemas[i]); + } + } + } else throw new RuntimeException("invalid mode: " + mode); + } catch (RuntimeException e) { + System.err.println("backup failed: " + e.getMessage()); + e.printStackTrace(System.err); + System.exit(1); + } + System.exit(0); + } + +}