redis.clients.jedis.commands.ProtocolCommand Java Examples

@Test
public void getErrorAfterConnectionReset() throws Exception {
  class TestConnection extends Connection {
    public TestConnection() {
      super("localhost", 6379);
    }

    @Override
    protected Connection sendCommand(ProtocolCommand cmd, byte[]... args) {
      return super.sendCommand(cmd, args);
    }
  }

  TestConnection conn = new TestConnection();

  try {
    conn.sendCommand(Command.HMSET, new byte[1024 * 1024 + 1][0]);
    fail("Should throw exception");
  } catch (JedisConnectionException jce) {
    assertEquals("ERR Protocol error: invalid multibulk length", jce.getMessage());
  }
}
 

protected Connection sendCommand(final ProtocolCommand cmd, final String... args) {
  final byte[][] bargs = new byte[args.length][];
  for (int i = 0; i < args.length; i++) {
    bargs[i] = SafeEncoder.encode(args[i]);
  }
  return sendCommand(cmd, bargs);
}
 

@Override
public Object sendCommand(ProtocolCommand cmd, byte[]... args) {
  Span span = helper.buildSpan("sendCommand");
  span.setTag("cmd", nullable(cmd));
  span.setTag("args", TracingHelper.toString(args));
  return helper.decorate(span, () -> wrapped.sendCommand(cmd, args));
}
 

@Override
public Object sendCommand(ProtocolCommand cmd,
    String... args) {
  Span span = helper.buildSpan("sendCommand");
  span.setTag("cmd", nullable(cmd));
  span.setTag("args", Arrays.toString(args));
  return helper.decorate(span, () -> wrapped.sendCommand(cmd, args));
}
 

@Override
public ProtocolCommand getAggregateCommand() {
    return ClusterCommand.AGGREGATE;
}
 

@Override
public ProtocolCommand getAliasDelCommand() {
    return ClusterCommand.ALIASDEL;
}
 

@Override
public ProtocolCommand getCursorCommand() {
  return ClusterCommand.CURSOR;
}
 

private BinaryClient sendCommand(Jedis conn, ProtocolCommand provider, String... args) {
    BinaryClient client = conn.getClient();
    client.sendCommand(provider, args);
    return client;
}
 

public static void sendCommand(final RedisOutputStream os, final ProtocolCommand command,
    final byte[]... args) {
  sendCommand(os, command.getRaw(), args);
}
 

protected Connection sendCommand(final ProtocolCommand cmd) {
  return sendCommand(cmd, EMPTY_ARGS);
}
 

@Override
public Object sendCommand(ProtocolCommand cmd) {
  Span span = helper.buildSpan("sendCommand");
  span.setTag("cmd", nullable(cmd));
  return helper.decorate(span, () -> wrapped.sendCommand(cmd));
}
 

private Connection sendCommand(Jedis conn, ProtocolCommand command, byte[]... args) {
  Connection client = conn.getClient();
  client.sendCommand(command, args);
  return client;
}
 

private Connection sendCommand(Jedis conn, String key, ProtocolCommand command, String ...args) {
  byte[][] fullArgs = new byte[args.length + 1][];
  fullArgs[0] = SafeEncoder.encode(key);
  System.arraycopy( SafeEncoder.encodeMany(args), 0, fullArgs, 1, args.length);
  return sendCommand(conn, command, fullArgs);
}
 

private void sendCommand(Connection conn, String key, ProtocolCommand command, String ...args) {
    String[] fullArgs = new String[args.length + 1];
    fullArgs[0] = key;
    System.arraycopy(args, 0, fullArgs, 1, args.length);
    conn.sendCommand(command, fullArgs);
}
 

private BinaryClient sendCommand(Jedis conn, ProtocolCommand provider, byte[]... args) {
    BinaryClient client = conn.getClient();
    client.sendCommand(provider, args);
    return client;
}
 

@Override
public ProtocolCommand getAliasUpdateCommand() {
    return ClusterCommand.ALIASUPDATE;
}
 

@Override
public ProtocolCommand getSynAddCommand() {
  return Command.SYNADD;
}
 

@Override
public ProtocolCommand getGetCommand() {
    return ClusterCommand.GET;
}
 

@Override
public ProtocolCommand getExplainCommand() {
    return ClusterCommand.EXPLAIN;
}
 

@Override
public ProtocolCommand getSearchCommand() {
    return ClusterCommand.SEARCH;
}
 

@Override
public ProtocolCommand getDropCommand() {
    return ClusterCommand.DROP;
}
 

@Override
public ProtocolCommand getInfoCommand() {
    return ClusterCommand.INFO;
}
 

@Override
public ProtocolCommand getDelCommand() {
    return ClusterCommand.DEL;
}
 

@Override
public ProtocolCommand getAddHashCommand() {
    return ClusterCommand.ADDHASH;
}
 

@Override
public ProtocolCommand getAddCommand() {
    return ClusterCommand.ADD;
}
 

@Override
public ProtocolCommand getAlterCommand() {
    return ClusterCommand.ALTER;
}
 

@Override
public ProtocolCommand getCreateCommand() {
    return ClusterCommand.CREATE;
}
 

@Override
public ProtocolCommand getSynDumpCommand() {
  return Command.SYNDUMP;
}
 

@Override
public ProtocolCommand getSynUpdateCommand() {
  return Command.SYNUPDATE;
}
 

@Override
public ProtocolCommand getMGetCommand() {
  return Command.MGET;
}