Merge "Create a user branch for every account"
This commit is contained in:
@@ -53,7 +53,7 @@ public class AccountCreator {
|
|||||||
private final Map<String, TestAccount> accounts;
|
private final Map<String, TestAccount> accounts;
|
||||||
|
|
||||||
private final SchemaFactory<ReviewDb> reviewDbProvider;
|
private final SchemaFactory<ReviewDb> reviewDbProvider;
|
||||||
private final AccountsUpdate accountsUpdate;
|
private final AccountsUpdate.Server accountsUpdate;
|
||||||
private final VersionedAuthorizedKeys.Accessor authorizedKeys;
|
private final VersionedAuthorizedKeys.Accessor authorizedKeys;
|
||||||
private final GroupCache groupCache;
|
private final GroupCache groupCache;
|
||||||
private final SshKeyCache sshKeyCache;
|
private final SshKeyCache sshKeyCache;
|
||||||
@@ -65,7 +65,7 @@ public class AccountCreator {
|
|||||||
@Inject
|
@Inject
|
||||||
AccountCreator(
|
AccountCreator(
|
||||||
SchemaFactory<ReviewDb> schema,
|
SchemaFactory<ReviewDb> schema,
|
||||||
AccountsUpdate accountsUpdate,
|
AccountsUpdate.Server accountsUpdate,
|
||||||
VersionedAuthorizedKeys.Accessor authorizedKeys,
|
VersionedAuthorizedKeys.Accessor authorizedKeys,
|
||||||
GroupCache groupCache,
|
GroupCache groupCache,
|
||||||
SshKeyCache sshKeyCache,
|
SshKeyCache sshKeyCache,
|
||||||
@@ -114,7 +114,7 @@ public class AccountCreator {
|
|||||||
Account a = new Account(id, TimeUtil.nowTs());
|
Account a = new Account(id, TimeUtil.nowTs());
|
||||||
a.setFullName(fullName);
|
a.setFullName(fullName);
|
||||||
a.setPreferredEmail(email);
|
a.setPreferredEmail(email);
|
||||||
accountsUpdate.insert(db, a);
|
accountsUpdate.create().insert(db, a);
|
||||||
|
|
||||||
if (groups != null) {
|
if (groups != null) {
|
||||||
for (String n : groups) {
|
for (String n : groups) {
|
||||||
|
|||||||
@@ -51,7 +51,6 @@ import com.google.gerrit.extensions.api.accounts.EmailInput;
|
|||||||
import com.google.gerrit.extensions.api.changes.AddReviewerInput;
|
import com.google.gerrit.extensions.api.changes.AddReviewerInput;
|
||||||
import com.google.gerrit.extensions.api.changes.ReviewInput;
|
import com.google.gerrit.extensions.api.changes.ReviewInput;
|
||||||
import com.google.gerrit.extensions.api.changes.StarsInput;
|
import com.google.gerrit.extensions.api.changes.StarsInput;
|
||||||
import com.google.gerrit.extensions.client.GeneralPreferencesInfo;
|
|
||||||
import com.google.gerrit.extensions.common.AccountInfo;
|
import com.google.gerrit.extensions.common.AccountInfo;
|
||||||
import com.google.gerrit.extensions.common.ChangeInfo;
|
import com.google.gerrit.extensions.common.ChangeInfo;
|
||||||
import com.google.gerrit.extensions.common.GpgKeyInfo;
|
import com.google.gerrit.extensions.common.GpgKeyInfo;
|
||||||
@@ -74,6 +73,7 @@ import com.google.gerrit.server.account.externalids.ExternalIds;
|
|||||||
import com.google.gerrit.server.account.externalids.ExternalIdsUpdate;
|
import com.google.gerrit.server.account.externalids.ExternalIdsUpdate;
|
||||||
import com.google.gerrit.server.config.AllUsersName;
|
import com.google.gerrit.server.config.AllUsersName;
|
||||||
import com.google.gerrit.server.git.ProjectConfig;
|
import com.google.gerrit.server.git.ProjectConfig;
|
||||||
|
import com.google.gerrit.server.notedb.rebuild.ChangeRebuilderImpl;
|
||||||
import com.google.gerrit.server.project.RefPattern;
|
import com.google.gerrit.server.project.RefPattern;
|
||||||
import com.google.gerrit.server.util.MagicBranch;
|
import com.google.gerrit.server.util.MagicBranch;
|
||||||
import com.google.gerrit.testutil.ConfigSuite;
|
import com.google.gerrit.testutil.ConfigSuite;
|
||||||
@@ -100,6 +100,8 @@ import org.eclipse.jgit.lib.Config;
|
|||||||
import org.eclipse.jgit.lib.Ref;
|
import org.eclipse.jgit.lib.Ref;
|
||||||
import org.eclipse.jgit.lib.RefUpdate;
|
import org.eclipse.jgit.lib.RefUpdate;
|
||||||
import org.eclipse.jgit.lib.Repository;
|
import org.eclipse.jgit.lib.Repository;
|
||||||
|
import org.eclipse.jgit.revwalk.RevCommit;
|
||||||
|
import org.eclipse.jgit.revwalk.RevWalk;
|
||||||
import org.eclipse.jgit.transport.PushCertificateIdent;
|
import org.eclipse.jgit.transport.PushCertificateIdent;
|
||||||
import org.eclipse.jgit.transport.PushResult;
|
import org.eclipse.jgit.transport.PushResult;
|
||||||
import org.eclipse.jgit.transport.RemoteRefUpdate;
|
import org.eclipse.jgit.transport.RemoteRefUpdate;
|
||||||
@@ -183,6 +185,26 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void create() throws Exception {
|
||||||
|
TestAccount foo = accounts.create("foo");
|
||||||
|
AccountInfo info = gApi.accounts().id(foo.id.get()).get();
|
||||||
|
assertThat(info.username).isEqualTo("foo");
|
||||||
|
|
||||||
|
// check user branch
|
||||||
|
try (Repository repo = repoManager.openRepository(allUsers);
|
||||||
|
RevWalk rw = new RevWalk(repo)) {
|
||||||
|
Ref ref = repo.exactRef(RefNames.refsUsers(foo.getId()));
|
||||||
|
assertThat(ref).isNotNull();
|
||||||
|
RevCommit c = rw.parseCommit(ref.getObjectId());
|
||||||
|
long timestampDiffMs =
|
||||||
|
Math.abs(
|
||||||
|
c.getCommitTime() * 1000L
|
||||||
|
- accountCache.get(foo.getId()).getAccount().getRegisteredOn().getTime());
|
||||||
|
assertThat(timestampDiffMs).isAtMost(ChangeRebuilderImpl.MAX_WINDOW_MS);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void get() throws Exception {
|
public void get() throws Exception {
|
||||||
AccountInfo info = gApi.accounts().id("admin").get();
|
AccountInfo info = gApi.accounts().id("admin").get();
|
||||||
@@ -553,7 +575,7 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
@Test
|
@Test
|
||||||
@Sandboxed
|
@Sandboxed
|
||||||
public void fetchUserBranch() throws Exception {
|
public void fetchUserBranch() throws Exception {
|
||||||
ensureUserBranchCreated(user);
|
setApiUser(user);
|
||||||
|
|
||||||
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers, user);
|
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers, user);
|
||||||
String userRefName = RefNames.refsUsers(user.id);
|
String userRefName = RefNames.refsUsers(user.id);
|
||||||
@@ -603,8 +625,6 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pushToUserBranch() throws Exception {
|
public void pushToUserBranch() throws Exception {
|
||||||
ensureUserBranchCreated(admin);
|
|
||||||
|
|
||||||
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
||||||
fetch(allUsersRepo, RefNames.refsUsers(admin.id) + ":userRef");
|
fetch(allUsersRepo, RefNames.refsUsers(admin.id) + ":userRef");
|
||||||
allUsersRepo.reset("userRef");
|
allUsersRepo.reset("userRef");
|
||||||
@@ -617,8 +637,6 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pushToUserBranchForReview() throws Exception {
|
public void pushToUserBranchForReview() throws Exception {
|
||||||
ensureUserBranchCreated(admin);
|
|
||||||
|
|
||||||
String userRefName = RefNames.refsUsers(admin.id);
|
String userRefName = RefNames.refsUsers(admin.id);
|
||||||
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
||||||
fetch(allUsersRepo, userRefName + ":userRef");
|
fetch(allUsersRepo, userRefName + ":userRef");
|
||||||
@@ -640,8 +658,6 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pushWatchConfigToUserBranch() throws Exception {
|
public void pushWatchConfigToUserBranch() throws Exception {
|
||||||
ensureUserBranchCreated(admin);
|
|
||||||
|
|
||||||
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
TestRepository<InMemoryRepository> allUsersRepo = cloneProject(allUsers);
|
||||||
fetch(allUsersRepo, RefNames.refsUsers(admin.id) + ":userRef");
|
fetch(allUsersRepo, RefNames.refsUsers(admin.id) + ":userRef");
|
||||||
allUsersRepo.reset("userRef");
|
allUsersRepo.reset("userRef");
|
||||||
@@ -683,8 +699,6 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
@Test
|
@Test
|
||||||
@Sandboxed
|
@Sandboxed
|
||||||
public void cannotDeleteUserBranch() throws Exception {
|
public void cannotDeleteUserBranch() throws Exception {
|
||||||
ensureUserBranchCreated(admin);
|
|
||||||
|
|
||||||
grant(
|
grant(
|
||||||
Permission.DELETE,
|
Permission.DELETE,
|
||||||
allUsers,
|
allUsers,
|
||||||
@@ -707,8 +721,6 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
@Test
|
@Test
|
||||||
@Sandboxed
|
@Sandboxed
|
||||||
public void deleteUserBranchWithAccessDatabaseCapability() throws Exception {
|
public void deleteUserBranchWithAccessDatabaseCapability() throws Exception {
|
||||||
ensureUserBranchCreated(admin);
|
|
||||||
|
|
||||||
allowGlobalCapabilities(REGISTERED_USERS, GlobalCapability.ACCESS_DATABASE);
|
allowGlobalCapabilities(REGISTERED_USERS, GlobalCapability.ACCESS_DATABASE);
|
||||||
grant(
|
grant(
|
||||||
Permission.DELETE,
|
Permission.DELETE,
|
||||||
@@ -1019,12 +1031,4 @@ public class AccountIT extends AbstractDaemonTest {
|
|||||||
assertThat(accounts).hasSize(1);
|
assertThat(accounts).hasSize(1);
|
||||||
assertThat(Iterables.getOnlyElement(accounts)).isEqualTo(expectedAccount.getId());
|
assertThat(Iterables.getOnlyElement(accounts)).isEqualTo(expectedAccount.getId());
|
||||||
}
|
}
|
||||||
|
|
||||||
private void ensureUserBranchCreated(TestAccount account) throws Exception {
|
|
||||||
// Change something in the user preferences to ensure that the user branch is created.
|
|
||||||
setApiUser(account);
|
|
||||||
GeneralPreferencesInfo input = new GeneralPreferencesInfo();
|
|
||||||
input.changesPerPage = GeneralPreferencesInfo.defaults().changesPerPage + 10;
|
|
||||||
gApi.accounts().self().setPreferences(input);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,68 @@
|
|||||||
|
// Copyright (C) 2017 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.pgm.init;
|
||||||
|
|
||||||
|
import static com.google.common.base.Preconditions.checkArgument;
|
||||||
|
|
||||||
|
import com.google.common.collect.ImmutableSet;
|
||||||
|
import com.google.gerrit.pgm.init.api.InitFlags;
|
||||||
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
||||||
|
import com.google.gerrit.server.GerritPersonIdentProvider;
|
||||||
|
import com.google.gerrit.server.account.AccountsUpdate;
|
||||||
|
import com.google.gerrit.server.config.SitePaths;
|
||||||
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import org.eclipse.jgit.internal.storage.file.FileRepository;
|
||||||
|
import org.eclipse.jgit.lib.ObjectInserter;
|
||||||
|
import org.eclipse.jgit.lib.PersonIdent;
|
||||||
|
import org.eclipse.jgit.lib.Repository;
|
||||||
|
import org.eclipse.jgit.lib.RepositoryCache.FileKey;
|
||||||
|
import org.eclipse.jgit.util.FS;
|
||||||
|
|
||||||
|
public class AccountsOnInit {
|
||||||
|
private final InitFlags flags;
|
||||||
|
private final SitePaths site;
|
||||||
|
private final String allUsers;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
public AccountsOnInit(InitFlags flags, SitePaths site, AllUsersNameOnInitProvider allUsers) {
|
||||||
|
this.flags = flags;
|
||||||
|
this.site = site;
|
||||||
|
this.allUsers = allUsers.get();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void insert(ReviewDb db, Account account) throws OrmException, IOException {
|
||||||
|
db.accounts().insert(ImmutableSet.of(account));
|
||||||
|
|
||||||
|
File path = getPath();
|
||||||
|
if (path != null) {
|
||||||
|
try (Repository repo = new FileRepository(path);
|
||||||
|
ObjectInserter oi = repo.newObjectInserter()) {
|
||||||
|
PersonIdent serverIdent = new GerritPersonIdentProvider(flags.cfg).get();
|
||||||
|
AccountsUpdate.createUserBranch(repo, oi, serverIdent, serverIdent, account);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private File getPath() {
|
||||||
|
Path basePath = site.resolve(flags.cfg.getString("gerrit", null, "basePath"));
|
||||||
|
checkArgument(basePath != null, "gerrit.basePath must be configured");
|
||||||
|
return FileKey.resolve(basePath.resolve(allUsers).toFile(), FS.DETECTED);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -29,7 +29,6 @@ import com.google.gerrit.reviewdb.client.AccountGroupName;
|
|||||||
import com.google.gerrit.reviewdb.client.AccountSshKey;
|
import com.google.gerrit.reviewdb.client.AccountSshKey;
|
||||||
import com.google.gerrit.reviewdb.server.ReviewDb;
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
||||||
import com.google.gerrit.server.account.AccountState;
|
import com.google.gerrit.server.account.AccountState;
|
||||||
import com.google.gerrit.server.account.AccountsUpdate;
|
|
||||||
import com.google.gerrit.server.account.externalids.ExternalId;
|
import com.google.gerrit.server.account.externalids.ExternalId;
|
||||||
import com.google.gerrit.server.index.account.AccountIndex;
|
import com.google.gerrit.server.index.account.AccountIndex;
|
||||||
import com.google.gerrit.server.index.account.AccountIndexCollection;
|
import com.google.gerrit.server.index.account.AccountIndexCollection;
|
||||||
@@ -48,7 +47,7 @@ import org.apache.commons.validator.routines.EmailValidator;
|
|||||||
public class InitAdminUser implements InitStep {
|
public class InitAdminUser implements InitStep {
|
||||||
private final ConsoleUI ui;
|
private final ConsoleUI ui;
|
||||||
private final InitFlags flags;
|
private final InitFlags flags;
|
||||||
private final AccountsUpdate accountsUpdate;
|
private final AccountsOnInit accounts;
|
||||||
private final VersionedAuthorizedKeysOnInit.Factory authorizedKeysFactory;
|
private final VersionedAuthorizedKeysOnInit.Factory authorizedKeysFactory;
|
||||||
private final ExternalIdsOnInit externalIds;
|
private final ExternalIdsOnInit externalIds;
|
||||||
private SchemaFactory<ReviewDb> dbFactory;
|
private SchemaFactory<ReviewDb> dbFactory;
|
||||||
@@ -58,12 +57,12 @@ public class InitAdminUser implements InitStep {
|
|||||||
InitAdminUser(
|
InitAdminUser(
|
||||||
InitFlags flags,
|
InitFlags flags,
|
||||||
ConsoleUI ui,
|
ConsoleUI ui,
|
||||||
AccountsUpdate accountsUpdate,
|
AccountsOnInit accounts,
|
||||||
VersionedAuthorizedKeysOnInit.Factory authorizedKeysFactory,
|
VersionedAuthorizedKeysOnInit.Factory authorizedKeysFactory,
|
||||||
ExternalIdsOnInit externalIds) {
|
ExternalIdsOnInit externalIds) {
|
||||||
this.flags = flags;
|
this.flags = flags;
|
||||||
this.ui = ui;
|
this.ui = ui;
|
||||||
this.accountsUpdate = accountsUpdate;
|
this.accounts = accounts;
|
||||||
this.authorizedKeysFactory = authorizedKeysFactory;
|
this.authorizedKeysFactory = authorizedKeysFactory;
|
||||||
this.externalIds = externalIds;
|
this.externalIds = externalIds;
|
||||||
}
|
}
|
||||||
@@ -110,7 +109,7 @@ public class InitAdminUser implements InitStep {
|
|||||||
Account a = new Account(id, TimeUtil.nowTs());
|
Account a = new Account(id, TimeUtil.nowTs());
|
||||||
a.setFullName(name);
|
a.setFullName(name);
|
||||||
a.setPreferredEmail(email);
|
a.setPreferredEmail(email);
|
||||||
accountsUpdate.insert(db, a);
|
accounts.insert(db, a);
|
||||||
|
|
||||||
AccountGroupName adminGroupName =
|
AccountGroupName adminGroupName =
|
||||||
db.accountGroupNames().get(new AccountGroup.NameKey("Administrators"));
|
db.accountGroupNames().get(new AccountGroup.NameKey("Administrators"));
|
||||||
|
|||||||
@@ -52,6 +52,7 @@ public class AccountManager {
|
|||||||
private static final Logger log = LoggerFactory.getLogger(AccountManager.class);
|
private static final Logger log = LoggerFactory.getLogger(AccountManager.class);
|
||||||
|
|
||||||
private final SchemaFactory<ReviewDb> schema;
|
private final SchemaFactory<ReviewDb> schema;
|
||||||
|
private final AccountsUpdate.Server accountsUpdateFactory;
|
||||||
private final AccountCache byIdCache;
|
private final AccountCache byIdCache;
|
||||||
private final AccountByEmailCache byEmailCache;
|
private final AccountByEmailCache byEmailCache;
|
||||||
private final Realm realm;
|
private final Realm realm;
|
||||||
@@ -67,6 +68,7 @@ public class AccountManager {
|
|||||||
@Inject
|
@Inject
|
||||||
AccountManager(
|
AccountManager(
|
||||||
SchemaFactory<ReviewDb> schema,
|
SchemaFactory<ReviewDb> schema,
|
||||||
|
AccountsUpdate.Server accountsUpdateFactory,
|
||||||
AccountCache byIdCache,
|
AccountCache byIdCache,
|
||||||
AccountByEmailCache byEmailCache,
|
AccountByEmailCache byEmailCache,
|
||||||
Realm accountMapper,
|
Realm accountMapper,
|
||||||
@@ -78,6 +80,7 @@ public class AccountManager {
|
|||||||
ExternalIds externalIds,
|
ExternalIds externalIds,
|
||||||
ExternalIdsUpdate.Server externalIdsUpdateFactory) {
|
ExternalIdsUpdate.Server externalIdsUpdateFactory) {
|
||||||
this.schema = schema;
|
this.schema = schema;
|
||||||
|
this.accountsUpdateFactory = accountsUpdateFactory;
|
||||||
this.byIdCache = byIdCache;
|
this.byIdCache = byIdCache;
|
||||||
this.byEmailCache = byEmailCache;
|
this.byEmailCache = byEmailCache;
|
||||||
this.realm = accountMapper;
|
this.realm = accountMapper;
|
||||||
@@ -229,7 +232,7 @@ public class AccountManager {
|
|||||||
awaitsFirstAccountCheck.getAndSet(false) && db.accounts().anyAccounts().toList().isEmpty();
|
awaitsFirstAccountCheck.getAndSet(false) && db.accounts().anyAccounts().toList().isEmpty();
|
||||||
|
|
||||||
try {
|
try {
|
||||||
db.accounts().upsert(Collections.singleton(account));
|
accountsUpdateFactory.create().upsert(db, account);
|
||||||
|
|
||||||
ExternalId existingExtId = externalIds.get(db, extId.key());
|
ExternalId existingExtId = externalIds.get(db, extId.key());
|
||||||
if (existingExtId != null && !existingExtId.accountId().equals(extId.accountId())) {
|
if (existingExtId != null && !existingExtId.accountId().equals(extId.accountId())) {
|
||||||
|
|||||||
@@ -14,22 +14,199 @@
|
|||||||
|
|
||||||
package com.google.gerrit.server.account;
|
package com.google.gerrit.server.account;
|
||||||
|
|
||||||
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
|
||||||
import com.google.common.collect.ImmutableSet;
|
import com.google.common.collect.ImmutableSet;
|
||||||
import com.google.gerrit.reviewdb.client.Account;
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
import com.google.gerrit.reviewdb.client.RefNames;
|
||||||
import com.google.gerrit.reviewdb.server.ReviewDb;
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
||||||
|
import com.google.gerrit.server.GerritPersonIdent;
|
||||||
|
import com.google.gerrit.server.IdentifiedUser;
|
||||||
|
import com.google.gerrit.server.config.AllUsersName;
|
||||||
|
import com.google.gerrit.server.git.GitRepositoryManager;
|
||||||
import com.google.gwtorm.server.OrmDuplicateKeyException;
|
import com.google.gwtorm.server.OrmDuplicateKeyException;
|
||||||
import com.google.gwtorm.server.OrmException;
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import com.google.inject.Provider;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
import org.eclipse.jgit.lib.CommitBuilder;
|
||||||
|
import org.eclipse.jgit.lib.Constants;
|
||||||
|
import org.eclipse.jgit.lib.ObjectId;
|
||||||
|
import org.eclipse.jgit.lib.ObjectInserter;
|
||||||
|
import org.eclipse.jgit.lib.PersonIdent;
|
||||||
|
import org.eclipse.jgit.lib.RefUpdate;
|
||||||
|
import org.eclipse.jgit.lib.RefUpdate.Result;
|
||||||
|
import org.eclipse.jgit.lib.Repository;
|
||||||
|
|
||||||
/** Updates accounts. */
|
/** Updates accounts. */
|
||||||
@Singleton
|
@Singleton
|
||||||
public class AccountsUpdate {
|
public class AccountsUpdate {
|
||||||
|
/**
|
||||||
|
* Factory to create an AccountsUpdate instance for updating accounts by the Gerrit server.
|
||||||
|
*
|
||||||
|
* <p>The Gerrit server identity will be used as author and committer for all commits that update
|
||||||
|
* the accounts.
|
||||||
|
*/
|
||||||
|
@Singleton
|
||||||
|
public static class Server {
|
||||||
|
private final GitRepositoryManager repoManager;
|
||||||
|
private final AllUsersName allUsersName;
|
||||||
|
private final Provider<PersonIdent> serverIdent;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
public Server(
|
||||||
|
GitRepositoryManager repoManager,
|
||||||
|
AllUsersName allUsersName,
|
||||||
|
@GerritPersonIdent Provider<PersonIdent> serverIdent) {
|
||||||
|
this.repoManager = repoManager;
|
||||||
|
this.allUsersName = allUsersName;
|
||||||
|
this.serverIdent = serverIdent;
|
||||||
|
}
|
||||||
|
|
||||||
|
public AccountsUpdate create() {
|
||||||
|
PersonIdent i = serverIdent.get();
|
||||||
|
return new AccountsUpdate(repoManager, allUsersName, i, i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Factory to create an AccountsUpdate instance for updating accounts by the current user.
|
||||||
|
*
|
||||||
|
* <p>The identity of the current user will be used as author for all commits that update the
|
||||||
|
* accounts. The Gerrit server identity will be used as committer.
|
||||||
|
*/
|
||||||
|
@Singleton
|
||||||
|
public static class User {
|
||||||
|
private final GitRepositoryManager repoManager;
|
||||||
|
private final AllUsersName allUsersName;
|
||||||
|
private final Provider<PersonIdent> serverIdent;
|
||||||
|
private final Provider<IdentifiedUser> identifiedUser;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
public User(
|
||||||
|
GitRepositoryManager repoManager,
|
||||||
|
AllUsersName allUsersName,
|
||||||
|
@GerritPersonIdent Provider<PersonIdent> serverIdent,
|
||||||
|
Provider<IdentifiedUser> identifiedUser) {
|
||||||
|
this.repoManager = repoManager;
|
||||||
|
this.allUsersName = allUsersName;
|
||||||
|
this.serverIdent = serverIdent;
|
||||||
|
this.identifiedUser = identifiedUser;
|
||||||
|
}
|
||||||
|
|
||||||
|
public AccountsUpdate create() {
|
||||||
|
PersonIdent i = serverIdent.get();
|
||||||
|
return new AccountsUpdate(
|
||||||
|
repoManager, allUsersName, createPersonIdent(i, identifiedUser.get()), i);
|
||||||
|
}
|
||||||
|
|
||||||
|
private PersonIdent createPersonIdent(PersonIdent ident, IdentifiedUser user) {
|
||||||
|
return user.newCommitterIdent(ident.getWhen(), ident.getTimeZone());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private final GitRepositoryManager repoManager;
|
||||||
|
private final AllUsersName allUsersName;
|
||||||
|
private final PersonIdent committerIdent;
|
||||||
|
private final PersonIdent authorIdent;
|
||||||
|
|
||||||
|
private AccountsUpdate(
|
||||||
|
GitRepositoryManager repoManager,
|
||||||
|
AllUsersName allUsersName,
|
||||||
|
PersonIdent committerIdent,
|
||||||
|
PersonIdent authorIdent) {
|
||||||
|
this.repoManager = checkNotNull(repoManager, "repoManager");
|
||||||
|
this.allUsersName = checkNotNull(allUsersName, "allUsersName");
|
||||||
|
this.committerIdent = checkNotNull(committerIdent, "committerIdent");
|
||||||
|
this.authorIdent = checkNotNull(authorIdent, "authorIdent");
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Inserts a new account.
|
* Inserts a new account.
|
||||||
*
|
*
|
||||||
* @throws OrmDuplicateKeyException if the account already exists
|
* @throws OrmDuplicateKeyException if the account already exists
|
||||||
|
* @throws IOException if updating the user branch fails
|
||||||
*/
|
*/
|
||||||
public void insert(ReviewDb db, Account account) throws OrmException {
|
public void insert(ReviewDb db, Account account) throws OrmException, IOException {
|
||||||
db.accounts().insert(ImmutableSet.of(account));
|
db.accounts().insert(ImmutableSet.of(account));
|
||||||
|
createUserBranch(account);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Inserts or updates an account.
|
||||||
|
*
|
||||||
|
* <p>If the account already exists, it is overwritten, otherwise it is inserted.
|
||||||
|
*/
|
||||||
|
public void upsert(ReviewDb db, Account account) throws OrmException, IOException {
|
||||||
|
db.accounts().upsert(ImmutableSet.of(account));
|
||||||
|
createUserBranchIfNeeded(account);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void createUserBranch(Account account) throws IOException {
|
||||||
|
try (Repository repo = repoManager.openRepository(allUsersName);
|
||||||
|
ObjectInserter oi = repo.newObjectInserter()) {
|
||||||
|
String refName = RefNames.refsUsers(account.getId());
|
||||||
|
if (repo.exactRef(refName) != null) {
|
||||||
|
throw new IOException(
|
||||||
|
String.format(
|
||||||
|
"User branch %s for newly created account %s already exists.",
|
||||||
|
refName, account.getId().get()));
|
||||||
|
}
|
||||||
|
createUserBranch(repo, oi, committerIdent, authorIdent, account);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void createUserBranchIfNeeded(Account account) throws IOException {
|
||||||
|
try (Repository repo = repoManager.openRepository(allUsersName);
|
||||||
|
ObjectInserter oi = repo.newObjectInserter()) {
|
||||||
|
if (repo.exactRef(RefNames.refsUsers(account.getId())) == null) {
|
||||||
|
createUserBranch(repo, oi, committerIdent, authorIdent, account);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void createUserBranch(
|
||||||
|
Repository repo,
|
||||||
|
ObjectInserter oi,
|
||||||
|
PersonIdent committerIdent,
|
||||||
|
PersonIdent authorIdent,
|
||||||
|
Account account)
|
||||||
|
throws IOException {
|
||||||
|
ObjectId id =
|
||||||
|
createInitialEmptyCommit(oi, committerIdent, authorIdent, account.getRegisteredOn());
|
||||||
|
|
||||||
|
String refName = RefNames.refsUsers(account.getId());
|
||||||
|
RefUpdate ru = repo.updateRef(refName);
|
||||||
|
ru.setExpectedOldObjectId(ObjectId.zeroId());
|
||||||
|
ru.setNewObjectId(id);
|
||||||
|
ru.setForceUpdate(true);
|
||||||
|
ru.setRefLogIdent(committerIdent);
|
||||||
|
ru.setRefLogMessage("Create Account", true);
|
||||||
|
Result result = ru.update();
|
||||||
|
if (result != Result.NEW) {
|
||||||
|
throw new IOException(String.format("Failed to update ref %s: %s", refName, result.name()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ObjectId createInitialEmptyCommit(
|
||||||
|
ObjectInserter oi,
|
||||||
|
PersonIdent committerIdent,
|
||||||
|
PersonIdent authorIdent,
|
||||||
|
Timestamp registrationDate)
|
||||||
|
throws IOException {
|
||||||
|
CommitBuilder cb = new CommitBuilder();
|
||||||
|
cb.setTreeId(emptyTree(oi));
|
||||||
|
cb.setCommitter(new PersonIdent(committerIdent, registrationDate));
|
||||||
|
cb.setAuthor(new PersonIdent(authorIdent, registrationDate));
|
||||||
|
cb.setMessage("Create Account");
|
||||||
|
ObjectId id = oi.insert(cb);
|
||||||
|
oi.flush();
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ObjectId emptyTree(ObjectInserter oi) throws IOException {
|
||||||
|
return oi.insert(Constants.OBJ_TREE, new byte[] {});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -69,7 +69,7 @@ public class CreateAccount implements RestModifyView<TopLevelResource, AccountIn
|
|||||||
private final VersionedAuthorizedKeys.Accessor authorizedKeys;
|
private final VersionedAuthorizedKeys.Accessor authorizedKeys;
|
||||||
private final SshKeyCache sshKeyCache;
|
private final SshKeyCache sshKeyCache;
|
||||||
private final AccountCache accountCache;
|
private final AccountCache accountCache;
|
||||||
private final AccountsUpdate accountsUpdate;
|
private final AccountsUpdate.User accountsUpdate;
|
||||||
private final AccountIndexer indexer;
|
private final AccountIndexer indexer;
|
||||||
private final AccountByEmailCache byEmailCache;
|
private final AccountByEmailCache byEmailCache;
|
||||||
private final AccountLoader.Factory infoLoader;
|
private final AccountLoader.Factory infoLoader;
|
||||||
@@ -87,7 +87,7 @@ public class CreateAccount implements RestModifyView<TopLevelResource, AccountIn
|
|||||||
VersionedAuthorizedKeys.Accessor authorizedKeys,
|
VersionedAuthorizedKeys.Accessor authorizedKeys,
|
||||||
SshKeyCache sshKeyCache,
|
SshKeyCache sshKeyCache,
|
||||||
AccountCache accountCache,
|
AccountCache accountCache,
|
||||||
AccountsUpdate accountsUpdate,
|
AccountsUpdate.User accountsUpdate,
|
||||||
AccountIndexer indexer,
|
AccountIndexer indexer,
|
||||||
AccountByEmailCache byEmailCache,
|
AccountByEmailCache byEmailCache,
|
||||||
AccountLoader.Factory infoLoader,
|
AccountLoader.Factory infoLoader,
|
||||||
@@ -175,7 +175,7 @@ public class CreateAccount implements RestModifyView<TopLevelResource, AccountIn
|
|||||||
Account a = new Account(id, TimeUtil.nowTs());
|
Account a = new Account(id, TimeUtil.nowTs());
|
||||||
a.setFullName(input.name);
|
a.setFullName(input.name);
|
||||||
a.setPreferredEmail(input.email);
|
a.setPreferredEmail(input.email);
|
||||||
accountsUpdate.insert(db, a);
|
accountsUpdate.create().insert(db, a);
|
||||||
|
|
||||||
for (AccountGroup.Id groupId : groups) {
|
for (AccountGroup.Id groupId : groups) {
|
||||||
AccountGroupMember m = new AccountGroupMember(new AccountGroupMember.Key(id, groupId));
|
AccountGroupMember m = new AccountGroupMember(new AccountGroupMember.Key(id, groupId));
|
||||||
|
|||||||
@@ -35,7 +35,7 @@ import java.util.concurrent.TimeUnit;
|
|||||||
/** A version of the database schema. */
|
/** A version of the database schema. */
|
||||||
public abstract class SchemaVersion {
|
public abstract class SchemaVersion {
|
||||||
/** The current schema version. */
|
/** The current schema version. */
|
||||||
public static final Class<Schema_145> C = Schema_145.class;
|
public static final Class<Schema_146> C = Schema_146.class;
|
||||||
|
|
||||||
public static int getBinaryVersion() {
|
public static int getBinaryVersion() {
|
||||||
return guessVersion(C);
|
return guessVersion(C);
|
||||||
|
|||||||
@@ -0,0 +1,156 @@
|
|||||||
|
// Copyright (C) 2017 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.schema;
|
||||||
|
|
||||||
|
import com.google.gerrit.reviewdb.client.Account;
|
||||||
|
import com.google.gerrit.reviewdb.client.RefNames;
|
||||||
|
import com.google.gerrit.reviewdb.server.ReviewDb;
|
||||||
|
import com.google.gerrit.server.GerritPersonIdent;
|
||||||
|
import com.google.gerrit.server.account.AccountsUpdate;
|
||||||
|
import com.google.gerrit.server.config.AllUsersName;
|
||||||
|
import com.google.gerrit.server.git.GitRepositoryManager;
|
||||||
|
import com.google.gwtorm.server.OrmException;
|
||||||
|
import com.google.inject.Inject;
|
||||||
|
import com.google.inject.Provider;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
import org.eclipse.jgit.lib.CommitBuilder;
|
||||||
|
import org.eclipse.jgit.lib.Constants;
|
||||||
|
import org.eclipse.jgit.lib.ObjectId;
|
||||||
|
import org.eclipse.jgit.lib.ObjectInserter;
|
||||||
|
import org.eclipse.jgit.lib.PersonIdent;
|
||||||
|
import org.eclipse.jgit.lib.Ref;
|
||||||
|
import org.eclipse.jgit.lib.RefUpdate;
|
||||||
|
import org.eclipse.jgit.lib.RefUpdate.Result;
|
||||||
|
import org.eclipse.jgit.lib.Repository;
|
||||||
|
import org.eclipse.jgit.revwalk.RevCommit;
|
||||||
|
import org.eclipse.jgit.revwalk.RevSort;
|
||||||
|
import org.eclipse.jgit.revwalk.RevWalk;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Make sure that for every account a user branch exists that has an initial empty commit with the
|
||||||
|
* registration date as commit time.
|
||||||
|
*
|
||||||
|
* <p>For accounts that don't have a user branch yet the user branch is created with an initial
|
||||||
|
* empty commit that has the registration date as commit time.
|
||||||
|
*
|
||||||
|
* <p>For accounts that already have a user branch the user branch is rewritten and an initial empty
|
||||||
|
* commit with the registration date as commit time is inserted (if such a commit doesn't exist
|
||||||
|
* yet).
|
||||||
|
*/
|
||||||
|
public class Schema_146 extends SchemaVersion {
|
||||||
|
private static final String CREATE_ACCOUNT_MSG = "Create Account";
|
||||||
|
|
||||||
|
private final GitRepositoryManager repoManager;
|
||||||
|
private final AllUsersName allUsersName;
|
||||||
|
private final PersonIdent serverIdent;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
Schema_146(
|
||||||
|
Provider<Schema_145> prior,
|
||||||
|
GitRepositoryManager repoManager,
|
||||||
|
AllUsersName allUsersName,
|
||||||
|
@GerritPersonIdent PersonIdent serverIdent) {
|
||||||
|
super(prior);
|
||||||
|
this.repoManager = repoManager;
|
||||||
|
this.allUsersName = allUsersName;
|
||||||
|
this.serverIdent = serverIdent;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void migrateData(ReviewDb db, UpdateUI ui) throws OrmException, SQLException {
|
||||||
|
try (Repository repo = repoManager.openRepository(allUsersName);
|
||||||
|
RevWalk rw = new RevWalk(repo);
|
||||||
|
ObjectInserter oi = repo.newObjectInserter()) {
|
||||||
|
ObjectId emptyTree = emptyTree(oi);
|
||||||
|
|
||||||
|
for (Account account : db.accounts().all()) {
|
||||||
|
String refName = RefNames.refsUsers(account.getId());
|
||||||
|
Ref ref = repo.exactRef(refName);
|
||||||
|
if (ref != null) {
|
||||||
|
rewriteUserBranch(repo, rw, oi, emptyTree, ref, account);
|
||||||
|
} else {
|
||||||
|
AccountsUpdate.createUserBranch(repo, oi, serverIdent, serverIdent, account);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new OrmException("Failed to rewrite user branches.", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void rewriteUserBranch(
|
||||||
|
Repository repo, RevWalk rw, ObjectInserter oi, ObjectId emptyTree, Ref ref, Account account)
|
||||||
|
throws IOException {
|
||||||
|
ObjectId current = createInitialEmptyCommit(oi, emptyTree, account.getRegisteredOn());
|
||||||
|
|
||||||
|
rw.reset();
|
||||||
|
rw.sort(RevSort.TOPO);
|
||||||
|
rw.sort(RevSort.REVERSE, true);
|
||||||
|
rw.markStart(rw.parseCommit(ref.getObjectId()));
|
||||||
|
|
||||||
|
RevCommit c;
|
||||||
|
while ((c = rw.next()) != null) {
|
||||||
|
if (isInitialEmptyCommit(emptyTree, c)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
CommitBuilder cb = new CommitBuilder();
|
||||||
|
cb.setParentId(current);
|
||||||
|
cb.setTreeId(c.getTree());
|
||||||
|
cb.setAuthor(c.getAuthorIdent());
|
||||||
|
cb.setCommitter(c.getCommitterIdent());
|
||||||
|
cb.setMessage(c.getFullMessage());
|
||||||
|
cb.setEncoding(c.getEncoding());
|
||||||
|
current = oi.insert(cb);
|
||||||
|
}
|
||||||
|
|
||||||
|
oi.flush();
|
||||||
|
|
||||||
|
RefUpdate ru = repo.updateRef(ref.getName());
|
||||||
|
ru.setExpectedOldObjectId(ref.getObjectId());
|
||||||
|
ru.setNewObjectId(current);
|
||||||
|
ru.setForceUpdate(true);
|
||||||
|
ru.setRefLogIdent(serverIdent);
|
||||||
|
ru.setRefLogMessage(getClass().getSimpleName(), true);
|
||||||
|
Result result = ru.update();
|
||||||
|
if (result != Result.FORCED) {
|
||||||
|
throw new IOException(
|
||||||
|
String.format("Failed to update ref %s: %s", ref.getName(), result.name()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private ObjectId createInitialEmptyCommit(
|
||||||
|
ObjectInserter oi, ObjectId emptyTree, Timestamp registrationDate) throws IOException {
|
||||||
|
PersonIdent ident = new PersonIdent(serverIdent, registrationDate);
|
||||||
|
|
||||||
|
CommitBuilder cb = new CommitBuilder();
|
||||||
|
cb.setTreeId(emptyTree);
|
||||||
|
cb.setCommitter(ident);
|
||||||
|
cb.setAuthor(ident);
|
||||||
|
cb.setMessage(CREATE_ACCOUNT_MSG);
|
||||||
|
return oi.insert(cb);
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean isInitialEmptyCommit(ObjectId emptyTree, RevCommit c) {
|
||||||
|
return c.getParentCount() == 0
|
||||||
|
&& c.getTree().equals(emptyTree)
|
||||||
|
&& c.getShortMessage().equals(CREATE_ACCOUNT_MSG);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ObjectId emptyTree(ObjectInserter oi) throws IOException {
|
||||||
|
return oi.insert(Constants.OBJ_TREE, new byte[] {});
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user