001/* 
002 * JKNIV, whinstone one contract to access your database.
003 * 
004 * Copyright (C) 2017, the original author or authors.
005 *
006 * This library is free software; you can redistribute it and/or
007 * modify it under the terms of the GNU Lesser General Public
008 * License as published by the Free Software Foundation; either
009 * version 2.1 of the License.
010 * 
011 * This library is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
014 * Lesser General Public License for more details.
015 * 
016 * You should have received a copy of the GNU Lesser General Public
017 * License along with this library; if not, write to the Free Software Foundation, Inc., 
018 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
019 */
020package net.sf.jkniv.whinstone.cassandra.commands;
021
022import com.datastax.driver.core.Row;
023
024import net.sf.jkniv.exception.HandleableException;
025import net.sf.jkniv.whinstone.Queryable;
026import net.sf.jkniv.whinstone.cassandra.statement.CassandraPreparedStatementAdapter;
027import net.sf.jkniv.whinstone.commands.Command;
028import net.sf.jkniv.whinstone.commands.CommandHandler;
029
030public class BulkCommand implements Command
031{
032    private final Queryable                            queryable;
033    private CassandraPreparedStatementAdapter<?, Row> stmt;
034    
035    public BulkCommand(Queryable queryable)
036    {
037        super();
038        this.queryable = queryable;
039    }
040    
041    @Override
042    public <T> Command with(T stmt)
043    {
044        this.stmt = (CassandraPreparedStatementAdapter) stmt;
045        return this;
046    }
047    
048    @Override
049    public Command with(HandleableException handleableException)
050    {
051        return this;
052    }
053    
054    @Override
055    public Command with(CommandHandler commandHandler)
056    {
057        return this;
058    }
059    
060    @Override
061    @SuppressWarnings("unchecked")
062    public <T> T execute()
063    {
064        Integer rows = queryable.bind(stmt).onBulk();
065        return (T) rows;
066    }
067}