Add Close() method to Project to release resources

Problem Observed

Goroutine leaks because there is no way to "close" projects

Problem Detail

NewProject() creates a defaultListener in it.
The listener starts a new goroutine in NewDefaultListener().
We currently don't have a method to stop this goroutine.

Suggested Resolution

This commit adds a new method Close() to release resources
tied to a Project.

Signed-off-by: Iwasaki Yudai <[email protected]>

Author: yudai

project/listener.go

@@ -39,13 +39,13 @@ type defaultListener struct {
 }
 
 // NewDefaultListener create a default listener for the specified project.
-func NewDefaultListener(p *Project) chan<- events.Event {
-	l := defaultListener{
+func NewDefaultListener(p *Project) *defaultListener {
+	l := &defaultListener{
 		listenChan: make(chan events.Event),
 		project:    p,
 	}
 	go l.start()
-	return l.listenChan
+	return l
 }
 
 func (d *defaultListener) start() {
@@ -79,3 +79,7 @@ func (d *defaultListener) start() {
 		}
 	}
 }
+
+func (d *defaultListener) close() {
+	close(d.listenChan)
+}

project/project.go

@@ -35,15 +35,15 @@ type Project struct {
 	ReloadCallback func() error
 	ParseOptions   *config.ParseOptions
 
-	runtime       RuntimeProject
-	networks      Networks
-	volumes       Volumes
-	configVersion string
-	context       *Context
-	reload        []string
-	upCount       int
-	listeners     []chan<- events.Event
-	hasListeners  bool
+	runtime         RuntimeProject
+	networks        Networks
+	volumes         Volumes
+	configVersion   string
+	context         *Context
+	reload          []string
+	upCount         int
+	listeners       []chan<- events.Event
+	defaultListener *defaultListener
 }
 
 // NewProject creates a new project with the specified context.
@@ -93,11 +93,18 @@ func NewProject(context *Context, runtime RuntimeProject, parseOptions *config.P
 
 	context.Project = p
 
-	p.listeners = []chan<- events.Event{NewDefaultListener(p)}
+	p.defaultListener = NewDefaultListener(p)
+	p.listeners = []chan<- events.Event{p.defaultListener.listenChan}
 
 	return p
 }
 
+// Close releases resources attached to the project
+func (p *Project) Close() error {
+	p.defaultListener.close()
+	return nil
+}
+
 // Parse populates project information based on its context. It sets up the name,
 // the composefile and the composebytes (the composefile content).
 func (p *Project) Parse() error {
@@ -511,11 +518,7 @@ func (p *Project) traverse(start bool, selected map[string]bool, wrappers map[st
 // AddListener adds the specified listener to the project.
 // This implements implicitly events.Emitter.
 func (p *Project) AddListener(c chan<- events.Event) {
-	if !p.hasListeners {
-		for _, l := range p.listeners {
-			close(l)
-		}
-		p.hasListeners = true
+	if len(p.listeners) == 1 && p.listeners[0] == p.defaultListener.listenChan {
 		p.listeners = []chan<- events.Event{c}
 	} else {
 		p.listeners = append(p.listeners, c)

project/project_test.go

@@ -64,6 +64,7 @@ func TestTwoCall(t *testing.T) {
 	p := NewProject(&Context{
 		ServiceFactory: factory,
 	}, nil, nil)
+	defer p.Close()
 	p.ServiceConfigs = config.NewServiceConfigs()
 	p.ServiceConfigs.Add("foo", &config.ServiceConfig{})
 
@@ -83,6 +84,7 @@ func TestTwoCall(t *testing.T) {
 func TestGetServiceConfig(t *testing.T) {
 
 	p := NewProject(&Context{}, nil, nil)
+	defer p.Close()
 	p.ServiceConfigs = config.NewServiceConfigs()
 	fooService := &config.ServiceConfig{}
 	p.ServiceConfigs.Add("foo", fooService)
@@ -112,6 +114,7 @@ func TestParseWithBadContent(t *testing.T) {
 			[]byte("garbage"),
 		},
 	}, nil, nil)
+	defer p.Close()
 
 	err := p.Parse()
 	if err == nil {
@@ -129,6 +132,7 @@ func TestParseWithGoodContent(t *testing.T) {
 			[]byte("not-garbage:\n  image: foo"),
 		},
 	}, nil, nil)
+	defer p.Close()
 
 	err := p.Parse()
 	if err != nil {
@@ -142,6 +146,7 @@ func TestParseWithDefaultEnvironmentLookup(t *testing.T) {
 			[]byte("not-garbage:\n  image: foo:${version}"),
 		},
 	}, nil, nil)
+	defer p.Close()
 
 	err := p.Parse()
 	if err != nil {
@@ -165,6 +170,7 @@ func TestEnvironmentResolve(t *testing.T) {
 		ServiceFactory:    factory,
 		EnvironmentLookup: &TestEnvironmentLookup{},
 	}, nil, nil)
+	defer p.Close()
 	p.ServiceConfigs = config.NewServiceConfigs()
 	p.ServiceConfigs.Add("foo", &config.ServiceConfig{
 		Environment: yaml.MaporEqualSlice([]string{
@@ -209,6 +215,7 @@ func TestParseWithMultipleComposeFiles(t *testing.T) {
 	p := NewProject(&Context{
 		ComposeBytes: [][]byte{configOne, configTwo},
 	}, nil, nil)
+	defer p.Close()
 
 	err := p.Parse()
 
@@ -222,6 +229,7 @@ func TestParseWithMultipleComposeFiles(t *testing.T) {
 	p = NewProject(&Context{
 		ComposeBytes: [][]byte{configTwo, configOne},
 	}, nil, nil)
+	defer p.Close()
 
 	err = p.Parse()
 
@@ -235,6 +243,7 @@ func TestParseWithMultipleComposeFiles(t *testing.T) {
 	p = NewProject(&Context{
 		ComposeBytes: [][]byte{configOne, configTwo, configThree},
 	}, nil, nil)
+	defer p.Close()
 
 	err = p.Parse()