/*
 * 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.iotdb.db.queryengine.plan.analyze.cache.schema.dualkeycache;

import javax.annotation.concurrent.GuardedBy;

import java.util.Map;
import java.util.function.BiFunction;
import java.util.function.Predicate;
import java.util.function.ToIntFunction;

/**
 * This interfaces defines the behaviour of a dual key cache. A dual key cache supports manage cache
 * values via two keys, first key and second key. Simply, the structure is like fk -> sk-> value.
 *
 * @param <FK> The first key of cache value
 * @param <SK> The second key of cache value
 * @param <V> The cache value
 */
public interface IDualKeyCache<FK, SK, V> {

  /** Get the cache value with given first key and second key. */
  V get(final FK firstKey, final SK secondKey);

  <R> boolean batchApply(
      final Map<FK, Map<SK, R>> inputMap, final BiFunction<V, R, Boolean> mappingFunction);

  /**
   * Update the existing value. The updater shall return the difference caused by the update,
   * because we do not want to call "valueSizeComputer" twice, which may include abundant useless
   * calculations.
   *
   * <p>Warning: This method is without any locks for performance concerns. The caller shall ensure
   * the concurrency safety for the value update.
   *
   * @param createIfNotExists put the value to cache iff it does not exist,
   */
  void update(
      final FK firstKey,
      final SK secondKey,
      final V value,
      final ToIntFunction<V> updater,
      final boolean createIfNotExists);

  /**
   * Update all the existing value with {@link SK} and a the {@link SK}s matching the given
   * predicate. The updater shall return the difference caused by the update, because we do not want
   * to call "valueSizeComputer" twice, which may include abundant useless calculations.
   *
   * <p>Warning: This method is without any locks for performance concerns. The caller shall ensure
   * the concurrency safety for the value update.
   */
  void update(
      final FK firstKey, final Predicate<SK> secondKeyChecker, final ToIntFunction<V> updater);

  /**
   * Update all the existing value with {@link SK} and a the {@link SK}s matching the given
   * predicate. The updater shall return the difference caused by the update, because we do not want
   * to call "valueSizeComputer" twice, which may include abundant useless calculations.
   *
   * <p>Warning: This method is without any locks for performance concerns. The caller shall ensure
   * the concurrency safety for the value update.
   */
  void update(
      final Predicate<FK> firstKeyChecker,
      final Predicate<SK> secondKeyChecker,
      final ToIntFunction<V> updater);

  /**
   * Invalidate all cache values in the cache and clear related cache keys. The cache status and
   * statistics won't be clear and they can still be accessed via cache.stats().
   */
  @GuardedBy("DataNodeSchemaCache#writeLock")
  void invalidateAll();

  /** Return all the current cache status and statistics. */
  IDualKeyCacheStats stats();

  /** remove all entries for firstKey */
  @GuardedBy("DataNodeSchemaCache#writeLock")
  void invalidate(final FK firstKey);

  /** remove matched entry */
  @GuardedBy("DataNodeSchemaCache#writeLock")
  void invalidate(final FK firstKey, final SK secondKey);

  @GuardedBy("DataNodeSchemaCache#writeLock")
  void invalidate(final FK firstKey, final Predicate<SK> secondKeyChecker);

  /** remove all entries matching the firstKey and the secondKey */
  @GuardedBy("DataNodeSchemaCache#writeLock")
  void invalidate(final Predicate<FK> firstKeyChecker, final Predicate<SK> secondKeyChecker);
}
