/// <summary>
/// Prints keys and values from <see cref="RocksDbSharp.RocksDb"/> located at <paramref name="rocksdbPath"/>.
/// </summary>
/// <param name="prefix">The prefix of keys to filter.</param>
/// <param name="inputFormat">An format of <paramref name="prefix"/>.
/// For instance, if <paramref name="inputFormat"/> is <see cref="InputFormat.Base64"/>, <paramref name="prefix"/>
/// should be base64 encoded value. If <paramref name="inputFormat"/> is <see cref="InputFormat.String"/>,
/// it treats <paramref name="prefix"/> as <see cref="Encoding.UTF8"/> encoded string.</param>
/// <param name="outputFormat">A format of output (i.e. keys and values).</param>
/// <param name="rocksdbPath">The path of <see cref="RocksDbSharp.RocksDb"/> to load.</param>
public void List(
[Option] string prefix = "",
[Option] InputFormat inputFormat = InputFormat.String,
[Option] OutputFormat outputFormat = OutputFormat.Hex,
[Option] string?rocksdbPath = null)
{
rocksdbPath ??= Directory.GetCurrentDirectory();
try
{
using var db = _rocksDbService.Load(rocksdbPath);
byte[] prefixBytes = inputFormat.Decode(prefix);
_inputOutputErrorContainer.Error.WriteLine("Key\tValue");
using var iterator = db.NewIterator();
for (iterator.Seek(prefixBytes);
iterator.Valid() && iterator.Key().Take(prefixBytes.Length).SequenceEqual(prefixBytes);
iterator.Next())
{
_inputOutputErrorContainer.Out.Write(outputFormat.Encode(iterator.Key()));
_inputOutputErrorContainer.Out.Write('\t');
_inputOutputErrorContainer.Out.WriteLine(outputFormat.Encode(iterator.Value()));
}
}
catch (Exception e)
{
_inputOutputErrorContainer.Error.WriteLine(e.Message);
}
}
/// <summary>
/// Sets the value corresponded to <paramref name="key"/> from rocksdb.
/// </summary>
/// <param name="key">A key string following <paramref name="inputFormat"/>,
/// to set into <see cref="RocksDbSharp.RocksDb"/> as a key.</param>
/// <param name="value">A value following <paramref name="inputFormat"/>,
/// to set into <see cref="RocksDbSharp.RocksDb"/> as a value.</param>
/// <param name="inputFormat">An format of <paramref name="key"/> and <paramref name="value"/>.
/// If <paramref name="inputFormat"/> is <see cref="InputFormat.Base64"/>, <paramref name="key"/>
/// and <paramref name="value"/> should be base64 encoded value. If <paramref name="inputFormat"/> is
/// <see cref="InputFormat.String"/>, it treats <paramref name="key"/> and <paramref name="value"/> as string.</param>
/// <param name="rocksdbPath">The path of <see cref="RocksDbSharp.RocksDb"/> to load.</param>
/// <returns>If it successes, returns 0. If it fails, returns -1.</returns>
public int Set([Argument] string key, [Argument] string value, [Option] InputFormat inputFormat = InputFormat.String, [Option] string?rocksdbPath = null)
{
rocksdbPath ??= Directory.GetCurrentDirectory();
try
{
using var db = _rocksDbService.Load(rocksdbPath);
byte[] keyBytes = inputFormat.Decode(key);
byte[] valueBytes = inputFormat.Decode(value);
db.Put(keyBytes, valueBytes);
return(0);
}
catch (Exception e)
{
_inputOutputErrorContainer.Error.WriteLine(e.Message);
}
return(-1);
}
[InlineData(InputFormat.String, "")] // empty
public void Remove(InputFormat inputFormat, string key)
{
_command.Remove(
key: key,
inputFormat: inputFormat,
rocksdbPath: _temporaryDirectory);
using var db = _rocksDbService.Load(_temporaryDirectory);
Assert.Null(db.Get(inputFormat.Decode(key)));
}
/// <summary>
/// Removes <paramref name="key"/> from <see cref="RocksDbSharp.RocksDb"/>
/// located at <paramref name="rocksdbPath"/>.
/// </summary>
/// <param name="key">The key of the value to remove.</param>
/// <param name="inputFormat">An format of <paramref name="key"/>.
/// For instance, if <paramref name="inputFormat"/> is <see cref="InputFormat.Base64"/>, <paramref name="key"/>
/// should be base64 encoded value. If <paramref name="inputFormat"/> is <see cref="InputFormat.String"/>,
/// it treats <paramref name="key"/> as <see cref="Encoding.UTF8"/> encoded string. If
/// <paramref name="inputFormat"/> is <see cref="InputFormat.Hex"/>, it treats as hexadecimal-string.</param>
/// <param name="rocksdbPath">The path of <see cref="RocksDbSharp.RocksDb"/> to load.</param>
public void Remove(
[Argument] string key,
[Option] InputFormat inputFormat = InputFormat.String,
[Option] string?rocksdbPath = null)
{
rocksdbPath ??= Directory.GetCurrentDirectory();
try
{
using var db = _rocksDbService.Load(rocksdbPath);
byte[] keyBytes = inputFormat.Decode(key);
db.Remove(keyBytes);
}
catch (Exception e)
{
_inputOutputErrorContainer.Error.WriteLine(e.Message);
}
}
/// <summary>
/// Gets the value corresponded to <paramref name="key"/> from rocksdb.
/// </summary>
/// <param name="key">A key string following <paramref name="inputFormat"/>,
/// to get from <see cref="RocksDbSharp.RocksDb"/> as a key.</param>
/// <param name="inputFormat">An format of <paramref name="key"/>.
/// If <paramref name="inputFormat"/> is <see cref="Base64"/>, <paramref name="key"/>
/// should be base64 encoded value. If <paramref name="inputFormat"/> is
/// <see cref="string"/>, it treats <paramref name="key"/> as <see cref="Encoding.UTF8"/> encoded
/// string.</param>
/// <param name="outputFormat">A format of output.</param>
/// <param name="rocksdbPath">The path of <see cref="RocksDbSharp.RocksDb"/> to load.</param>
/// <returns>If it successes, returns 0. If it fails, returns -1.</returns>
public int Get([Argument] string key, [Option] InputFormat inputFormat = InputFormat.String, [Option] OutputFormat outputFormat = OutputFormat.Base64, [Option] string?rocksdbPath = null)
{
rocksdbPath ??= Directory.GetCurrentDirectory();
try
{
using var db = _rocksDbService.Load(rocksdbPath);
byte[] keyBytes = inputFormat.Decode(key);
if (db.Get(keyBytes) is { } valueBytes)
{
string value = outputFormat.Encode(valueBytes);
_inputOutputErrorContainer.Out.Write(value);
return(0);
}
}
catch (Exception e)
{
_inputOutputErrorContainer.Error.WriteLine(e.Message);
}
return(-1);
}
