Method prepareRequests

test/core/TestMultiGetQuery.java:286–373

()

  }

  @Test
  public void prepareRequests() throws Exception {
    MultiGetQuery mgq = new MultiGetQuery(tsdb, query, METRIC_BYTES, q_tags, 
        start_ts, end_ts, tsdb.dataTable(), spans, null, 0, null, query_stats, 
        0, max_bytes, false, multiget_no_meta);
    List<Long> timestamps = mgq.prepareRowBaseTimes();
    ByteMap<ByteMap<List<GetRequest>>> row_map = mgq.prepareRequests(timestamps, q_tags);
    ByteSet tsuids = new ByteSet();
    for (ByteMap<List<GetRequest>> rows : row_map.values()) {
      tsuids.addAll(rows.keySet());
    }
    assertEquals(3, tsuids.size());
    
    List<GetRequest> rows = new ArrayList<GetRequest>();
    for (Entry<byte[], ByteMap<List<GetRequest>>> salt_entry : row_map.entrySet()) {
      System.out.println(salt_entry.getValue());
      rows.addAll(salt_entry.getValue().get(getTSUID(METRIC_STRING, TAGK_STRING, 
         "A", TAGK_B_STRING, "D")));
    }

    assertEquals(timestamps.size(), rows.size());
    for (int i = 0; i < timestamps.size(); i++) {
      byte[] key = getRowKey(METRIC_STRING, timestamps.get(i).intValue(), 
          TAGK_STRING, "A", TAGK_B_STRING, "D");
      assertArrayEquals(key, rows.get(i).key());
    }
    
    rows = new ArrayList<GetRequest>();
    for (Entry<byte[], ByteMap<List<GetRequest>>> salt_entry : row_map.entrySet()) {
      rows.addAll(salt_entry.getValue().get(getTSUID(METRIC_STRING, TAGK_STRING, 
         "B", TAGK_B_STRING, "D")));
    }
    assertEquals(timestamps.size(), rows.size());
    for (int i = 0; i < timestamps.size(); i++) {
      byte[] key = getRowKey(METRIC_STRING, timestamps.get(i).intValue(), 
          TAGK_STRING, "B", TAGK_B_STRING, "D");
      assertArrayEquals(key, rows.get(i).key());
    }
    
    rows = new ArrayList<GetRequest>();
    for (Entry<byte[], ByteMap<List<GetRequest>>> salt_entry : row_map.entrySet()) {
      rows.addAll(salt_entry.getValue().get(getTSUID(METRIC_STRING, TAGK_STRING, 
         "C", TAGK_B_STRING, "D")));
    }
    assertEquals(timestamps.size(), rows.size());
    for (int i = 0; i < timestamps.size(); i++) {
      byte[] key = getRowKey(METRIC_STRING, timestamps.get(i).intValue(), 
          TAGK_STRING, "C", TAGK_B_STRING, "D");
      assertArrayEquals(key, rows.get(i).key());
    }
    
    rows = new ArrayList<GetRequest>();
    for (Entry<byte[], ByteMap<List<GetRequest>>> salt_entry : row_map.entrySet()) {
      rows.addAll(salt_entry.getValue().get(getTSUID(METRIC_STRING, TAGK_STRING, 
          "A", TAGK_B_STRING, "D")));
    }
   
    assertEquals(timestamps.size(), rows.size());