org.projectnessie.nessie.combined.CombinedCreateReference Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nessie-combined-cs Show documentation

There is a newer version: 0.97.1

/*
 * Copyright (C) 2022 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.nessie.combined;

import org.projectnessie.api.v2.TreeApi;
import org.projectnessie.client.builder.BaseCreateReferenceBuilder;
import org.projectnessie.error.NessieConflictException;
import org.projectnessie.error.NessieNotFoundException;
import org.projectnessie.model.Branch;
import org.projectnessie.model.Detached;
import org.projectnessie.model.Reference;

public class CombinedCreateReference extends BaseCreateReferenceBuilder {

  private final TreeApi treeApi;

  protected CombinedCreateReference(TreeApi treeApi) {
    this.treeApi = treeApi;
  }

  @Override
  public Reference create() throws NessieNotFoundException, NessieConflictException {
    Reference source;
    if (sourceRefName == null) {
      if (reference.getHash() == null) {
        source = null;
      } else {
        source = Detached.of(reference.getHash());
      }
    } else {
      source = Branch.of(sourceRefName, reference.getHash());
    }

    try {
      return treeApi
          .createReference(reference.getName(), reference.getType().name(), source)
          .getReference();
    } catch (RuntimeException e) {
      throw CombinedClientImpl.maybeWrapException(e);
    }
  }
}