/*
 * Copyright Terracotta, Inc.
 *
 * 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.ehcache.integration;

import org.ehcache.CacheManager;
import org.ehcache.PersistentCacheManager;
import org.ehcache.config.builders.CacheManagerBuilder;
import org.ehcache.config.units.MemoryUnit;
import org.ehcache.impl.copy.SerializingCopier;
import org.ehcache.integration.domain.Person;
import org.ehcache.spi.persistence.StateRepository;
import org.ehcache.spi.serialization.Serializer;
import org.ehcache.spi.serialization.SerializerException;
import org.ehcache.spi.serialization.StatefulSerializer;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.junit.rules.TemporaryFolder;

import java.nio.ByteBuffer;

import static org.ehcache.config.builders.CacheConfigurationBuilder.newCacheConfigurationBuilder;
import static org.ehcache.config.builders.CacheManagerBuilder.newCacheManagerBuilder;
import static org.ehcache.config.builders.CacheManagerBuilder.persistence;
import static org.ehcache.config.builders.ResourcePoolsBuilder.newResourcePoolsBuilder;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;

public class SerializersTest {

  @Rule
  public TemporaryFolder temporaryFolder = new TemporaryFolder();
  @Rule
  public ExpectedException expectedException = ExpectedException.none();

  @Test
  public void testStatefulSerializer() throws Exception {
    StatefulSerializerImpl<Long> serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithByRefHeapCache(serializer);
    assertThat(serializer.initCount, is(0));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithByValueHeapCache(serializer);
    assertThat(serializer.initCount, is(1));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithOffheapCache(serializer);
    assertThat(serializer.initCount, is(1));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithHeapOffheapCache(serializer);
    assertThat(serializer.initCount, is(1));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithDiskCache(serializer);
    assertThat(serializer.initCount, is(1));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithHeapDiskCache(serializer);
    assertThat(serializer.initCount, is(1));

    serializer = new StatefulSerializerImpl<Long>();
    testSerializerWithThreeTierCache(serializer);
    assertThat(serializer.initCount, is(1));

  }

  private void testSerializerWithByRefHeapCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<CacheManager> cmBuilder =
      newCacheManagerBuilder()
        .withCache("heapByRefCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().heap(10))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithByValueHeapCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<CacheManager> cmBuilder =
      newCacheManagerBuilder()
        .withCache("heapByValueCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().heap(10))
            .withKeyCopier(SerializingCopier.<Long>asCopierClass())
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithOffheapCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<CacheManager> cmBuilder =
      newCacheManagerBuilder()
        .withCache("offheapCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().offheap(2, MemoryUnit.MB))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithHeapOffheapCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<CacheManager> cmBuilder =
      newCacheManagerBuilder()
        .withCache("heapOffheapCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().heap(10).offheap(2, MemoryUnit.MB))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithDiskCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<PersistentCacheManager> cmBuilder =
      newCacheManagerBuilder()
        .with(persistence(temporaryFolder.newFolder().getAbsolutePath()))
        .withCache("diskCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().disk(8, MemoryUnit.MB, true))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithHeapDiskCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<PersistentCacheManager> cmBuilder =
      newCacheManagerBuilder()
        .with(persistence(temporaryFolder.newFolder().getAbsolutePath()))
        .withCache("heapDiskCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().heap(10).disk(8, MemoryUnit.MB, true))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  private void testSerializerWithThreeTierCache(Serializer<Long> serializer) throws Exception {
    CacheManagerBuilder<PersistentCacheManager> cmBuilder =
      newCacheManagerBuilder()
        .with(persistence(temporaryFolder.newFolder().getAbsolutePath()))
        .withCache("heapOffheapDiskCache",
          newCacheConfigurationBuilder(Long.class, Person.class, newResourcePoolsBuilder().heap(10).offheap(2, MemoryUnit.MB).disk(8, MemoryUnit.MB, true))
            .withKeySerializer(serializer)
        );
    cmBuilder.build(true);
  }

  public static class StatefulSerializerImpl<T> implements StatefulSerializer<T> {

    private int initCount = 0;

    @Override
    public void init(final StateRepository stateRepository) {
      initCount++;
    }

    @Override
    public ByteBuffer serialize(final T object) throws SerializerException {
      return null;
    }

    @Override
    public T read(final ByteBuffer binary) throws ClassNotFoundException, SerializerException {
      return null;
    }

    @Override
    public boolean equals(final T object, final ByteBuffer binary) throws ClassNotFoundException, SerializerException {
      return false;
    }
  }

}