|
|
|
@ -4,6 +4,7 @@
|
|
|
|
|
// (c) Vitaliy Filippov, 2024
|
|
|
|
|
// License: Mozilla Public License 2.0 or Vitastor Network Public License 1.1
|
|
|
|
|
|
|
|
|
|
const fs = require('fs');
|
|
|
|
|
const fsp = require('fs').promises;
|
|
|
|
|
const http = require('http');
|
|
|
|
|
const https = require('https');
|
|
|
|
@ -15,13 +16,14 @@ License: Mozilla Public License 2.0 or Vitastor Network Public License 1.1
|
|
|
|
|
Usage:
|
|
|
|
|
|
|
|
|
|
anticli.js [OPTIONS] put <key> [<value>]
|
|
|
|
|
anticli.js [OPTIONS] get <key> [-p|--prefix] [-v|--print-value-only] [-k|--keys-only]
|
|
|
|
|
anticli.js [OPTIONS] get <key> [-p|--prefix] [-v|--print-value-only] [-k|--keys-only] [--no-temp]
|
|
|
|
|
anticli.js [OPTIONS] del <key> [-p|--prefix]
|
|
|
|
|
anticli.js [OPTIONS] load [--with-lease] < dump.json
|
|
|
|
|
|
|
|
|
|
Options:
|
|
|
|
|
|
|
|
|
|
[--endpoints|-e http://node1:2379,http://node2:2379,http://node3:2379]
|
|
|
|
|
[--cert cert.pem] [--key key.pem] [--timeout 1000]
|
|
|
|
|
[--cert cert.pem] [--key key.pem] [--timeout 1000] [--json]
|
|
|
|
|
`;
|
|
|
|
|
|
|
|
|
|
class AntiEtcdCli
|
|
|
|
@ -54,6 +56,19 @@ class AntiEtcdCli
|
|
|
|
|
{
|
|
|
|
|
options['keys_only'] = true;
|
|
|
|
|
}
|
|
|
|
|
else if (arg == '--with_lease')
|
|
|
|
|
{
|
|
|
|
|
options['with_lease'] = true;
|
|
|
|
|
}
|
|
|
|
|
else if (arg == '--write_out' && args[i+1] == 'json')
|
|
|
|
|
{
|
|
|
|
|
i++;
|
|
|
|
|
options['json'] = true;
|
|
|
|
|
}
|
|
|
|
|
else if (arg == '--json' || arg == '--write_out=json')
|
|
|
|
|
{
|
|
|
|
|
options['json'] = true;
|
|
|
|
|
}
|
|
|
|
|
else if (arg[0] == '-' && arg[1] !== '-')
|
|
|
|
|
{
|
|
|
|
|
process.stderr.write('Unknown option '+arg);
|
|
|
|
@ -68,9 +83,9 @@ class AntiEtcdCli
|
|
|
|
|
cmd.push(arg);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (!cmd.length || cmd[0] != 'get' && cmd[0] != 'put' && cmd[0] != 'del')
|
|
|
|
|
if (!cmd.length || cmd[0] != 'get' && cmd[0] != 'put' && cmd[0] != 'del' && cmd[0] != 'load')
|
|
|
|
|
{
|
|
|
|
|
process.stderr.write('Supported commands: get, put, del. Use --help to see details\n');
|
|
|
|
|
process.stderr.write('Supported commands: get, put, del, load. Use --help to see details\n');
|
|
|
|
|
process.exit(1);
|
|
|
|
|
}
|
|
|
|
|
return [ cmd, options ];
|
|
|
|
@ -102,12 +117,51 @@ class AntiEtcdCli
|
|
|
|
|
{
|
|
|
|
|
await this.del(cmd.slice(1));
|
|
|
|
|
}
|
|
|
|
|
else if (cmd[0] == 'load')
|
|
|
|
|
{
|
|
|
|
|
await this.load();
|
|
|
|
|
}
|
|
|
|
|
// wait until output is fully flushed
|
|
|
|
|
await new Promise(ok => process.stdout.write('', ok));
|
|
|
|
|
await new Promise(ok => process.stderr.write('', ok));
|
|
|
|
|
process.exit(0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async load()
|
|
|
|
|
{
|
|
|
|
|
const dump = JSON.parse(await new Promise((ok, no) => fs.readFile(0, { encoding: 'utf-8' }, (err, res) => err ? no(err) : ok(res))));
|
|
|
|
|
if (!dump.responses && !dump.kvs)
|
|
|
|
|
{
|
|
|
|
|
console.error('dump should be /kv/txn or /kv/range response in json format');
|
|
|
|
|
process.exit(1);
|
|
|
|
|
}
|
|
|
|
|
const success = [];
|
|
|
|
|
for (const r of (dump.responses
|
|
|
|
|
? dump.responses.map(r => r.response_range).filter(r => r)
|
|
|
|
|
: [ dump ]))
|
|
|
|
|
{
|
|
|
|
|
for (const kv of r.kvs)
|
|
|
|
|
{
|
|
|
|
|
if (kv.value == null)
|
|
|
|
|
{
|
|
|
|
|
console.error('dump should contain values');
|
|
|
|
|
process.exit(1);
|
|
|
|
|
}
|
|
|
|
|
success.push({ request_put: { key: kv.key, value: kv.value, lease: this.options.with_lease ? kv.lease||undefined : undefined } });
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
const res = await this.request('/v3/kv/txn', { success });
|
|
|
|
|
if (this.options.json)
|
|
|
|
|
{
|
|
|
|
|
process.stdout.write(JSON.stringify(res));
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (res.succeeded)
|
|
|
|
|
{
|
|
|
|
|
process.stdout.write('OK, loaded '+success.length+' values\n');
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async get(keys)
|
|
|
|
|
{
|
|
|
|
|
if (this.options.prefix)
|
|
|
|
@ -116,6 +170,22 @@ class AntiEtcdCli
|
|
|
|
|
}
|
|
|
|
|
const txn = { success: keys.map(key => ({ request_range: this.options.prefix ? { key: b64(key+'/'), range_end: b64(key+'0') } : { key: b64(key) } })) };
|
|
|
|
|
const res = await this.request('/v3/kv/txn', txn);
|
|
|
|
|
if (this.options.notemp)
|
|
|
|
|
{
|
|
|
|
|
// Skip temporary values (values with lease)
|
|
|
|
|
for (const r of res.responses||[])
|
|
|
|
|
{
|
|
|
|
|
if (r.response_range)
|
|
|
|
|
{
|
|
|
|
|
r.response_range.kvs = r.response_range.kvs.filter(kv => !kv.lease);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (this.options.json)
|
|
|
|
|
{
|
|
|
|
|
process.stdout.write(JSON.stringify(keys.length == 1 ? res.responses[0].response_range : res));
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
for (const r of res.responses||[])
|
|
|
|
|
{
|
|
|
|
|
if (r.response_range)
|
|
|
|
@ -139,7 +209,7 @@ class AntiEtcdCli
|
|
|
|
|
{
|
|
|
|
|
if (value === undefined)
|
|
|
|
|
{
|
|
|
|
|
value = await fsp.readFile(0, { encoding: 'utf-8' });
|
|
|
|
|
value = await new Promise((ok, no) => fs.readFile(0, { encoding: 'utf-8' }, (err, res) => err ? no(err) : ok(res)));
|
|
|
|
|
}
|
|
|
|
|
const res = await this.request('/v3/kv/put', { key: b64(key), value: b64(value) });
|
|
|
|
|
if (res.header)
|
|
|
|
@ -175,18 +245,18 @@ class AntiEtcdCli
|
|
|
|
|
{
|
|
|
|
|
if (res.json.error)
|
|
|
|
|
{
|
|
|
|
|
process.stderr.write(cur_url+': '+res.json.error);
|
|
|
|
|
process.stderr.write(cur_url+': '+res.json.error+'\n');
|
|
|
|
|
process.exit(1);
|
|
|
|
|
}
|
|
|
|
|
return res.json;
|
|
|
|
|
}
|
|
|
|
|
if (res.body)
|
|
|
|
|
{
|
|
|
|
|
process.stderr.write(cur_url+': '+res.body);
|
|
|
|
|
process.stderr.write(cur_url+': '+res.body+'\n');
|
|
|
|
|
}
|
|
|
|
|
if (res.error)
|
|
|
|
|
{
|
|
|
|
|
process.stderr.write(cur_url+': '+res.error);
|
|
|
|
|
process.stderr.write(cur_url+': '+res.error+'\n');
|
|
|
|
|
if (!res.response || !res.response.statusCode)
|
|
|
|
|
{
|
|
|
|
|
// This URL is unavailable
|
|
|
|
|