| // Copyright (C) 2016 The Android Open Source Project |
| // |
| // 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.google.gerrit.server.query.account; |
| |
| import com.google.common.base.Joiner; |
| import com.google.common.collect.Lists; |
| import com.google.gerrit.reviewdb.client.Project; |
| import com.google.gerrit.server.account.AccountState; |
| import com.google.gerrit.server.account.ExternalId; |
| import com.google.gerrit.server.index.IndexConfig; |
| import com.google.gerrit.server.index.account.AccountIndexCollection; |
| import com.google.gerrit.server.query.InternalQuery; |
| import com.google.gwtorm.server.OrmException; |
| import com.google.inject.Inject; |
| import java.util.List; |
| import java.util.Set; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| public class InternalAccountQuery extends InternalQuery<AccountState> { |
| private static final Logger log = LoggerFactory.getLogger(InternalAccountQuery.class); |
| |
| @Inject |
| InternalAccountQuery( |
| AccountQueryProcessor queryProcessor, |
| AccountIndexCollection indexes, |
| IndexConfig indexConfig) { |
| super(queryProcessor, indexes, indexConfig); |
| } |
| |
| @Override |
| public InternalAccountQuery setLimit(int n) { |
| super.setLimit(n); |
| return this; |
| } |
| |
| @Override |
| public InternalAccountQuery enforceVisibility(boolean enforce) { |
| super.enforceVisibility(enforce); |
| return this; |
| } |
| |
| @Override |
| public InternalAccountQuery setRequestedFields(Set<String> fields) { |
| super.setRequestedFields(fields); |
| return this; |
| } |
| |
| @Override |
| public InternalAccountQuery noFields() { |
| super.noFields(); |
| return this; |
| } |
| |
| public List<AccountState> byDefault(String query) throws OrmException { |
| return query(AccountPredicates.defaultPredicate(query)); |
| } |
| |
| public List<AccountState> byEmailPrefix(String emailPrefix) throws OrmException { |
| return query(AccountPredicates.email(emailPrefix)); |
| } |
| |
| public List<AccountState> byExternalId(String scheme, String id) throws OrmException { |
| return byExternalId(ExternalId.Key.create(scheme, id)); |
| } |
| |
| public List<AccountState> byExternalId(ExternalId.Key externalId) throws OrmException { |
| return query(AccountPredicates.externalId(externalId.toString())); |
| } |
| |
| public AccountState oneByExternalId(String externalId) throws OrmException { |
| return oneByExternalId(ExternalId.Key.parse(externalId)); |
| } |
| |
| public AccountState oneByExternalId(String scheme, String id) throws OrmException { |
| return oneByExternalId(ExternalId.Key.create(scheme, id)); |
| } |
| |
| public AccountState oneByExternalId(ExternalId.Key externalId) throws OrmException { |
| List<AccountState> accountStates = byExternalId(externalId); |
| if (accountStates.size() == 1) { |
| return accountStates.get(0); |
| } else if (accountStates.size() > 0) { |
| StringBuilder msg = new StringBuilder(); |
| msg.append("Ambiguous external ID ").append(externalId).append("for accounts: "); |
| Joiner.on(", ") |
| .appendTo(msg, Lists.transform(accountStates, AccountState.ACCOUNT_ID_FUNCTION)); |
| log.warn(msg.toString()); |
| } |
| return null; |
| } |
| |
| public List<AccountState> byFullName(String fullName) throws OrmException { |
| return query(AccountPredicates.fullName(fullName)); |
| } |
| |
| public List<AccountState> byWatchedProject(Project.NameKey project) throws OrmException { |
| return query(AccountPredicates.watchedProject(project)); |
| } |
| } |