enable outputing lmap only

Author: OneOfOne

cmd/cmap-gen/main.go

@@ -26,6 +26,7 @@ var (
 	valueType = flag.String("vt", "interface{}", "value type")
 	hashFn    = flag.String("hfn", "cmap.DefaultKeyHasher", "hash func")
 	pkgPath   = flag.String("p", "", "package path")
+	lmapOnly  = flag.Bool("lmap", false, "generate LMap only")
 
 	toStdout = flag.Bool("stdout", false, "write the output to stdout rather than creating a package")
 
@@ -65,7 +66,12 @@ func main() {
 		}
 	}
 
-	r, closeReaders := multiFileReader(filepath.Join(cmapBase, "cmap.go"), filepath.Join(cmapBase, "lmap.go"))
+	files := []string{filepath.Join(cmapBase, "cmap.go"), filepath.Join(cmapBase, "lmap.go")}
+	if *lmapOnly {
+		files = files[1:]
+	}
+
+	r, closeReaders := multiFileReader(files...)
 	var (
 		filters = getLineFilters()
 		buf     bytes.Buffer
@@ -100,7 +106,12 @@ package %s
 		log.Fatalf("error compiling code: \n\t%s\n\n%s", strings.Join(errs, "\n\t"), src)
 	}
 
-	fp := filepath.Join(*pkgPath, "cmap.go")
+	var fp string
+	if *lmapOnly {
+		fp = filepath.Join(*pkgPath, "lmap.go")
+	} else {
+		fp = filepath.Join(*pkgPath, "cmap.go")
+	}
 
 	if *verbose {
 		log.Printf("writing typed cmap %s.CMap[%s][%s] (using HashFn: %s) to file %s...", *pkgName, *keyType, *valueType, *hashFn, fp)