src/main/java/com/google/gerrit/elasticsearch/ElasticStoredValue.java

// Copyright (C) 2021 The Android Open Source Project
//
// Licensed 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 com.google.gerrit.elasticsearch;

import static com.google.common.collect.ImmutableList.toImmutableList;

import com.google.gerrit.index.StoredValue;
import com.google.gson.JsonElement;
import com.google.protobuf.MessageLite;
import java.sql.Timestamp;
import java.time.Instant;
import java.time.format.DateTimeFormatter;
import java.util.stream.StreamSupport;

/** Bridge to recover fields from the elastic index. */
public class ElasticStoredValue implements StoredValue {
  private final JsonElement field;

  ElasticStoredValue(JsonElement field) {
    this.field = field;
  }

  @Override
  public String asString() {
    return field.getAsString();
  }

  @Override
  public Iterable<String> asStrings() {
    return StreamSupport.stream(field.getAsJsonArray().spliterator(), false)
        .map(f -> f.getAsString())
        .collect(toImmutableList());
  }

  @Override
  public Integer asInteger() {
    return field.getAsInt();
  }

  @Override
  public Iterable<Integer> asIntegers() {
    return StreamSupport.stream(field.getAsJsonArray().spliterator(), false)
        .map(f -> f.getAsInt())
        .collect(toImmutableList());
  }

  @Override
  public Long asLong() {
    return field.getAsLong();
  }

  @Override
  public Iterable<Long> asLongs() {
    return StreamSupport.stream(field.getAsJsonArray().spliterator(), false)
        .map(f -> f.getAsLong())
        .collect(toImmutableList());
  }

  @Override
  public Timestamp asTimestamp() {
    return Timestamp.from(Instant.from(DateTimeFormatter.ISO_INSTANT.parse(field.getAsString())));
  }

  @Override
  public byte[] asByteArray() {
    return AbstractElasticIndex.decodeBase64(field.getAsString());
  }

  @Override
  public Iterable<byte[]> asByteArrays() {
    return StreamSupport.stream(field.getAsJsonArray().spliterator(), false)
        .map(f -> AbstractElasticIndex.decodeBase64(f.getAsString()))
        .collect(toImmutableList());
  }

  @Override
  public MessageLite asProto() {
    // Elasticsearch does not store protos
    return null;
  }

  @Override
  public Iterable<MessageLite> asProtos() {
    // Elasticsearch does not store protos
    return null;
  }
}