it.f2informatica.mongodb.repositories.impl.ConsultantRepositoryImpl.java Source code

Introduction

Here is the source code for it.f2informatica.mongodb.repositories.impl.ConsultantRepositoryImpl.java

Source

/*
 * =============================================================================
 *
 *   Copyright (c) 2014, Fernando Aspiazu
 *
 *   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 it.f2informatica.mongodb.repositories.impl;

import com.mongodb.WriteResult;
import it.f2informatica.mongodb.domain.Consultant;
import it.f2informatica.mongodb.domain.Education;
import it.f2informatica.mongodb.domain.Experience;
import it.f2informatica.mongodb.domain.Language;
import it.f2informatica.mongodb.repositories.custom.ConsultantRepositoryCustom;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationResults;
import org.springframework.data.mongodb.core.aggregation.Fields;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Component;

import java.util.List;

import static org.springframework.data.mongodb.core.aggregation.TypedAggregation.*;
import static org.springframework.data.mongodb.core.query.Criteria.where;

@Component
public class ConsultantRepositoryImpl implements ConsultantRepositoryCustom {
    private static final String ID = "id";
    private static final String EXPERIENCES = "experiences";
    private static final String LANGUAGES = "languages";
    private static final String SKILLS = "skills";
    private static final String EDUCATIONS = "educationList";

    @Autowired
    private MongoTemplate mongoTemplate;

    @Override
    public int updateConsultantsPersonalDetails(Update updateFields, String consultantId) {
        return updateConsultant(whereConsultantIdIs(consultantId), updateFields).getN();
    }

    @Override
    public Experience findExperience(String consultantId, String experienceId) {
        Aggregation aggregation = newAggregation(match(where("id").is(consultantId)), group("experiences"),
                unwind(previousOperation()),
                match(where(previousOperation() + "." + Fields.UNDERSCORE_ID).is(experienceId)));
        AggregationResults<Experience> aggregationResults = mongoTemplate.aggregate(aggregation, Consultant.class,
                Experience.class);
        return aggregationResults.getUniqueMappedResult();
    }

    @Override
    public Education findEducation(String consultantId, String educationId) {
        Aggregation aggregation = newAggregation(match(where("id").is(consultantId)), group("educationList"),
                unwind(previousOperation()),
                match(where(previousOperation() + "." + Fields.UNDERSCORE_ID).is(educationId)));
        AggregationResults<Education> aggregationResults = mongoTemplate.aggregate(aggregation, Consultant.class,
                Education.class);
        return aggregationResults.getUniqueMappedResult();
    }

    @Override
    public int addExperience(Experience experience, String consultantId) {
        Query query = whereConsultantIdIs(consultantId);
        Update update = new Update().addToSet(EXPERIENCES, experience);
        return updateConsultant(query, update).getN();
    }

    @Override
    public int updateExperience(Experience experience, String consultantId) {
        Query query = new Query(
                where(ID).is(consultantId).and(EXPERIENCES + "." + Fields.UNDERSCORE_ID).is(experience.getId()));
        Update update = new Update().set(EXPERIENCES + ".$", experience);
        return updateConsultant(query, update).getN();
    }

    @Override
    public int removeExperience(String consultantId, String experienceId) {
        Query query = new Query(
                where(ID).is(consultantId).and(EXPERIENCES + "." + Fields.UNDERSCORE_ID).is(experienceId));
        Update update = new Update().pull(EXPERIENCES, findExperience(consultantId, experienceId));
        return updateConsultant(query, update).getN();
    }

    @Override
    public int addLanguages(List<Language> languages, String consultantId) {
        Update update = new Update().set(LANGUAGES, languages);
        return updateConsultant(whereConsultantIdIs(consultantId), update).getN();
    }

    @Override
    public int addSkills(String[] skills, String consultantId) {
        Update update = new Update().set(SKILLS, skills);
        return updateConsultant(whereConsultantIdIs(consultantId), update).getN();
    }

    @Override
    public int addEducation(Education education, String consultantId) {
        Query query = whereConsultantIdIs(consultantId);
        Update update = new Update().addToSet(EDUCATIONS, education);
        return updateConsultant(query, update).getN();
    }

    @Override
    public int updateEducation(Education education, String consultantId) {
        Query query = new Query(
                where(ID).is(consultantId).and(EDUCATIONS + "." + Fields.UNDERSCORE_ID).is(education.getId()));
        Update update = new Update().set(EDUCATIONS + ".$", education);
        return updateConsultant(query, update).getN();
    }

    @Override
    public int removeEducation(String consultantId, String educationId) {
        Query query = new Query(
                where(ID).is(consultantId).and(EDUCATIONS + "." + Fields.UNDERSCORE_ID).is(educationId));
        Update update = new Update().pull(EDUCATIONS, findEducation(consultantId, educationId));
        return updateConsultant(query, update).getN();
    }

    private Query whereConsultantIdIs(String consultantId) {
        return new Query(where(ID).is(consultantId));
    }

    private WriteResult updateConsultant(Query query, Update update) {
        return mongoTemplate.updateFirst(query, update, Consultant.class);
    }

}