Java tutorial
/* * Copyright 2012 Nodeable Inc * * 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.streamreduce.core.dao; import com.google.code.morphia.Datastore; import com.google.code.morphia.query.Query; import com.streamreduce.core.model.APIAuthenticationToken; import com.streamreduce.core.model.Account; import com.streamreduce.core.model.User; import java.util.HashSet; import java.util.List; import java.util.Set; import java.util.regex.Pattern; import org.bson.types.ObjectId; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; import org.springframework.stereotype.Repository; import org.springframework.util.Assert; @Repository("userDAO") public class UserDAO extends SobaObjectDAO<User> { @Autowired protected UserDAO(@Qualifier(value = "businessDBDatastore") Datastore datastore) { super(datastore); } public User findUser(String username) { Assert.hasText(username); return ds.createQuery(entityClazz).field("username") .equal(Pattern.compile("^\\Q" + username + "\\E$", Pattern.CASE_INSENSITIVE)).get(); } public User findUser(String signupKey, String userId) { Assert.hasText(signupKey); Assert.hasText(userId); return ds.createQuery(entityClazz).field("secretKey").equal(signupKey).field("_id") .equal(new ObjectId(userId)).get(); } public User findInvitedUser(String inviteKey, String accountId) { Assert.hasText(inviteKey); Assert.hasText(accountId); User user = ds.createQuery(entityClazz).field("secretKey").equal(inviteKey).get(); if (user != null) { if ((user.getAccount().getId().toString().equals(accountId))) { return user; } } return null; } public List<User> allEnabledUsersForAccount(Account account) { return ds.createQuery(entityClazz).field("account").equal(account).field("userStatus") .notEqual(User.UserStatus.DISABLED).asList(); } public User findUserForAlias(Account account, String alias) { return ds.createQuery(entityClazz).field("alias") .equal(Pattern.compile("^\\Q" + alias + "\\E$", Pattern.CASE_INSENSITIVE)).field("account") .equal(account).get(); } public User findUserForUsername(Account account, String username) { return ds.createQuery(entityClazz).field("username") .equal(Pattern.compile("^\\Q" + username + "\\E$", Pattern.CASE_INSENSITIVE)).field("account") .equal(account).get(); } public User findUserInAccount(Account account, String name) { Query<User> query = ds.createQuery(entityClazz); query.field("account").equal(account); query.or(query.criteria("username").equal(name), query.criteria("alias").equal(Pattern.compile("^\\Q" + name + "\\E$", Pattern.CASE_INSENSITIVE))); return query.get(); } public APIAuthenticationToken findAuthToken(String authenticationToken) { User user = findByAuthToken(authenticationToken); if (user != null) { return user.getAuthenticationToken(); } return null; } public User findByAuthToken(String authenticationToken) { Assert.hasText(authenticationToken); APIAuthenticationToken authToken = new APIAuthenticationToken(authenticationToken); return ds.createQuery(entityClazz).field("authenticationToken").equal(authToken).get(); } public Set<User> getActiveLoggedInUsers(Account account, Long maxInactivity) { Query<User> q = createQuery(); q.field("account").equal(account); q.field("lastActivity").greaterThanOrEq(System.currentTimeMillis() - maxInactivity); return new HashSet<>(q.asList()); } }