int
util_write(WT_SESSION *session, int argc, char *argv[])
{
WT_CURSOR *cursor;
uint64_t recno;
int append, ch, overwrite, rkey, ret;
const char *uri;
char config[100];
append = overwrite = ret = 0;
while ((ch = util_getopt(argc, argv, "ao")) != EOF)
switch (ch) {
case 'a':
append = 1;
break;
case 'o':
overwrite = 1;
break;
case '?':
default:
return (usage());
}
argc -= util_optind;
argv += util_optind;
/*
* The remaining arguments are a uri followed by a list of values (if
* append is set), or key/value pairs (if append is not set).
*/
if (append) {
if (argc < 2)
return (usage());
} else
if (argc < 3 || ((argc - 1) % 2 != 0))
return (usage());
if ((uri =
util_name(*argv, "table", UTIL_FILE_OK | UTIL_TABLE_OK)) == NULL)
return (1);
/* Open the object. */
(void)snprintf(config, sizeof(config), "%s,%s",
append ? "append=true" : "", overwrite ? "overwrite=true" : "");
if ((ret = session->open_cursor(
session, uri, NULL, config, &cursor)) != 0)
return (util_err(ret, "%s: session.open", uri));
/*
* A simple search only makes sense if the key format is a string or a
* record number, and the value format is a single string.
*/
if (strcmp(cursor->key_format, "r") != 0 &&
strcmp(cursor->key_format, "S") != 0) {
fprintf(stderr,
"%s: write command only possible when the key format is "
"a record number or string\n",
progname);
return (1);
}
rkey = strcmp(cursor->key_format, "r") == 0 ? 1 : 0;
if (strcmp(cursor->value_format, "S") != 0) {
fprintf(stderr,
"%s: write command only possible when the value format is "
"a string\n",
progname);
return (1);
}
/* Run through the values or key/value pairs. */
while (*++argv != NULL) {
if (!append) {
if (rkey) {
if (util_str2recno(*argv, &recno))
return (1);
cursor->set_key(cursor, recno);
} else
cursor->set_key(cursor, *argv);
++argv;
}
cursor->set_value(cursor, *argv);
if ((ret = cursor->insert(cursor)) != 0)
return (util_cerr(uri, "search", ret));
}
return (0);
}
int
util_read(WT_SESSION *session, int argc, char *argv[])
{
WT_CURSOR *cursor;
WT_DECL_RET;
uint64_t recno;
int ch;
bool rkey, rval;
char *uri, *value;
uri = NULL;
while ((ch = __wt_getopt(progname, argc, argv, "")) != EOF)
switch (ch) {
case '?':
default:
return (usage());
}
argc -= __wt_optind;
argv += __wt_optind;
/* The remaining arguments are a uri followed by a list of keys. */
if (argc < 2)
return (usage());
if ((uri = util_uri(session, *argv, "table")) == NULL)
return (1);
/*
* Open the object; free allocated memory immediately to simplify
* future error handling.
*/
if ((ret =
session->open_cursor(session, uri, NULL, NULL, &cursor)) != 0)
(void)util_err(session, ret, "%s: session.open_cursor", uri);
free(uri);
if (ret != 0)
return (ret);
/*
* A simple search only makes sense if the key format is a string or a
* record number, and the value format is a single string.
*/
if (strcmp(cursor->key_format, "r") != 0 &&
strcmp(cursor->key_format, "S") != 0) {
fprintf(stderr,
"%s: read command only possible when the key format is "
"a record number or string\n",
progname);
return (1);
}
rkey = strcmp(cursor->key_format, "r") == 0;
if (strcmp(cursor->value_format, "S") != 0) {
fprintf(stderr,
"%s: read command only possible when the value format is "
"a string\n",
progname);
return (1);
}
/*
* Run through the keys, returning non-zero on error or if any requested
* key isn't found.
*/
for (rval = false; *++argv != NULL;) {
if (rkey) {
if (util_str2recno(session, *argv, &recno))
return (1);
cursor->set_key(cursor, recno);
} else
cursor->set_key(cursor, *argv);
switch (ret = cursor->search(cursor)) {
case 0:
if ((ret = cursor->get_value(cursor, &value)) != 0)
return (util_cerr(cursor, "get_value", ret));
if (printf("%s\n", value) < 0)
return (util_err(session, EIO, NULL));
break;
case WT_NOTFOUND:
(void)util_err(session, 0, "%s: not found", *argv);
rval = true;
break;
default:
return (util_cerr(cursor, "search", ret));
}
}
return (rval ? 1 : 0);
}
