Java MapRDB类代码示例

import com.mapr.db.MapRDB; //导入依赖的package包/类
@GET
@Produces(APPLICATION_JSON)
public Response getData() throws JsonProcessingException {

    List<Document> items = new ArrayList<>();
    ObjectMapper object = new ObjectMapper();
    table = dao.getStatsTable();

    QueryCondition c = MapRDB.newCondition().is("simulationId", QueryCondition.Op.EQUAL, dao.getLastSimulationID());
    DocumentStream rs = table.find(c);

    if (rs != null) {
        for(Document doc : rs) {
            items.add(doc);
            System.out.println(doc.asJsonString());
        }
        rs.close();
    }
    return Response.ok(object.writeValueAsString(items)).build();
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void addInitialStats() {
    long lastSimulationID = getLastSimulationID();
    for(int i=1; i<=Config.TOWER_COUNT; i++) {
        Document towerDoc = MapRDB.newDocument().
                set("_id", lastSimulationID+"/tower"+i).
                set("towerId", i).
                set("towerAllInfo", 0).
                set("towerFails", 0).
                set("towerDurations", 0).
                set("sessions", 0).
                set("time", 1).
                set("simulationId", lastSimulationID);
        statsTable.insert(towerDoc);
    }
    statsTable.flush();
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
public void mergeScanSpec(String functionName, JsonScanSpec scanSpec) {

    if (this.condition != null && scanSpec.getCondition() != null) {
      QueryCondition newCond = MapRDB.newCondition();
      switch (functionName) {
      case "booleanAnd":
        newCond.and();
        break;
      case "booleanOr":
        newCond.or();
        break;
        default:
          assert(false);
      }

      newCond.condition(this.condition)
             .condition(scanSpec.getCondition())
             .close()
             .build();

      this.condition = newCond;
    } else if (scanSpec.getCondition() != null){
      this.condition = scanSpec.getCondition();
    }
  }
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void init() {
  logger.debug("Getting tablet locations");
  try {
    Configuration conf = new Configuration();

    // Fetch table and tabletInfo only once and cache.
    table = MapRDB.getTable(scanSpec.getTableName());
    tabletInfos = table.getTabletInfos(scanSpec.getCondition());

    // Calculate totalRowCount for the table from tabletInfos estimatedRowCount.
    // This will avoid calling expensive MapRDBTableStats API to get total rowCount, avoiding
    // duplicate work and RPCs to MapR DB server.
    for (TabletInfo tabletInfo : tabletInfos) {
      totalRowCount += tabletInfo.getEstimatedNumRows();
    }

    computeRegionsToScan();

  } catch (Exception e) {
    throw new DrillRuntimeException("Error getting region info for table: " + scanSpec.getTableName(), e);
  }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void queryByAge() {

    QueryCondition condition = MapRDB.newCondition()
            .is("age", QueryCondition.Op.GREATER , 50)
            .build();

    int counter = 0;
    try(DocumentStream documentStream = table.find(condition)) {
      for (Document doc : documentStream ) {
        System.out.print(".");
        counter++;
      }
    }
    System.out.println("\n\nTotal documents : "+ counter + " for "+ condition);

  }
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void initTable(String tableName, boolean createTable) throws MapRJsonDocumentLoaderException {
  // if we've encountered this table name already, it's open.
  if(theTables.containsKey(tableName)) {
    theTables.get(tableName);
  }

  // open table (optionally create it)
  // and add to the group of open tables.
  try {
    theTables.put(tableName, MapRDB.getTable(tableName));
  } catch (DBException ex) {
    if (createTable) {
      try {
        theTables.put(tableName, MapRDB.createTable(tableName));
      } catch (DBException ex2) {
        throw new MapRJsonDocumentLoaderException(
            "Encountered error creating table " + tableName, ex2);
      }
    } else {
      throw new MapRJsonDocumentLoaderException(
          "MapR DB table " + tableName + "does not exist, and not configured to create", ex);
    }
  }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private int recordCount() {
  Table tab;
  try {
    tab = MapRDB.getTable(TABLE_NAME);
    DocumentStream stream = tab.find();
    Iterator<Document> iter = stream.iterator();
    int i = 0 ;
    while(iter.hasNext()) {
      iter.next();    // don't care about the returned Document.
      i++;
    }
    tab.close();
    return i;

  } catch (DBException ex) {
    throw ex;
  }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private Table getTable(String tableName) {
    Table table;
    System.out.println("Check DB");
    synchronized (lock) {
        if (!MapRDB.tableExists(tableName)) {
            table = MapRDB.createTable(tableName);
        } else {
            table = MapRDB.getTable(tableName);
        }
    }
    return table;
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
public void newSimulation(){
    DocumentMutation mutation = MapRDB.newMutation().
            setOrReplace("time", new Date().getTime()).
            increment("simulationId", 1);
    simulationTable.update(SIMULATION_KEY, mutation);
    addInitialStats();
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
public void addCDR(String cdrJson) {
    Document document = MapRDB.newDocument(cdrJson);
    document.set("simulationId", getLastSimulationID());
    cdrsTable.insert(String.format("%08d", document.getLong("simulationId"))+"/00000"+document.getDouble("time") +"/"+ document.getString("callerId")+"/"
            + document.getString("towerId") +"/"+document.getString("state"), document);
    System.out.println("inserted " + document.toString());
    cdrsTable.flush();
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void receive() {
    long lastId = DAO.getInstance().getLastSimulationID();
    QueryCondition c = MapRDB.newCondition().is("simulationId", QueryCondition.Op.EQUAL, lastId);
    for (Document document : table.find(c)) {
        store(document.toString());
    }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private static Table getTable(String tablePath) {

    if ( ! MapRDB.tableExists(tablePath)) {
      return MapRDB.createTable(tablePath);
    } else {
      return MapRDB.getTable(tablePath);
    }

  }
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
@Override
public void setup(OperatorContext context, OutputMutator output) throws ExecutionSetupException {
  this.vectorWriter = new VectorContainerWriter(output, unionEnabled);
  this.operatorContext = context;

  try {
    table = MapRDB.getTable(tableName);
    table.setOption(TableOption.EXCLUDEID, !includeId);
    documentStream = table.find(condition, projectedFields);
    documentReaderIterators = documentStream.documentReaders().iterator();
  } catch (DBException e) {
    throw new ExecutionSetupException(e);
  }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
public static void createJsonTables() throws IOException {
  admin = MapRDB.newAdmin();
  if (admin.tableExists(TMP_BUSINESS_TABLE)) {
    admin.deleteTable(TMP_BUSINESS_TABLE);
  }

  try (Table table = admin.createTable(TMP_BUSINESS_TABLE);
       InputStream in = getJsonStream("json/business.json");
       DocumentStream stream = Json.newDocumentStream(in)) {
    for (Document document : stream) {
      table.insert(document, "business_id");
    }
    table.flush();
  }
}
 

import com.mapr.db.MapRDB; //导入依赖的package包/类
private void createTable() throws IOException {
  // delete table
  if ( MapRDB.tableExists(TABLE_PATH)) {
    MapRDB.deleteTable(TABLE_PATH);
  }


  // Admin Tool
  Admin admin = MapRDB.newAdmin();

  // Create a table descriptor
  TableDescriptor tableDescriptor = MapRDB.newTableDescriptor()
          .setPath(TABLE_PATH)  // set the Path of the table in MapR-FS
          .setSplitSize(512)    // Size in mebibyte (Mega Binary Bytes)
          .setBulkLoad(false);   // Created with Bulk mode by default

  // Configuration of the default Column Family, used to store JSON element by default
  FamilyDescriptor familyDesc = MapRDB.newDefaultFamilyDescriptor()
          .setCompression(FamilyDescriptor.Compression.None)
          .setInMemory(true); // To tell the DB to keep these value in RAM as much as possible
  tableDescriptor.addFamily(familyDesc);


  // Create a new colmn family to store specific JSON attributes
  familyDesc = MapRDB.newFamilyDescriptor()
          .setName("clicks")
          .setJsonFieldPath("clicks")
          .setCompression(FamilyDescriptor.Compression.ZLIB)  // compression for this CF
          .setInMemory(false);

  tableDescriptor.addFamily(familyDesc);

  Table table = admin.createTable(tableDescriptor);
}