summaryrefslogtreecommitdiff
path: root/examples/c/csv/query_main.c
blob: eb15636750a9ca7ffb3a718d0219609d12f0abda (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*-
 * See the file LICENSE for redistribution information.
 *
 * Copyright (c) 2005, 2015 Oracle and/or its affiliates.  All rights reserved.
 *
 * $Id$
 */

#include "csv.h"
#include "csv_local.h"
#include "csv_extern.h"

static int usage(void);

/*
 * Globals
 */
DB_ENV	 *dbenv;			/* Database environment */
DB	 *db;				/* Primary database */
int	  verbose;			/* Program verbosity */
char	 *progname;			/* Program name */

int
main(int argc, char *argv[])
{
	int ch, done, ret, t_ret;
	char **clist, **clp, *home;

	/* Initialize globals. */
	dbenv = NULL;
	db = NULL;
	if ((progname = strrchr(argv[0], '/')) == NULL)
		progname = argv[0];
	else
		++progname;
	verbose = 0;

	/* Initialize arguments. */
	home = NULL;
	ret = 0;

	/* Allocate enough room for command-list arguments. */
	if ((clp = clist =
	    (char **)calloc((size_t)argc + 1, sizeof(char *))) == NULL) {
		fprintf(stderr, "%s: %s\n", progname, strerror(ENOMEM));
		return (EXIT_FAILURE);
	}

	/* Process arguments. */
	while ((ch = getopt(argc, argv, "c:h:v")) != EOF)
		switch (ch) {
		case 'c':
			*clp++ = optarg;
			break;
		case 'h':
			home = optarg;
			break;
		case 'v':
			++verbose;
			break;
		case '?':
		default:
			return (usage());
		}
	argc -= optind;
	argv += optind;

	if (*argv != NULL)
		return (usage());

	/* Create or join the database environment. */
	if (csv_env_open(home, 1) != 0)
		return (EXIT_FAILURE);

	/* Handle the queries. */
	if (clp == clist)
		ret = query_interactive();
	else
		for (clp = clist, done = 0; *clp != NULL && !done; ++clp)
			if ((ret = query(*clp, &done)) != 0)
				break;

	/* Close the database environment. */
	if ((t_ret = csv_env_close()) != 0 && ret == 0)
		ret = t_ret;

	return (ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
}

/*
 * usage --
 *	Program usage message.
 */
static int
usage(void)
{
	(void)fprintf(stderr, "usage: %s [-v] [-c cmd] [-h home]\n", progname);
	return (EXIT_FAILURE);
}