com.testquack.dal.impl.ProjectRepositoryCustomImpl Maven / Gradle / Ivy

Go to download
package com.testquack.dal.impl;

import com.testquack.dal.ProjectRepositoryCustom;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import com.testquack.beans.Project;

import java.util.List;

public class ProjectRepositoryCustomImpl extends CommonRepositoryImpl
        implements ProjectRepositoryCustom {

    @Autowired
    MongoOperations mongoOperations;

    @Override
    public List findByOrganizationId(String id) {
        Query query = new Query().with(new Sort(Sort.Direction.ASC, "id"));
        query.addCriteria(Criteria.where("organizationId").in(id));
        return mongoOperations.find(query, Project.class);
    }

    @Override
    public Class getEntityClass() {
        return Project.class;
    }

    @Override
    protected String getCollectionName(String projectId) {
        return "projects";
    }
}