Merge changes from topic 'account-index'
* changes: Make LimitPredicate generic Add AccountIndexRewriter Add IndexedAccountQuery Reindex account whenever account is evicted from cache Reindex: Allow to specify index name for reindex operation Reindex: Add --list option to list available indices Initial implementation of account index in Lucene
This commit is contained in:
commit
649d7849f8
|
@ -24,6 +24,13 @@ Rebuilds the secondary index.
|
||||||
--dry-run::
|
--dry-run::
|
||||||
Dry run. Don't write anything to index.
|
Dry run. Don't write anything to index.
|
||||||
|
|
||||||
|
--list::
|
||||||
|
List available index names.
|
||||||
|
|
||||||
|
--index::
|
||||||
|
Reindex only index with given name. This option can be supplied
|
||||||
|
more than once to reindex multiple indices.
|
||||||
|
|
||||||
== CONTEXT
|
== CONTEXT
|
||||||
The secondary index must be enabled. See
|
The secondary index must be enabled. See
|
||||||
link:config-gerrit.html#index.type[index.type].
|
link:config-gerrit.html#index.type[index.type].
|
||||||
|
|
|
@ -66,6 +66,7 @@ import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
@ -154,7 +155,7 @@ public class GerritPublicKeyCheckerTest {
|
||||||
return userFactory.create(id);
|
return userFactory.create(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
private IdentifiedUser reloadUser() {
|
private IdentifiedUser reloadUser() throws IOException {
|
||||||
accountCache.evict(userId);
|
accountCache.evict(userId);
|
||||||
user = userFactory.create(userId);
|
user = userFactory.create(userId);
|
||||||
return user;
|
return user;
|
||||||
|
|
|
@ -219,7 +219,7 @@ class BecomeAnyAccountLoginServlet extends HttpServlet {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private AuthResult create() {
|
private AuthResult create() throws IOException {
|
||||||
String fakeId = AccountExternalId.SCHEME_UUID + UUID.randomUUID();
|
String fakeId = AccountExternalId.SCHEME_UUID + UUID.randomUUID();
|
||||||
try {
|
try {
|
||||||
return accountManager.authenticate(new AuthRequest(fakeId));
|
return accountManager.authenticate(new AuthRequest(fakeId));
|
||||||
|
|
|
@ -153,7 +153,7 @@ class HttpLoginServlet extends HttpServlet {
|
||||||
}
|
}
|
||||||
|
|
||||||
private void updateRemoteExternalId(AuthResult arsp, String remoteAuthToken)
|
private void updateRemoteExternalId(AuthResult arsp, String remoteAuthToken)
|
||||||
throws AccountException, OrmException {
|
throws AccountException, OrmException, IOException {
|
||||||
AccountExternalId remoteAuthExtId =
|
AccountExternalId remoteAuthExtId =
|
||||||
new AccountExternalId(arsp.getAccountId(), new AccountExternalId.Key(
|
new AccountExternalId(arsp.getAccountId(), new AccountExternalId.Key(
|
||||||
SCHEME_EXTERNAL, remoteAuthToken));
|
SCHEME_EXTERNAL, remoteAuthToken));
|
||||||
|
|
|
@ -27,6 +27,8 @@ import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.gwtorm.server.OrmRuntimeException;
|
import com.google.gwtorm.server.OrmRuntimeException;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
/** Support for services which require a {@link ReviewDb} instance. */
|
/** Support for services which require a {@link ReviewDb} instance. */
|
||||||
public class BaseServiceImplementation {
|
public class BaseServiceImplementation {
|
||||||
private final Provider<ReviewDb> schema;
|
private final Provider<ReviewDb> schema;
|
||||||
|
@ -86,6 +88,8 @@ public class BaseServiceImplementation {
|
||||||
handleOrmException(callback, ex);
|
handleOrmException(callback, ex);
|
||||||
} catch (OrmException e) {
|
} catch (OrmException e) {
|
||||||
handleOrmException(callback, e);
|
handleOrmException(callback, e);
|
||||||
|
} catch (IOException e) {
|
||||||
|
callback.onFailure(e);
|
||||||
} catch (Failure e) {
|
} catch (Failure e) {
|
||||||
if (e.getCause() instanceof NoSuchProjectException
|
if (e.getCause() instanceof NoSuchProjectException
|
||||||
|| e.getCause() instanceof NoSuchChangeException) {
|
|| e.getCause() instanceof NoSuchChangeException) {
|
||||||
|
@ -132,6 +136,6 @@ public class BaseServiceImplementation {
|
||||||
* @throws InvalidQueryException
|
* @throws InvalidQueryException
|
||||||
*/
|
*/
|
||||||
T run(ReviewDb db) throws OrmException, Failure, NoSuchProjectException,
|
T run(ReviewDb db) throws OrmException, Failure, NoSuchProjectException,
|
||||||
NoSuchGroupException, InvalidQueryException;
|
NoSuchGroupException, InvalidQueryException, IOException;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -40,6 +40,7 @@ import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
@ -99,7 +100,8 @@ class AccountSecurityImpl extends BaseServiceImplementation implements
|
||||||
final AsyncCallback<Account> callback) {
|
final AsyncCallback<Account> callback) {
|
||||||
run(callback, new Action<Account>() {
|
run(callback, new Action<Account>() {
|
||||||
@Override
|
@Override
|
||||||
public Account run(ReviewDb db) throws OrmException, Failure {
|
public Account run(ReviewDb db)
|
||||||
|
throws OrmException, Failure, IOException {
|
||||||
IdentifiedUser self = user.get();
|
IdentifiedUser self = user.get();
|
||||||
final Account me = db.accounts().get(self.getAccountId());
|
final Account me = db.accounts().get(self.getAccountId());
|
||||||
final String oldEmail = me.getPreferredEmail();
|
final String oldEmail = me.getPreferredEmail();
|
||||||
|
@ -134,7 +136,8 @@ class AccountSecurityImpl extends BaseServiceImplementation implements
|
||||||
final AsyncCallback<VoidResult> callback) {
|
final AsyncCallback<VoidResult> callback) {
|
||||||
run(callback, new Action<VoidResult>() {
|
run(callback, new Action<VoidResult>() {
|
||||||
@Override
|
@Override
|
||||||
public VoidResult run(final ReviewDb db) throws OrmException, Failure {
|
public VoidResult run(final ReviewDb db)
|
||||||
|
throws OrmException, Failure, IOException {
|
||||||
ContributorAgreement ca = projectCache.getAllProjects().getConfig()
|
ContributorAgreement ca = projectCache.getAllProjects().getConfig()
|
||||||
.getContributorAgreement(agreementName);
|
.getContributorAgreement(agreementName);
|
||||||
if (ca == null) {
|
if (ca == null) {
|
||||||
|
|
|
@ -24,6 +24,7 @@ import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.assistedinject.Assisted;
|
import com.google.inject.assistedinject.Assisted;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
|
@ -60,7 +61,7 @@ class DeleteExternalIds extends Handler<Set<AccountExternalId.Key>> {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Set<AccountExternalId.Key> call() throws OrmException {
|
public Set<AccountExternalId.Key> call() throws OrmException, IOException {
|
||||||
final Map<AccountExternalId.Key, AccountExternalId> have = have();
|
final Map<AccountExternalId.Key, AccountExternalId> have = have();
|
||||||
|
|
||||||
List<AccountExternalId> toDelete = new ArrayList<>();
|
List<AccountExternalId> toDelete = new ArrayList<>();
|
||||||
|
|
|
@ -0,0 +1,199 @@
|
||||||
|
// 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.lucene;
|
||||||
|
|
||||||
|
import static com.google.gerrit.server.index.account.AccountField.ID;
|
||||||
|
|
||||||
|
import com.google.common.collect.ImmutableSet;
|
||||||
|
import com.google.common.collect.Sets;
|
||||||
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
import com.google.gerrit.server.account.AccountCache;
|
||||||
|
import com.google.gerrit.server.account.AccountState;
|
||||||
|
import com.google.gerrit.server.config.GerritServerConfig;
|
||||||
|
import com.google.gerrit.server.config.SitePaths;
|
||||||
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
|
import com.google.gerrit.server.index.Schema;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndex;
|
||||||
|
import com.google.gerrit.server.query.DataSource;
|
||||||
|
import com.google.gerrit.server.query.Predicate;
|
||||||
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
import com.google.gwtorm.server.ResultSet;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import com.google.inject.assistedinject.Assisted;
|
||||||
|
|
||||||
|
import org.apache.lucene.document.Document;
|
||||||
|
import org.apache.lucene.index.Term;
|
||||||
|
import org.apache.lucene.search.IndexSearcher;
|
||||||
|
import org.apache.lucene.search.Query;
|
||||||
|
import org.apache.lucene.search.ScoreDoc;
|
||||||
|
import org.apache.lucene.search.SearcherFactory;
|
||||||
|
import org.apache.lucene.search.Sort;
|
||||||
|
import org.apache.lucene.search.SortField;
|
||||||
|
import org.apache.lucene.search.TopFieldDocs;
|
||||||
|
import org.apache.lucene.store.Directory;
|
||||||
|
import org.apache.lucene.store.FSDirectory;
|
||||||
|
import org.apache.lucene.store.RAMDirectory;
|
||||||
|
import org.eclipse.jgit.lib.Config;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.concurrent.ExecutionException;
|
||||||
|
|
||||||
|
public class LuceneAccountIndex
|
||||||
|
extends AbstractLuceneIndex<Account.Id, AccountState>
|
||||||
|
implements AccountIndex {
|
||||||
|
private static final Logger log =
|
||||||
|
LoggerFactory.getLogger(LuceneAccountIndex.class);
|
||||||
|
|
||||||
|
private static final String ACCOUNTS = "accounts";
|
||||||
|
|
||||||
|
private static final String ID_SORT_FIELD = sortFieldName(ID);
|
||||||
|
|
||||||
|
private static Term idTerm(AccountState as) {
|
||||||
|
return idTerm(as.getAccount().getId());
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Term idTerm(Account.Id id) {
|
||||||
|
return QueryBuilder.intTerm(ID.getName(), id.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
private final GerritIndexWriterConfig indexWriterConfig;
|
||||||
|
private final QueryBuilder<AccountState> queryBuilder;
|
||||||
|
private final AccountCache accountCache;
|
||||||
|
|
||||||
|
private static Directory dir(Schema<AccountState> schema, Config cfg,
|
||||||
|
SitePaths sitePaths) throws IOException {
|
||||||
|
if (LuceneIndexModule.isInMemoryTest(cfg)) {
|
||||||
|
return new RAMDirectory();
|
||||||
|
}
|
||||||
|
Path indexDir =
|
||||||
|
LuceneVersionManager.getDir(sitePaths, ACCOUNTS + "_", schema);
|
||||||
|
return FSDirectory.open(indexDir);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
LuceneAccountIndex(
|
||||||
|
@GerritServerConfig Config cfg,
|
||||||
|
SitePaths sitePaths,
|
||||||
|
AccountCache accountCache,
|
||||||
|
@Assisted Schema<AccountState> schema) throws IOException {
|
||||||
|
super(schema, sitePaths, dir(schema, cfg, sitePaths), ACCOUNTS, null,
|
||||||
|
new GerritIndexWriterConfig(cfg, ACCOUNTS), new SearcherFactory());
|
||||||
|
this.accountCache = accountCache;
|
||||||
|
|
||||||
|
indexWriterConfig =
|
||||||
|
new GerritIndexWriterConfig(cfg, ACCOUNTS);
|
||||||
|
queryBuilder = new QueryBuilder<>(schema, indexWriterConfig.getAnalyzer());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void replace(AccountState as) throws IOException {
|
||||||
|
try {
|
||||||
|
// No parts of FillArgs are currently required, just use null.
|
||||||
|
replace(idTerm(as), toDocument(as, null)).get();
|
||||||
|
} catch (ExecutionException | InterruptedException e) {
|
||||||
|
throw new IOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void delete(Account.Id key) throws IOException {
|
||||||
|
try {
|
||||||
|
delete(idTerm(key)).get();
|
||||||
|
} catch (ExecutionException | InterruptedException e) {
|
||||||
|
throw new IOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DataSource<AccountState> getSource(Predicate<AccountState> p,
|
||||||
|
QueryOptions opts) throws QueryParseException {
|
||||||
|
return new QuerySource(
|
||||||
|
opts,
|
||||||
|
queryBuilder.toQuery(p),
|
||||||
|
new Sort(
|
||||||
|
new SortField(ID_SORT_FIELD, SortField.Type.LONG, true)));
|
||||||
|
}
|
||||||
|
|
||||||
|
private class QuerySource implements DataSource<AccountState> {
|
||||||
|
private final QueryOptions opts;
|
||||||
|
private final Query query;
|
||||||
|
private final Sort sort;
|
||||||
|
|
||||||
|
private QuerySource(QueryOptions opts, Query query, Sort sort) {
|
||||||
|
this.opts = opts;
|
||||||
|
this.query = query;
|
||||||
|
this.sort = sort;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int getCardinality() {
|
||||||
|
// TODO(dborowitz): In contrast to the comment in
|
||||||
|
// LuceneChangeIndex.QuerySource#getCardinality, at this point I actually
|
||||||
|
// think we might just want to remove getCardinality.
|
||||||
|
return 10;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ResultSet<AccountState> read() throws OrmException {
|
||||||
|
IndexSearcher searcher = null;
|
||||||
|
try {
|
||||||
|
searcher = acquire();
|
||||||
|
int realLimit = opts.start() + opts.limit();
|
||||||
|
TopFieldDocs docs = searcher.search(query, realLimit, sort);
|
||||||
|
List<AccountState> result = new ArrayList<>(docs.scoreDocs.length);
|
||||||
|
for (int i = opts.start(); i < docs.scoreDocs.length; i++) {
|
||||||
|
ScoreDoc sd = docs.scoreDocs[i];
|
||||||
|
Document doc = searcher.doc(sd.doc, fields(opts));
|
||||||
|
result.add(toAccountState(doc));
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new OrmException(e);
|
||||||
|
} finally {
|
||||||
|
if (searcher != null) {
|
||||||
|
try {
|
||||||
|
release(searcher);
|
||||||
|
} catch (IOException e) {
|
||||||
|
log.warn("cannot release Lucene searcher", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private Set<String> fields(QueryOptions opts) {
|
||||||
|
Set<String> fs = opts.fields();
|
||||||
|
return fs.contains(ID.getName())
|
||||||
|
? fs
|
||||||
|
: Sets.union(fs, ImmutableSet.of(ID.getName()));
|
||||||
|
}
|
||||||
|
|
||||||
|
private AccountState toAccountState(Document doc) {
|
||||||
|
Account.Id id =
|
||||||
|
new Account.Id(doc.getField(ID.getName()).numericValue().intValue());
|
||||||
|
// Use the AccountCache rather than depending on any stored fields in the
|
||||||
|
// document (of which there shouldn't be any. The most expensive part to
|
||||||
|
// compute anyway is the effective group IDs, and we don't have a good way
|
||||||
|
// to reindex when those change.
|
||||||
|
return accountCache.get(id);
|
||||||
|
}
|
||||||
|
}
|
|
@ -21,8 +21,8 @@ import static com.google.gerrit.server.git.QueueProvider.QueueType.INTERACTIVE;
|
||||||
import static com.google.gerrit.server.index.change.ChangeField.CHANGE;
|
import static com.google.gerrit.server.index.change.ChangeField.CHANGE;
|
||||||
import static com.google.gerrit.server.index.change.ChangeField.LEGACY_ID;
|
import static com.google.gerrit.server.index.change.ChangeField.LEGACY_ID;
|
||||||
import static com.google.gerrit.server.index.change.ChangeField.PROJECT;
|
import static com.google.gerrit.server.index.change.ChangeField.PROJECT;
|
||||||
import static com.google.gerrit.server.index.change.IndexRewriter.CLOSED_STATUSES;
|
import static com.google.gerrit.server.index.change.ChangeIndexRewriter.CLOSED_STATUSES;
|
||||||
import static com.google.gerrit.server.index.change.IndexRewriter.OPEN_STATUSES;
|
import static com.google.gerrit.server.index.change.ChangeIndexRewriter.OPEN_STATUSES;
|
||||||
|
|
||||||
import com.google.common.base.Function;
|
import com.google.common.base.Function;
|
||||||
import com.google.common.collect.ArrayListMultimap;
|
import com.google.common.collect.ArrayListMultimap;
|
||||||
|
@ -51,7 +51,7 @@ import com.google.gerrit.server.index.change.ChangeField.ChangeProtoField;
|
||||||
import com.google.gerrit.server.index.change.ChangeField.PatchSetApprovalProtoField;
|
import com.google.gerrit.server.index.change.ChangeField.PatchSetApprovalProtoField;
|
||||||
import com.google.gerrit.server.index.change.ChangeField.PatchSetProtoField;
|
import com.google.gerrit.server.index.change.ChangeField.PatchSetProtoField;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndex;
|
import com.google.gerrit.server.index.change.ChangeIndex;
|
||||||
import com.google.gerrit.server.index.change.IndexRewriter;
|
import com.google.gerrit.server.index.change.ChangeIndexRewriter;
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
import com.google.gerrit.server.query.change.ChangeData;
|
import com.google.gerrit.server.query.change.ChangeData;
|
||||||
|
@ -239,7 +239,7 @@ public class LuceneChangeIndex implements ChangeIndex {
|
||||||
@Override
|
@Override
|
||||||
public ChangeDataSource getSource(Predicate<ChangeData> p, QueryOptions opts)
|
public ChangeDataSource getSource(Predicate<ChangeData> p, QueryOptions opts)
|
||||||
throws QueryParseException {
|
throws QueryParseException {
|
||||||
Set<Change.Status> statuses = IndexRewriter.getPossibleStatus(p);
|
Set<Change.Status> statuses = ChangeIndexRewriter.getPossibleStatus(p);
|
||||||
List<ChangeSubIndex> indexes = Lists.newArrayListWithCapacity(2);
|
List<ChangeSubIndex> indexes = Lists.newArrayListWithCapacity(2);
|
||||||
if (!Sets.intersection(statuses, OPEN_STATUSES).isEmpty()) {
|
if (!Sets.intersection(statuses, OPEN_STATUSES).isEmpty()) {
|
||||||
indexes.add(openIndex);
|
indexes.add(openIndex);
|
||||||
|
|
|
@ -23,6 +23,7 @@ import com.google.gerrit.server.index.IndexConfig;
|
||||||
import com.google.gerrit.server.index.IndexDefinition;
|
import com.google.gerrit.server.index.IndexDefinition;
|
||||||
import com.google.gerrit.server.index.IndexModule;
|
import com.google.gerrit.server.index.IndexModule;
|
||||||
import com.google.gerrit.server.index.Schema;
|
import com.google.gerrit.server.index.Schema;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndex;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndex;
|
import com.google.gerrit.server.index.change.ChangeIndex;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.Provides;
|
import com.google.inject.Provides;
|
||||||
|
@ -74,6 +75,10 @@ public class LuceneIndexModule extends LifecycleModule {
|
||||||
new FactoryModuleBuilder()
|
new FactoryModuleBuilder()
|
||||||
.implement(ChangeIndex.class, LuceneChangeIndex.class)
|
.implement(ChangeIndex.class, LuceneChangeIndex.class)
|
||||||
.build(ChangeIndex.Factory.class));
|
.build(ChangeIndex.Factory.class));
|
||||||
|
install(
|
||||||
|
new FactoryModuleBuilder()
|
||||||
|
.implement(AccountIndex.class, LuceneAccountIndex.class)
|
||||||
|
.build(AccountIndex.Factory.class));
|
||||||
|
|
||||||
install(new IndexModule(threads));
|
install(new IndexModule(threads));
|
||||||
if (singleVersions == null) {
|
if (singleVersions == null) {
|
||||||
|
|
|
@ -17,6 +17,10 @@ package com.google.gerrit.pgm;
|
||||||
import static com.google.common.base.Preconditions.checkNotNull;
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
import static com.google.gerrit.server.schema.DataSourceProvider.Context.MULTI_USER;
|
import static com.google.gerrit.server.schema.DataSourceProvider.Context.MULTI_USER;
|
||||||
|
|
||||||
|
import com.google.common.base.Function;
|
||||||
|
import com.google.common.collect.FluentIterable;
|
||||||
|
import com.google.common.collect.Ordering;
|
||||||
|
import com.google.common.collect.Sets;
|
||||||
import com.google.gerrit.common.Die;
|
import com.google.gerrit.common.Die;
|
||||||
import com.google.gerrit.extensions.config.FactoryModule;
|
import com.google.gerrit.extensions.config.FactoryModule;
|
||||||
import com.google.gerrit.lifecycle.LifecycleManager;
|
import com.google.gerrit.lifecycle.LifecycleManager;
|
||||||
|
@ -47,6 +51,8 @@ import java.util.Collection;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.TreeSet;
|
||||||
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
public class Reindex extends SiteProgram {
|
public class Reindex extends SiteProgram {
|
||||||
|
@ -60,6 +66,12 @@ public class Reindex extends SiteProgram {
|
||||||
@Option(name = "--verbose", usage = "Output debug information for each change")
|
@Option(name = "--verbose", usage = "Output debug information for each change")
|
||||||
private boolean verbose;
|
private boolean verbose;
|
||||||
|
|
||||||
|
@Option(name = "--list", usage = "List supported indices and exit")
|
||||||
|
private boolean list;
|
||||||
|
|
||||||
|
@Option(name = "--index", usage = "Only reindex specified indices")
|
||||||
|
private List<String> indices = new ArrayList<>();
|
||||||
|
|
||||||
private Injector dbInjector;
|
private Injector dbInjector;
|
||||||
private Injector sysInjector;
|
private Injector sysInjector;
|
||||||
private Config globalConfig;
|
private Config globalConfig;
|
||||||
|
@ -86,12 +98,10 @@ public class Reindex extends SiteProgram {
|
||||||
sysManager.add(sysInjector);
|
sysManager.add(sysInjector);
|
||||||
sysManager.start();
|
sysManager.start();
|
||||||
sysInjector.injectMembers(this);
|
sysInjector.injectMembers(this);
|
||||||
|
checkIndicesOption();
|
||||||
|
|
||||||
try {
|
try {
|
||||||
boolean ok = true;
|
boolean ok = list ? list() : reindex();
|
||||||
for (IndexDefinition<?, ?, ?> def : indexDefs) {
|
|
||||||
ok &= reindex(def);
|
|
||||||
}
|
|
||||||
return ok ? 0 : 1;
|
return ok ? 0 : 1;
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
throw die(e.getMessage(), e);
|
throw die(e.getMessage(), e);
|
||||||
|
@ -101,6 +111,46 @@ public class Reindex extends SiteProgram {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private boolean list() {
|
||||||
|
for (IndexDefinition<?, ?, ?> def : indexDefs) {
|
||||||
|
System.out.format("%s\n", def.getName());
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean reindex() throws IOException {
|
||||||
|
boolean ok = true;
|
||||||
|
for (IndexDefinition<?, ?, ?> def : indexDefs) {
|
||||||
|
if (indices.isEmpty() || indices.contains(def.getName())) {
|
||||||
|
ok &= reindex(def);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void checkIndicesOption() throws Die {
|
||||||
|
if (indices.isEmpty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
checkNotNull(indexDefs, "Called this method before injectMembers?");
|
||||||
|
Set<String> valid = FluentIterable.from(indexDefs).transform(
|
||||||
|
new Function<IndexDefinition<?, ?, ?>, String>() {
|
||||||
|
@Override
|
||||||
|
public String apply(IndexDefinition<?, ?, ?> input) {
|
||||||
|
return input.getName();
|
||||||
|
}
|
||||||
|
}).toSortedSet(Ordering.natural());
|
||||||
|
|
||||||
|
Set<String> invalid = Sets.difference(Sets.newHashSet(indices), valid);
|
||||||
|
if (invalid.isEmpty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
throw die("invalid index name(s): " + new TreeSet<>(invalid)
|
||||||
|
+ " available indices are: " + valid);
|
||||||
|
}
|
||||||
|
|
||||||
private void checkNotSlaveMode() throws Die {
|
private void checkNotSlaveMode() throws Die {
|
||||||
if (globalConfig.getBoolean("container", "slave", false)) {
|
if (globalConfig.getBoolean("container", "slave", false)) {
|
||||||
throw die("Cannot run reindex in slave mode");
|
throw die("Cannot run reindex in slave mode");
|
||||||
|
|
|
@ -16,6 +16,8 @@ package com.google.gerrit.server.account;
|
||||||
|
|
||||||
import com.google.gerrit.reviewdb.client.Account;
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
/** Caches important (but small) account state to avoid database hits. */
|
/** Caches important (but small) account state to avoid database hits. */
|
||||||
public interface AccountCache {
|
public interface AccountCache {
|
||||||
AccountState get(Account.Id accountId);
|
AccountState get(Account.Id accountId);
|
||||||
|
@ -24,7 +26,7 @@ public interface AccountCache {
|
||||||
|
|
||||||
AccountState getByUsername(String username);
|
AccountState getByUsername(String username);
|
||||||
|
|
||||||
void evict(Account.Id accountId);
|
void evict(Account.Id accountId) throws IOException;
|
||||||
|
|
||||||
void evictByUsername(String username);
|
void evictByUsername(String username);
|
||||||
}
|
}
|
||||||
|
|
|
@ -26,6 +26,8 @@ import com.google.gerrit.reviewdb.client.AccountGroup;
|
||||||
import com.google.gerrit.reviewdb.client.AccountGroupMember;
|
import com.google.gerrit.reviewdb.client.AccountGroupMember;
|
||||||
import com.google.gerrit.reviewdb.server.ReviewDb;
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
||||||
import com.google.gerrit.server.cache.CacheModule;
|
import com.google.gerrit.server.cache.CacheModule;
|
||||||
|
import com.google.gerrit.server.index.Index;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndexCollection;
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.gwtorm.server.SchemaFactory;
|
import com.google.gwtorm.server.SchemaFactory;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
|
@ -74,12 +76,15 @@ public class AccountCacheImpl implements AccountCache {
|
||||||
|
|
||||||
private final LoadingCache<Account.Id, AccountState> byId;
|
private final LoadingCache<Account.Id, AccountState> byId;
|
||||||
private final LoadingCache<String, Optional<Account.Id>> byName;
|
private final LoadingCache<String, Optional<Account.Id>> byName;
|
||||||
|
private final AccountIndexCollection indexes;
|
||||||
|
|
||||||
@Inject
|
@Inject
|
||||||
AccountCacheImpl(@Named(BYID_NAME) LoadingCache<Account.Id, AccountState> byId,
|
AccountCacheImpl(@Named(BYID_NAME) LoadingCache<Account.Id, AccountState> byId,
|
||||||
@Named(BYUSER_NAME) LoadingCache<String, Optional<Account.Id>> byUsername) {
|
@Named(BYUSER_NAME) LoadingCache<String, Optional<Account.Id>> byUsername,
|
||||||
|
AccountIndexCollection indexes) {
|
||||||
this.byId = byId;
|
this.byId = byId;
|
||||||
this.byName = byUsername;
|
this.byName = byUsername;
|
||||||
|
this.indexes = indexes;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -109,9 +114,16 @@ public class AccountCacheImpl implements AccountCache {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void evict(Account.Id accountId) {
|
public void evict(Account.Id accountId) throws IOException {
|
||||||
if (accountId != null) {
|
if (accountId != null) {
|
||||||
byId.invalidate(accountId);
|
byId.invalidate(accountId);
|
||||||
|
index(accountId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void index(Account.Id id) throws IOException {
|
||||||
|
for (Index<?, AccountState> i : indexes.getWriteIndexes()) {
|
||||||
|
i.replace(get(id));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -37,6 +37,7 @@ import com.google.inject.Singleton;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
@ -101,7 +102,8 @@ public class AccountManager {
|
||||||
* @throws AccountException the account does not exist, and cannot be created,
|
* @throws AccountException the account does not exist, and cannot be created,
|
||||||
* or exists, but cannot be located, or is inactive.
|
* or exists, but cannot be located, or is inactive.
|
||||||
*/
|
*/
|
||||||
public AuthResult authenticate(AuthRequest who) throws AccountException {
|
public AuthResult authenticate(AuthRequest who)
|
||||||
|
throws AccountException, IOException {
|
||||||
who = realm.authenticate(who);
|
who = realm.authenticate(who);
|
||||||
try {
|
try {
|
||||||
try (ReviewDb db = schema.open()) {
|
try (ReviewDb db = schema.open()) {
|
||||||
|
@ -152,7 +154,8 @@ public class AccountManager {
|
||||||
}
|
}
|
||||||
|
|
||||||
private void update(ReviewDb db, AuthRequest who, AccountExternalId extId)
|
private void update(ReviewDb db, AuthRequest who, AccountExternalId extId)
|
||||||
throws OrmException, NameAlreadyUsedException, InvalidUserNameException {
|
throws OrmException, NameAlreadyUsedException, InvalidUserNameException,
|
||||||
|
IOException {
|
||||||
IdentifiedUser user = userFactory.create(extId.getAccountId());
|
IdentifiedUser user = userFactory.create(extId.getAccountId());
|
||||||
Account toUpdate = null;
|
Account toUpdate = null;
|
||||||
|
|
||||||
|
@ -214,7 +217,7 @@ public class AccountManager {
|
||||||
}
|
}
|
||||||
|
|
||||||
private AuthResult create(ReviewDb db, AuthRequest who)
|
private AuthResult create(ReviewDb db, AuthRequest who)
|
||||||
throws OrmException, AccountException {
|
throws OrmException, AccountException, IOException {
|
||||||
Account.Id newId = new Account.Id(db.nextAccountId());
|
Account.Id newId = new Account.Id(db.nextAccountId());
|
||||||
Account account = new Account(newId, TimeUtil.nowTs());
|
Account account = new Account(newId, TimeUtil.nowTs());
|
||||||
AccountExternalId extId = createId(newId, who);
|
AccountExternalId extId = createId(newId, who);
|
||||||
|
@ -340,7 +343,7 @@ public class AccountManager {
|
||||||
* cannot be linked at this time.
|
* cannot be linked at this time.
|
||||||
*/
|
*/
|
||||||
public AuthResult link(Account.Id to, AuthRequest who)
|
public AuthResult link(Account.Id to, AuthRequest who)
|
||||||
throws AccountException, OrmException {
|
throws AccountException, OrmException, IOException {
|
||||||
try (ReviewDb db = schema.open()) {
|
try (ReviewDb db = schema.open()) {
|
||||||
AccountExternalId.Key key = id(who);
|
AccountExternalId.Key key = id(who);
|
||||||
AccountExternalId extId = getAccountExternalId(db, key);
|
AccountExternalId extId = getAccountExternalId(db, key);
|
||||||
|
@ -392,7 +395,7 @@ public class AccountManager {
|
||||||
* cannot be linked at this time.
|
* cannot be linked at this time.
|
||||||
*/
|
*/
|
||||||
public AuthResult updateLink(Account.Id to, AuthRequest who) throws OrmException,
|
public AuthResult updateLink(Account.Id to, AuthRequest who) throws OrmException,
|
||||||
AccountException {
|
AccountException, IOException {
|
||||||
try (ReviewDb db = schema.open()) {
|
try (ReviewDb db = schema.open()) {
|
||||||
AccountExternalId.Key key = id(who);
|
AccountExternalId.Key key = id(who);
|
||||||
List<AccountExternalId.Key> filteredKeysByScheme =
|
List<AccountExternalId.Key> filteredKeysByScheme =
|
||||||
|
@ -429,7 +432,7 @@ public class AccountManager {
|
||||||
* cannot be unlinked at this time.
|
* cannot be unlinked at this time.
|
||||||
*/
|
*/
|
||||||
public AuthResult unlink(Account.Id from, AuthRequest who)
|
public AuthResult unlink(Account.Id from, AuthRequest who)
|
||||||
throws AccountException, OrmException {
|
throws AccountException, OrmException, IOException {
|
||||||
try (ReviewDb db = schema.open()) {
|
try (ReviewDb db = schema.open()) {
|
||||||
AccountExternalId.Key key = id(who);
|
AccountExternalId.Key key = id(who);
|
||||||
AccountExternalId extId = getAccountExternalId(db, key);
|
AccountExternalId extId = getAccountExternalId(db, key);
|
||||||
|
|
|
@ -29,6 +29,7 @@ import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.assistedinject.Assisted;
|
import com.google.inject.assistedinject.Assisted;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
@ -71,7 +72,7 @@ public class ChangeUserName implements Callable<VoidResult> {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public VoidResult call() throws OrmException, NameAlreadyUsedException,
|
public VoidResult call() throws OrmException, NameAlreadyUsedException,
|
||||||
InvalidUserNameException {
|
InvalidUserNameException, IOException {
|
||||||
final Collection<AccountExternalId> old = old();
|
final Collection<AccountExternalId> old = old();
|
||||||
if (!old.isEmpty()) {
|
if (!old.isEmpty()) {
|
||||||
throw new IllegalStateException(USERNAME_CANNOT_BE_CHANGED);
|
throw new IllegalStateException(USERNAME_CANNOT_BE_CHANGED);
|
||||||
|
|
|
@ -39,6 +39,8 @@ import org.apache.commons.validator.routines.EmailValidator;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
public class CreateEmail implements RestModifyView<AccountResource, EmailInput> {
|
public class CreateEmail implements RestModifyView<AccountResource, EmailInput> {
|
||||||
private static final Logger log = LoggerFactory.getLogger(CreateEmail.class);
|
private static final Logger log = LoggerFactory.getLogger(CreateEmail.class);
|
||||||
|
|
||||||
|
@ -75,7 +77,7 @@ public class CreateEmail implements RestModifyView<AccountResource, EmailInput>
|
||||||
public Response<EmailInfo> apply(AccountResource rsrc, EmailInput input)
|
public Response<EmailInfo> apply(AccountResource rsrc, EmailInput input)
|
||||||
throws AuthException, BadRequestException, ResourceConflictException,
|
throws AuthException, BadRequestException, ResourceConflictException,
|
||||||
ResourceNotFoundException, OrmException, EmailException,
|
ResourceNotFoundException, OrmException, EmailException,
|
||||||
MethodNotAllowedException {
|
MethodNotAllowedException, IOException {
|
||||||
if (self.get() != rsrc.getUser()
|
if (self.get() != rsrc.getUser()
|
||||||
&& !self.get().getCapabilities().canModifyAccount()) {
|
&& !self.get().getCapabilities().canModifyAccount()) {
|
||||||
throw new AuthException("not allowed to add email address");
|
throw new AuthException("not allowed to add email address");
|
||||||
|
@ -104,7 +106,7 @@ public class CreateEmail implements RestModifyView<AccountResource, EmailInput>
|
||||||
public Response<EmailInfo> apply(IdentifiedUser user, EmailInput input)
|
public Response<EmailInfo> apply(IdentifiedUser user, EmailInput input)
|
||||||
throws AuthException, BadRequestException, ResourceConflictException,
|
throws AuthException, BadRequestException, ResourceConflictException,
|
||||||
ResourceNotFoundException, OrmException, EmailException,
|
ResourceNotFoundException, OrmException, EmailException,
|
||||||
MethodNotAllowedException {
|
MethodNotAllowedException, IOException {
|
||||||
if (input.email != null && !email.equals(input.email)) {
|
if (input.email != null && !email.equals(input.email)) {
|
||||||
throw new BadRequestException("email address must match URL");
|
throw new BadRequestException("email address must match URL");
|
||||||
}
|
}
|
||||||
|
|
|
@ -27,6 +27,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
|
||||||
@RequiresCapability(GlobalCapability.MODIFY_ACCOUNT)
|
@RequiresCapability(GlobalCapability.MODIFY_ACCOUNT)
|
||||||
|
@ -46,7 +47,7 @@ public class DeleteActive implements RestModifyView<AccountResource, Input> {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Response<?> apply(AccountResource rsrc, Input input)
|
public Response<?> apply(AccountResource rsrc, Input input)
|
||||||
throws ResourceNotFoundException, OrmException {
|
throws ResourceNotFoundException, OrmException, IOException {
|
||||||
Account a = dbProvider.get().accounts().get(rsrc.getUser().getAccountId());
|
Account a = dbProvider.get().accounts().get(rsrc.getUser().getAccountId());
|
||||||
if (a == null) {
|
if (a == null) {
|
||||||
throw new ResourceNotFoundException("account not found");
|
throw new ResourceNotFoundException("account not found");
|
||||||
|
|
|
@ -31,6 +31,8 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
@Singleton
|
@Singleton
|
||||||
public class DeleteEmail implements RestModifyView<AccountResource.Email, Input> {
|
public class DeleteEmail implements RestModifyView<AccountResource.Email, Input> {
|
||||||
public static class Input {
|
public static class Input {
|
||||||
|
@ -53,7 +55,8 @@ public class DeleteEmail implements RestModifyView<AccountResource.Email, Input>
|
||||||
@Override
|
@Override
|
||||||
public Response<?> apply(AccountResource.Email rsrc, Input input)
|
public Response<?> apply(AccountResource.Email rsrc, Input input)
|
||||||
throws AuthException, ResourceNotFoundException,
|
throws AuthException, ResourceNotFoundException,
|
||||||
ResourceConflictException, MethodNotAllowedException, OrmException {
|
ResourceConflictException, MethodNotAllowedException, OrmException,
|
||||||
|
IOException {
|
||||||
if (self.get() != rsrc.getUser()
|
if (self.get() != rsrc.getUser()
|
||||||
&& !self.get().getCapabilities().canModifyAccount()) {
|
&& !self.get().getCapabilities().canModifyAccount()) {
|
||||||
throw new AuthException("not allowed to delete email address");
|
throw new AuthException("not allowed to delete email address");
|
||||||
|
@ -63,7 +66,7 @@ public class DeleteEmail implements RestModifyView<AccountResource.Email, Input>
|
||||||
|
|
||||||
public Response<?> apply(IdentifiedUser user, String email)
|
public Response<?> apply(IdentifiedUser user, String email)
|
||||||
throws ResourceNotFoundException, ResourceConflictException,
|
throws ResourceNotFoundException, ResourceConflictException,
|
||||||
MethodNotAllowedException, OrmException {
|
MethodNotAllowedException, OrmException, IOException {
|
||||||
if (!realm.allowsEdit(FieldName.REGISTER_NEW_EMAIL)) {
|
if (!realm.allowsEdit(FieldName.REGISTER_NEW_EMAIL)) {
|
||||||
throw new MethodNotAllowedException("realm does not allow deleting emails");
|
throw new MethodNotAllowedException("realm does not allow deleting emails");
|
||||||
}
|
}
|
||||||
|
|
|
@ -27,6 +27,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
|
||||||
@RequiresCapability(GlobalCapability.MODIFY_ACCOUNT)
|
@RequiresCapability(GlobalCapability.MODIFY_ACCOUNT)
|
||||||
|
@ -46,7 +47,7 @@ public class PutActive implements RestModifyView<AccountResource, Input> {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Response<String> apply(AccountResource rsrc, Input input)
|
public Response<String> apply(AccountResource rsrc, Input input)
|
||||||
throws ResourceNotFoundException, OrmException {
|
throws ResourceNotFoundException, OrmException, IOException {
|
||||||
Account a = dbProvider.get().accounts().get(rsrc.getUser().getAccountId());
|
Account a = dbProvider.get().accounts().get(rsrc.getUser().getAccountId());
|
||||||
if (a == null) {
|
if (a == null) {
|
||||||
throw new ResourceNotFoundException("account not found");
|
throw new ResourceNotFoundException("account not found");
|
||||||
|
|
|
@ -34,6 +34,7 @@ import com.google.inject.Singleton;
|
||||||
|
|
||||||
import org.apache.commons.codec.binary.Base64;
|
import org.apache.commons.codec.binary.Base64;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.security.NoSuchAlgorithmException;
|
import java.security.NoSuchAlgorithmException;
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
@ -69,8 +70,9 @@ public class PutHttpPassword implements RestModifyView<AccountResource, Input> {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Response<String> apply(AccountResource rsrc, Input input) throws AuthException,
|
public Response<String> apply(AccountResource rsrc, Input input)
|
||||||
ResourceNotFoundException, ResourceConflictException, OrmException {
|
throws AuthException, ResourceNotFoundException,
|
||||||
|
ResourceConflictException, OrmException, IOException {
|
||||||
if (input == null) {
|
if (input == null) {
|
||||||
input = new Input();
|
input = new Input();
|
||||||
}
|
}
|
||||||
|
@ -101,7 +103,8 @@ public class PutHttpPassword implements RestModifyView<AccountResource, Input> {
|
||||||
}
|
}
|
||||||
|
|
||||||
public Response<String> apply(IdentifiedUser user, String newPassword)
|
public Response<String> apply(IdentifiedUser user, String newPassword)
|
||||||
throws ResourceNotFoundException, ResourceConflictException, OrmException {
|
throws ResourceNotFoundException, ResourceConflictException, OrmException,
|
||||||
|
IOException {
|
||||||
if (user.getUserName() == null) {
|
if (user.getUserName() == null) {
|
||||||
throw new ResourceConflictException("username must be set");
|
throw new ResourceConflictException("username must be set");
|
||||||
}
|
}
|
||||||
|
|
|
@ -36,6 +36,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
|
||||||
@Singleton
|
@Singleton
|
||||||
|
@ -62,7 +63,7 @@ public class PutName implements RestModifyView<AccountResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public Response<String> apply(AccountResource rsrc, Input input)
|
public Response<String> apply(AccountResource rsrc, Input input)
|
||||||
throws AuthException, MethodNotAllowedException,
|
throws AuthException, MethodNotAllowedException,
|
||||||
ResourceNotFoundException, OrmException {
|
ResourceNotFoundException, OrmException, IOException {
|
||||||
if (self.get() != rsrc.getUser()
|
if (self.get() != rsrc.getUser()
|
||||||
&& !self.get().getCapabilities().canModifyAccount()) {
|
&& !self.get().getCapabilities().canModifyAccount()) {
|
||||||
throw new AuthException("not allowed to change name");
|
throw new AuthException("not allowed to change name");
|
||||||
|
@ -71,7 +72,8 @@ public class PutName implements RestModifyView<AccountResource, Input> {
|
||||||
}
|
}
|
||||||
|
|
||||||
public Response<String> apply(IdentifiedUser user, Input input)
|
public Response<String> apply(IdentifiedUser user, Input input)
|
||||||
throws MethodNotAllowedException, ResourceNotFoundException, OrmException {
|
throws MethodNotAllowedException, ResourceNotFoundException, OrmException,
|
||||||
|
IOException {
|
||||||
if (input == null) {
|
if (input == null) {
|
||||||
input = new Input();
|
input = new Input();
|
||||||
}
|
}
|
||||||
|
|
|
@ -28,6 +28,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
|
||||||
@Singleton
|
@Singleton
|
||||||
|
@ -50,7 +51,8 @@ public class PutPreferred implements
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Response<String> apply(AccountResource.Email rsrc, Input input)
|
public Response<String> apply(AccountResource.Email rsrc, Input input)
|
||||||
throws AuthException, ResourceNotFoundException, OrmException {
|
throws AuthException, ResourceNotFoundException, OrmException,
|
||||||
|
IOException {
|
||||||
if (self.get() != rsrc.getUser()
|
if (self.get() != rsrc.getUser()
|
||||||
&& !self.get().getCapabilities().canModifyAccount()) {
|
&& !self.get().getCapabilities().canModifyAccount()) {
|
||||||
throw new AuthException("not allowed to set preferred email address");
|
throw new AuthException("not allowed to set preferred email address");
|
||||||
|
@ -59,7 +61,7 @@ public class PutPreferred implements
|
||||||
}
|
}
|
||||||
|
|
||||||
public Response<String> apply(IdentifiedUser user, String email)
|
public Response<String> apply(IdentifiedUser user, String email)
|
||||||
throws ResourceNotFoundException, OrmException {
|
throws ResourceNotFoundException, OrmException, IOException {
|
||||||
Account a = dbProvider.get().accounts().get(user.getAccountId());
|
Account a = dbProvider.get().accounts().get(user.getAccountId());
|
||||||
if (a == null) {
|
if (a == null) {
|
||||||
throw new ResourceNotFoundException("account not found");
|
throw new ResourceNotFoundException("account not found");
|
||||||
|
|
|
@ -30,6 +30,8 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
@Singleton
|
@Singleton
|
||||||
public class PutUsername implements RestModifyView<AccountResource, Input> {
|
public class PutUsername implements RestModifyView<AccountResource, Input> {
|
||||||
public static class Input {
|
public static class Input {
|
||||||
|
@ -56,7 +58,7 @@ public class PutUsername implements RestModifyView<AccountResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public String apply(AccountResource rsrc, Input input) throws AuthException,
|
public String apply(AccountResource rsrc, Input input) throws AuthException,
|
||||||
MethodNotAllowedException, UnprocessableEntityException,
|
MethodNotAllowedException, UnprocessableEntityException,
|
||||||
ResourceConflictException, OrmException {
|
ResourceConflictException, OrmException, IOException {
|
||||||
if (self.get() != rsrc.getUser()
|
if (self.get() != rsrc.getUser()
|
||||||
&& !self.get().getCapabilities().canAdministrateServer()) {
|
&& !self.get().getCapabilities().canAdministrateServer()) {
|
||||||
throw new AuthException("not allowed to set username");
|
throw new AuthException("not allowed to set username");
|
||||||
|
|
|
@ -311,7 +311,7 @@ public class AccountApiImpl implements AccountApi {
|
||||||
new AccountResource.Email(account.getUser(), input.email);
|
new AccountResource.Email(account.getUser(), input.email);
|
||||||
try {
|
try {
|
||||||
createEmailFactory.create(input.email).apply(rsrc, input);
|
createEmailFactory.create(input.email).apply(rsrc, input);
|
||||||
} catch (EmailException | OrmException e) {
|
} catch (EmailException | OrmException | IOException e) {
|
||||||
throw new RestApiException("Cannot add email", e);
|
throw new RestApiException("Cannot add email", e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -44,6 +44,7 @@ import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.inject.assistedinject.Assisted;
|
import com.google.inject.assistedinject.Assisted;
|
||||||
import com.google.inject.assistedinject.AssistedInject;
|
import com.google.inject.assistedinject.AssistedInject;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -217,7 +218,7 @@ class GroupApiImpl implements GroupApi {
|
||||||
try {
|
try {
|
||||||
addMembers.apply(
|
addMembers.apply(
|
||||||
rsrc, AddMembers.Input.fromMembers(Arrays.asList(members)));
|
rsrc, AddMembers.Input.fromMembers(Arrays.asList(members)));
|
||||||
} catch (OrmException e) {
|
} catch (OrmException | IOException e) {
|
||||||
throw new RestApiException("Cannot add group members", e);
|
throw new RestApiException("Cannot add group members", e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -227,7 +228,7 @@ class GroupApiImpl implements GroupApi {
|
||||||
try {
|
try {
|
||||||
deleteMembers.apply(
|
deleteMembers.apply(
|
||||||
rsrc, AddMembers.Input.fromMembers(Arrays.asList(members)));
|
rsrc, AddMembers.Input.fromMembers(Arrays.asList(members)));
|
||||||
} catch (OrmException e) {
|
} catch (OrmException | IOException e) {
|
||||||
throw new RestApiException("Cannot remove group members", e);
|
throw new RestApiException("Cannot remove group members", e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -90,7 +90,7 @@ class GroupsImpl implements Groups {
|
||||||
GroupInfo info = createGroup.create(in.name)
|
GroupInfo info = createGroup.create(in.name)
|
||||||
.apply(TopLevelResource.INSTANCE, in);
|
.apply(TopLevelResource.INSTANCE, in);
|
||||||
return id(info.id);
|
return id(info.id);
|
||||||
} catch (OrmException e) {
|
} catch (OrmException | IOException e) {
|
||||||
throw new RestApiException("Cannot create group " + in.name, e);
|
throw new RestApiException("Cannot create group " + in.name, e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -32,6 +32,8 @@ import org.kohsuke.args4j.spi.OptionHandler;
|
||||||
import org.kohsuke.args4j.spi.Parameters;
|
import org.kohsuke.args4j.spi.Parameters;
|
||||||
import org.kohsuke.args4j.spi.Setter;
|
import org.kohsuke.args4j.spi.Setter;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
public class AccountIdHandler extends OptionHandler<Account.Id> {
|
public class AccountIdHandler extends OptionHandler<Account.Id> {
|
||||||
private final AccountResolver accountResolver;
|
private final AccountResolver accountResolver;
|
||||||
private final AccountManager accountManager;
|
private final AccountManager accountManager;
|
||||||
|
@ -76,7 +78,7 @@ public class AccountIdHandler extends OptionHandler<Account.Id> {
|
||||||
throw new CmdLineException(owner, "user \"" + token + "\" not found");
|
throw new CmdLineException(owner, "user \"" + token + "\" not found");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} catch (OrmException e) {
|
} catch (OrmException | IOException e) {
|
||||||
throw new CmdLineException(owner, "database is down");
|
throw new CmdLineException(owner, "database is down");
|
||||||
}
|
}
|
||||||
setter.addValue(accountId);
|
setter.addValue(accountId);
|
||||||
|
@ -84,7 +86,7 @@ public class AccountIdHandler extends OptionHandler<Account.Id> {
|
||||||
}
|
}
|
||||||
|
|
||||||
private Account.Id createAccountByLdap(String user)
|
private Account.Id createAccountByLdap(String user)
|
||||||
throws CmdLineException {
|
throws CmdLineException, IOException {
|
||||||
if (!user.matches(Account.USER_NAME_PATTERN)) {
|
if (!user.matches(Account.USER_NAME_PATTERN)) {
|
||||||
throw new CmdLineException(owner, "user \"" + user + "\" not found");
|
throw new CmdLineException(owner, "user \"" + user + "\" not found");
|
||||||
}
|
}
|
||||||
|
|
|
@ -30,6 +30,8 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
@Singleton
|
@Singleton
|
||||||
public class ConfirmEmail implements RestModifyView<ConfigResource, Input> {
|
public class ConfirmEmail implements RestModifyView<ConfigResource, Input> {
|
||||||
public static class Input {
|
public static class Input {
|
||||||
|
@ -53,7 +55,7 @@ public class ConfirmEmail implements RestModifyView<ConfigResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public Response<?> apply(ConfigResource rsrc, Input input)
|
public Response<?> apply(ConfigResource rsrc, Input input)
|
||||||
throws AuthException, UnprocessableEntityException, AccountException,
|
throws AuthException, UnprocessableEntityException, AccountException,
|
||||||
OrmException {
|
OrmException, IOException {
|
||||||
CurrentUser user = self.get();
|
CurrentUser user = self.get();
|
||||||
if (!user.isIdentifiedUser()) {
|
if (!user.isIdentifiedUser()) {
|
||||||
throw new AuthException("Authentication required");
|
throw new AuthException("Authentication required");
|
||||||
|
|
|
@ -45,6 +45,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
|
@ -114,7 +115,7 @@ public class AddMembers implements RestModifyView<GroupResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public List<AccountInfo> apply(GroupResource resource, Input input)
|
public List<AccountInfo> apply(GroupResource resource, Input input)
|
||||||
throws AuthException, MethodNotAllowedException,
|
throws AuthException, MethodNotAllowedException,
|
||||||
UnprocessableEntityException, OrmException {
|
UnprocessableEntityException, OrmException, IOException {
|
||||||
AccountGroup internalGroup = resource.toAccountGroup();
|
AccountGroup internalGroup = resource.toAccountGroup();
|
||||||
if (internalGroup == null) {
|
if (internalGroup == null) {
|
||||||
throw new MethodNotAllowedException();
|
throw new MethodNotAllowedException();
|
||||||
|
@ -142,7 +143,7 @@ public class AddMembers implements RestModifyView<GroupResource, Input> {
|
||||||
}
|
}
|
||||||
|
|
||||||
private Account findAccount(String nameOrEmail) throws AuthException,
|
private Account findAccount(String nameOrEmail) throws AuthException,
|
||||||
UnprocessableEntityException, OrmException {
|
UnprocessableEntityException, OrmException, IOException {
|
||||||
try {
|
try {
|
||||||
return accounts.parse(nameOrEmail).getAccount();
|
return accounts.parse(nameOrEmail).getAccount();
|
||||||
} catch (UnprocessableEntityException e) {
|
} catch (UnprocessableEntityException e) {
|
||||||
|
@ -174,7 +175,8 @@ public class AddMembers implements RestModifyView<GroupResource, Input> {
|
||||||
}
|
}
|
||||||
|
|
||||||
public void addMembers(AccountGroup.Id groupId,
|
public void addMembers(AccountGroup.Id groupId,
|
||||||
Collection<? extends Account.Id> newMemberIds) throws OrmException {
|
Collection<? extends Account.Id> newMemberIds)
|
||||||
|
throws OrmException, IOException {
|
||||||
Map<Account.Id, AccountGroupMember> newAccountGroupMembers = new HashMap<>();
|
Map<Account.Id, AccountGroupMember> newAccountGroupMembers = new HashMap<>();
|
||||||
for (Account.Id accId : newMemberIds) {
|
for (Account.Id accId : newMemberIds) {
|
||||||
if (!newAccountGroupMembers.containsKey(accId)) {
|
if (!newAccountGroupMembers.containsKey(accId)) {
|
||||||
|
@ -197,7 +199,7 @@ public class AddMembers implements RestModifyView<GroupResource, Input> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private Account createAccountByLdap(String user) {
|
private Account createAccountByLdap(String user) throws IOException {
|
||||||
if (!user.matches(Account.USER_NAME_PATTERN)) {
|
if (!user.matches(Account.USER_NAME_PATTERN)) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
@ -238,7 +240,7 @@ public class AddMembers implements RestModifyView<GroupResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public AccountInfo apply(GroupResource resource, PutMember.Input input)
|
public AccountInfo apply(GroupResource resource, PutMember.Input input)
|
||||||
throws AuthException, MethodNotAllowedException,
|
throws AuthException, MethodNotAllowedException,
|
||||||
ResourceNotFoundException, OrmException {
|
ResourceNotFoundException, OrmException, IOException {
|
||||||
AddMembers.Input in = new AddMembers.Input();
|
AddMembers.Input in = new AddMembers.Input();
|
||||||
in._oneMember = id;
|
in._oneMember = id;
|
||||||
try {
|
try {
|
||||||
|
|
|
@ -50,6 +50,7 @@ import com.google.inject.assistedinject.Assisted;
|
||||||
import org.eclipse.jgit.lib.Config;
|
import org.eclipse.jgit.lib.Config;
|
||||||
import org.eclipse.jgit.lib.PersonIdent;
|
import org.eclipse.jgit.lib.PersonIdent;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Locale;
|
import java.util.Locale;
|
||||||
|
@ -98,7 +99,7 @@ public class CreateGroup implements RestModifyView<TopLevelResource, GroupInput>
|
||||||
@Override
|
@Override
|
||||||
public GroupInfo apply(TopLevelResource resource, GroupInput input)
|
public GroupInfo apply(TopLevelResource resource, GroupInput input)
|
||||||
throws BadRequestException, UnprocessableEntityException,
|
throws BadRequestException, UnprocessableEntityException,
|
||||||
ResourceConflictException, OrmException {
|
ResourceConflictException, OrmException, IOException {
|
||||||
if (input == null) {
|
if (input == null) {
|
||||||
input = new GroupInput();
|
input = new GroupInput();
|
||||||
}
|
}
|
||||||
|
@ -138,7 +139,7 @@ public class CreateGroup implements RestModifyView<TopLevelResource, GroupInput>
|
||||||
}
|
}
|
||||||
|
|
||||||
private AccountGroup createGroup(CreateGroupArgs createGroupArgs)
|
private AccountGroup createGroup(CreateGroupArgs createGroupArgs)
|
||||||
throws OrmException, ResourceConflictException {
|
throws OrmException, ResourceConflictException, IOException {
|
||||||
|
|
||||||
// Do not allow creating groups with the same name as system groups
|
// Do not allow creating groups with the same name as system groups
|
||||||
List<String> sysGroupNames = SystemGroupBackend.getNames();
|
List<String> sysGroupNames = SystemGroupBackend.getNames();
|
||||||
|
|
|
@ -34,6 +34,7 @@ import com.google.inject.Inject;
|
||||||
import com.google.inject.Provider;
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
@ -62,7 +63,7 @@ public class DeleteMembers implements RestModifyView<GroupResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public Response<?> apply(GroupResource resource, Input input)
|
public Response<?> apply(GroupResource resource, Input input)
|
||||||
throws AuthException, MethodNotAllowedException,
|
throws AuthException, MethodNotAllowedException,
|
||||||
UnprocessableEntityException, OrmException {
|
UnprocessableEntityException, OrmException, IOException {
|
||||||
AccountGroup internalGroup = resource.toAccountGroup();
|
AccountGroup internalGroup = resource.toAccountGroup();
|
||||||
if (internalGroup == null) {
|
if (internalGroup == null) {
|
||||||
throw new MethodNotAllowedException();
|
throw new MethodNotAllowedException();
|
||||||
|
@ -125,7 +126,7 @@ public class DeleteMembers implements RestModifyView<GroupResource, Input> {
|
||||||
@Override
|
@Override
|
||||||
public Response<?> apply(MemberResource resource, Input input)
|
public Response<?> apply(MemberResource resource, Input input)
|
||||||
throws AuthException, MethodNotAllowedException,
|
throws AuthException, MethodNotAllowedException,
|
||||||
UnprocessableEntityException, OrmException {
|
UnprocessableEntityException, OrmException, IOException {
|
||||||
AddMembers.Input in = new AddMembers.Input();
|
AddMembers.Input in = new AddMembers.Input();
|
||||||
in._oneMember = resource.getMember().getAccountId().toString();
|
in._oneMember = resource.getMember().getAccountId().toString();
|
||||||
return delete.get().apply(resource, in);
|
return delete.get().apply(resource, in);
|
||||||
|
|
|
@ -14,6 +14,8 @@
|
||||||
|
|
||||||
package com.google.gerrit.server.index;
|
package com.google.gerrit.server.index;
|
||||||
|
|
||||||
|
import com.google.gerrit.server.account.AccountState;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndex;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndex;
|
import com.google.gerrit.server.index.change.ChangeIndex;
|
||||||
import com.google.gerrit.server.index.change.DummyChangeIndex;
|
import com.google.gerrit.server.index.change.DummyChangeIndex;
|
||||||
import com.google.gerrit.server.query.change.ChangeData;
|
import com.google.gerrit.server.query.change.ChangeData;
|
||||||
|
@ -27,11 +29,19 @@ public class DummyIndexModule extends AbstractModule {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private static class DummyAccountIndexFactory implements AccountIndex.Factory {
|
||||||
|
@Override
|
||||||
|
public AccountIndex create(Schema<AccountState> schema) {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void configure() {
|
protected void configure() {
|
||||||
install(new IndexModule(1));
|
install(new IndexModule(1));
|
||||||
bind(IndexConfig.class).toInstance(IndexConfig.createDefault());
|
bind(IndexConfig.class).toInstance(IndexConfig.createDefault());
|
||||||
bind(Index.class).toInstance(new DummyChangeIndex());
|
bind(Index.class).toInstance(new DummyChangeIndex());
|
||||||
|
bind(AccountIndex.Factory.class).toInstance(new DummyAccountIndexFactory());
|
||||||
bind(ChangeIndex.Factory.class).toInstance(new DummyChangeIndexFactory());
|
bind(ChangeIndex.Factory.class).toInstance(new DummyChangeIndexFactory());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -26,11 +26,15 @@ import com.google.common.util.concurrent.MoreExecutors;
|
||||||
import com.google.gerrit.lifecycle.LifecycleModule;
|
import com.google.gerrit.lifecycle.LifecycleModule;
|
||||||
import com.google.gerrit.server.config.GerritServerConfig;
|
import com.google.gerrit.server.config.GerritServerConfig;
|
||||||
import com.google.gerrit.server.git.WorkQueue;
|
import com.google.gerrit.server.git.WorkQueue;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndexCollection;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndexDefinition;
|
||||||
|
import com.google.gerrit.server.index.account.AccountIndexRewriter;
|
||||||
|
import com.google.gerrit.server.index.account.AccountSchemaDefinitions;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndexDefinition;
|
import com.google.gerrit.server.index.change.ChangeIndexDefinition;
|
||||||
|
import com.google.gerrit.server.index.change.ChangeIndexRewriter;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndexer;
|
import com.google.gerrit.server.index.change.ChangeIndexer;
|
||||||
import com.google.gerrit.server.index.change.ChangeSchemaDefinitions;
|
import com.google.gerrit.server.index.change.ChangeSchemaDefinitions;
|
||||||
import com.google.gerrit.server.index.change.IndexRewriter;
|
|
||||||
import com.google.inject.Injector;
|
import com.google.inject.Injector;
|
||||||
import com.google.inject.Key;
|
import com.google.inject.Key;
|
||||||
import com.google.inject.Provides;
|
import com.google.inject.Provides;
|
||||||
|
@ -55,6 +59,7 @@ public class IndexModule extends LifecycleModule {
|
||||||
|
|
||||||
public static final ImmutableCollection<SchemaDefinitions<?>> ALL_SCHEMA_DEFS =
|
public static final ImmutableCollection<SchemaDefinitions<?>> ALL_SCHEMA_DEFS =
|
||||||
ImmutableList.<SchemaDefinitions<?>> of(
|
ImmutableList.<SchemaDefinitions<?>> of(
|
||||||
|
AccountSchemaDefinitions.INSTANCE,
|
||||||
ChangeSchemaDefinitions.INSTANCE);
|
ChangeSchemaDefinitions.INSTANCE);
|
||||||
|
|
||||||
/** Type of secondary index. */
|
/** Type of secondary index. */
|
||||||
|
@ -83,7 +88,11 @@ public class IndexModule extends LifecycleModule {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void configure() {
|
protected void configure() {
|
||||||
bind(IndexRewriter.class);
|
bind(AccountIndexRewriter.class);
|
||||||
|
bind(AccountIndexCollection.class);
|
||||||
|
listener().to(AccountIndexCollection.class);
|
||||||
|
|
||||||
|
bind(ChangeIndexRewriter.class);
|
||||||
bind(ChangeIndexCollection.class);
|
bind(ChangeIndexCollection.class);
|
||||||
listener().to(ChangeIndexCollection.class);
|
listener().to(ChangeIndexCollection.class);
|
||||||
factory(ChangeIndexer.Factory.class);
|
factory(ChangeIndexer.Factory.class);
|
||||||
|
@ -91,9 +100,12 @@ public class IndexModule extends LifecycleModule {
|
||||||
|
|
||||||
@Provides
|
@Provides
|
||||||
Collection<IndexDefinition<?, ?, ?>> getIndexDefinitions(
|
Collection<IndexDefinition<?, ?, ?>> getIndexDefinitions(
|
||||||
|
AccountIndexDefinition accounts,
|
||||||
ChangeIndexDefinition changes) {
|
ChangeIndexDefinition changes) {
|
||||||
Collection<IndexDefinition<?, ?, ?>> result =
|
Collection<IndexDefinition<?, ?, ?>> result =
|
||||||
ImmutableList.<IndexDefinition<?, ?, ?>> of(changes);
|
ImmutableList.<IndexDefinition<?, ?, ?>> of(
|
||||||
|
accounts,
|
||||||
|
changes);
|
||||||
Set<String> expected = FluentIterable.from(ALL_SCHEMA_DEFS)
|
Set<String> expected = FluentIterable.from(ALL_SCHEMA_DEFS)
|
||||||
.transform(new Function<SchemaDefinitions<?>, String>() {
|
.transform(new Function<SchemaDefinitions<?>, String>() {
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -0,0 +1,24 @@
|
||||||
|
// 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.index;
|
||||||
|
|
||||||
|
import com.google.gerrit.server.query.Predicate;
|
||||||
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
|
||||||
|
public interface IndexRewriter<T> {
|
||||||
|
|
||||||
|
Predicate<T> rewrite(Predicate<T> in, QueryOptions opts)
|
||||||
|
throws QueryParseException;
|
||||||
|
}
|
|
@ -0,0 +1,183 @@
|
||||||
|
// 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.index;
|
||||||
|
|
||||||
|
import com.google.common.base.Function;
|
||||||
|
import com.google.common.base.MoreObjects;
|
||||||
|
import com.google.common.collect.ImmutableList;
|
||||||
|
import com.google.common.collect.Iterables;
|
||||||
|
import com.google.gerrit.server.query.DataSource;
|
||||||
|
import com.google.gerrit.server.query.Paginated;
|
||||||
|
import com.google.gerrit.server.query.Predicate;
|
||||||
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
import com.google.gwtorm.server.ResultSet;
|
||||||
|
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Wrapper combining an {@link IndexPredicate} together with a
|
||||||
|
* {@link DataSource} that returns matching results from the index.
|
||||||
|
* <p>
|
||||||
|
* Appropriate to return as the rootmost predicate that can be processed using
|
||||||
|
* the secondary index; such predicates must also implement {@link DataSource}
|
||||||
|
* to be chosen by the query processor.
|
||||||
|
*
|
||||||
|
* @param <I> The type of the IDs by which the entities are stored in the index.
|
||||||
|
* @param <T> The type of the entities that are stored in the index.
|
||||||
|
*/
|
||||||
|
public class IndexedQuery<I, T> extends Predicate<T>
|
||||||
|
implements DataSource<T>, Paginated<T> {
|
||||||
|
protected final Index<I, T> index;
|
||||||
|
|
||||||
|
private QueryOptions opts;
|
||||||
|
private final Predicate<T> pred;
|
||||||
|
private DataSource<T> source;
|
||||||
|
private final Map<T, DataSource<T>> fromSource;
|
||||||
|
|
||||||
|
public IndexedQuery(Index<I, T> index, Predicate<T> pred,
|
||||||
|
QueryOptions opts) throws QueryParseException {
|
||||||
|
this.index = index;
|
||||||
|
this.opts = opts;
|
||||||
|
this.pred = pred;
|
||||||
|
this.source = index.getSource(pred, this.opts);
|
||||||
|
this.fromSource = new HashMap<>();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int getChildCount() {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Predicate<T> getChild(int i) {
|
||||||
|
if (i == 0) {
|
||||||
|
return pred;
|
||||||
|
}
|
||||||
|
throw new ArrayIndexOutOfBoundsException(i);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<Predicate<T>> getChildren() {
|
||||||
|
return ImmutableList.of(pred);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public QueryOptions getOptions() {
|
||||||
|
return opts;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int getCardinality() {
|
||||||
|
return source != null ? source.getCardinality() : opts.limit();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ResultSet<T> read() throws OrmException {
|
||||||
|
final DataSource<T> currSource = source;
|
||||||
|
final ResultSet<T> rs = currSource.read();
|
||||||
|
|
||||||
|
return new ResultSet<T>() {
|
||||||
|
@Override
|
||||||
|
public Iterator<T> iterator() {
|
||||||
|
return Iterables.transform(
|
||||||
|
rs,
|
||||||
|
new Function<T, T>() {
|
||||||
|
@Override
|
||||||
|
public
|
||||||
|
T apply(T t) {
|
||||||
|
fromSource.put(t, currSource);
|
||||||
|
return t;
|
||||||
|
}
|
||||||
|
}).iterator();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<T> toList() {
|
||||||
|
List<T> r = rs.toList();
|
||||||
|
for (T t : r) {
|
||||||
|
fromSource.put(t, currSource);
|
||||||
|
}
|
||||||
|
return r;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close() {
|
||||||
|
rs.close();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ResultSet<T> restart(int start) throws OrmException {
|
||||||
|
opts = opts.withStart(start);
|
||||||
|
try {
|
||||||
|
source = index.getSource(pred, opts);
|
||||||
|
} catch (QueryParseException e) {
|
||||||
|
// Don't need to show this exception to the user; the only thing that
|
||||||
|
// changed about pred was its start, and any other QPEs that might happen
|
||||||
|
// should have already thrown from the constructor.
|
||||||
|
throw new OrmException(e);
|
||||||
|
}
|
||||||
|
// Don't convert start to a limit, since the caller of this method (see
|
||||||
|
// AndSource) has calculated the actual number to skip.
|
||||||
|
return read();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Predicate<T> copy(Collection<? extends Predicate<T>> children) {
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean match(T t) throws OrmException {
|
||||||
|
return (source != null && fromSource.get(t) == source) || pred.match(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int getCost() {
|
||||||
|
// Index queries are assumed to be cheaper than any other type of query, so
|
||||||
|
// so try to make sure they get picked. Note that pred's cost may be higher
|
||||||
|
// because it doesn't know whether it's being used in an index query or not.
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int hashCode() {
|
||||||
|
return pred.hashCode();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean equals(Object other) {
|
||||||
|
if (other == null || getClass() != other.getClass()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
IndexedQuery<?, ?> o = (IndexedQuery<?, ?>) other;
|
||||||
|
return pred.equals(o.pred)
|
||||||
|
&& opts.equals(o.opts);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString() {
|
||||||
|
return MoreObjects.toStringHelper("index")
|
||||||
|
.add("p", pred)
|
||||||
|
.add("opts", opts)
|
||||||
|
.toString();
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,44 @@
|
||||||
|
// 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.index.account;
|
||||||
|
|
||||||
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
|
||||||
|
import com.google.gerrit.server.account.AccountState;
|
||||||
|
import com.google.gerrit.server.index.IndexRewriter;
|
||||||
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
|
import com.google.gerrit.server.query.Predicate;
|
||||||
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import com.google.inject.Singleton;
|
||||||
|
|
||||||
|
@Singleton
|
||||||
|
public class AccountIndexRewriter implements IndexRewriter<AccountState> {
|
||||||
|
|
||||||
|
private final AccountIndexCollection indexes;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
AccountIndexRewriter(AccountIndexCollection indexes) {
|
||||||
|
this.indexes = indexes;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Predicate<AccountState> rewrite(Predicate<AccountState> in,
|
||||||
|
QueryOptions opts) throws QueryParseException {
|
||||||
|
AccountIndex index = indexes.getSearchIndex();
|
||||||
|
checkNotNull(index, "no active search index configured for accounts");
|
||||||
|
return new IndexedAccountQuery(index, in, opts);
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,34 @@
|
||||||
|
// 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.index.account;
|
||||||
|
|
||||||
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
import com.google.gerrit.server.account.AccountState;
|
||||||
|
import com.google.gerrit.server.index.Index;
|
||||||
|
import com.google.gerrit.server.index.IndexedQuery;
|
||||||
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
|
import com.google.gerrit.server.query.DataSource;
|
||||||
|
import com.google.gerrit.server.query.Predicate;
|
||||||
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
|
||||||
|
public class IndexedAccountQuery extends IndexedQuery<Account.Id, AccountState>
|
||||||
|
implements DataSource<AccountState> {
|
||||||
|
|
||||||
|
public IndexedAccountQuery(Index<Account.Id, AccountState> index,
|
||||||
|
Predicate<AccountState> pred, QueryOptions opts)
|
||||||
|
throws QueryParseException {
|
||||||
|
super(index, pred, opts);
|
||||||
|
}
|
||||||
|
}
|
|
@ -20,16 +20,18 @@ import com.google.gerrit.reviewdb.client.Change;
|
||||||
import com.google.gerrit.reviewdb.client.Change.Status;
|
import com.google.gerrit.reviewdb.client.Change.Status;
|
||||||
import com.google.gerrit.server.index.IndexConfig;
|
import com.google.gerrit.server.index.IndexConfig;
|
||||||
import com.google.gerrit.server.index.IndexPredicate;
|
import com.google.gerrit.server.index.IndexPredicate;
|
||||||
|
import com.google.gerrit.server.index.IndexRewriter;
|
||||||
import com.google.gerrit.server.index.QueryOptions;
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
import com.google.gerrit.server.query.AndPredicate;
|
import com.google.gerrit.server.query.AndPredicate;
|
||||||
|
import com.google.gerrit.server.query.LimitPredicate;
|
||||||
import com.google.gerrit.server.query.NotPredicate;
|
import com.google.gerrit.server.query.NotPredicate;
|
||||||
import com.google.gerrit.server.query.OrPredicate;
|
import com.google.gerrit.server.query.OrPredicate;
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
import com.google.gerrit.server.query.change.AndSource;
|
import com.google.gerrit.server.query.change.AndSource;
|
||||||
import com.google.gerrit.server.query.change.ChangeData;
|
import com.google.gerrit.server.query.change.ChangeData;
|
||||||
|
import com.google.gerrit.server.query.change.ChangeQueryBuilder;
|
||||||
import com.google.gerrit.server.query.change.ChangeStatusPredicate;
|
import com.google.gerrit.server.query.change.ChangeStatusPredicate;
|
||||||
import com.google.gerrit.server.query.change.LimitPredicate;
|
|
||||||
import com.google.gerrit.server.query.change.OrSource;
|
import com.google.gerrit.server.query.change.OrSource;
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
@ -43,7 +45,7 @@ import java.util.Set;
|
||||||
|
|
||||||
/** Rewriter that pushes boolean logic into the secondary index. */
|
/** Rewriter that pushes boolean logic into the secondary index. */
|
||||||
@Singleton
|
@Singleton
|
||||||
public class IndexRewriter {
|
public class ChangeIndexRewriter implements IndexRewriter<ChangeData> {
|
||||||
/** Set of all open change statuses. */
|
/** Set of all open change statuses. */
|
||||||
public static final Set<Change.Status> OPEN_STATUSES;
|
public static final Set<Change.Status> OPEN_STATUSES;
|
||||||
|
|
||||||
|
@ -124,12 +126,13 @@ public class IndexRewriter {
|
||||||
private final IndexConfig config;
|
private final IndexConfig config;
|
||||||
|
|
||||||
@Inject
|
@Inject
|
||||||
IndexRewriter(ChangeIndexCollection indexes,
|
ChangeIndexRewriter(ChangeIndexCollection indexes,
|
||||||
IndexConfig config) {
|
IndexConfig config) {
|
||||||
this.indexes = indexes;
|
this.indexes = indexes;
|
||||||
this.config = config;
|
this.config = config;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
public Predicate<ChangeData> rewrite(Predicate<ChangeData> in,
|
public Predicate<ChangeData> rewrite(Predicate<ChangeData> in,
|
||||||
QueryOptions opts) throws QueryParseException {
|
QueryOptions opts) throws QueryParseException {
|
||||||
ChangeIndex index = indexes.getSearchIndex();
|
ChangeIndex index = indexes.getSearchIndex();
|
||||||
|
@ -172,7 +175,7 @@ public class IndexRewriter {
|
||||||
// Replace any limits with the limit provided by the caller. The caller
|
// Replace any limits with the limit provided by the caller. The caller
|
||||||
// should have already searched the predicate tree for limit predicates
|
// should have already searched the predicate tree for limit predicates
|
||||||
// and included that in their limit computation.
|
// and included that in their limit computation.
|
||||||
return new LimitPredicate(opts.limit());
|
return new LimitPredicate<>(ChangeQueryBuilder.FIELD_LIMIT, opts.limit());
|
||||||
} else if (!isRewritePossible(in)) {
|
} else if (!isRewritePossible(in)) {
|
||||||
return null; // magic to indicate "in" cannot be rewritten
|
return null; // magic to indicate "in" cannot be rewritten
|
||||||
}
|
}
|
|
@ -18,28 +18,19 @@ import static com.google.gerrit.server.index.change.ChangeField.CHANGE;
|
||||||
import static com.google.gerrit.server.index.change.ChangeField.PROJECT;
|
import static com.google.gerrit.server.index.change.ChangeField.PROJECT;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
import com.google.common.base.Function;
|
|
||||||
import com.google.common.base.MoreObjects;
|
|
||||||
import com.google.common.collect.ImmutableList;
|
|
||||||
import com.google.common.collect.ImmutableSet;
|
import com.google.common.collect.ImmutableSet;
|
||||||
import com.google.common.collect.Iterables;
|
|
||||||
import com.google.common.primitives.Ints;
|
import com.google.common.primitives.Ints;
|
||||||
|
import com.google.gerrit.reviewdb.client.Change;
|
||||||
import com.google.gerrit.server.index.IndexConfig;
|
import com.google.gerrit.server.index.IndexConfig;
|
||||||
import com.google.gerrit.server.index.IndexPredicate;
|
import com.google.gerrit.server.index.IndexPredicate;
|
||||||
|
import com.google.gerrit.server.index.IndexedQuery;
|
||||||
import com.google.gerrit.server.index.QueryOptions;
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
import com.google.gerrit.server.query.DataSource;
|
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
import com.google.gerrit.server.query.change.ChangeData;
|
import com.google.gerrit.server.query.change.ChangeData;
|
||||||
import com.google.gerrit.server.query.change.ChangeDataSource;
|
import com.google.gerrit.server.query.change.ChangeDataSource;
|
||||||
import com.google.gerrit.server.query.change.Paginated;
|
|
||||||
import com.google.gwtorm.server.OrmException;
|
|
||||||
import com.google.gwtorm.server.ResultSet;
|
|
||||||
|
|
||||||
import java.util.Collection;
|
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
import java.util.Iterator;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -50,8 +41,8 @@ import java.util.Set;
|
||||||
* the secondary index; such predicates must also implement
|
* the secondary index; such predicates must also implement
|
||||||
* {@link ChangeDataSource} to be chosen by the query processor.
|
* {@link ChangeDataSource} to be chosen by the query processor.
|
||||||
*/
|
*/
|
||||||
public class IndexedChangeQuery extends Predicate<ChangeData>
|
public class IndexedChangeQuery extends IndexedQuery<Change.Id, ChangeData>
|
||||||
implements ChangeDataSource, Paginated {
|
implements ChangeDataSource {
|
||||||
public static QueryOptions oneResult() {
|
public static QueryOptions oneResult() {
|
||||||
return createOptions(IndexConfig.createDefault(), 0, 1,
|
return createOptions(IndexConfig.createDefault(), 0, 1,
|
||||||
ImmutableSet.<String> of());
|
ImmutableSet.<String> of());
|
||||||
|
@ -78,144 +69,13 @@ public class IndexedChangeQuery extends Predicate<ChangeData>
|
||||||
return IndexedChangeQuery.createOptions(opts.config(), 0, limit, opts.fields());
|
return IndexedChangeQuery.createOptions(opts.config(), 0, limit, opts.fields());
|
||||||
}
|
}
|
||||||
|
|
||||||
private final ChangeIndex index;
|
|
||||||
|
|
||||||
private QueryOptions opts;
|
|
||||||
private Predicate<ChangeData> pred;
|
|
||||||
private DataSource<ChangeData> source;
|
|
||||||
|
|
||||||
public IndexedChangeQuery(ChangeIndex index, Predicate<ChangeData> pred,
|
public IndexedChangeQuery(ChangeIndex index, Predicate<ChangeData> pred,
|
||||||
QueryOptions opts) throws QueryParseException {
|
QueryOptions opts) throws QueryParseException {
|
||||||
this.index = index;
|
super(index, pred, convertOptions(opts));
|
||||||
this.opts = convertOptions(opts);
|
|
||||||
this.pred = pred;
|
|
||||||
this.source = index.getSource(pred, this.opts);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public int getChildCount() {
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Predicate<ChangeData> getChild(int i) {
|
|
||||||
if (i == 0) {
|
|
||||||
return pred;
|
|
||||||
}
|
|
||||||
throw new ArrayIndexOutOfBoundsException(i);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<Predicate<ChangeData>> getChildren() {
|
|
||||||
return ImmutableList.of(pred);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public QueryOptions getOptions() {
|
|
||||||
return opts;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public int getCardinality() {
|
|
||||||
return source != null ? source.getCardinality() : opts.limit();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean hasChange() {
|
public boolean hasChange() {
|
||||||
return index.getSchema().hasField(ChangeField.CHANGE);
|
return index.getSchema().hasField(ChangeField.CHANGE);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public ResultSet<ChangeData> read() throws OrmException {
|
|
||||||
final DataSource<ChangeData> currSource = source;
|
|
||||||
final ResultSet<ChangeData> rs = currSource.read();
|
|
||||||
|
|
||||||
return new ResultSet<ChangeData>() {
|
|
||||||
@Override
|
|
||||||
public Iterator<ChangeData> iterator() {
|
|
||||||
return Iterables.transform(
|
|
||||||
rs,
|
|
||||||
new Function<ChangeData, ChangeData>() {
|
|
||||||
@Override
|
|
||||||
public
|
|
||||||
ChangeData apply(ChangeData input) {
|
|
||||||
input.cacheFromSource(currSource);
|
|
||||||
return input;
|
|
||||||
}
|
|
||||||
}).iterator();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<ChangeData> toList() {
|
|
||||||
List<ChangeData> r = rs.toList();
|
|
||||||
for (ChangeData cd : r) {
|
|
||||||
cd.cacheFromSource(currSource);
|
|
||||||
}
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void close() {
|
|
||||||
rs.close();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public ResultSet<ChangeData> restart(int start) throws OrmException {
|
|
||||||
opts = opts.withStart(start);
|
|
||||||
try {
|
|
||||||
source = index.getSource(pred, opts);
|
|
||||||
} catch (QueryParseException e) {
|
|
||||||
// Don't need to show this exception to the user; the only thing that
|
|
||||||
// changed about pred was its start, and any other QPEs that might happen
|
|
||||||
// should have already thrown from the constructor.
|
|
||||||
throw new OrmException(e);
|
|
||||||
}
|
|
||||||
// Don't convert start to a limit, since the caller of this method (see
|
|
||||||
// AndSource) has calculated the actual number to skip.
|
|
||||||
return read();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Predicate<ChangeData> copy(
|
|
||||||
Collection<? extends Predicate<ChangeData>> children) {
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean match(ChangeData cd) throws OrmException {
|
|
||||||
return (source != null && cd.isFromSource(source)) || pred.match(cd);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public int getCost() {
|
|
||||||
// Index queries are assumed to be cheaper than any other type of query, so
|
|
||||||
// so try to make sure they get picked. Note that pred's cost may be higher
|
|
||||||
// because it doesn't know whether it's being used in an index query or not.
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public int hashCode() {
|
|
||||||
return pred.hashCode();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean equals(Object other) {
|
|
||||||
if (other == null || getClass() != other.getClass()) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
IndexedChangeQuery o = (IndexedChangeQuery) other;
|
|
||||||
return pred.equals(o.pred)
|
|
||||||
&& opts.equals(o.opts);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String toString() {
|
|
||||||
return MoreObjects.toStringHelper("index")
|
|
||||||
.add("p", pred)
|
|
||||||
.add("opts", opts)
|
|
||||||
.toString();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -12,31 +12,24 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
package com.google.gerrit.server.query.change;
|
package com.google.gerrit.server.query;
|
||||||
|
|
||||||
import static com.google.gerrit.server.query.change.ChangeQueryBuilder.FIELD_LIMIT;
|
public class LimitPredicate<T> extends IntPredicate<T> {
|
||||||
|
|
||||||
import com.google.gerrit.server.query.IntPredicate;
|
|
||||||
import com.google.gerrit.server.query.Predicate;
|
|
||||||
import com.google.gerrit.server.query.QueryBuilder;
|
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
|
||||||
|
|
||||||
public class LimitPredicate extends IntPredicate<ChangeData> {
|
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
public static Integer getLimit(Predicate<ChangeData> p) {
|
public static Integer getLimit(String fieldName, Predicate<?> p) {
|
||||||
IntPredicate<?> ip = QueryBuilder.find(p, IntPredicate.class, FIELD_LIMIT);
|
IntPredicate<?> ip = QueryBuilder.find(p, IntPredicate.class, fieldName);
|
||||||
return ip != null ? ip.intValue() : null;
|
return ip != null ? ip.intValue() : null;
|
||||||
}
|
}
|
||||||
|
|
||||||
public LimitPredicate(int limit) throws QueryParseException {
|
public LimitPredicate(String fieldName, int limit) throws QueryParseException {
|
||||||
super(ChangeQueryBuilder.FIELD_LIMIT, limit);
|
super(fieldName, limit);
|
||||||
if (limit <= 0) {
|
if (limit <= 0) {
|
||||||
throw new QueryParseException("limit must be positive: " + limit);
|
throw new QueryParseException("limit must be positive: " + limit);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean match(ChangeData object) {
|
public boolean match(T object) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
|
@ -12,14 +12,14 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
package com.google.gerrit.server.query.change;
|
package com.google.gerrit.server.query;
|
||||||
|
|
||||||
import com.google.gerrit.server.index.QueryOptions;
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.gwtorm.server.ResultSet;
|
import com.google.gwtorm.server.ResultSet;
|
||||||
|
|
||||||
public interface Paginated {
|
public interface Paginated<T> {
|
||||||
QueryOptions getOptions();
|
QueryOptions getOptions();
|
||||||
|
|
||||||
ResultSet<ChangeData> restart(int start) throws OrmException;
|
ResultSet<T> restart(int start) throws OrmException;
|
||||||
}
|
}
|
|
@ -22,6 +22,7 @@ import com.google.common.collect.FluentIterable;
|
||||||
import com.google.common.collect.ImmutableList;
|
import com.google.common.collect.ImmutableList;
|
||||||
import com.google.common.collect.Iterables;
|
import com.google.common.collect.Iterables;
|
||||||
import com.google.gerrit.server.query.AndPredicate;
|
import com.google.gerrit.server.query.AndPredicate;
|
||||||
|
import com.google.gerrit.server.query.Paginated;
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gwtorm.server.ListResultSet;
|
import com.google.gwtorm.server.ListResultSet;
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
@ -126,7 +127,8 @@ public class AndSource extends AndPredicate<ChangeData>
|
||||||
// least one of its results, we may not have filled the full
|
// least one of its results, we may not have filled the full
|
||||||
// limit the caller wants. Restart the source and continue.
|
// limit the caller wants. Restart the source and continue.
|
||||||
//
|
//
|
||||||
Paginated p = (Paginated) source;
|
@SuppressWarnings("unchecked")
|
||||||
|
Paginated<ChangeData> p = (Paginated<ChangeData>) source;
|
||||||
while (skipped && r.size() < p.getOptions().limit() + start) {
|
while (skipped && r.size() < p.getOptions().limit() + start) {
|
||||||
skipped = false;
|
skipped = false;
|
||||||
ResultSet<ChangeData> next = p.restart(nextStart);
|
ResultSet<ChangeData> next = p.restart(nextStart);
|
||||||
|
|
|
@ -64,7 +64,6 @@ import com.google.gerrit.server.project.ChangeControl;
|
||||||
import com.google.gerrit.server.project.NoSuchChangeException;
|
import com.google.gerrit.server.project.NoSuchChangeException;
|
||||||
import com.google.gerrit.server.project.ProjectCache;
|
import com.google.gerrit.server.project.ProjectCache;
|
||||||
import com.google.gerrit.server.project.SubmitRuleEvaluator;
|
import com.google.gerrit.server.project.SubmitRuleEvaluator;
|
||||||
import com.google.gerrit.server.query.DataSource;
|
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
import com.google.gwtorm.server.ResultSet;
|
import com.google.gwtorm.server.ResultSet;
|
||||||
import com.google.inject.assistedinject.Assisted;
|
import com.google.inject.assistedinject.Assisted;
|
||||||
|
@ -323,7 +322,6 @@ public class ChangeData {
|
||||||
private final MergeabilityCache mergeabilityCache;
|
private final MergeabilityCache mergeabilityCache;
|
||||||
private final StarredChangesUtil starredChangesUtil;
|
private final StarredChangesUtil starredChangesUtil;
|
||||||
private final Change.Id legacyId;
|
private final Change.Id legacyId;
|
||||||
private DataSource<ChangeData> returnedBySource;
|
|
||||||
private Project.NameKey project;
|
private Project.NameKey project;
|
||||||
private Change change;
|
private Change change;
|
||||||
private ChangeNotes notes;
|
private ChangeNotes notes;
|
||||||
|
@ -551,14 +549,6 @@ public class ChangeData {
|
||||||
return db;
|
return db;
|
||||||
}
|
}
|
||||||
|
|
||||||
public boolean isFromSource(DataSource<ChangeData> s) {
|
|
||||||
return s == returnedBySource;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void cacheFromSource(DataSource<ChangeData> s) {
|
|
||||||
returnedBySource = s;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setCurrentFilePaths(List<String> filePaths) throws OrmException {
|
public void setCurrentFilePaths(List<String> filePaths) throws OrmException {
|
||||||
PatchSet ps = currentPatchSet();
|
PatchSet ps = currentPatchSet();
|
||||||
if (ps != null) {
|
if (ps != null) {
|
||||||
|
|
|
@ -57,12 +57,13 @@ import com.google.gerrit.server.index.Schema;
|
||||||
import com.google.gerrit.server.index.change.ChangeField;
|
import com.google.gerrit.server.index.change.ChangeField;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndex;
|
import com.google.gerrit.server.index.change.ChangeIndex;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
||||||
import com.google.gerrit.server.index.change.IndexRewriter;
|
import com.google.gerrit.server.index.change.ChangeIndexRewriter;
|
||||||
import com.google.gerrit.server.notedb.ChangeNotes;
|
import com.google.gerrit.server.notedb.ChangeNotes;
|
||||||
import com.google.gerrit.server.patch.PatchListCache;
|
import com.google.gerrit.server.patch.PatchListCache;
|
||||||
import com.google.gerrit.server.project.ChangeControl;
|
import com.google.gerrit.server.project.ChangeControl;
|
||||||
import com.google.gerrit.server.project.ListChildProjects;
|
import com.google.gerrit.server.project.ListChildProjects;
|
||||||
import com.google.gerrit.server.project.ProjectCache;
|
import com.google.gerrit.server.project.ProjectCache;
|
||||||
|
import com.google.gerrit.server.query.LimitPredicate;
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gerrit.server.query.QueryBuilder;
|
import com.google.gerrit.server.query.QueryBuilder;
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
|
@ -158,7 +159,7 @@ public class ChangeQueryBuilder extends QueryBuilder<ChangeData> {
|
||||||
public static class Arguments {
|
public static class Arguments {
|
||||||
final Provider<ReviewDb> db;
|
final Provider<ReviewDb> db;
|
||||||
final Provider<InternalChangeQuery> queryProvider;
|
final Provider<InternalChangeQuery> queryProvider;
|
||||||
final IndexRewriter rewriter;
|
final ChangeIndexRewriter rewriter;
|
||||||
final DynamicMap<ChangeOperatorFactory> opFactories;
|
final DynamicMap<ChangeOperatorFactory> opFactories;
|
||||||
final IdentifiedUser.GenericFactory userFactory;
|
final IdentifiedUser.GenericFactory userFactory;
|
||||||
final CapabilityControl.Factory capabilityControlFactory;
|
final CapabilityControl.Factory capabilityControlFactory;
|
||||||
|
@ -190,7 +191,7 @@ public class ChangeQueryBuilder extends QueryBuilder<ChangeData> {
|
||||||
@VisibleForTesting
|
@VisibleForTesting
|
||||||
public Arguments(Provider<ReviewDb> db,
|
public Arguments(Provider<ReviewDb> db,
|
||||||
Provider<InternalChangeQuery> queryProvider,
|
Provider<InternalChangeQuery> queryProvider,
|
||||||
IndexRewriter rewriter,
|
ChangeIndexRewriter rewriter,
|
||||||
DynamicMap<ChangeOperatorFactory> opFactories,
|
DynamicMap<ChangeOperatorFactory> opFactories,
|
||||||
IdentifiedUser.GenericFactory userFactory,
|
IdentifiedUser.GenericFactory userFactory,
|
||||||
Provider<CurrentUser> self,
|
Provider<CurrentUser> self,
|
||||||
|
@ -229,7 +230,7 @@ public class ChangeQueryBuilder extends QueryBuilder<ChangeData> {
|
||||||
private Arguments(
|
private Arguments(
|
||||||
Provider<ReviewDb> db,
|
Provider<ReviewDb> db,
|
||||||
Provider<InternalChangeQuery> queryProvider,
|
Provider<InternalChangeQuery> queryProvider,
|
||||||
IndexRewriter rewriter,
|
ChangeIndexRewriter rewriter,
|
||||||
DynamicMap<ChangeOperatorFactory> opFactories,
|
DynamicMap<ChangeOperatorFactory> opFactories,
|
||||||
IdentifiedUser.GenericFactory userFactory,
|
IdentifiedUser.GenericFactory userFactory,
|
||||||
Provider<CurrentUser> self,
|
Provider<CurrentUser> self,
|
||||||
|
@ -843,7 +844,8 @@ public class ChangeQueryBuilder extends QueryBuilder<ChangeData> {
|
||||||
|
|
||||||
@Operator
|
@Operator
|
||||||
public Predicate<ChangeData> limit(String limit) throws QueryParseException {
|
public Predicate<ChangeData> limit(String limit) throws QueryParseException {
|
||||||
return new LimitPredicate(Integer.parseInt(limit));
|
return new LimitPredicate<>(ChangeQueryBuilder.FIELD_LIMIT,
|
||||||
|
Integer.parseInt(limit));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Operator
|
@Operator
|
||||||
|
|
|
@ -15,6 +15,7 @@
|
||||||
package com.google.gerrit.server.query.change;
|
package com.google.gerrit.server.query.change;
|
||||||
|
|
||||||
import static com.google.common.base.Preconditions.checkState;
|
import static com.google.common.base.Preconditions.checkState;
|
||||||
|
import static com.google.gerrit.server.query.change.ChangeQueryBuilder.FIELD_LIMIT;
|
||||||
import static com.google.gerrit.server.query.change.ChangeStatusPredicate.open;
|
import static com.google.gerrit.server.query.change.ChangeStatusPredicate.open;
|
||||||
|
|
||||||
import com.google.common.base.Throwables;
|
import com.google.common.base.Throwables;
|
||||||
|
@ -32,10 +33,11 @@ import com.google.gerrit.server.index.IndexPredicate;
|
||||||
import com.google.gerrit.server.index.QueryOptions;
|
import com.google.gerrit.server.index.QueryOptions;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndex;
|
import com.google.gerrit.server.index.change.ChangeIndex;
|
||||||
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
import com.google.gerrit.server.index.change.ChangeIndexCollection;
|
||||||
import com.google.gerrit.server.index.change.IndexRewriter;
|
import com.google.gerrit.server.index.change.ChangeIndexRewriter;
|
||||||
import com.google.gerrit.server.index.change.IndexedChangeQuery;
|
import com.google.gerrit.server.index.change.IndexedChangeQuery;
|
||||||
import com.google.gerrit.server.notedb.ChangeNotes;
|
import com.google.gerrit.server.notedb.ChangeNotes;
|
||||||
import com.google.gerrit.server.project.ChangeControl;
|
import com.google.gerrit.server.project.ChangeControl;
|
||||||
|
import com.google.gerrit.server.query.LimitPredicate;
|
||||||
import com.google.gerrit.server.query.Predicate;
|
import com.google.gerrit.server.query.Predicate;
|
||||||
import com.google.gerrit.server.query.QueryParseException;
|
import com.google.gerrit.server.query.QueryParseException;
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
@ -54,7 +56,7 @@ public class QueryProcessor {
|
||||||
private final ChangeControl.GenericFactory changeControlFactory;
|
private final ChangeControl.GenericFactory changeControlFactory;
|
||||||
private final ChangeNotes.Factory notesFactory;
|
private final ChangeNotes.Factory notesFactory;
|
||||||
private final ChangeIndexCollection indexes;
|
private final ChangeIndexCollection indexes;
|
||||||
private final IndexRewriter rewriter;
|
private final ChangeIndexRewriter rewriter;
|
||||||
private final IndexConfig indexConfig;
|
private final IndexConfig indexConfig;
|
||||||
private final Metrics metrics;
|
private final Metrics metrics;
|
||||||
|
|
||||||
|
@ -69,7 +71,7 @@ public class QueryProcessor {
|
||||||
ChangeControl.GenericFactory changeControlFactory,
|
ChangeControl.GenericFactory changeControlFactory,
|
||||||
ChangeNotes.Factory notesFactory,
|
ChangeNotes.Factory notesFactory,
|
||||||
ChangeIndexCollection indexes,
|
ChangeIndexCollection indexes,
|
||||||
IndexRewriter rewriter,
|
ChangeIndexRewriter rewriter,
|
||||||
IndexConfig indexConfig,
|
IndexConfig indexConfig,
|
||||||
Metrics metrics) {
|
Metrics metrics) {
|
||||||
this.db = db;
|
this.db = db;
|
||||||
|
@ -245,7 +247,7 @@ public class QueryProcessor {
|
||||||
if (limitFromCaller > 0) {
|
if (limitFromCaller > 0) {
|
||||||
possibleLimits.add(limitFromCaller);
|
possibleLimits.add(limitFromCaller);
|
||||||
}
|
}
|
||||||
Integer limitFromPredicate = LimitPredicate.getLimit(p);
|
Integer limitFromPredicate = LimitPredicate.getLimit(FIELD_LIMIT, p);
|
||||||
if (limitFromPredicate != null) {
|
if (limitFromPredicate != null) {
|
||||||
possibleLimits.add(limitFromPredicate);
|
possibleLimits.add(limitFromPredicate);
|
||||||
}
|
}
|
||||||
|
|
|
@ -45,13 +45,13 @@ import java.util.Arrays;
|
||||||
import java.util.EnumSet;
|
import java.util.EnumSet;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
|
||||||
public class IndexRewriterTest extends GerritBaseTests {
|
public class ChangeIndexRewriterTest extends GerritBaseTests {
|
||||||
private static final IndexConfig CONFIG = IndexConfig.createDefault();
|
private static final IndexConfig CONFIG = IndexConfig.createDefault();
|
||||||
|
|
||||||
private FakeChangeIndex index;
|
private FakeChangeIndex index;
|
||||||
private ChangeIndexCollection indexes;
|
private ChangeIndexCollection indexes;
|
||||||
private ChangeQueryBuilder queryBuilder;
|
private ChangeQueryBuilder queryBuilder;
|
||||||
private IndexRewriter rewrite;
|
private ChangeIndexRewriter rewrite;
|
||||||
|
|
||||||
@Before
|
@Before
|
||||||
public void setUp() throws Exception {
|
public void setUp() throws Exception {
|
||||||
|
@ -59,7 +59,7 @@ public class IndexRewriterTest extends GerritBaseTests {
|
||||||
indexes = new ChangeIndexCollection();
|
indexes = new ChangeIndexCollection();
|
||||||
indexes.setSearchIndex(index);
|
indexes.setSearchIndex(index);
|
||||||
queryBuilder = new FakeQueryBuilder(indexes);
|
queryBuilder = new FakeQueryBuilder(indexes);
|
||||||
rewrite = new IndexRewriter(indexes,
|
rewrite = new ChangeIndexRewriter(indexes,
|
||||||
IndexConfig.create(0, 0, 3));
|
IndexConfig.create(0, 0, 3));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -292,6 +292,6 @@ public class IndexRewriterTest extends GerritBaseTests {
|
||||||
}
|
}
|
||||||
|
|
||||||
private Set<Change.Status> status(String query) throws QueryParseException {
|
private Set<Change.Status> status(String query) throws QueryParseException {
|
||||||
return IndexRewriter.getPossibleStatus(parse(query));
|
return ChangeIndexRewriter.getPossibleStatus(parse(query));
|
||||||
}
|
}
|
||||||
}
|
}
|
|
@ -38,6 +38,7 @@ import com.google.inject.Inject;
|
||||||
import org.kohsuke.args4j.Argument;
|
import org.kohsuke.args4j.Argument;
|
||||||
import org.kohsuke.args4j.Option;
|
import org.kohsuke.args4j.Option;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
|
||||||
|
@ -88,7 +89,7 @@ final class CreateGroupCommand extends SshCommand {
|
||||||
private AddIncludedGroups addIncludedGroups;
|
private AddIncludedGroups addIncludedGroups;
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void run() throws Failure, OrmException {
|
protected void run() throws Failure, OrmException, IOException {
|
||||||
try {
|
try {
|
||||||
GroupResource rsrc = createGroup();
|
GroupResource rsrc = createGroup();
|
||||||
|
|
||||||
|
@ -104,7 +105,8 @@ final class CreateGroupCommand extends SshCommand {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private GroupResource createGroup() throws RestApiException, OrmException {
|
private GroupResource createGroup()
|
||||||
|
throws RestApiException, OrmException, IOException {
|
||||||
GroupInput input = new GroupInput();
|
GroupInput input = new GroupInput();
|
||||||
input.description = groupDescription;
|
input.description = groupDescription;
|
||||||
input.visibleToAll = visibleToAll;
|
input.visibleToAll = visibleToAll;
|
||||||
|
@ -120,7 +122,7 @@ final class CreateGroupCommand extends SshCommand {
|
||||||
}
|
}
|
||||||
|
|
||||||
private void addMembers(GroupResource rsrc) throws RestApiException,
|
private void addMembers(GroupResource rsrc) throws RestApiException,
|
||||||
OrmException {
|
OrmException, IOException {
|
||||||
AddMembers.Input input =
|
AddMembers.Input input =
|
||||||
AddMembers.Input.fromMembers(FluentIterable
|
AddMembers.Input.fromMembers(FluentIterable
|
||||||
.from(initialMembers)
|
.from(initialMembers)
|
||||||
|
|
|
@ -256,8 +256,8 @@ final class SetAccountCommand extends SshCommand {
|
||||||
new AccountResource.SshKey(user, sshKey), null);
|
new AccountResource.SshKey(user, sshKey), null);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void addEmail(String email) throws UnloggedFailure, RestApiException,
|
private void addEmail(String email)
|
||||||
OrmException {
|
throws UnloggedFailure, RestApiException, OrmException, IOException {
|
||||||
EmailInput in = new EmailInput();
|
EmailInput in = new EmailInput();
|
||||||
in.email = email;
|
in.email = email;
|
||||||
in.noConfirmation = true;
|
in.noConfirmation = true;
|
||||||
|
@ -268,7 +268,8 @@ final class SetAccountCommand extends SshCommand {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private void deleteEmail(String email) throws RestApiException, OrmException {
|
private void deleteEmail(String email)
|
||||||
|
throws RestApiException, OrmException, IOException {
|
||||||
if (email.equals("ALL")) {
|
if (email.equals("ALL")) {
|
||||||
List<EmailInfo> emails = getEmails.apply(rsrc);
|
List<EmailInfo> emails = getEmails.apply(rsrc);
|
||||||
for (EmailInfo e : emails) {
|
for (EmailInfo e : emails) {
|
||||||
|
@ -281,8 +282,8 @@ final class SetAccountCommand extends SshCommand {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private void putPreferred(String email) throws RestApiException,
|
private void putPreferred(String email)
|
||||||
OrmException {
|
throws RestApiException, OrmException, IOException {
|
||||||
for (EmailInfo e : getEmails.apply(rsrc)) {
|
for (EmailInfo e : getEmails.apply(rsrc)) {
|
||||||
if (e.email.equals(email)) {
|
if (e.email.equals(email)) {
|
||||||
putPreferred.apply(new AccountResource.Email(user, email), null);
|
putPreferred.apply(new AccountResource.Email(user, email), null);
|
||||||
|
|
Loading…
Reference in New Issue