full/src/main/java/de/ids_mannheim/korap/handlers/RowMapperFactory.java

package de.ids_mannheim.korap.handlers;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.util.Map;

import org.springframework.jdbc.core.RowMapper;

import de.ids_mannheim.korap.config.Attributes;
import de.ids_mannheim.korap.config.URIParam;
import de.ids_mannheim.korap.user.KorAPUser;
import de.ids_mannheim.korap.user.User;

/**
 * @author hanl
 * @date 14/01/2014
 */
public class RowMapperFactory {

    public static class UserMapMapper implements RowMapper<Map<?,?>> {

        @Override
        public Map<?, ?> mapRow (ResultSet rs, int rowNum) throws SQLException {
            User user = new UserMapper().mapRow(rs, rowNum);
            return user.toMap();
        }
    }

    public static class UserMapper implements RowMapper<User> {

        @Override
        public User mapRow (ResultSet rs, int rowNum) throws SQLException {
            User user;
            switch (rs.getInt("type")) {
                case 0:
                    user = getKorAP(rs);
                    break;
//                case 1:
//                    user = getShib(rs);
//                    break;
                default:
                    user = User.UserFactory.getDemoUser();
                    user.setId(rs.getInt("id"));
                    user.setAccountCreation(rs.getTimestamp(
                            Attributes.ACCOUNT_CREATION).getTime());
                    return user;
            }
            return user;
        }


        private KorAPUser getKorAP (ResultSet rs) throws SQLException {
            KorAPUser user = User.UserFactory.getUser(rs
                    .getString(Attributes.USERNAME));
            user.setPassword(rs.getString(Attributes.PASSWORD));
            user.setId(rs.getInt(Attributes.ID));
            user.setAccountLocked(rs.getBoolean(Attributes.ACCOUNTLOCK));
            user.setAccountCreation(rs.getLong(Attributes.ACCOUNT_CREATION));
            user.setAccountLink(rs.getString(Attributes.ACCOUNTLINK));
            long l = rs.getLong(Attributes.URI_EXPIRATION);

            URIParam param = new URIParam(
                    rs.getString(Attributes.URI_FRAGMENT), l == 0 ? -1
                            : new Timestamp(l).getTime());
            user.addField(param);
            return user;
        }


//        private ShibbolethUser getShib (ResultSet rs) throws SQLException {
//            ShibbolethUser user = User.UserFactory.getShibInstance(
//                    rs.getString(Attributes.USERNAME),
//                    rs.getString(Attributes.MAIL), rs.getString(Attributes.CN));
//            user.setId(rs.getInt(Attributes.ID));
//            return user;
//        }

    }

}