All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.redislabs.redisgraph.impl.ResultSetImpl Maven / Gradle / Ivy

    There is a newer version: 2.6.0-rc3
    Show newest version
    package com.redislabs.redisgraph.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.stream.Collectors;

import com.redislabs.redisgraph.Record;
import com.redislabs.redisgraph.ResultSet;
import com.redislabs.redisgraph.Statistics;

import redis.clients.jedis.util.SafeEncoder;

public class ResultSetImpl implements ResultSet{
  
  private final int totalResults;
  private final List header;
  private final List results;
  private final Statistics statistics;
  private int position = 0;

  public ResultSetImpl(List resp) {
    
    this.statistics = new StatisticsImpl((List)resp.get(1));
    
    ArrayList> result = (ArrayList>) resp.get(0);
    
    // Empty result set
    if(result == null || result.isEmpty()) {
      header = new ArrayList<>(0);
      totalResults = 0;
      results = new ArrayList<>(0);
    } else {
      ArrayList headers = (ArrayList)result.get(0);
      header = headers.stream().map( String::new).collect(Collectors.toList());

      // First row is a header row
      totalResults = result.size()-1;
      results = new ArrayList<>(totalResults);
      // Skips last row (runtime info)
      for (int i = 1; i <= totalResults; i++) {
        ArrayList row = result.get(i);
        Record record = new RecordImpl(header, row.stream().map( obj -> {
          if(obj instanceof byte[]) {
            return SafeEncoder.encode((byte[])obj);  
          } 
          return obj;
        }).collect(Collectors.toList()));
        results.add(record);
      }
    }
  }

  @Override
  public List getHeader(){
    return header;
  }

  @Override
  public boolean hasNext() {
    return position < results.size();
  }

  @Override
  public Record next() {
    if (!hasNext())
      throw new NoSuchElementException();
    return results.get(position++);
  }

  @Override
  public Statistics getStatistics() {
    return statistics;
  }

  @Override
  public String toString() {
    return this.header + "\n" + this.results + "\n" + this.statistics;
  }
}
    
    
    
    

    




    
    
    © 2015 - 2024 Weber Informatics LLC | Privacy Policy