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 DefaultCommand implements Command
031{
032    protected final Queryable queryable;
033    protected CassandraPreparedStatementAdapter<Number, Row> stmt;
034    protected HandleableException handlerException;
035    
036    public DefaultCommand(Queryable queryable)
037    {
038        super();
039        this.queryable = queryable;
040    }
041    @Override
042    public <T> Command with(T stmt)
043    {
044        this.stmt = (CassandraPreparedStatementAdapter<Number, Row>)stmt;
045        return this;
046    }
047    
048    @Override
049    public Command with(HandleableException handlerException)
050    {
051        this.handlerException = handlerException;
052        return this;
053    }
054    
055    @Override
056    public Command with(CommandHandler commandHandler)
057    {
058        return this;
059    }
060    
061    @Override
062    @SuppressWarnings("unchecked")
063    public <T> T execute()
064    {
065        queryable.bind(stmt).on();
066        Integer rows = stmt.execute();
067        return (T) rows;
068    }
069}