geode-redis/src/main/java/org/apache/geode/redis/internal/executor/RedisResponse.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license
 * agreements. See the NOTICE file distributed with this work for additional information regarding
 * copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance with the License. You may obtain a
 * copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 *
 */

package org.apache.geode.redis.internal.executor;

import java.math.BigInteger;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.function.Function;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.buffer.UnpooledByteBufAllocator;

import org.apache.geode.redis.internal.netty.Coder;
import org.apache.geode.redis.internal.netty.CoderException;

public class RedisResponse {

  private final Function<ByteBufAllocator, ByteBuf> coderCallback;

  private Runnable afterWriteCallback;

  private RedisResponse(Function<ByteBufAllocator, ByteBuf> coderCallback) {
    this.coderCallback = coderCallback;
  }

  public void setAfterWriteCallback(Runnable callback) {
    afterWriteCallback = callback;
  }

  public void afterWrite() {
    if (afterWriteCallback != null) {
      afterWriteCallback.run();
    }
  }

  public ByteBuf encode(ByteBufAllocator allocator) {
    return coderCallback.apply(allocator);
  }

  public static RedisResponse integer(long numericValue) {
    return new RedisResponse((bba) -> Coder.getIntegerResponse(bba, numericValue));
  }

  public static RedisResponse integer(boolean exists) {
    return new RedisResponse((bba) -> Coder.getIntegerResponse(bba, exists ? 1 : 0));
  }

  public static RedisResponse string(String stringValue) {
    return new RedisResponse((bba) -> Coder.getSimpleStringResponse(bba, stringValue));
  }

  public static RedisResponse string(byte[] byteArray) {
    return new RedisResponse((bba) -> Coder.getSimpleStringResponse(bba, byteArray));
  }

  public static RedisResponse bulkString(Object value) {
    return new RedisResponse((bba) -> {
      try {
        return Coder.getBulkStringResponse(bba, value);
      } catch (CoderException e) {
        return Coder.getErrorResponse(bba, "Internal server error: " + e.getMessage());
      }
    });
  }

  public static RedisResponse ok() {
    return new RedisResponse((bba) -> Coder.getSimpleStringResponse(bba, "OK"));
  }

  public static RedisResponse nil() {
    return new RedisResponse(Coder::getNilResponse);
  }

  public static RedisResponse flattenedArray(Collection<Collection<?>> nestedCollection) {
    return new RedisResponse((bba) -> {
      try {
        return Coder.getFlattenedArrayResponse(bba, nestedCollection);
      } catch (CoderException e) {
        return Coder.getErrorResponse(bba, "Internal server error: " + e.getMessage());
      }
    });
  }

  public static RedisResponse array(Collection<?> collection) {
    return new RedisResponse((bba) -> {
      try {
        return Coder.getArrayResponse(bba, collection);
      } catch (CoderException e) {
        return Coder.getErrorResponse(bba, "Internal server error: " + e.getMessage());
      }
    });
  }

  public static RedisResponse emptyArray() {
    return new RedisResponse(Coder::getEmptyArrayResponse);
  }

  public static RedisResponse emptyString() {
    return new RedisResponse(Coder::getEmptyStringResponse);
  }

  public static RedisResponse error(String error) {
    return new RedisResponse((bba) -> Coder.getErrorResponse(bba, error));
  }

  public static RedisResponse customError(String error) {
    return new RedisResponse((bba) -> Coder.getCustomErrorResponse(bba, error));
  }

  public static RedisResponse wrongType(String error) {
    return new RedisResponse((bba) -> Coder.getWrongTypeResponse(bba, error));
  }

  public static RedisResponse scan(BigInteger cursor, List<Object> scanResult) {
    return new RedisResponse((bba) -> Coder.getScanResponse(bba, cursor, scanResult));
  }

  public static RedisResponse emptyScan() {
    return scan(new BigInteger("0"), new ArrayList<>());
  }

  /**
   * Be aware that this implementation will create extra garbage since it allocates from the heap.
   */
  public String toString() {
    return encode(new UnpooledByteBufAllocator(false)).toString(Charset.defaultCharset());
  }

  public static RedisResponse doubleValue(double numericValue) {
    return new RedisResponse((bba) -> Coder.getDoubleResponse(bba, numericValue));
  }

}