com.datastrato.gravitino.storage.FunctionUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core Show documentation

Gravitino is a high-performance, geo-distributed and federated metadata lake.

There is a newer version: 0.5.1

/*
 * Copyright 2023 Datastrato Pvt Ltd.
 * This software is licensed under the Apache License version 2.
 */

package com.datastrato.gravitino.storage;

import com.datastrato.gravitino.storage.kv.TransactionalKvBackend;
import com.datastrato.gravitino.utils.Executable;
import java.io.IOException;
import java.util.concurrent.locks.ReentrantReadWriteLock;

/** Tools for executing functions with locks and transaction. */
public class FunctionUtils {

  private FunctionUtils() {}

  @FunctionalInterface
  public interface IOExecutable {

    R execute() throws IOException;
  }

  public static  R executeWithReadLock(IOExecutable executable, ReentrantReadWriteLock lock)
      throws IOException {
    // It already held the write lock
    if (lock.isWriteLockedByCurrentThread()) {
      return executable.execute();
    }

    lock.readLock().lock();
    try {
      return executable.execute();
    } finally {
      lock.readLock().unlock();
    }
  }

  public static  R executeWithWriteLock(IOExecutable executable, ReentrantReadWriteLock lock)
      throws IOException {
    lock.writeLock().lock();
    try {
      return executable.execute();
    } finally {
      lock.writeLock().unlock();
    }
  }

  public static  R executeInTransaction(
      Executable executable, TransactionalKvBackend transactionalKvBackend)
      throws E, IOException {

    transactionalKvBackend.begin();
    try {
      R r = executable.execute();
      transactionalKvBackend.commit();
      return r;
    } finally {
      // Let GC do the roll-back work
      transactionalKvBackend.closeTransaction();
    }
  }
}