zishuo/onepanel
1
0
Fork 0
mirror of https://github.com/onepanelio/onepanel.git synced 2025-11-03 10:01:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adding systemConfig to function.

Browse Source 
- Also added error return
This commit is contained in:
Aleksandr Melnikov
2020-09-18 15:50:28 -07:00
parent aaf20b4ab6
commit 31076bc70d
2 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/workflow_execution.go
View File

@@ -324,8 +324,14 @@ func (c *Client) createWorkflow(namespace string, workflowTemplateID uint64, wor
if err = c.injectAutomatedFields(namespace, wf, opts); err != nil {
return nil, err
}
wf = ensureWorkflowRunsOnDedicatedNode(wf)
systemConfig, err := c.GetSystemConfig()
if err != nil {
return nil, err
}
wf, err = ensureWorkflowRunsOnDedicatedNode(wf, systemConfig)
if err != nil {
return nil, err
}
createdArgoWorkflow, err := c.ArgoprojV1alpha1().Workflows(namespace).Create(wf)
if err != nil {
return nil, err
@@ -355,7 +361,7 @@ func (c *Client) createWorkflow(namespace string, workflowTemplateID uint64, wor
return
}
func ensureWorkflowRunsOnDedicatedNode(wf *wfv1.Workflow) *wfv1.Workflow {
func ensureWorkflowRunsOnDedicatedNode(wf *wfv1.Workflow, config SystemConfig) (*wfv1.Workflow, error) {
antiAffinityLabelKey := "onepanel.io/reserves-instance-type"
nodeSelectorVal := "singular-workflow"
for i := range wf.Spec.Templates {