import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
protected int readImpl(TxnHandle txn, String table, String key, Set<String> fields,
        HashMap<String, ByteIterator> result) throws WrongTypeException, NoSuchObjectException,
        VersionNotFoundException, NetworkException {
    @SuppressWarnings("unchecked")
    final LWWStringMapRegisterCRDT register = txn.get(new CRDTIdentifier(table, key), false,
            LWWStringMapRegisterCRDT.class, notificationsSubscriber);
    final HashMap<String, String> value = register.getValue();
    if (value == null) {
        return ERROR_NOT_FOUND;
    }
    if (fields == null) {
        StringByteIterator.putAllAsByteIterators(result, value);
    } else {
        for (final String field : fields) {
            result.put(field, new StringByteIterator(value.get(field)));
        }
    }
    return 0;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
protected int updateImpl(TxnHandle txn, String table, String key, HashMap<String, ByteIterator> values)
        throws WrongTypeException, NoSuchObjectException, VersionNotFoundException, NetworkException {
    @SuppressWarnings("unchecked")
    final LWWStringMapRegisterCRDT register = txn.get(new CRDTIdentifier(table, key), false,
            LWWStringMapRegisterCRDT.class, notificationsSubscriber);
    final HashMap<String, String> value = register.getValue();
    if (value == null) {
        return ERROR_NOT_FOUND;
    }
    final HashMap<String, String> mutableValue = new HashMap<String, String>(value);
    StringByteIterator.putAllAsStrings(mutableValue, values);
    register.set(mutableValue);

    return 0;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
protected int readImpl(TxnHandle txn, String table, String key, Set<String> fields,
        HashMap<String, ByteIterator> result) throws WrongTypeException, NoSuchObjectException,
        VersionNotFoundException, NetworkException {
    @SuppressWarnings("unchecked")
    final PutOnlyLWWStringMapCRDT map = txn.get(new CRDTIdentifier(table, key), false,
            PutOnlyLWWStringMapCRDT.class, notificationsSubscriber);
    if (fields == null) {
        final HashMap<String, String> value = map.getValue();
        StringByteIterator.putAllAsByteIterators(result, value);
    } else {
        for (final String field : fields) {
            final String keyValue = map.get(field);
            if (keyValue == null) {
                return ERROR_NOT_FOUND;
            }
            result.put(field, new StringByteIterator(keyValue));
        }
    }
    return 0;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
protected int readImpl(TxnHandle txn, String table, String key, Set<String> fields,
        HashMap<String, ByteIterator> result) throws WrongTypeException, NoSuchObjectException,
        VersionNotFoundException, NetworkException {
    final HashMap<CRDTIdentifier, String> ids = getFieldIds(txn, table, key, fields, false);

    // Just trigger parallel reads.
    txn.bulkGet(notificationsSubscriber != null, ids.keySet(), null);
    // Then do separate reads, to trigger notifications too.
    for (final Entry<CRDTIdentifier, String> entry : ids.entrySet()) {
        @SuppressWarnings("unchecked")
        final LWWStringRegisterCRDT register = txn.get(entry.getKey(), false, LWWStringRegisterCRDT.class,
                notificationsSubscriber);
        final String value = register.getValue();
        if (value == null) {
            return ERROR_NOT_FOUND;
        }
        result.put(entry.getValue(), new StringByteIterator(register.getValue()));
    }
    return 0;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
@SuppressWarnings({ "rawtypes" })
public Status insert(final String table, final String key, final HashMap<String, ByteIterator> values) {
	HashMap fields;
	if (byteFormat) {
		fields = convertContent(values);
	} else {
		fields = StringByteIterator.getStringMap(values);
	}
	try {
		if (xRepo.getDocumentManagement().storeDocument(key, fields, insertProps) == null) {
			logger.debug("insert; document was not created for some reason; key: {}", key);
			return Status.UNEXPECTED_STATE;
		}
		return Status.OK;
	} catch (Exception ex) {
		logger.error("insert.error", ex);
		return Status.ERROR;
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
@SuppressWarnings({ "rawtypes" })
public Status update(final String table, final String key, final HashMap<String, ByteIterator> values) {
	HashMap fields;
	if (byteFormat) {
		fields = convertContent(values);
	} else {
		fields = StringByteIterator.getStringMap(values);
	}
	try {
		if (xRepo.getDocumentManagement().storeDocument(key, fields, updateProps) == null) {
			logger.debug("update; document was not updated for some reason; key: {}", key);
			return Status.UNEXPECTED_STATE;
		}
		return Status.OK;
	} catch (Exception ex) {
		logger.error("update.error", ex);
		return Status.ERROR;
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
@SuppressWarnings({ "unchecked", "rawtypes" })
public Status insert(final String table, final String key, final HashMap<String, ByteIterator> values) {
	HashMap<String, Object> params = new HashMap<>();
	params.put("uri", URI.create(key));
	HashMap content = StringByteIterator.getStringMap(values);
	content.put("key", key);
	params.put("content", content);
	params.put("props", insertProps);
	try (ResultCursor cursor = xRepo.getQueryManagement().executeQuery(qStore, params, queryProps)) {
		if (cursor.isEmpty()) {
			logger.debug("insert; document was not created for some reason; key: {}", key);
			return Status.UNEXPECTED_STATE;
		} else {
			return Status.OK;
		}
	} catch (Exception ex) {
		logger.error("insert.error; key: {}", key, ex);
		return Status.ERROR;
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
@SuppressWarnings({ "unchecked", "rawtypes" })
public Status update(final String table, final String key, final HashMap<String, ByteIterator> values) {
	HashMap<String, Object> params = new HashMap<>();
	params.put("uri", URI.create(key));
	HashMap content = StringByteIterator.getStringMap(values);
	content.put("key", key);
	params.put("content", content);
	params.put("props", updateProps);
	try (ResultCursor cursor = xRepo.getQueryManagement().executeQuery(qStore, params, queryProps)) {
		if (cursor.isEmpty()) {
			logger.debug("update; document was not updated for some reason; key: {}", key);
			return Status.UNEXPECTED_STATE;
		} else {
			return Status.OK;
		}
	} catch (Exception ex) {
		logger.error("update.error; key: {}", key, ex);
		return Status.ERROR;
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
protected void populateStringResult(final Map<String, Object> document, final Set<String> fields, 
		final HashMap<String, ByteIterator> result) {
	// fill results
	if (fields == null) {
		for (Map.Entry<String, Object> entry: document.entrySet()) {
			result.put(entry.getKey(), new StringByteIterator(entry.getValue().toString()));
		}
	} else {
		for (String field: fields) {
			Object value = document.get(field);
			if (value != null) {
				result.put(field, new StringByteIterator(value.toString()));
			}
		}
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
public int read(String table, String key, Set<String> fields, HashMap<String, ByteIterator> result) {
    try {
       Map<String, String> row;
       if (clustered) {
          row = AtomicMapLookup.getAtomicMap(infinispanManager.getCache(table), key, false);
       } else {
          Cache<String, Map<String, String>> cache = infinispanManager.getCache(table);
          row = cache.get(key);
       }
       if (row != null) {
          result.clear();
          if (fields == null || fields.isEmpty()) {
StringByteIterator.putAllAsByteIterators(result, row);
          } else {
      for (String field : fields) result.put(field, new StringByteIterator(row.get(field)));
          }
       }
       return OK;
    } catch (Exception e) {
       return ERROR;
    }
 }
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
public int scan(String table, String startkey, int recordcount,
        Set<String> fields, Vector<HashMap<String, ByteIterator>> result) {
    try {
        //XXX what to pass in for nulls / zeros?
        RecordListResponse res = c.scan(table, ScanOrder.Ascending, bufStr(startkey), true, null, false, recordcount, 0);
        int ret = ycsbThriftRet(res.responseCode, ResponseCode.Success, ResponseCode.ScanEnded);
        if(ret == 0) {
            for(Record r : res.records) {
                HashMap<String, ByteIterator> tuple = new HashMap<String, ByteIterator>();
                // Note: r.getKey() and r.getValue() call special helper methods that trim the buffer
                // to an appropriate length, and memcpy it to a byte[].  Trying to manipulate the ByteBuffer
                // directly leads to trouble.
                tuple.put("key", new StringByteIterator(new String(r.getKey())));
                decode(fields, new String(r.getValue())/*strBuf(r.bufferForValue())*/, tuple);
                result.add(tuple);
            }
        }
        return ret;
    } catch(TException e) {
        e.printStackTrace();
        return 2;
    }
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
public int read(String table, String key, Set<String> fields,
		HashMap<String, ByteIterator> result) {
	if ( checkStore(table) == ERROR ) {
		return ERROR;
	}
	
	Versioned<HashMap<String, String>> versionedValue = storeClient.get(key);
	
	if ( versionedValue == null )
		return NOT_FOUND;
	
	if ( fields != null ) {
		for (String field : fields) {
			String val = versionedValue.getValue().get(field);
			if ( val != null )
			    result.put(field, new StringByteIterator(val));
		}
	} else {
		StringByteIterator.putAllAsByteIterators(result, versionedValue.getValue());
	}
	return OK;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
public Status read(String table, String key, Set<String> fields, HashMap<String, ByteIterator> result) {
	HashMap<String, String> r = dmap.get(key);
	if(r == null){
		return Status.NOT_FOUND;
	}else{
		if(fields != null){
			for(Entry<String, ByteIterator> e : StringByteIterator.getByteIteratorMap(r).entrySet()){
				if(fields.contains(e.getKey())){
					result.put(e.getKey(),e.getValue());
				}
			}
		}else{
			result.putAll(StringByteIterator.getByteIteratorMap(r));
		}
		return Status.OK;
	}
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
public Status scan(String table, String startkey, int recordcount,	Set<String> fields, Vector<HashMap<String, ByteIterator>> result) {
   	Command cmd = new Command(cmd_id.incrementAndGet(),CommandType.GETRANGE,startkey,("").getBytes(),recordcount);
	try {
		Response r = client.send(cmd);
		if(r != null){
			List<Command> lc = r.getResponse(10000);
			if(lc.isEmpty()){ return Status.ERROR; }
   			for(Command c : lc){
    			if(c.getType() == CommandType.RESPONSE){
    				HashMap<String, ByteIterator> tuple = new HashMap<String, ByteIterator>();
    				tuple.put("key", new StringByteIterator(c.getKey()));
    				if(c.getValue() != null && c.getValue().length > 0){
    					decode(fields, new String(c.getValue()), tuple);
    					result.add(tuple);
    				}
    			}			    				
   			}
			return Status.OK;
		}
	} catch (Exception e) {
		e.printStackTrace();
	}
	return Status.ERROR;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
/**
 * Insert a record in the database. Any field/value pairs in the specified values HashMap will be written into the record with the specified
 * record key.
 *
 * @param table The name of the table
 * @param key The record key of the record to insert.
 * @param values A HashMap of field/value pairs to insert in the record
 * @return Zero on success, a non-zero error code on error. See this class's description for a discussion of error codes.
 */
public int insert(String table, String key, HashMap<String, ByteIterator> values) {
  try {
    final ODocument document = new ODocument(CLASS);
    for (Entry<String, String> entry : StringByteIterator.getStringMap(values).entrySet())
      document.field(entry.getKey(), entry.getValue());
    document.save();
    dictionary.put(key, document);

    return 0;
  } catch (Exception e) {
    e.printStackTrace();
  }
  return 1;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
/**
 * Update a record in the database. Any field/value pairs in the specified values HashMap will be written into the record with the specified
 * record key, overwriting any existing values with the same field name.
 *
 * @param table The name of the table
 * @param key The record key of the record to write.
 * @param values A HashMap of field/value pairs to update in the record
 * @return Zero on success, a non-zero error code on error. See this class's description for a discussion of error codes.
 */
public int update(String table, String key, HashMap<String, ByteIterator> values) {
  try {
    final ODocument document = dictionary.get(key);
    if (document != null) {
      for (Entry<String, String> entry : StringByteIterator.getStringMap(values).entrySet())
        document.field(entry.getKey(), entry.getValue());
      document.save();
      return 0;
    }
  } catch (Exception e) {
    e.printStackTrace();
  }
  return 1;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
/**
 * Insert a record in the database. Any field/value pairs in the specified
 * values HashMap will be written into the record with the specified record
 * key.
 *
 * @param table The name of the table
 * @param key The record key of the record to insert.
 * @param values A HashMap of field/value pairs to insert in the record
 * @return Zero on success, a non-zero error code on error. See this class's
 * description for a discussion of error codes.
 */
@Override
public int insert(String table, String key, HashMap<String, ByteIterator> values) {
    try {
        final XContentBuilder doc = jsonBuilder().startObject();

        for (Entry<String, String> entry : StringByteIterator.getStringMap(values).entrySet()) {
            doc.field(entry.getKey(), entry.getValue());
        }

        doc.endObject();

        client.prepareIndex(indexKey, table, key)
                .setSource(doc)
                .execute()
                .actionGet();

        return 0;
    } catch (Exception e) {
        e.printStackTrace();
    }
    return 1;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
/**
 * Update a record in the database. Any field/value pairs in the specified
 * values HashMap will be written into the record with the specified record
 * key, overwriting any existing values with the same field name.
 *
 * @param table The name of the table
 * @param key The record key of the record to write.
 * @param values A HashMap of field/value pairs to update in the record
 * @return Zero on success, a non-zero error code on error. See this class's
 * description for a discussion of error codes.
 */
@Override
public int update(String table, String key, HashMap<String, ByteIterator> values) {
    try {
        final GetResponse response = client.prepareGet(indexKey, table, key)
                .execute()
                .actionGet();

        if (response.isExists()) {
            for (Entry<String, String> entry : StringByteIterator.getStringMap(values).entrySet()) {
                response.getSource().put(entry.getKey(), entry.getValue());
            }

            client.prepareIndex(indexKey, table, key)
                    .setSource(response.getSource())
                    .execute()
                    .actionGet();

            return 0;
        }

    } catch (Exception e) {
        e.printStackTrace();
    }
    return 1;
}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
/**
 * Test of update method, of class ElasticSearchClient.
 */
@Test
public void testUpdate() {
    System.out.println("update");
    int i;
    HashMap<String, ByteIterator> newValues = new HashMap<String, ByteIterator>(10);

    for (i = 1; i <= 10; i++) {
        newValues.put("field" + i, new StringByteIterator("newvalue" + i));
    }

    int expResult = 0;
    int result = instance.update(MOCK_TABLE, MOCK_KEY1, newValues);
    assertEquals(expResult, result);

    //validate that the values changed
    HashMap<String, ByteIterator> resultParam = new HashMap<String, ByteIterator>(10);
    instance.read(MOCK_TABLE, MOCK_KEY1, MOCK_DATA.keySet(), resultParam);

    for (i = 1; i <= 10; i++) {
        assertEquals("newvalue" + i, resultParam.get("field" + i).toString());
    }

}
 

import com.yahoo.ycsb.StringByteIterator; //导入依赖的package包/类
@Override
protected int insertImpl(TxnHandle txn, String table, String key, HashMap<String, ByteIterator> values)
        throws WrongTypeException, NoSuchObjectException, VersionNotFoundException, NetworkException {
    @SuppressWarnings("unchecked")
    final LWWStringMapRegisterCRDT register = txn.get(new CRDTIdentifier(table, key), true,
            LWWStringMapRegisterCRDT.class, notificationsSubscriber);
    final HashMap<String, String> value = StringByteIterator.getStringMap(values);
    register.set(value);

    return 0;
}