Skip to content

Commit 54734dc

Browse files
feat(worker): expose SSH port for pam_slurm_adopt
Add SSH port (22) to worker container specification to enable users to SSH into worker nodes where they have running jobs. This works with updated slurmd images that include pam_slurm_adopt for access control.
1 parent ad50184 commit 54734dc

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

internal/builder/worker_app.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import (
2121

2222
const (
2323
SlurmdPort = 6818
24+
SshPort = 22
2425

2526
slurmdUser = "root"
2627

@@ -113,6 +114,11 @@ func (b *Builder) slurmdContainer(nodeset *slinkyv1alpha1.NodeSet, controller *s
113114
ContainerPort: SlurmdPort,
114115
Protocol: corev1.ProtocolTCP,
115116
},
117+
{
118+
Name: "ssh",
119+
ContainerPort: SshPort,
120+
Protocol: corev1.ProtocolTCP,
121+
},
116122
},
117123
StartupProbe: &corev1.Probe{
118124
ProbeHandler: corev1.ProbeHandler{

internal/builder/worker_app_test.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,18 @@ func TestBuilder_BuildWorkerPodTemplate(t *testing.T) {
9393
t.Errorf("Containers[0].Ports[0].ContainerPort = %v , want = %v",
9494
got.Spec.Containers[0].Ports[0].Name, SlurmdPort)
9595

96+
case len(got.Spec.Containers[0].Ports) != 2:
97+
t.Errorf("len(Containers[0].Ports) = %v , want = 2 (slurmd + ssh)",
98+
len(got.Spec.Containers[0].Ports))
99+
100+
case got.Spec.Containers[0].Ports[1].Name != "ssh":
101+
t.Errorf("Containers[0].Ports[1].Name = %v , want = ssh",
102+
got.Spec.Containers[0].Ports[1].Name)
103+
104+
case got.Spec.Containers[0].Ports[1].ContainerPort != SshPort:
105+
t.Errorf("Containers[0].Ports[1].ContainerPort = %v , want = %v",
106+
got.Spec.Containers[0].Ports[1].ContainerPort, SshPort)
107+
96108
case got.Spec.Subdomain == "":
97109
t.Errorf("Subdomain = %v , want = non-empty", got.Spec.Subdomain)
98110

0 commit comments

Comments
 (0)