Break core structs down into obj and leak packages

leak/finder.go

@@ -0,0 +1,81 @@
+package leak
+
+import (
+	"fmt"
+
+	"github.com/jimeh/rbheapleak/obj"
+)
+
+func NewFinder(file1, file2, file3 string) *Finder {
+	return &Finder{
+		FilePaths: [3]string{file1, file2, file3},
+	}
+}
+
+type Finder struct {
+	FilePaths [3]string
+	Dumps     [3]*obj.Dump
+	Leaks     []*string
+	Verbose   bool
+}
+
+func (s *Finder) Process() error {
+	for i, filePath := range s.FilePaths {
+		s.log(fmt.Sprintf("Parsing %s", filePath))
+		dump := obj.NewDump(filePath)
+
+		err := dump.Process()
+		if err != nil {
+			return err
+		}
+
+		s.Dumps[i] = dump
+		s.log(fmt.Sprintf("Parsed %d objects", len(dump.Index)))
+	}
+
+	return nil
+}
+
+func (s *Finder) PrintLeakedAddresses() {
+	s.log("\nLeaked Addresses:")
+	s.Dumps[1].PrintEntryAddress(s.FindLeaks())
+}
+
+func (s *Finder) PrintLeakedObjects() {
+	s.log("\nLeaked Objects:")
+	s.Dumps[1].PrintEntryJSON(s.FindLeaks())
+}
+
+func (s *Finder) FindLeaks() []*string {
+	if s.Leaks != nil {
+		return s.Leaks
+	}
+
+	mapA := map[string]bool{}
+	mapC := map[string]bool{}
+
+	for _, x := range s.Dumps[0].Index {
+		mapA[*x] = true
+	}
+
+	for _, x := range s.Dumps[2].Index {
+		mapC[*x] = true
+	}
+
+	for _, x := range s.Dumps[1].Index {
+		_, okA := mapA[*x]
+		_, okC := mapC[*x]
+
+		if !okA && okC {
+			s.Leaks = append(s.Leaks, x)
+		}
+	}
+
+	return s.Leaks
+}
+
+func (s *Finder) log(msg string) {
+	if s.Verbose {
+		fmt.Println(msg)
+	}
+}