com.github.oscerd.component.cassandra.ResultSetFormatStrategies Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of camel-cassandra Show documentation

Camel Cassandra Component

There is a newer version: 3.1.0

package com.github.oscerd.component.cassandra;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;

public class ResultSetFormatStrategies {

    private static final IResultSetFormatStrategy rowsList = new IResultSetFormatStrategy() {

		@Override
		public Object getResult(ResultSet resultSet) {
    		List rows = new ArrayList();
    		Iterator rowIter = resultSet.iterator();
    		while (rowIter.hasNext()) {
    			rows.add(rowIter.next());
    		}
    		return rows;
		}
    };
    
    private static final IResultSetFormatStrategy normalResultSet = new IResultSetFormatStrategy() {

		@Override
		public Object getResult(ResultSet resultSet) {
			return resultSet;
		}
    };
    
    public static IResultSetFormatStrategy rowsList() {
        return rowsList;
    }
    
    public static IResultSetFormatStrategy normalResultSet() {
        return normalResultSet;
    }
	
    public static IResultSetFormatStrategy fromName(String name) {
        if (name.equals("normalResultSet")) {
        	return ResultSetFormatStrategies.normalResultSet();
        }
        if (name.equals("rowsList")) {
            return ResultSetFormatStrategies.rowsList();
        }
		return null;
    }
}