Best code snippets using org.apache.gora.store.DataStore.get(Showing top 15 results out of 315)

public static void testPutMixedMapTypes(DataStore<String, WebPage> store) {
 WebPage webpage = createWebPage();
 webpage.getByteData().put(new Utf8("byteData"), ByteBuffer.wrap(ByteUtils.toBytes("hello map")));
 webpage.getStringData().put(new Utf8("stringData"), "hello map");
 store.createSchema();
 store.put(webpage.getUrl().toString(), webpage);
 store.flush();
 assertNotNull(store.get(webpage.getUrl().toString()));
}

public static void testGetEmployee(DataStore<String, Employee> dataStore)
 throws Exception {
 dataStore.createSchema();
 Employee employee = DataStoreTestUtil.createEmployee();
 String ssn = employee.getSsn().toString();
 dataStore.put(ssn, employee);
 dataStore.flush();
 Employee after = dataStore.get(ssn, AvroUtils.getSchemaFieldNames(Employee.SCHEMA$));
 assertEqualEmployeeObjects(employee, after);
}

@Test
public void testPutGet() throws Exception {
 String key = "org.apache.gora:http:/";
 DataStore<String, WebPage> store = new MemStore<>();
 assumeTrue(store.get(key, new String[0]) == null);
 store.put(key, WebPage.newBuilder().build());
 assertNotNull(store.get(key, new String[0]));
 store.close();
}

private static void assertTokenCount(DataStore<String, TokenDatum> outStore,
  String token, int count) throws Exception {
 TokenDatum datum = outStore.get(token, null);
 assertNotNull("token:" + token + " cannot be found in datastore", datum);
 assertEquals("count for token:" + token + " is wrong", count, datum.getCount().intValue());
}

/** Fetches a single pageview object and prints it*/
private void get(long key) throws Exception {
 Pageview pageview = dataStore.get(key);
 printPageview(pageview);
}

@Override
public T load(K key) throws CacheLoaderException {
 T persistent = null;
 try {
  persistent = dataStore.get(key);
  LOG.info("Loaded data bean from persistent datastore on key {}.", key.toString());
 } catch (CacheLoaderException ex) {
  throw ex;
 }
 return persistent;
}

/**
 * Fetches a single pageview object with required fields and prints it
 */
private void get(long key, String[] fields) throws Exception {
 Pageview pageview = cacheStore.get(key, fields);
 printPageview(pageview);
}

public static void testGetEmployeeNonExisting(DataStore<String, Employee> dataStore)
 throws Exception {
 Employee employee = dataStore.get("_NON_EXISTING_SSN_FOR_EMPLOYEE_");
 assertNull(employee);
}

public static void testGetEmployeeRecursive(DataStore<String, Employee> dataStore)
 throws Exception {
 Employee employee = DataStoreTestUtil.createEmployee();
 Employee boss = DataStoreTestUtil.createBoss();
 employee.setBoss(boss);
 
 String ssn = employee.getSsn().toString();
 dataStore.put(ssn, employee);
 dataStore.flush();
 Employee after = dataStore.get(ssn, AvroUtils.getSchemaFieldNames(Employee.SCHEMA$));
 assertEqualEmployeeObjects(employee, after);
}

private static void testGetWebPage(DataStore<String, WebPage> store, String[] fields)
 throws Exception {
 createWebPageData(store);
 for(int i=0; i<URLS.length; i++) {
  WebPage page = store.get(URLS[i], fields);
  assertWebPage(page, i);
 }
}

@Override
protected boolean nextInner() throws IOException {
 if (!iterator.hasNext()) {
  return false;
 }
 key = iterator.next();
 LOG.info("Results set pointer is now moved to key {}.", key);
 persistent = dataStore.get(key);
 this.current++;
 return true;
}

public static void testGetEmployee3UnionField(DataStore<String, Employee> dataStore)
 throws Exception {
 Employee employee = DataStoreTestUtil.createEmployee();
 employee.setBoss(new Utf8("Real boss")) ;
 String ssn = employee.getSsn().toString();
 dataStore.put(ssn, employee);
 dataStore.flush();
 Employee after = dataStore.get(ssn, AvroUtils.getSchemaFieldNames(Employee.SCHEMA$));
 assertEqualEmployeeObjects(employee, after);
 assertEquals("Real boss", after.getBoss().toString()) ;
}

@Test
public void testGetMissingValue() {
 DataStore<String, WebPage> store = new MemStore<>();
 WebPage nullWebPage = store.get("missing", new String[0]);
 assertNull(nullWebPage);
 store.close();
}

/**
 * Fetches a single pageview object and prints it
 */
private void get(long key, boolean isCacheEnabled) throws Exception {
 if (!isCacheEnabled) {
  Pageview pageview = dataStore.get(key);
  printPageview(pageview);
 } else {
  Pageview pageview = cacheStore.get(key);
  printPageview(pageview);
 }
}

@Override
public Map<K, T> loadAll(Iterable<? extends K> keys) throws CacheLoaderException {
 Map<K, T> loaded = new HashMap<K, T>();
 for (K key : keys) {
  T persistent = dataStore.get(key);
  LOG.info("Loaded data bean from persistent datastore on key {}.", key.toString());
  if (persistent != null) {
   loaded.put(key, persistent);
  }
 }
 return loaded;
}