com.rambird.miles.repository.jdbc.JdbcMileRepositoryImpl.java Source code

Introduction

Here is the source code for com.rambird.miles.repository.jdbc.JdbcMileRepositoryImpl.java

Source

/*
 * Copyright 2002-2013 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.rambird.miles.repository.jdbc;

import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.sql.DataSource;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.namedparam.BeanPropertySqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.core.simple.ParameterizedBeanPropertyRowMapper;
import org.springframework.jdbc.core.simple.SimpleJdbcInsert;
import org.springframework.orm.ObjectRetrievalFailureException;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Repository;
import org.springframework.util.StringUtils;

import com.rambird.miles.model.Category;
import com.rambird.miles.model.MyMile;
import com.rambird.miles.model.Owner;
import com.rambird.miles.model.Pet;
import com.rambird.miles.model.SearchMiles;
import com.rambird.miles.model.Visit;
import com.rambird.miles.repository.MileRepository;
import com.rambird.miles.repository.OwnerRepository;

/**
 * A simple JDBC-based implementation of the {@link OwnerRepository} interface.
 *
 * @author Ken Krebs
 * @author Juergen Hoeller
 * @author Rob Harrop
 * @author Sam Brannen
 * @author Thomas Risberg
 * @author Mark Fisher
 */
@Repository
public class JdbcMileRepositoryImpl implements MileRepository {

    private final String SELECT_QUERY = "SELECT mileid, catg, milestone, comments, finish_by, priority, user_name, home FROM mymiles";

    private NamedParameterJdbcTemplate namedParameterJdbcTemplate;

    private SimpleJdbcInsert insertMile;

    @Autowired
    public JdbcMileRepositoryImpl(DataSource dataSource, NamedParameterJdbcTemplate namedParameterJdbcTemplate) {

        this.insertMile = new SimpleJdbcInsert(dataSource).withTableName("mymiles")
                .usingGeneratedKeyColumns("mileid");

        this.namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
    }

    /**
     * Loads the {@link Owner} with the supplied <code>id</code>; also loads the {@link Pet Pets} and {@link Visit Visits}
     * for the corresponding owner, if not already loaded.
     */
    @Override
    public MyMile findById(int id) throws DataAccessException {
        MyMile mile;
        try {
            Map<String, Object> params = new HashMap<String, Object>();
            params.put("id", id);
            mile = this.namedParameterJdbcTemplate.queryForObject(SELECT_QUERY + " WHERE mileid= :id", params,
                    ParameterizedBeanPropertyRowMapper.newInstance(MyMile.class));
        } catch (EmptyResultDataAccessException ex) {
            throw new ObjectRetrievalFailureException(MyMile.class, id);
        }
        return mile;
    }

    /**
     * Loads {@link Owner Owners} from the data store by last name, returning all owners whose last name <i>starts</i> with
     * the given name; also loads the {@link Pet Pets} and {@link Visit Visits} for the corresponding owners, if not
     * already loaded.
     */
    @Override
    public Collection<MyMile> findAll(SearchMiles searchMiles) throws DataAccessException {
        Authentication auth = SecurityContextHolder.getContext().getAuthentication();
        Map<String, Object> params = new HashMap<String, Object>();
        params.put("userName", auth.getName());
        String searchIndexFilter = StringUtils.isEmpty(searchMiles.getSearchText()) ? ""
                : " and search_index @@ to_tsquery('" + searchMiles.getSearchText() + "') ";

        List<MyMile> miles = this.namedParameterJdbcTemplate.query(
                SELECT_QUERY + " where user_name=:userName " + searchIndexFilter, params,
                ParameterizedBeanPropertyRowMapper.newInstance(MyMile.class));

        return miles;
    }

    @Override
    public void save(MyMile mile) throws DataAccessException {
        BeanPropertySqlParameterSource parameterSource = new BeanPropertySqlParameterSource(mile);
        Authentication auth = SecurityContextHolder.getContext().getAuthentication();
        mile.setUserName(auth.getName()); //get logged in username

        if (mile.isNew()) {
            Number newKey = this.insertMile.executeAndReturnKey(parameterSource);
            mile.setMileId(newKey.intValue());
        } else {
            this.namedParameterJdbcTemplate.update(
                    "UPDATE mymiles SET catg=:catg, milestone=:milestone, user_name=:userName WHERE mileid=:mileId",
                    parameterSource);
        }
    }
}