Skip to content

Commit e00fcbd

Browse files
move private networking to profile setting
1 parent 1fe25e3 commit e00fcbd

4 files changed

Lines changed: 6 additions & 9 deletions

File tree

CloudController.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ private async Task<string> GenerateName()
6464
return name;
6565
}
6666

67-
public async Task<Machine> CreateNewRunner(string arch, string size, string runnerToken, string targetName, bool isCustom = false, string profileName = "default", bool usePrivateNetworks = false)
67+
public async Task<Machine> CreateNewRunner(string arch, string size, string runnerToken, string targetName, bool isCustom = false, string profileName = "default")
6868
{
6969

7070
// Select VM size for job - All AMD
@@ -164,7 +164,7 @@ public async Task<Machine> CreateNewRunner(string arch, string size, string runn
164164
}
165165
try
166166
{
167-
var privateNetworks = usePrivateNetworks ? networks.Select(x => x.Id).ToList() : new List<long>();
167+
var privateNetworks = profile.UsePrivateNetworks ? networks.Select(x => x.Id).ToList() : new List<long>();
168168

169169
newSrv = await _client.Server.Create(dataCenters[ct], imageId.Value, name, srvType.Value, userData: cloudInitcontent, sshKeysIds: srvKeys, privateNetoworksIds: privateNetworks);
170170
success = true;

Models/RunnerProfile.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,5 @@ public class RunnerProfile
77
public string OsImageName { get; set; }
88
public bool IsCustomImage { get; set; }
99
public string Name { get; set; }
10+
public bool UsePrivateNetworks { get; set; }
1011
}

PoolManager.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -558,7 +558,7 @@ private async Task<bool> CreateRunner(CreateRunnerTask rt)
558558
TargetType.Organization => runner.Owner,
559559
_ => throw new ArgumentOutOfRangeException()
560560
};
561-
Machine newRunner = await _cc.CreateNewRunner(runner.Arch, runner.Size, rt.RunnerToken, targetName, runner.IsCustom, runner.Profile, runner.UsePrivateNetwork);
561+
Machine newRunner = await _cc.CreateNewRunner(runner.Arch, runner.Size, rt.RunnerToken, targetName, runner.IsCustom, runner.Profile);
562562
_logger.LogInformation($"New Runner {newRunner.Name} [{runner.Size} on {runner.Arch}] entering pool for {targetName}.");
563563
MachineCreatedCount.Labels(runner.Owner, runner.Size).Inc();
564564

Program.cs

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -340,10 +340,7 @@ private static async Task<IResult> RunnerStateReportHandler(HttpRequest request,
340340
Arch = runner.Arch,
341341
IPv4 = string.Empty,
342342
IsCustom = runner.IsCustom,
343-
Owner = runner.Owner,
344-
UsePrivateNetwork = runner.UsePrivateNetwork
345-
346-
343+
Owner = runner.Owner
347344
};
348345
await db.Runners.AddAsync(newRunner);
349346
await db.SaveChangesAsync();
@@ -632,8 +629,7 @@ private static async Task JobQueued(ILogger<Program> logger, string repoName, Li
632629
Arch = arch,
633630
IPv4 = string.Empty,
634631
IsCustom = isCustom,
635-
Owner = owner,
636-
UsePrivateNetwork = labels.Contains("use-cache-server")
632+
Owner = owner
637633

638634
};
639635
await db.Runners.AddAsync(newRunner);

0 commit comments

Comments
 (0)