com.epam.ta.reportportal.database.dao.ProjectRepositoryCustomImpl.java Source code

Introduction

Here is the source code for com.epam.ta.reportportal.database.dao.ProjectRepositoryCustomImpl.java

Source

/*
 * Copyright 2016 EPAM Systems
 * 
 * 
 * This file is part of EPAM Report Portal.
 * https://github.com/reportportal/commons-dao
 * 
 * Report Portal is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Report Portal is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Report Portal.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.epam.ta.reportportal.database.dao;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.function.Predicate;
import java.util.stream.Collectors;

import com.epam.ta.reportportal.database.entity.ProjectRole;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;

import com.epam.ta.reportportal.database.entity.Project;
import com.epam.ta.reportportal.database.entity.Project.UserConfig;

/**
 * Project repository routines custom implementation
 *
 * @author Andrei_Ramanchuk
 */
class ProjectRepositoryCustomImpl implements ProjectRepositoryCustom {

    private static final String PROJECT_ID = "_id";
    private static final String PROJECT_TYPE = "configuration.entryType";

    @Autowired
    private MongoTemplate mongoTemplate;

    @Override
    public boolean isAssignedToProject(String project, String login) {
        return mongoTemplate.exists(
                Query.query(Criteria.where("name").is(project.toLowerCase())).addCriteria(userExists(login)),
                Project.class);
    }

    @Override
    public List<Project> findUserProjects(String login) {
        return mongoTemplate.find(Query.query(userExists(login)), Project.class);
    }

    @Override
    public List<String> findProjectUsers(String projectName, String value) {
        Query query = Query.query(Criteria.where("name").is(projectName));
        query.fields().include("users");
        Project p = mongoTemplate.findOne(query, Project.class);
        return p == null ? null
                : p.getUsers().keySet().stream().filter(userNameContains(value)).collect(Collectors.toList());
    }

    @Override
    public List<Project> findUserProjects(String login, String projectType) {
        Query query = Query.query(userExists(login)).addCriteria(Criteria.where(PROJECT_TYPE).is(projectType));
        return mongoTemplate.find(query, Project.class);
    }

    @Override
    public void removeUserFromProjects(String userId) {
        StringBuilder builder = new StringBuilder("users.");
        builder.append(userId);
        Query query = Query.query(Criteria.where(builder.toString()).exists(true));
        mongoTemplate.updateMulti(query, new Update().unset(builder.toString()), Project.class);
    }

    @Override
    public void addUsers(String projectId, Map<String, UserConfig> users) {
        Query query = Query.query(Criteria.where("_id").is(projectId));
        Update update = new Update();
        // TODO possible bug, only one update!
        for (Map.Entry<String, UserConfig> entry : users.entrySet()) {
            update.set("users." + entry.getKey(), entry.getValue());
        }
        mongoTemplate.updateFirst(query, update, Project.class);
    }

    @Override
    public List<String> findAllProjectNames() {
        Query query = new Query();
        query.fields().include("_id");
        return mongoTemplate.find(query, Project.class).stream().map(Project::getId).collect(Collectors.toList());
    }

    @Override
    public synchronized void clearExternalSystems(String projectId) {
        Query query = Query.query(Criteria.where(PROJECT_ID).is(projectId));
        Project first = mongoTemplate.findOne(query, Project.class);
        first.getConfiguration().setExternalSystem(Collections.emptyList());
        mongoTemplate.save(first);
    }

    @Override
    public Map<String, ProjectRole> findProjectRoles(String login) {
        final Query q = Query.query(userExists(login));
        q.fields().include("users");
        return mongoTemplate.find(q, Project.class).stream()
                .collect(Collectors.toMap(Project::getName, p -> p.getUsers().get(login).getProjectRole()));
    }

    private Criteria userExists(String login) {
        return Criteria.where("users." + login).exists(true);
    }

    /**
     * Users with usernames contains specified value
     *
     * @param value
     * @return
     */
    private Predicate<String> userNameContains(final String value) {
        return input -> input.contains(value);
    }
}