ActiveWebSocketUserRepositoryImpl.java

/*
 * Copyright 2014 the original author or authors.
 *
 * 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 com.bearchoke.platform.domain.user.repositories.impl;

import com.bearchoke.platform.domain.user.document.ActiveWebSocketUser;
import com.bearchoke.platform.domain.user.repositories.ActiveWebSocketUserRepositoryCustom;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.mapping.BasicMongoPersistentEntity;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.repository.query.MongoEntityInformation;
import org.springframework.data.mongodb.repository.support.MappingMongoEntityInformation;
import org.springframework.data.mongodb.repository.support.MongoRepositoryFactory;
import org.springframework.data.mongodb.repository.support.SimpleMongoRepository;
import org.springframework.data.util.ClassTypeInformation;
import org.springframework.stereotype.Repository;

import java.util.List;

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

/**
 * Created by Bjorn Harvold
 * Date: 1/9/14
 * Time: 8:05 PM
 * Responsibility:
 */
public class ActiveWebSocketUserRepositoryImpl implements ActiveWebSocketUserRepositoryCustom {

    private final MongoOperations operations;

    @Autowired
    public ActiveWebSocketUserRepositoryImpl(MongoOperations operations) {
        this.operations = operations;
    }


    public ActiveWebSocketUser findUserByUsername(String username) {
        ActiveWebSocketUser result = null;

        Query q = query(where("username").is(username));

        List<ActiveWebSocketUser> list = operations.find(q, ActiveWebSocketUser.class);

        if (list != null && !list.isEmpty()) {
            result = list.get(0);
        }

        return result;
    }

    @Override
    public ActiveWebSocketUser findUserBySessionId(String sessionId) {
        ActiveWebSocketUser result = null;

        Query q = query(where("sessionId").is(sessionId));

        List<ActiveWebSocketUser> list = operations.find(q, ActiveWebSocketUser.class);

        if (list != null && !list.isEmpty()) {
            result = list.get(0);
        }

        return result;
    }
}