Implement Lazy Loading of NameTable

Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>

perflib/nametable.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"fmt"
 	"strconv"
+	"sync"
 )
 
 // Initialize global name tables
@@ -18,54 +19,63 @@ func (p *perfObjectType) LookupName() string {
 }
 
 type NameTable struct {
-	byIndex  map[uint32]string
-	byString map[string]uint32
+	once sync.Once
+
+	name string
+
+	table struct {
+		index  map[uint32]string
+		string map[string]uint32
+	}
 }
 
 func (t *NameTable) LookupString(index uint32) string {
-	return t.byIndex[index]
+	t.initialize()
+	return t.table.index[index]
 }
 
 func (t *NameTable) LookupIndex(str string) uint32 {
-	return t.byString[str]
+	t.initialize()
+	return t.table.string[str]
 }
 
 // QueryNameTable Query a perflib name table from the registry. Specify the type and the language
 // code (i.e. "Counter 009" or "Help 009") for English language.
 func QueryNameTable(tableName string) *NameTable {
-	nameTable := new(NameTable)
-	nameTable.byIndex = make(map[uint32]string)
-
-	buffer, err := queryRawData(tableName)
-	if err != nil {
-		panic(err)
+	return &NameTable{
+		name: tableName,
 	}
-	r := bytes.NewReader(buffer)
-	for {
-		index, err := readUTF16String(r)
-		if err != nil {
-			break
-		}
-
-		desc, err := readUTF16String(r)
-		if err != nil {
-			break
-		}
-
-		indexInt, _ := strconv.Atoi(index)
-
-		if err != nil {
-			panic(fmt.Sprint("Invalid index ", index))
-		}
-
-		nameTable.byIndex[uint32(indexInt)] = desc
-	}
-
-	nameTable.byString = make(map[string]uint32)
-
-	for k, v := range nameTable.byIndex {
-		nameTable.byString[v] = k
-	}
-
-	return nameTable
+}
+
+func (t *NameTable) initialize() {
+	t.once.Do(func() {
+		t.table.index = make(map[uint32]string)
+		t.table.string = make(map[string]uint32)
+
+		buffer, err := queryRawData(t.name)
+		if err != nil {
+			panic(err)
+		}
+		r := bytes.NewReader(buffer)
+		for {
+			index, err := readUTF16String(r)
+			if err != nil {
+				break
+			}
+
+			desc, err := readUTF16String(r)
+			if err != nil {
+				break
+			}
+
+			if err != nil {
+				panic(fmt.Sprint("Invalid index ", index))
+			}
+
+			indexInt, _ := strconv.Atoi(index)
+
+			t.table.index[uint32(indexInt)] = desc
+			t.table.string[desc] = uint32(indexInt)
+		}
+	})
 }