com.linksinnovation.elearning.repository.ViewAmountRepository.java Source code

Introduction

Here is the source code for com.linksinnovation.elearning.repository.ViewAmountRepository.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.linksinnovation.elearning.repository;

import com.linksinnovation.elearning.dto.BasicConditionDTO;
import com.linksinnovation.elearning.model.report.ViewAmount;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.Query;
import org.springframework.stereotype.Repository;

/**
 *
 * @author Jirawong Wongdokpuang <jirawong@linksinnovation.com>
 */
@Repository
public class ViewAmountRepository {
    @PersistenceContext
    private EntityManager em;

    private static final String queryString = "SELECT update_date,user,lecture FROM report_amount_view";

    public List<ViewAmount> findByCondition(BasicConditionDTO amountDTO) {
        if (null == amountDTO.getCategory() && null == amountDTO.getStart()) {
            String queryAppend = queryString + " ORDER BY update_date,user ASC";
            Query query = em.createNativeQuery(queryAppend);
            List<Object[]> resultList = query.getResultList();
            return mapObject(resultList);
        } else if (null != amountDTO.getCategory() && null == amountDTO.getStart()) {
            String queryAppend = queryString + " WHERE category_id=:cat";
            if (null != amountDTO.getSubCategory()) {
                queryAppend = queryAppend + " AND sub_category_id=:subcat";
            }
            queryAppend = queryAppend + " ORDER BY update_date,user ASC";
            Query query = em.createNativeQuery(queryAppend);
            query.setParameter("cat", amountDTO.getCategory());
            if (null != amountDTO.getSubCategory()) {
                query.setParameter("subcat", amountDTO.getSubCategory());
            }
            List<Object[]> resultList = query.getResultList();
            return mapObject(resultList);
        } else if (null != amountDTO.getCategory() && null != amountDTO.getStart() && null != amountDTO.getEnd()) {
            String queryAppend = queryString + " WHERE update_date BETWEEN :start AND :end AND category_id=:cat";
            if (null != amountDTO.getSubCategory()) {
                queryAppend = queryAppend + " AND sub_category_id=:subcat";
            }
            queryAppend = queryAppend + " ORDER BY update_date,user ASC";
            Query query = em.createNativeQuery(queryAppend);
            query.setParameter("start", amountDTO.getStart());
            query.setParameter("end", amountDTO.getEnd());
            query.setParameter("cat", amountDTO.getCategory());

            if (null != amountDTO.getSubCategory()) {
                query.setParameter("subcat", amountDTO.getSubCategory());
            }
            List<Object[]> resultList = query.getResultList();
            return mapObject(resultList);
        } else if (null == amountDTO.getCategory() && null != amountDTO.getStart() && null != amountDTO.getEnd()) {
            String queryAppend = queryString + " WHERE update_date BETWEEN :start AND :end";
            queryAppend = queryAppend + " ORDER BY update_date,user ASC";
            Query query = em.createNativeQuery(queryAppend);
            query.setParameter("start", amountDTO.getStart());
            query.setParameter("end", amountDTO.getEnd());
            List<Object[]> resultList = query.getResultList();
            return mapObject(resultList);
        } else {
            String queryAppend = queryString + " ORDER BY update_date,user ASC";
            Query query = em.createNativeQuery(queryAppend);
            List<Object[]> resultList = query.getResultList();
            return mapObject(resultList);
        }
    }

    private List<ViewAmount> mapObject(List<Object[]> resultList) {
        List<ViewAmount> viewAmounts = new ArrayList<>();
        resultList.stream().map((o) -> new ViewAmount((Date) o[0], o[1].toString(), o[2].toString()))
                .forEach((videoAmount) -> {
                    viewAmounts.add(videoAmount);
                });
        return viewAmounts;
    }
}