package in.partake.model.dao.postgres9.impl;
import in.partake.base.TimeUtil;
import in.partake.model.dao.DAOException;
import in.partake.model.dao.DataIterator;
import in.partake.model.dao.MapperDataIterator;
import in.partake.model.dao.PartakeConnection;
import in.partake.model.dao.access.IUserImageAccess;
import in.partake.model.dao.postgres9.Postgres9Connection;
import in.partake.model.dao.postgres9.Postgres9Dao;
import in.partake.model.dao.postgres9.Postgres9Entity;
import in.partake.model.dao.postgres9.Postgres9EntityDao;
import in.partake.model.dao.postgres9.Postgres9EntityDataMapper;
import in.partake.model.dao.postgres9.Postgres9IndexDao;
import in.partake.model.dao.postgres9.Postgres9StatementAndResultSet;
import in.partake.model.dto.UserImage;
import java.io.IOException;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import org.codehaus.jackson.JsonParseException;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.node.ObjectNode;
class EntityImageMapper extends Postgres9EntityDataMapper<UserImage> {
public UserImage map(Postgres9Entity entity) throws DAOException {
if (entity == null)
return null;
ObjectNode obj;
try {
obj = new ObjectMapper().readValue(new String(entity.getBody(), UTF8), ObjectNode.class);
} catch (JsonParseException e) {
throw new IllegalArgumentException(e);
} catch (JsonMappingException e) {
throw new IllegalArgumentException(e);
} catch (IOException e) {
throw new DAOException(e);
}
UserImage imageData = new UserImage(obj);
imageData.setData(entity.getOpt());
return imageData.freeze();
}
}
public class Postgres9UserImageDao extends Postgres9Dao implements IUserImageAccess {
static final String ENTITY_TABLE_NAME = "UserImageEntities";
static final String USER_INDEX_TABLE_NAME = "UserImageIndex";
static final int CURRENT_VERSION = 1;
private final Postgres9EntityDao entityDao;
private final Postgres9IndexDao indexDao;
private final EntityImageMapper mapper;
public Postgres9UserImageDao() {
this.entityDao = new Postgres9EntityDao(ENTITY_TABLE_NAME);
this.indexDao = new Postgres9IndexDao(USER_INDEX_TABLE_NAME);
this.mapper = new EntityImageMapper();
}
@Override
public void initialize(PartakeConnection con) throws DAOException {
Postgres9Connection pcon = (Postgres9Connection) con;
entityDao.initialize(pcon);
if (!existsTable(pcon, USER_INDEX_TABLE_NAME)) {
indexDao.createIndexTable(pcon, "CREATE TABLE " + USER_INDEX_TABLE_NAME + "(id TEXT PRIMARY KEY, userId TEXT NOT NULL, createdAt TIMESTAMP)");
indexDao.createIndex(pcon, "CREATE INDEX " + USER_INDEX_TABLE_NAME + "UserId" + " ON " + USER_INDEX_TABLE_NAME + "(userId)");
indexDao.createIndex(pcon, "CREATE INDEX " + USER_INDEX_TABLE_NAME + "CreatedAt" + " ON " + USER_INDEX_TABLE_NAME + "(createdAt)");
}
}
@Override
public void truncate(PartakeConnection con) throws DAOException {
entityDao.truncate((Postgres9Connection) con);
indexDao.truncate((Postgres9Connection) con);
}
@Override
public void put(PartakeConnection con, UserImage imageData) throws DAOException {
Postgres9Connection pcon = (Postgres9Connection) con;
Postgres9Entity entity = new Postgres9Entity(imageData.getId(), CURRENT_VERSION, imageData.toJSON().toString().getBytes(UTF8), imageData.getData(), TimeUtil.getCurrentDateTime());
if (entityDao.exists(pcon, imageData.getId()))
entityDao.update(pcon, entity);
else
entityDao.insert(pcon, entity);
indexDao.put(pcon, new String[] {"id", "userId", "createdAt"}, new Object[] { imageData.getId(), imageData.getUserId(), imageData.getCreatedAt() });
}
@Override
public UserImage find(PartakeConnection con, String id) throws DAOException {
return mapper.map(entityDao.find((Postgres9Connection) con, id));
}
@Override
public boolean exists(PartakeConnection con, String id) throws DAOException {
return entityDao.exists((Postgres9Connection) con, id);
}
@Override
public void remove(PartakeConnection con, String id) throws DAOException {
entityDao.remove((Postgres9Connection) con, id);
indexDao.remove((Postgres9Connection) con, "id", id);
}
@Override
public DataIterator<UserImage> getIterator(PartakeConnection con) throws DAOException {
return new MapperDataIterator<Postgres9Entity, UserImage>(mapper, entityDao.getIterator((Postgres9Connection) con));
}
@Override
public List<String> findIdsByUserId(PartakeConnection con, String userId, int offset, int limit) throws DAOException {
Postgres9StatementAndResultSet psars = indexDao.select((Postgres9Connection) con,
"SELECT id FROM " + USER_INDEX_TABLE_NAME + " WHERE userId = ? ORDER BY createdAt DESC OFFSET ? LIMIT ?",
new Object[] { userId, offset, limit });
ArrayList<String> result = new ArrayList<String>();
try {
ResultSet rs = psars.getResultSet();
while (rs.next())
result.add(rs.getString(1));
} catch (SQLException e) {
throw new DAOException(e);
} finally {
psars.close();
}
return result;
}
@Override
public int countByUserId(PartakeConnection con, String userId) throws DAOException {
return indexDao.count((Postgres9Connection) con, "userId", userId);
}
@Override
public String getFreshId(PartakeConnection con) throws DAOException {
return entityDao.getFreshId((Postgres9Connection) con);
}
@Override
public int count(PartakeConnection con) throws DAOException {
return entityDao.count((Postgres9Connection) con);
}
}