taler-merchant-dbinit.c (4105B)
1 /* 2 This file is part of TALER 3 Copyright (C) 2014, 2015, 2020 Taler Systems SA 4 5 TALER is free software; you can redistribute it and/or modify it under the 6 terms of the GNU General Public License as published by the Free Software 7 Foundation; either version 3, or (at your option) any later version. 8 9 TALER is distributed in the hope that it will be useful, but WITHOUT ANY 10 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR 11 A PARTICULAR PURPOSE. See the GNU General Public License for more details. 12 13 You should have received a copy of the GNU General Public License along with 14 TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/> 15 */ 16 /** 17 * @file src/merchant-tools/taler-merchant-dbinit.c 18 * @brief Create tables for the merchant database. 19 * @author Florian Dold 20 * @author Marcello Stanisci 21 */ 22 #include "platform.h" 23 #include <taler/taler_util.h> 24 #include <gnunet/gnunet_util_lib.h> 25 #include "taler/taler_merchant_util.h" 26 #include "merchantdb_lib.h" 27 #include "merchant-database/create_tables.h" 28 #include "merchant-database/drop_tables.h" 29 #include "merchant-database/gc.h" 30 31 32 /** 33 * Return value from main(). 34 */ 35 static int global_ret; 36 37 /** 38 * -r option: do full DB reset 39 */ 40 static int reset_db; 41 42 /** 43 * -g option: garbage collect DB 44 */ 45 static int gc_db; 46 47 /** 48 * Main function that will be run. 49 * 50 * @param cls closure 51 * @param args remaining command-line arguments 52 * @param cfgfile name of the configuration file used (for saving, can be NULL!) 53 * @param cfg configuration 54 */ 55 static void 56 run (void *cls, 57 char *const *args, 58 const char *cfgfile, 59 const struct GNUNET_CONFIGURATION_Handle *cfg) 60 { 61 struct TALER_MERCHANTDB_PostgresContext *pg; 62 63 /* Reset retains precedence over GC. Scheduled GC must never migrate. */ 64 if (reset_db) 65 gc_db = 0; 66 pg = gc_db 67 ? TALER_MERCHANTDB_connect (cfg) 68 : TALER_MERCHANTDB_connect_admin (cfg); 69 if (NULL == pg) 70 { 71 fprintf (stderr, 72 "Failed to initialize database connection.\n"); 73 global_ret = 1; 74 return; 75 } 76 if (reset_db) 77 { 78 if (GNUNET_OK != 79 TALER_MERCHANTDB_drop_tables (pg)) 80 { 81 global_ret = 1; 82 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 83 "Failed to reset the database\n"); 84 TALER_MERCHANTDB_disconnect (pg); 85 return; 86 } 87 gc_db = 0; /* not useful */ 88 } 89 if (gc_db) 90 { 91 if (GNUNET_OK != 92 TALER_MERCHANTDB_gc (pg)) 93 { 94 global_ret = 1; 95 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 96 "Failed to garbage collect database\n"); 97 } 98 TALER_MERCHANTDB_disconnect (pg); 99 return; 100 } 101 if (GNUNET_OK != 102 TALER_MERCHANTDB_create_tables (pg)) 103 { 104 global_ret = 1; 105 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 106 "Failed to initialize tables\n"); 107 TALER_MERCHANTDB_disconnect (pg); 108 return; 109 } 110 TALER_MERCHANTDB_disconnect (pg); 111 } 112 113 114 /** 115 * The main function of the database initialization tool. 116 * Used to initialize the Taler Exchange's database. 117 * 118 * @param argc number of arguments from the command line 119 * @param argv command line arguments 120 * @return 0 ok, 1 on error 121 */ 122 int 123 main (int argc, 124 char *const *argv) 125 { 126 struct GNUNET_GETOPT_CommandLineOption options[] = { 127 GNUNET_GETOPT_option_flag ('g', 128 "gc", 129 "garbage collect database", 130 &gc_db), 131 GNUNET_GETOPT_option_flag ('r', 132 "reset", 133 "reset database (DANGEROUS: all existing data is lost!)", 134 &reset_db), 135 136 GNUNET_GETOPT_option_version (PACKAGE_VERSION), 137 GNUNET_GETOPT_OPTION_END 138 }; 139 enum GNUNET_GenericReturnValue ret; 140 141 ret = GNUNET_PROGRAM_run ( 142 TALER_MERCHANT_project_data (), 143 argc, argv, 144 "taler-merchant-dbinit", 145 gettext_noop ("Initialize Taler merchant database"), 146 options, 147 &run, NULL); 148 if (GNUNET_SYSERR == ret) 149 return 3; 150 if (GNUNET_NO == ret) 151 return 0; 152 return global_ret; 153 } 154 155 156 /* end of taler-merchant-dbinit.c */