org.jnosql.diana.couchbase.key.CouchbaseBucketManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of couchbase-driver Show documentation
The Eclipse JNoSQL communication layer, Diana, to Couchbase
The newest version!
/*
 *  Copyright (c) 2017 Otávio Santana and others
 *   All rights reserved. This program and the accompanying materials
 *   are made available under the terms of the Eclipse Public License v1.0
 *   and Apache License v2.0 which accompanies this distribution.
 *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
 *
 *   You may elect to redistribute this code under either of these licenses.
 *
 *   Contributors:
 *
 *   Otavio Santana
 */
package org.jnosql.diana.couchbase.key;


import com.couchbase.client.java.Bucket;
import com.couchbase.client.java.document.JsonDocument;
import com.couchbase.client.java.document.RawJsonDocument;
import com.couchbase.client.java.document.json.JsonObject;
import com.couchbase.client.java.document.json.JsonValue;
import com.couchbase.client.java.error.DocumentDoesNotExistException;
import org.jnosql.diana.api.Value;
import org.jnosql.diana.api.key.BucketManager;
import org.jnosql.diana.api.key.KeyValueEntity;
import org.jnosql.diana.driver.JsonbSupplier;
import org.jnosql.diana.driver.ValueJSON;

import javax.json.bind.Jsonb;
import java.time.Duration;
import java.util.Objects;
import java.util.Optional;
import java.util.logging.Logger;

import static java.util.Objects.requireNonNull;
import static java.util.stream.Collectors.toList;
import static java.util.stream.StreamSupport.stream;
import static org.jnosql.diana.driver.ValueUtil.convert;

/**
 * The couchbase implementation to {@link BucketManager}
 */
public class CouchbaseBucketManager implements BucketManager {

    private static final Logger LOGGER = Logger.getLogger(CouchbaseBucketManager.class.getName());

    private static final Jsonb JSONB = JsonbSupplier.getInstance().get();

    private final Bucket bucket;

    private final String bucketName;

    CouchbaseBucketManager(Bucket bucket, String bucketName) {
        this.bucket = bucket;
        this.bucketName = bucketName;
    }

    @Override
    public  void put(K key, V value) {
        requireNonNull(key, "key is required");
        requireNonNull(value, "value is required");

        if (JsonValue.checkType(value)) {
            bucket.upsert(RawJsonDocument.create(key.toString(), JSONB.toJson(value.toString())));
        } else {
            bucket.upsert(JsonDocument.create(key.toString(), JsonObjectCouchbaseUtil.toJson(JSONB, value)));
        }

    }

    @Override
    public  void put(KeyValueEntity entity) throws NullPointerException {
        requireNonNull(entity, "entity is required");
        put(entity.getKey(), convert(entity.getValue()));
    }

    @Override
    public  void put(KeyValueEntity entity, Duration ttl) {
        requireNonNull(entity, "entity is required");
        requireNonNull(ttl, "ttl is required");


        if (JsonValue.checkType(entity.get())) {
            RawJsonDocument jsonDocument = RawJsonDocument.create(entity.getKey().toString(), (int) ttl.getSeconds(),
                    JSONB.toJson(entity.get().toString()));

            bucket.upsert(jsonDocument);
        } else {
            JsonObject jsonObject = JsonObjectCouchbaseUtil.toJson(JSONB, entity.get());
            JsonDocument jsonDocument = JsonDocument.create(entity.getKey().toString(), (int) ttl.getSeconds(), jsonObject);
            bucket.upsert(jsonDocument);
        }

    }

    @Override
    public  void put(Iterable> keyValueEntities) {
        requireNonNull(keyValueEntities, "keyValueEntities is required");
        keyValueEntities.forEach(this::put);
    }

    @Override
    public  void put(Iterable> keyValueEntities, Duration ttl) {
        requireNonNull(keyValueEntities, "keyValueEntities is required");
        requireNonNull(ttl, "ttl is required");
        keyValueEntities.forEach(k -> this.put(k, ttl));
    }

    @Override
    public  Optional get(K key) throws NullPointerException {
        requireNonNull(key, "key is required");
        RawJsonDocument jsonDocument = bucket.get(key.toString(), RawJsonDocument.class);
        if (Objects.isNull(jsonDocument)) {
            return Optional.empty();
        }
        Object value = jsonDocument.content();
        return Optional.of(ValueJSON.of(value.toString()));
    }

    @Override
    public  Iterable get(Iterable keys) {
        requireNonNull(keys, "keys is required");
        return stream(keys.spliterator(), false)
                .map(this::get)
                .filter(Optional::isPresent)
                .map(Optional::get)
                .collect(toList());
    }

    @Override
    public  void remove(K key) {
        requireNonNull(key, "key is required");
        try {
            bucket.remove(key.toString());
        } catch (DocumentDoesNotExistException e) {
            LOGGER.info("Not found any document with the key " + key);
        }
    }

    @Override
    public  void remove(Iterable keys) {
        requireNonNull(keys, "keys is required");
        keys.forEach(this::remove);
    }

    @Override
    public void close() {
        bucket.close();
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder("CouchbaseBucketManager{");
        sb.append("bucket=").append(bucket);
        sb.append(", bucketName='").append(bucketName).append('\'');
        sb.append('}');
        return sb.toString();
    }
}