Example #1
0
  public ModelAndView query(HttpServletRequest request, HttpServletResponse response) {
    ModelAndView mv = new ModelAndView();
    int curPage = ServletRequestUtils.getIntParameter(request, "curPage", 0);

    int pageSize = ServletRequestUtils.getIntParameter(request, "pageSize", 100);

    if (pageSize > 10000) {
      throw new RuntimeException("max pageSize can not be more than 10000");
    }

    PageBean pageBean = null;
    if (curPage != 0) {
      pageBean = new PageBean(curPage, pageSize);
    }

    String errorMsg = "";

    final SearchParam searchParam = new SearchParam();
    try {
      request.setCharacterEncoding("utf-8");
      String tableName = request.getParameter("table");

      String select = request.getParameter("select");
      String where = request.getParameter("where");
      String keys = request.getParameter("keys");
      String startKey = request.getParameter("startKey");
      String endKey = request.getParameter("endKey");
      String family = request.getParameter("family");
      String hbaseFilter = request.getParameter("hbaseFilter");

      if (startKey == null) startKey = "";

      if (endKey == null) endKey = "";

      String tenantId = request.getParameter("tenantId");

      if (startKey != null && startKey.contains("$")) {
        if (tenantId != null) {
          String[] tenantIdArray = tenantId.split(",");
          startKey = startKey.replaceFirst(REGEX, tenantIdArray[0]);
        }
      }

      if (endKey != null && endKey.contains("$")) {
        if (tenantId != null) {
          String[] tenantIdArray = tenantId.split(",");
          endKey = endKey.replaceFirst(REGEX, tenantIdArray[0]) + "_z";
        }
      }

      searchParam.setTableName(tableName);
      searchParam.setFamily(family);
      searchParam.setStartKey(startKey);
      searchParam.setEndKey(endKey);
      searchParam.setKeys(keys);
      searchParam.setWhere(where);
      searchParam.setHbaseFilter(hbaseFilter);
      searchParam.setSelect(select);

    } catch (Exception e) {
      logger.error(e.getMessage(), e);
      errorMsg += e.getMessage();
    }

    try {
      pageBean = getDataMapTemp(searchParam, curPage, pageSize);
    } catch (IOException e) {
      errorMsg += e.getMessage();
    }

    @SuppressWarnings("unchecked")
    List<Map<String, String>> rows = pageBean.getResults();

    List<String> titleList = new ArrayList<String>();
    if (null != rows && rows.size() > 0) {
      for (Iterator<String> iter = rows.get(0).keySet().iterator(); iter.hasNext(); ) {
        String next = iter.next();
        if ("_id".equals(next)) {
          titleList.add(0, next);
        } else {
          titleList.add(next);
        }
      }
    }

    DataSource dataSource = new DataSource();
    dataSource.setRows(pageBean.getResults());
    dataSource.setPage(pageBean.getCurrentPage());
    dataSource.setTotal(pageBean.getTotalRecords());
    dataSource.setRecords(pageBean.getLength());
    dataSource.setStatus("success");

    mv.addObject("status", errorMsg);
    mv.addObject("rows", rows);
    mv.addObject("keyList", titleList);
    mv.addObject("result", dataSource);

    return mv;
  }
Example #2
0
  private PageBean getDataMapTemp(SearchParam searchParam, Integer currentPage, Integer pageSize)
      throws IOException {
    String tableName = searchParam.getTableName();
    String startRow = searchParam.getStartKey();
    String stopRow = searchParam.getEndKey();

    List<Map<String, String>> mapList = null;
    mapList = new LinkedList<Map<String, String>>();
    ResultScanner scanner = null;
    // 为分页创建的封装类对象,下面有给出具体属性 TBData tbData = null;
    PageBean tbData = new PageBean();
    Scan scan = null;
    HbaseWhereEngine whereEngine = null;
    HTableInterface table = HbaseTool.getTable(tableName); // 获取筛选对象
    if ("ca_summary_optimize".equals(tableName) && compoundFieldMap.isEmpty()) {
      compoundFieldMap = CompoundFieldConfigService.getParentComField(tableName);
    }
    try {
      if (StringUtils.isNotEmpty(searchParam.getWhere())) {
        whereEngine =
            new HbaseWhereEngine(tableName, searchParam.getFamily(), searchParam.getWhere());
      }

      String[] selectArray = null;
      if (StringUtils.isNotEmpty(searchParam.getSelect())) {
        selectArray = searchParam.getSelect().split(",");
      }
      byte[] cf = Bytes.toBytes(searchParam.getFamily());

      // if keys is not null,then we can sure which records to be selected!
      if (StringUtils.isNotEmpty(searchParam.getKeys())) {
        List<Get> getKeysList = new ArrayList<Get>();

        for (String key : searchParam.getKeys().split(",")) {
          Get get = new Get(Bytes.toBytes(key));
          getKeysList.add(get);

          if (selectArray != null) {
            for (String field : selectArray) {
              String temp[] = processField(field, searchParam.getFamily());
              if ("ca_summary_optimize".equals(tableName)
                  && compoundFieldMap.containsKey(temp[1])) {
                get.addColumn(Bytes.toBytes(temp[0]), Bytes.toBytes(compoundFieldMap.get(temp[1])));
              } else {
                get.addColumn(Bytes.toBytes(temp[0]), Bytes.toBytes(temp[1]));
              }
            }
          }

          if (selectArray != null && whereEngine != null) {
            Set<String> varSet = whereEngine.getVarSet();
            for (String var : varSet) {
              get.addColumn(cf, Bytes.toBytes(var));
            }
          }
        }

        Result[] resultsFromKeys = table.get(getKeysList);

        for (Result rr : resultsFromKeys) {
          if (!rr.isEmpty()) {
            if (whereEngine != null && !whereEngine.meetCondition(rr)) {
              continue;
            }
            Map<String, String> map = new TreeMap<String, String>();

            map.put("_id", Bytes.toString(rr.getRow()));
            for (String field : selectArray) {
              String value = HbaseWhereEngine.getColumnValue(tableName, cf, rr, field);
              if (!field.equals("id")) {
                map.put(field, value);
              }
            }
            mapList.add(map);
          }
        }
        pageSize = mapList.size();
        tbData.setCurrentPage(currentPage);
        tbData.setLength(pageSize);
        tbData.setTotalRecords(mapList.size());
      } else { // if keys is null,we select some records between startKey and end key or top
               // pageSize !
        // 获取最大返回结果数量
        if (pageSize == null || pageSize == 0L) pageSize = 100;
        if (currentPage == null || currentPage == 0) currentPage = 1;
        // 计算起始页和结束页

        Integer firstPage = (currentPage - 1) * pageSize;

        Integer endPage = firstPage + pageSize;

        scan = getScan(startRow, stopRow);
        // 给筛选对象放入过滤器(true标识分页,具体方法在下面)
        scan.setFilter(packageFilters(searchParam, true));

        // 缓存1000条数据
        scan.setCaching(1000);
        scan.setCacheBlocks(false);

        if (selectArray != null) {
          for (String field : selectArray) {
            String temp[] = processField(field, searchParam.getFamily());
            if ("ca_summary_optimize".equals(tableName) && compoundFieldMap.containsKey(temp[1])) {
              scan.addColumn(Bytes.toBytes(temp[0]), Bytes.toBytes(compoundFieldMap.get(temp[1])));
            } else {
              scan.addColumn(Bytes.toBytes(temp[0]), Bytes.toBytes(temp[1]));
            }
          }
        }

        if (selectArray != null && whereEngine != null) {
          Set<String> varSet = whereEngine.getVarSet();
          for (String var : varSet) {
            scan.addColumn(cf, Bytes.toBytes(var));
          }
        }

        scanner = table.getScanner(scan);
        int i = 0;
        List<byte[]> rowList = new LinkedList<byte[]>();
        // 遍历扫描器对象, 并将需要查询出来的数据row key取出
        for (Result result : scanner) {
          String row = toStr(result.getRow());
          if (i >= firstPage && i < endPage) { // filter firstPage
            rowList.add(getBytes(row));
          }
          if (i >= endPage) {
            break;
          }
          i++;
        }
        // 获取取出的row key的GET对象
        List<Get> getList = getList(rowList, cf, selectArray, tableName);
        Result[] results = table.get(getList);
        for (Result result : results) {
          if (whereEngine != null && !whereEngine.meetCondition(result)) {
            continue;
          }
          Map<byte[], byte[]> fmap = packFamilyMap(tableName, result, cf, selectArray);
          Map<String, String> rmap = packRowMap(fmap);
          rmap.put("_id", toStr(result.getRow()));
          mapList.add(rmap);
        }

        // 封装分页对象

        tbData.setCurrentPage(currentPage);
        tbData.setLength(pageSize); // (pageSize);
        tbData.setTotalRecords(i);
      }
      tbData.setResults(mapList);
    } catch (Exception e) {
    } finally {
      if (table != null) {
        table.close();
      }
      closeScanner(scanner);
    }
    compoundFieldMap.clear();
    return tbData;
  }