Merge pull request #59 from mostlikelee/puppet-file-descriptor

bugfix: ensure file descriptor close on puppet file

Author: grahamgilbert

tables/puppet/yaml.go

@@ -19,27 +19,28 @@ func yamlPath() string {
 }
 
 func getPuppetYaml() (*PuppetInfo, error) {
-
 	var yamlData PuppetInfo
 
 	yamlFile, err := os.Open(yamlPath())
 	if err != nil {
 		log.Print(err)
 		return &yamlData, err
 	}
+	defer yamlFile.Close()
 
 	buf := new(bytes.Buffer)
 	_, err = buf.ReadFrom(yamlFile)
 	if err != nil {
 		return &yamlData, err
 	}
+
 	yamlString := buf.String()
 	yamlString = strings.Replace(yamlString, "\r", "\n", -1)
 
 	err = yaml.Unmarshal([]byte(yamlString), &yamlData)
 	if err != nil {
 		return &yamlData, err
 	}
-	yamlFile.Close()
+
 	return &yamlData, nil
 }