uk.co.caeldev.springsecuritymongo.repositories.MongoApprovalRepositoryImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-security-mongo Show documentation

A Commons Library to provide the repositories for mongo db

There is a newer version: 3.0.2

Show newest version

package uk.co.caeldev.springsecuritymongo.repositories;

import com.mongodb.WriteResult;
import org.joda.time.LocalDate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Component;
import uk.co.caeldev.springsecuritymongo.domain.MongoApproval;

import java.util.Collection;
import java.util.List;

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

@Component
public class MongoApprovalRepositoryImpl implements MongoApprovalRepositoryBase {

    private final MongoTemplate mongoTemplate;

    @Autowired
    public MongoApprovalRepositoryImpl(final MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }

    @Override
    public boolean updateOrCreate(final Collection mongoApprovals) {
        boolean result = true;
        for (MongoApproval mongoApproval : mongoApprovals) {
            final Update update = Update.update("expiresAt", mongoApproval.getExpiresAt())
                    .addToSet("status", mongoApproval.getStatus())
                    .addToSet("lastModifiedAt", mongoApproval.getLastUpdatedAt());

            final WriteResult writeResult = mongoTemplate.upsert(byUserIdAndClientIdAndScope(mongoApproval), update, MongoApproval.class);

            if (writeResult.getN() != 1) {
                result = false;
            }
        }
        return result;
    }

    @Override
    public boolean updateExpiresAt(final LocalDate expiresAt,
                                   final MongoApproval mongoApproval) {
        final Update update = Update.update("expiresAt", expiresAt);

        final WriteResult writeResult = mongoTemplate.updateFirst(byUserIdAndClientIdAndScope(mongoApproval),
                update,
                MongoApproval.class);

        return writeResult.getN() == 1;
    }

    @Override
    public boolean deleteByUserIdAndClientIdAndScope(final MongoApproval mongoApproval) {
        final WriteResult writeResult = mongoTemplate.remove(byUserIdAndClientIdAndScope(mongoApproval),
                MongoApproval.class);

        return writeResult.getN() == 1;
    }

    @Override
    public List findByUserIdAndClientId(final String userId,
                                                       final String clientId) {
        final Query query = Query.query(where("userId").is(userId)
                .andOperator(where("clientId").is(clientId)));
        return mongoTemplate.find(query, MongoApproval.class);
    }

    private Query byUserIdAndClientIdAndScope(final MongoApproval mongoApproval) {
        return Query.query(where("userId").is(mongoApproval.getUserId())
                    .andOperator(where("clientId").is(mongoApproval.getClientId())
                            .andOperator(where("scope").is(mongoApproval.getScope()))));
    }
}