All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.projectnessie.versioned.mongodb.MongoBaseValue Maven / Gradle / Ivy

There is a newer version: 0.9.2
Show newest version
/*
 * Copyright (C) 2020 Dremio
 *
 * 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 org.projectnessie.versioned.mongodb;

import static org.projectnessie.versioned.mongodb.MongoSerDe.deserializeId;

import com.google.common.base.Preconditions;
import java.util.HashSet;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.stream.Stream;
import org.bson.BsonBinary;
import org.bson.BsonWriter;
import org.bson.Document;
import org.projectnessie.versioned.store.Id;
import org.projectnessie.versioned.store.Store;
import org.projectnessie.versioned.tiered.BaseValue;

abstract class MongoBaseValue> implements BaseValue {

  static final String ID = Store.KEY_NAME;
  static final String DT = Store.DT_NAME;

  final BsonWriter bsonWriter;

  private final Set properties = new HashSet<>();

  static > C produceBase(Document document, C v) {
    return v.id(deserializeId(document, ID)).dt(document.getLong(DT));
  }

  protected MongoBaseValue(BsonWriter bsonWriter) {
    // empty
    this.bsonWriter = bsonWriter;
  }

  @SuppressWarnings("unchecked")
  @Override
  public C id(Id id) {
    if (!properties.contains(ID)) {
      // MongoSerDe calls this method as the very first one during serialization to satisfiy the
      // deserialization requirement that the ID field is the first one being deserialized.
      serializeId(ID, id);
    }
    return (C) this;
  }

  @SuppressWarnings("unchecked")
  @Override
  public C dt(long dt) {
    if (properties.contains(DT)) {
      throw new IllegalStateException(String.format("Property '%s' already serialized.", DT));
    }
    serializeLong(DT, dt);
    return (C) this;
  }

  BsonWriter build() {
    checkPresent(ID, "id");
    checkPresent(DT, "dt");

    return bsonWriter;
  }

  void addProperty(String id) {
    if (!properties.add(id)) {
      throw new IllegalStateException(String.format("Property '%s' already serialized.", id));
    }
  }

  void checkPresent(String id, String name) {
    Preconditions.checkArgument(
        properties.contains(id),
        String.format(
            "Method %s of consumer %s has not been called", name, getClass().getSimpleName()));
  }

  void checkNotPresent(String id, String name) {
    Preconditions.checkArgument(
        !properties.contains(id),
        String.format(
            "Method %s of consumer %s must not be called", name, getClass().getSimpleName()));
  }

  void serializeId(String property, Id id) {
    addProperty(property);
    MongoSerDe.serializeId(bsonWriter, property, id);
  }

  void serializeLong(String property, long value) {
    addProperty(property);
    bsonWriter.writeInt64(property, value);
  }

  void serializeIds(String property, Stream ids) {
    addProperty(property);
    bsonWriter.writeStartArray(property);

    ids.forEach(id -> bsonWriter.writeBinaryData(new BsonBinary(id.toBytes())));

    bsonWriter.writeEndArray();
  }

  void serializeString(String property, String str) {
    addProperty(property);
    bsonWriter.writeString(property, str);
  }

   void serializeArray(String prop, Stream src, BiConsumer inner) {
    addProperty(prop);
    MongoSerDe.serializeArray(bsonWriter, prop, src, inner);
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy