GitRemoteFeaturesBase.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
 
namespace Tgstation.Server.Host.Components.Repository
{
    abstract class GitRemoteFeaturesBase : IGitRemoteFeatures
    {
        public abstract string TestMergeRefSpecFormatter { get; }
 
        public abstract string TestMergeLocalBranchNameFormatter { get; }
 
        public abstract RemoteGitProvider? RemoteGitProvider { get; }
 
        public string RemoteRepositoryOwner { get; }
 
        public string RemoteRepositoryName { get; }
 
        protected ILogger<GitRemoteFeaturesBase> Logger { get; }
 
        readonly Dictionary<TestMergeParameters, Models.TestMerge> cachedLookups;
 
        public GitRemoteFeaturesBase(ILogger<GitRemoteFeaturesBase> logger, Uri remoteUrl)
        {
            Logger = logger ?? throw new ArgumentNullException(nameof(logger));
            ArgumentNullException.ThrowIfNull(remoteUrl);
 
            RemoteRepositoryOwner = remoteUrl.Segments[1].TrimEnd('/');
            RemoteRepositoryName = remoteUrl.Segments[2].TrimEnd('/');
            if (RemoteRepositoryName.EndsWith(".git", StringComparison.OrdinalIgnoreCase))
                RemoteRepositoryName = RemoteRepositoryName[0..^4];
 
            cachedLookups = new Dictionary<TestMergeParameters, Models.TestMerge>();
        }
 
        public async ValueTask<Models.TestMerge> GetTestMerge(
            TestMergeParameters parameters,
            RepositorySettings repositorySettings,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(parameters);
            ArgumentNullException.ThrowIfNull(repositorySettings);
 
            Models.TestMerge? result;
            lock (cachedLookups)
                if (cachedLookups.TryGetValue(parameters, out result))
                    Logger.LogTrace("Using cache for test merge #{testMergeNumber}", parameters.Number);
 
            if (result == null)
            {
                Logger.LogTrace("Retrieving metadata for test merge #{testMergeNumber}...", parameters.Number);
                result = await GetTestMergeImpl(parameters, repositorySettings, cancellationToken);
                lock (cachedLookups)
                    if (!cachedLookups.TryAdd(parameters, result))
                        Logger.LogError("Race condition on adding test merge #{testMergeNumber}!", parameters.Number);
            }
 
            return result;
        }
 
        protected abstract ValueTask<Models.TestMerge> GetTestMergeImpl(
            TestMergeParameters parameters,
            RepositorySettings repositorySettings,
            CancellationToken cancellationToken);
    }
}