コード例 #1
0
  @Test
  @SuppressWarnings("unchecked")
  public void canUseAnnotations() throws Exception {
    CorfuRuntime r = getDefaultRuntime();
    TestClassUsingAnnotation test =
        r.getObjectsView()
            .build()
            .setStreamName("test")
            .setType(TestClassUsingAnnotation.class)
            .open();

    assertThat(test.testFn1()).isTrue();

    assertThat(test.testIncrement()).isTrue();

    assertThat(test.getValue()).isNotZero();

    // clear the cache, forcing a new object to be built.
    r.getObjectsView().getObjectCache().clear();

    TestClassUsingAnnotation test2 =
        r.getObjectsView()
            .build()
            .setStreamName("test")
            .setType(TestClassUsingAnnotation.class)
            .open();

    assertThat(test).isNotSameAs(test2);

    assertThat(test2.getValue()).isNotZero();
  }
コード例 #2
0
  @Test
  @SuppressWarnings("unchecked")
  public void multipleWritesConsistencyTestConcurrent() throws Exception {
    getDefaultRuntime().connect();

    Map<String, String> testMap =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);
    testMap.clear();
    int num_threads = 5;
    int num_records = 1_000;

    scheduleConcurrently(
        num_threads,
        threadNumber -> {
          int base = threadNumber * num_records;
          for (int i = base; i < base + num_records; i++) {
            assertThat(testMap.put(Integer.toString(i), Integer.toString(i))).isEqualTo(null);
          }
        });
    executeScheduled(num_threads, 50, TimeUnit.SECONDS);

    Map<String, String> testMap2 =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);

    scheduleConcurrently(
        num_threads,
        threadNumber -> {
          int base = threadNumber * num_records;
          for (int i = base; i < base + num_records; i++) {
            assertThat(testMap2.get(Integer.toString(i))).isEqualTo(Integer.toString(i));
          }
        });
    executeScheduled(num_threads, 50, TimeUnit.SECONDS);
  }
コード例 #3
0
 @Test
 @SuppressWarnings("unchecked")
 public void canWrapObjectWithPrimitiveTypes() throws Exception {
   // begin tests
   CorfuRuntime r = getDefaultRuntime().connect();
   TestClassWithPrimitives test = r.getObjectsView().open("test", TestClassWithPrimitives.class);
   test.setPrimitive("hello world".getBytes());
   assertThat(test.getPrimitive()).isEqualTo("hello world".getBytes());
 }
コード例 #4
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
 @Test
 @SuppressWarnings("unchecked")
 public void canContainNullObjects() throws Exception {
   Map<String, String> testMap =
       getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("a"), SMRMap.class);
   testMap.clear();
   testMap.put("z", null);
   assertThat(testMap.get("z")).isEqualTo(null);
   Map<String, String> testMap2 =
       getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("a"), SMRMap.class);
   assertThat(testMap2.get("z")).isEqualTo(null);
 }
コード例 #5
0
 /**
  * Deserialize an object from a given byte buffer.
  *
  * @param b The bytebuf to deserialize.
  * @return The deserialized object.
  */
 @Override
 public Object deserialize(ByteBuf b, CorfuRuntime rt) {
   int classNameLength = b.readShort();
   byte[] classNameBytes = new byte[classNameLength];
   b.readBytes(classNameBytes, 0, classNameLength);
   String className = new String(classNameBytes);
   if (className.equals("null")) {
     return null;
   } else if (className.equals("CorfuObject")) {
     int SMRClassNameLength = b.readShort();
     byte[] SMRClassNameBytes = new byte[SMRClassNameLength];
     b.readBytes(SMRClassNameBytes, 0, SMRClassNameLength);
     String SMRClassName = new String(SMRClassNameBytes);
     try {
       return rt.getObjectsView()
           .build()
           .setStreamID(new UUID(b.readLong(), b.readLong()))
           .setType(Class.forName(SMRClassName))
           .open();
     } catch (ClassNotFoundException cnfe) {
       log.error("Exception during deserialization!", cnfe);
       throw new RuntimeException(cnfe);
     }
   } else {
     try (ByteBufInputStream bbis = new ByteBufInputStream(b)) {
       try (InputStreamReader r = new InputStreamReader(bbis)) {
         return gson.fromJson(r, Class.forName(className));
       }
     } catch (IOException | ClassNotFoundException ie) {
       log.error("Exception during deserialization!", ie);
       throw new RuntimeException(ie);
     }
   }
 }
コード例 #6
0
  @Test
  @SuppressWarnings("unchecked")
  public void canReadWriteToSingle() throws Exception {
    getDefaultRuntime();

    Map<String, String> testMap =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);
    testMap.clear();
    assertThat(testMap.put("a", "a")).isNull();
    assertThat(testMap.put("a", "b")).isEqualTo("a");
    assertThat(testMap.get("a")).isEqualTo("b");

    Map<String, String> testMap2 =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);
    assertThat(testMap2.get("a")).isEqualTo("b");
  }
コード例 #7
0
  @Test
  @SuppressWarnings("unchecked")
  public void canUsePrimitiveSerializer() throws Exception {
    // begin tests
    CorfuRuntime r = getDefaultRuntime().connect();
    TestClassWithPrimitives test =
        r.getObjectsView()
            .build()
            .setType(TestClassWithPrimitives.class)
            .setStreamName("test")
            .setSerializer(null)
            .open();

    test.setPrimitive("hello world".getBytes());
    assertThat(test.getPrimitive()).isEqualTo("hello world".getBytes());
  }
コード例 #8
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
 @Test
 @SuppressWarnings("unchecked")
 public void modificationDuringTransactionCausesAbort() throws Exception {
   Map<String, String> testMap =
       getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("A"), SMRMap.class);
   assertThat(testMap.put("a", "z"));
   getRuntime().getObjectsView().TXBegin();
   assertThat(testMap.put("a", "a")).isEqualTo("z");
   assertThat(testMap.put("a", "b")).isEqualTo("a");
   assertThat(testMap.get("a")).isEqualTo("b");
   CompletableFuture cf =
       CompletableFuture.runAsync(
           () -> {
             Map<String, String> testMap2 =
                 getRuntime()
                     .getObjectsView()
                     .open(
                         UUID.nameUUIDFromBytes("A".getBytes()),
                         SMRMap.class,
                         null,
                         EnumSet.of(ObjectOpenOptions.NO_CACHE),
                         SerializerType.JSON);
             testMap2.put("a", "f");
           });
   cf.join();
   assertThatThrownBy(() -> getRuntime().getObjectsView().TXEnd())
       .isInstanceOf(TransactionAbortedException.class);
 }
コード例 #9
0
  @Test
  public void postHandlersFire() throws Exception {
    CorfuRuntime r = getDefaultRuntime();

    Map<String, String> test =
        r.getObjectsView().build().setType(SMRMap.class).setStreamName("test").open();

    ICorfuSMRObject cObj = (ICorfuSMRObject) test;
    final AtomicInteger ai = new AtomicInteger(0);
    cObj.registerPostHandler(
        (String method, Object[] args, Object state) -> {
          ai.incrementAndGet();
        });
    test.put("a", "b");
    test.get("a");
    assertThat(ai.get()).isEqualTo(1);
  }
コード例 #10
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
  @Test
  @SuppressWarnings("unchecked")
  public void unusedMutatorAccessor() throws Exception {
    Map<String, String> testMap =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("A"), SMRMap.class);

    testMap.put("a", "z");
  }
コード例 #11
0
  @Test
  public void canOpenObjectWithTwoRuntimes() throws Exception {
    getDefaultRuntime();

    TestClass testClass =
        getRuntime().getObjectsView().build().setStreamName("test").setType(TestClass.class).open();

    testClass.set(52);
    assertThat(testClass.get()).isEqualTo(52);

    CorfuRuntime runtime2 = new CorfuRuntime(getDefaultEndpoint());
    runtime2.connect();

    TestClass testClass2 =
        runtime2.getObjectsView().build().setStreamName("test").setType(TestClass.class).open();

    assertThat(testClass2.get()).isEqualTo(52);
  }
コード例 #12
0
  @Test
  @SuppressWarnings("unchecked")
  public void multipleWritesConsistencyTest() throws Exception {
    getDefaultRuntime().connect();

    Map<String, String> testMap =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);
    testMap.clear();

    for (int i = 0; i < 10_000; i++) {
      assertThat(testMap.put(Integer.toString(i), Integer.toString(i))).isNull();
    }

    Map<String, String> testMap2 =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("test"), TreeMap.class);
    for (int i = 0; i < 10_000; i++) {
      assertThat(testMap2.get(Integer.toString(i))).isEqualTo(Integer.toString(i));
    }
  }
コード例 #13
0
  @Test
  @SuppressWarnings("unchecked")
  public void canUseCustomSerializer() throws Exception {
    // Register a custom serializer and use it with an SMR object
    Serializers.setCustomSerializer(SerializerType.JSON.getSerializer());
    CorfuRuntime r = getDefaultRuntime();

    Map<String, String> test =
        r.getObjectsView()
            .build()
            .setType(SMRMap.class)
            .setStreamName("test")
            .setSerializer(SerializerType.CUSTOM)
            .open();

    test.put("a", "b");
    test.get("a");
    assertThat(test.get("a")).isEqualTo("b");
  }
コード例 #14
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
  @Test
  @SuppressWarnings("unchecked")
  public void canContainOtherCorfuObjects() throws Exception {
    Map<String, String> testMap =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("a"), SMRMap.class);
    testMap.clear();
    testMap.put("z", "e");
    Map<String, Map<String, String>> testMap2 =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("b"), SMRMap.class);
    testMap2.put("a", testMap);

    assertThat(testMap2.get("a").get("z")).isEqualTo("e");

    testMap2.get("a").put("y", "f");

    assertThat(testMap.get("y")).isEqualTo("f");

    Map<String, String> testMap3 =
        getRuntime().getObjectsView().open(CorfuRuntime.getStreamID("a"), SMRMap.class);

    assertThat(testMap3.get("y")).isEqualTo("f");
  }
コード例 #15
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
  @Test
  @SuppressWarnings("unchecked")
  public void canUpdateSingleObjectTransacationallyWhenCached() throws Exception {
    r.setCacheDisabled(false);

    Map<String, String> testMap =
        getRuntime().getObjectsView().open(UUID.randomUUID(), SMRMap.class);
    getRuntime().getObjectsView().TXBegin();
    assertThat(testMap.put("a", "a")).isNull();
    assertThat(testMap.put("a", "b")).isEqualTo("a");
    assertThat(testMap.get("a")).isEqualTo("b");
    getRuntime().getObjectsView().TXEnd();
    assertThat(testMap.get("a")).isEqualTo("b");
  }
コード例 #16
0
ファイル: SMRMapTest.java プロジェクト: medhavidhawan/CorfuDB
  @Test
  @SuppressWarnings("unchecked")
  public void loadsFollowedByGetsConcurrent() throws Exception {
    r.setBackpointersDisabled(true);

    Map<String, String> testMap =
        getRuntime().getObjectsView().open(UUID.randomUUID(), SMRMap.class);

    final int num_threads = 5;
    final int num_records = 1_000;
    testMap.clear();

    scheduleConcurrently(
        num_threads,
        threadNumber -> {
          int base = threadNumber * num_records;
          for (int i = base; i < base + num_records; i++) {
            assertThat(testMap.put(Integer.toString(i), Integer.toString(i))).isEqualTo(null);
          }
        });

    long startTime = System.currentTimeMillis();
    executeScheduled(num_threads, 30, TimeUnit.SECONDS);
    calculateRequestsPerSecond("WPS", num_records * num_threads, startTime);

    scheduleConcurrently(
        num_threads,
        threadNumber -> {
          int base = threadNumber * num_records;
          for (int i = base; i < base + num_records; i++) {
            assertThat(testMap.get(Integer.toString(i))).isEqualTo(Integer.toString(i));
          }
        });

    startTime = System.currentTimeMillis();
    executeScheduled(num_threads, 30, TimeUnit.SECONDS);
    calculateRequestsPerSecond("RPS", num_records * num_threads, startTime);
  }