2 * Copyright 2000-2018 JetBrains s.r.o.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package jetbrains.buildServer.buildTriggers.vcs.git;
19 import jetbrains.buildServer.log.Loggers;
20 import jetbrains.buildServer.serverSide.BuildServerAdapter;
21 import jetbrains.buildServer.serverSide.BuildServerListener;
22 import jetbrains.buildServer.serverSide.ServerResponsibility;
23 import jetbrains.buildServer.serverSide.TeamCityProperties;
24 import jetbrains.buildServer.util.EventDispatcher;
25 import jetbrains.buildServer.util.ThreadUtil;
26 import jetbrains.buildServer.util.executors.ExecutorsFactory;
27 import jetbrains.buildServer.vcs.*;
28 import org.eclipse.jgit.lib.Repository;
29 import org.jetbrains.annotations.NotNull;
31 import java.util.HashSet;
33 import java.util.concurrent.ConcurrentHashMap;
34 import java.util.concurrent.ExecutorService;
36 public class GitClonesUpdater {
37 private final ConcurrentHashMap<VcsRoot, RepositoryStateData> myScheduledForUpdate = new ConcurrentHashMap<>();
38 private final GitVcsSupport myVcs;
39 private final RepositoryManager myRepositoryManager;
40 private volatile ExecutorService myExecutor;
42 public GitClonesUpdater(@NotNull EventDispatcher<RepositoryStateListener> eventDispatcher,
43 @NotNull EventDispatcher<BuildServerListener> serverEventDispatcher,
44 @NotNull ServerResponsibility serverResponsibility,
45 @NotNull GitVcsSupport gitVcsSupport,
46 @NotNull RepositoryManager repositoryManager) {
47 myVcs = gitVcsSupport;
48 myRepositoryManager = repositoryManager;
50 eventDispatcher.addListener(new RepositoryStateListenerAdapter() {
52 public void repositoryStateChanged(@NotNull final VcsRoot root,
53 @NotNull final RepositoryState oldState,
54 @NotNull final RepositoryState newState) {
55 if (serverResponsibility.canCheckForChanges()) return;
57 if (!TeamCityProperties.getBooleanOrTrue("teamcity.git.localClones.updateIfNoCheckingForChangesResponsibility")) return;
59 if (root.getVcsName().equals(Constants.VCS_NAME)) {
60 myScheduledForUpdate.put(root, RepositoryStateFactory.toData(newState));
61 synchronized (myScheduledForUpdate) {
62 if (myExecutor == null) {
63 myExecutor = ExecutorsFactory.newFixedDaemonExecutor("Git local clones updater",
64 TeamCityProperties.getInteger("teamcity.git.localClones.maxParallelUpdateThreads", 2));
67 myExecutor.submit(GitClonesUpdater.this::processVcsRootsScheduledForUpdate);
72 serverEventDispatcher.addListener(new BuildServerAdapter() {
74 public void serverShutdown() {
75 if (myExecutor != null) {
76 ThreadUtil.shutdownGracefully(myExecutor, "Git local clones updater");
82 private void processVcsRootsScheduledForUpdate() {
83 Set<VcsRoot> vcsRoots = new HashSet<>(myScheduledForUpdate.keySet());
84 for (VcsRoot root: vcsRoots) {
85 RepositoryStateData state = myScheduledForUpdate.remove(root);
86 if (state == null) continue;
88 OperationContext context = myVcs.createContext(root, "updating local clone");
90 GitVcsRoot gitRoot = context.getGitRoot();
91 myRepositoryManager.runWithDisabledRemove(gitRoot.getRepositoryDir(), () -> {
92 Repository repo = context.getRepository();
94 myVcs.getCollectChangesPolicy().ensureRepositoryStateLoadedFor(context, repo, true, state);
95 } catch (Exception e1) {
96 throw new VcsException(e1);
99 } catch (VcsException e1) {
100 Loggers.VCS.warnAndDebugDetails("Could not update local clone for: " + LogUtil.describe(root), e1);