Skip to content

Commit

Permalink
Adds an AfterFiles field to nodetasks.File and makes sure CoreOS uses it
Browse files Browse the repository at this point in the history
  • Loading branch information
louismunro committed Mar 26, 2018
1 parent 4d2bedc commit 487dc33
Show file tree
Hide file tree
Showing 4 changed files with 80 additions and 3 deletions.
7 changes: 4 additions & 3 deletions nodeup/pkg/model/docker.go
Original file line number Diff line number Diff line change
Expand Up @@ -718,9 +718,10 @@ func (b *DockerBuilder) buildContainerOSConfigurationDropIn(c *fi.ModelBuilderCo
contents := strings.Join(lines, "\n")

c.AddTask(&nodetasks.File{
Path: "/etc/systemd/system/docker.service.d/10-kops.conf",
Contents: fi.NewStringResource(contents),
Type: nodetasks.FileType_File,
AfterFiles: []string{"/etc/sysconfig/docker"},
Path: "/etc/systemd/system/docker.service.d/10-kops.conf",
Contents: fi.NewStringResource(contents),
Type: nodetasks.FileType_File,
OnChangeExecute: [][]string{
{"systemctl", "daemon-reload"},
{"systemctl", "restart", "docker.service"},
Expand Down
1 change: 1 addition & 0 deletions upup/pkg/fi/nodeup/nodetasks/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ go_test(
srcs = [
"archive_test.go",
"bindmount_test.go",
"file_test.go",
"loadimage_test.go",
"service_test.go",
],
Expand Down
12 changes: 12 additions & 0 deletions upup/pkg/fi/nodeup/nodetasks/file.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ const FileType_Directory = "directory"
const FileType_File = "file"

type File struct {
AfterFiles []string `json:"afterfiles,omitempty"`
Contents fi.Resource `json:"contents,omitempty"`
Group *string `json:"group,omitempty"`
IfNotExists bool `json:"ifNotExists,omitempty"`
Expand Down Expand Up @@ -103,6 +104,17 @@ func (e *File) GetDependencies(tasks map[string]fi.Task) []fi.Task {
deps = append(deps, v)
}

// Requires other files to be created first
for _, f := range e.AfterFiles {
for _, v := range tasks {
if file, ok := v.(*File); ok {
if file.Path == f {
deps = append(deps, v)
}
}
}
}

return deps
}

Expand Down
63 changes: 63 additions & 0 deletions upup/pkg/fi/nodeup/nodetasks/file_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
/*
Copyright 2017 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package nodetasks

import (
"testing"

"k8s.io/kops/upup/pkg/fi"
)

func TestFileDependencies(t *testing.T) {
parentFileName := "/dependedon"
childFileName := "/dependant"

grid := []struct {
parent fi.Task
child fi.Task
}{
{
parent: &File{
Path: parentFileName,
Contents: fi.NewStringResource("I am depended on by " + childFileName),
Type: FileType_File,
},
child: &File{
AfterFiles: []string{parentFileName},
Path: childFileName,
Contents: fi.NewStringResource("I depend on " + parentFileName),
Type: FileType_File,
},
},
}

for _, g := range grid {
allTasks := make(map[string]fi.Task)
allTasks["parent"] = g.parent
allTasks["child"] = g.child

deps := g.parent.(fi.HasDependencies).GetDependencies(allTasks)
if len(deps) != 0 {
t.Errorf("found unexpected dependencies for parent: %v %v", g.parent, deps)
}

childDeps := g.child.(fi.HasDependencies).GetDependencies(allTasks)
if len(childDeps) != 1 {
t.Errorf("found unexpected dependencies for child: %v %v", g.child, childDeps)
}
}
}

0 comments on commit 487dc33

Please sign in to comment.