Search in sources :

Example 26 with Bucket

use of com.couchbase.client.java.Bucket in project tutorials by eugenp.

the class ClusterServiceImpl method openBucket.

@Override
public synchronized Bucket openBucket(String name, String password) {
    if (!buckets.containsKey(name)) {
        Bucket bucket = cluster.openBucket(name, password);
        buckets.put(name, bucket);
    }
    return buckets.get(name);
}
Also used : Bucket(com.couchbase.client.java.Bucket)

Example 27 with Bucket

use of com.couchbase.client.java.Bucket in project tutorials by eugenp.

the class PersonServiceImplIntegrationTest method setupBeforeClass.

@BeforeClass
public static void setupBeforeClass() {
    final Cluster cluster = CouchbaseCluster.create(MultiBucketCouchbaseConfig.NODE_LIST);
    final Bucket bucket = cluster.openBucket(MultiBucketCouchbaseConfig.DEFAULT_BUCKET_NAME, MultiBucketCouchbaseConfig.DEFAULT_BUCKET_PASSWORD);
    bucket.upsert(JsonDocument.create(johnSmithId, jsonJohnSmith));
    bucket.upsert(JsonDocument.create(foobarId, jsonFooBar));
    bucket.close();
    cluster.disconnect();
}
Also used : Bucket(com.couchbase.client.java.Bucket) CouchbaseCluster(com.couchbase.client.java.CouchbaseCluster) Cluster(com.couchbase.client.java.Cluster) BeforeClass(org.junit.BeforeClass)

Example 28 with Bucket

use of com.couchbase.client.java.Bucket in project tutorials by eugenp.

the class StudentServiceImplIntegrationTest method setupBeforeClass.

@BeforeClass
public static void setupBeforeClass() {
    Cluster cluster = CouchbaseCluster.create(MultiBucketCouchbaseConfig.NODE_LIST);
    Bucket bucket = cluster.openBucket(MultiBucketCouchbaseConfig.DEFAULT_BUCKET_NAME, MultiBucketCouchbaseConfig.DEFAULT_BUCKET_PASSWORD);
    bucket.upsert(JsonDocument.create(joeCollegeId, jsonJoeCollege));
    bucket.upsert(JsonDocument.create(judyJetsonId, jsonJudyJetson));
    bucket.close();
    cluster.disconnect();
}
Also used : Bucket(com.couchbase.client.java.Bucket) CouchbaseCluster(com.couchbase.client.java.CouchbaseCluster) Cluster(com.couchbase.client.java.Cluster) BeforeClass(org.junit.BeforeClass)

Example 29 with Bucket

use of com.couchbase.client.java.Bucket in project jnosql-diana-driver by eclipse.

the class DefaultCouchbaseDocumentCollectionManager method delete.

@Override
public void delete(DocumentDeleteQuery query) {
    QueryConverter.QueryConverterResult delete = QueryConverter.delete(query, database);
    if (nonNull(delete.getStatement())) {
        ParameterizedN1qlQuery n1qlQuery = N1qlQuery.parameterized(delete.getStatement(), delete.getParams());
        bucket.query(n1qlQuery);
    }
    if (!delete.getKeys().isEmpty()) {
        delete.getKeys().stream().map(s -> getPrefix(query.getDocumentCollection(), s)).forEach(bucket::remove);
    }
}
Also used : Document(org.jnosql.diana.api.document.Document) EntityConverter.getPrefix(org.jnosql.diana.couchbase.document.EntityConverter.getPrefix) Statement(com.couchbase.client.java.query.Statement) ParameterizedN1qlQuery(com.couchbase.client.java.query.ParameterizedN1qlQuery) JsonObject(com.couchbase.client.java.document.json.JsonObject) ArrayList(java.util.ArrayList) JsonDocument(com.couchbase.client.java.document.JsonDocument) Duration(java.time.Duration) Objects.requireNonNull(java.util.Objects.requireNonNull) StreamSupport(java.util.stream.StreamSupport) Collectors.toSet(java.util.stream.Collectors.toSet) N1qlQueryResult(com.couchbase.client.java.query.N1qlQueryResult) KEY_FIELD(org.jnosql.diana.couchbase.document.EntityConverter.KEY_FIELD) Set(java.util.Set) DocumentDeleteQuery(org.jnosql.diana.api.document.DocumentDeleteQuery) DocumentQuery(org.jnosql.diana.api.document.DocumentQuery) Bucket(com.couchbase.client.java.Bucket) SearchQueryRow(com.couchbase.client.java.search.result.SearchQueryRow) List(java.util.List) DocumentEntity(org.jnosql.diana.api.document.DocumentEntity) N1qlQuery(com.couchbase.client.java.query.N1qlQuery) SearchQuery(com.couchbase.client.java.search.SearchQuery) Objects.nonNull(java.util.Objects.nonNull) ID_FIELD(org.jnosql.diana.couchbase.document.EntityConverter.ID_FIELD) EntityConverter.convert(org.jnosql.diana.couchbase.document.EntityConverter.convert) SearchQueryResult(com.couchbase.client.java.search.result.SearchQueryResult) ParameterizedN1qlQuery(com.couchbase.client.java.query.ParameterizedN1qlQuery)

Example 30 with Bucket

use of com.couchbase.client.java.Bucket in project samza by apache.

the class TestCouchbaseBucketRegistry method testCloseTwoBucketsInSameCluster.

/**
 * This unit test simulates closing two buckets within one cluster. The cluster should only be disconnected when all
 * buckets has been closed.
 */
@Test
public void testCloseTwoBucketsInSameCluster() {
    String bucketName1 = "bucket1";
    String bucketName2 = "bucket2";
    List<String> clusterNodes = Arrays.asList("cluster");
    CouchbaseEnvironmentConfigs configs = new CouchbaseEnvironmentConfigs();
    CouchbaseCluster cluster = mock(CouchbaseCluster.class);
    Bucket bucket1 = mock(Bucket.class);
    Bucket bucket2 = mock(Bucket.class);
    when(bucket1.close()).thenReturn(true).thenReturn(false);
    when(bucket2.close()).thenReturn(true).thenReturn(false);
    when(cluster.openBucket(bucketName1)).thenReturn(bucket1);
    when(cluster.openBucket(bucketName2)).thenReturn(bucket2);
    when(cluster.disconnect()).thenReturn(true).thenReturn(false);
    mockStatic(CouchbaseCluster.class);
    when(CouchbaseCluster.create(any(CouchbaseEnvironment.class), eq(clusterNodes))).thenReturn(cluster);
    CouchbaseBucketRegistry registry = new CouchbaseBucketRegistry();
    registry.getBucket(bucketName1, clusterNodes, configs);
    registry.getBucket(bucketName2, clusterNodes, configs);
    assertTrue(registry.closeBucket(bucketName1, clusterNodes));
    assertTrue(registry.closeBucket(bucketName2, clusterNodes));
    // Cluster should have been disconnected. Should return false.
    assertFalse(cluster.disconnect());
    // Buckets should have been closed. Should return false.
    assertFalse(cluster.disconnect());
}
Also used : Bucket(com.couchbase.client.java.Bucket) CouchbaseCluster(com.couchbase.client.java.CouchbaseCluster) CouchbaseEnvironment(com.couchbase.client.java.env.CouchbaseEnvironment) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest) Test(org.junit.Test)

Aggregations

Bucket (com.couchbase.client.java.Bucket)72 Test (org.junit.Test)53 Matchers.anyString (org.mockito.Matchers.anyString)21 JsonObject (com.couchbase.client.java.document.json.JsonObject)18 PrepareForTest (org.powermock.core.classloader.annotations.PrepareForTest)16 N1qlQueryResult (com.couchbase.client.java.query.N1qlQueryResult)15 MockFlowFile (org.apache.nifi.util.MockFlowFile)15 AsyncBucket (com.couchbase.client.java.AsyncBucket)13 TimeUnit (java.util.concurrent.TimeUnit)12 CouchbaseCluster (com.couchbase.client.java.CouchbaseCluster)11 CouchbaseException (com.couchbase.client.core.CouchbaseException)9 Cluster (com.couchbase.client.java.Cluster)8 RawJsonDocument (com.couchbase.client.java.document.RawJsonDocument)8 JsonDocument (com.couchbase.client.java.document.JsonDocument)7 CouchbaseEnvironment (com.couchbase.client.java.env.CouchbaseEnvironment)7 StringSerde (org.apache.samza.serializers.StringSerde)7 HashMap (java.util.HashMap)6 JsonNode (com.fasterxml.jackson.databind.JsonNode)5 N1qlQuery (com.couchbase.client.java.query.N1qlQuery)4 N1qlQueryRow (com.couchbase.client.java.query.N1qlQueryRow)4