TradeableInstrumentDao.java

package com.activequant.dao.mybatis;

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

import org.apache.ibatis.session.SqlSessionFactory;

import com.activequant.dao.mybatis.mapper.GenericRowMapper;
import com.activequant.domainmodel.Instrument;
import com.activequant.domainmodel.TradeableInstrument;
import com.activequant.interfaces.dao.ITradeableInstrumentDao;

public class TradeableInstrumentDao extends GenericMapperDao<TradeableInstrument> implements ITradeableInstrumentDao {

    // private Logger log = Logger.getLogger(InstrumentDao.class);
    private static final String tableName = "TradeableInstrument";

    public TradeableInstrumentDao(GenericRowMapper mapper, SqlSessionFactory s) {
        super(s, mapper, TradeableInstrument.class, tableName);
    }

    public TradeableInstrument[] findFor(Instrument instrument) {
        // dirty. would have to use a field mapper table to ensure that
        // INSTRUMENTID is always the same as in MarketDataInstrument.
        // trade-off.

        List<String> ids = mapper.findByString(tableName, "INSTRUMENTID", instrument.getId());
        List<TradeableInstrument> mdis = new ArrayList<TradeableInstrument>();
        for (String id : ids) {
            mdis.add(this.load(id));
        }
        return mdis.toArray(new TradeableInstrument[] {});
    }
    
    
    public List<String> findIdsFor(String instrumentId){
    	List<String> ids = mapper.findByString(tableName, "INSTRUMENTID", instrumentId);
    	return ids; 
    }
    
    
    public TradeableInstrument findFor(String providerId, Instrument instrument) {
        // dirty. would have to use a field mapper table to ensure that
        // INSTRUMENTID is always the same as in MarketDataInstrument.
        // trade-off.

        List<String> ids = mapper.findByString(tableName, "INSTRUMENTID", instrument.getId());
        List<TradeableInstrument> mdis = new ArrayList<TradeableInstrument>();
        for (String id : ids) {
            TradeableInstrument tdi = this.load(id);
            if(tdi.getTradingProvider().equals(providerId))
            	return tdi;
        }
        return null;
    }
    

    

    public TradeableInstrument findByProvId(String providerId, String provSpecInstId) {
        List<String> insts = mapper.findBy2StringVals(tableName, "TradingProvider".toUpperCase(), providerId,
                "ProviderSpecificId".toUpperCase(), provSpecInstId);
        if (insts.size() > 1) {
            throw new RuntimeException("Ambigous tradeable! "  + providerId + "/"+provSpecInstId);
        }
        if (insts.size() == 0)
            return null;
        String id = insts.get(0);
        TradeableInstrument tdi = load(id);
        return tdi;
    }
    
}