package com.excelian.mache.couchbase;

import com.couchbase.client.java.Bucket;
import com.couchbase.client.java.Cluster;
import com.couchbase.client.java.CouchbaseCluster;
import com.couchbase.client.java.cluster.BucketSettings;
import com.couchbase.client.java.cluster.ClusterManager;
import com.couchbase.client.java.env.CouchbaseEnvironment;
import com.excelian.mache.core.AbstractCacheLoader;
import com.excelian.mache.core.SchemaOptions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.couchbase.core.CouchbaseTemplate;

import java.util.List;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * An implementation of the Mache CacheLoader for Couchbase Server. Utilises the newer Spring Data Couchbase
 * that
 *
 * @param <K> Cache key type.
 * @param <V> Cache value type.
 */
public class CouchbaseCacheLoader<K, V> extends AbstractCacheLoader<K, V, Cluster> {

    private static final Logger LOG = LoggerFactory.getLogger(CouchbaseCacheLoader.class);

    private Class<K> keyType;
    private Class<V> valueType;
    private final Cluster cluster;
    private ClusterManager manager;
    private CouchbaseTemplate template;
    private BucketSettings bucketSettings;

    private List<String> nodes;
    private String adminUser;
    private String adminPassword;
    private SchemaOptions schemaOptions;

    /**
     * @param keyType The class type of the cache key.
     * @param valueType The class type of the cache value.
     * @param bucketSettings Bucket that will hold cached objects.
     * @param cluster Cluster connection.
     * @param adminUser Administration user for Couchbase cluster.
     * @param adminPassword Password for Administration user for Couchbase cluster.
     * @param schemaOptions Determine whether to create/drop bucket.
     */
    public CouchbaseCacheLoader(Class<K> keyType, Class<V> valueType, BucketSettings bucketSettings,
                                Cluster cluster, String adminUser,
                                String adminPassword, SchemaOptions schemaOptions) {
        this.keyType = keyType;
        this.valueType = valueType;
        this.bucketSettings = bucketSettings;
        this.cluster = cluster;
        this.adminUser = adminUser;
        this.adminPassword = adminPassword;
        this.schemaOptions = schemaOptions;

        if(cluster==null)
        {
            throw new NullPointerException("Cluster cannot be null");
        }
    }

    @Override
    public void create() {
        synchronized (this) {
            if (manager == null) {
                LOG.info("Attempting to connect to authenticate to Couchbase cluster as {}", adminUser);
                manager = cluster.clusterManager(adminUser, adminPassword);

                dropBucketIfRequired();
                Bucket bucket = createBucketIfRequired();
                template = new CouchbaseTemplate(manager.info(), bucket);
                LOG.info("Using Couchbase bucket: {}", bucket);
            }
        }
    }

    @Override
    public V load(K key) throws Exception {
        checkNotNull(key);
        return template.findById(key.toString(), valueType);
    }

    @Override
    public void put(K key, V value) {
        template.save(value);
    }

    @Override
    public void remove(K key) {
        template.remove(key);
    }

    @Override
    public void close() {
        if (cluster != null) {
            synchronized (this) {
                if (cluster != null) {
                    dropBucketIfRequired();
                    LOG.info("Disconnecting from Couchbase cluster {}", nodes);
                    cluster.disconnect();
                    cluster = null;
                }
            }
        }
    }

    @Override
    public Cluster getDriverSession() {
        return cluster;
    }

    @Override
    public String getName() {
        return getClass().getSimpleName();
    }

    private void dropBucketIfRequired() {
        if (schemaOptions.shouldDropSchema() && manager.hasBucket(bucketSettings.name())) {
            LOG.info("Removing bucket {}", bucketSettings.name());
            manager.removeBucket(bucketSettings.name());
        }
    }

    private Bucket createBucketIfRequired() {
        if (schemaOptions.shouldCreateSchema() && !manager.hasBucket(bucketSettings.name())) {
            LOG.info("Creating bucket {}", bucketSettings.name());
            manager.insertBucket(bucketSettings);
        }
        return cluster.openBucket(bucketSettings.name());
    }

    @Override
    public String toString() {
        return "CouchbaseCacheLoader{"
                + "cluster=" + cluster
                + ", manager=" + manager
                + ", template=" + template
                + ", bucketSettings=" + bucketSettings
                + ", nodes=" + nodes
                + ", adminUser='" + adminUser + '\''
                + ", adminPassword='" + adminPassword + '\''
                + ", schemaOptions=" + schemaOptions
                + ", keyType=" + keyType
                + ", valueType=" + valueType
                + '}';
    }
}


@@ -3404,37 +3404,37 @@
 ) %7B%0A        if (
-clust
+manag
 er != null) %7B%0A  
@@ -3485,21 +3485,21 @@
     if (
-clust
+manag
 er != nu
@@ -3552,166 +3552,8 @@
 ();%0A
-                    LOG.info(%22Disconnecting from Couchbase cluster %7B%7D%22, nodes);%0A                    cluster.disconnect();%0A                    cluster = null;%0A
     

Update CouchbaseCacheLoader.java