Search in sources :

Example 61 with Bucket

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

the class N1QLIntegrationTest method givenDeleteStatement_whenQuery_thenDelete.

@Test
public void givenDeleteStatement_whenQuery_thenDelete() {
    Bucket bucket = bucketFactory.getTravelSampleBucket();
    String query4 = "DELETE FROM `travel-sample` USE KEYS \"cust_50\"";
    N1qlQueryResult result4 = bucket.query(N1qlQuery.simple(query4));
}
Also used : Bucket(com.couchbase.client.java.Bucket) N1qlQueryResult(com.couchbase.client.java.query.N1qlQueryResult) Test(org.junit.Test)

Example 62 with Bucket

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

the class N1QLIntegrationTest method givenDocuments_whenBatchInsert_thenResult.

@Test
public void givenDocuments_whenBatchInsert_thenResult() {
    Bucket bucket = bucketFactory.getTravelSampleBucket();
    List<JsonDocument> documents = IntStream.rangeClosed(0, 10).mapToObj(i -> {
        JsonObject content = JsonObject.create().put("id", i).put("type", "airline").put("name", "Sample Airline " + i);
        return JsonDocument.create("cust_" + i, content);
    }).collect(Collectors.toList());
    List<JsonDocument> r5 = Observable.from(documents).flatMap(doc -> bucket.async().insert(doc)).toList().last().toBlocking().single();
    r5.forEach(System.out::println);
}
Also used : IntStream(java.util.stream.IntStream) Statement(com.couchbase.client.java.query.Statement) RunWith(org.junit.runner.RunWith) Autowired(org.springframework.beans.factory.annotation.Autowired) JsonObject(com.couchbase.client.java.document.json.JsonObject) Observable(rx.Observable) JsonDocument(com.couchbase.client.java.document.JsonDocument) SpringJUnit4ClassRunner(org.springframework.test.context.junit4.SpringJUnit4ClassRunner) Select.select(com.couchbase.client.java.query.Select.select) JsonNode(com.fasterxml.jackson.databind.JsonNode) N1qlQueryResult(com.couchbase.client.java.query.N1qlQueryResult) N1qlQueryRow(com.couchbase.client.java.query.N1qlQueryRow) Assert.assertNotNull(org.junit.Assert.assertNotNull) Test(org.junit.Test) UUID(java.util.UUID) Collectors(java.util.stream.Collectors) Bucket(com.couchbase.client.java.Bucket) List(java.util.List) Stream(java.util.stream.Stream) Cluster(com.couchbase.client.java.Cluster) ContextConfiguration(org.springframework.test.context.ContextConfiguration) Expression(com.couchbase.client.java.query.dsl.Expression) N1qlQuery(com.couchbase.client.java.query.N1qlQuery) CodeSnippets.extractJsonResult(com.baeldung.couchbase.n1ql.CodeSnippets.extractJsonResult) JsonArray(com.couchbase.client.java.document.json.JsonArray) Bucket(com.couchbase.client.java.Bucket) JsonObject(com.couchbase.client.java.document.json.JsonObject) JsonDocument(com.couchbase.client.java.document.JsonDocument) Test(org.junit.Test)

Example 63 with Bucket

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

the class PersonServiceIntegrationTest method setupBeforeClass.

@BeforeClass
public static void setupBeforeClass() {
    final Cluster cluster = CouchbaseCluster.create(MyCouchbaseConfig.NODE_LIST);
    final Bucket bucket = cluster.openBucket(MyCouchbaseConfig.BUCKET_NAME, MyCouchbaseConfig.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 64 with Bucket

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

the class StudentServiceIntegrationTest method setupBeforeClass.

@BeforeClass
public static void setupBeforeClass() {
    Cluster cluster = CouchbaseCluster.create(MyCouchbaseConfig.NODE_LIST);
    Bucket bucket = cluster.openBucket(MyCouchbaseConfig.BUCKET_NAME, MyCouchbaseConfig.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 65 with Bucket

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

the class TestCouchbaseBucketRegistry method testCloseBucket.

/**
 * This unit test simulates 10 tasks using the same bucket. Each task will call registry.getBucket once. Then
 * each task will also call registry.closeBucket once. After that, registry.closeBucket should return false if we
 * close the bucket one more time. And the bucket should have already been closed.
 */
@Test
public void testCloseBucket() {
    String bucketName = "bucket";
    List<String> clusterNodes = Arrays.asList("cluster");
    CouchbaseEnvironmentConfigs configs = new CouchbaseEnvironmentConfigs();
    CouchbaseCluster cluster = mock(CouchbaseCluster.class);
    Bucket bucket = mock(Bucket.class);
    when(bucket.close()).thenReturn(true).thenReturn(false);
    when(cluster.openBucket(bucketName)).thenReturn(bucket);
    when(cluster.disconnect()).thenReturn(true).thenReturn(false);
    mockStatic(CouchbaseCluster.class);
    when(CouchbaseCluster.create(any(CouchbaseEnvironment.class), eq(clusterNodes))).thenReturn(cluster);
    CouchbaseBucketRegistry registry = new CouchbaseBucketRegistry();
    int numOfThreads = 10;
    for (int i = 0; i < numOfThreads; i++) {
        registry.getBucket(bucketName, clusterNodes, configs);
    }
    for (int i = 0; i < numOfThreads; i++) {
        assertTrue(registry.closeBucket(bucketName, clusterNodes));
    }
    // Close one more time. Should return false.
    assertFalse(registry.closeBucket(bucketName, clusterNodes));
    // Bucket should has been closed
    assertFalse(bucket.close());
}
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