From 5d1265926680b30058f0314f48cea7bf01633627 Mon Sep 17 00:00:00 2001
From: whyrusleeping <why@ipfs.io>
Date: Tue, 10 Dec 2019 14:01:17 +0100
Subject: [PATCH] allow lotus-bench to benchmark existing sectorbuilders

---
 cmd/lotus-bench/main.go | 71 ++++++++++++++++++++++++++---------------
 cmd/lotus-seed/main.go  |  7 ++++
 2 files changed, 52 insertions(+), 26 deletions(-)

diff --git a/cmd/lotus-bench/main.go b/cmd/lotus-bench/main.go
index 7964186f2..c6c58e8fe 100644
--- a/cmd/lotus-bench/main.go
+++ b/cmd/lotus-bench/main.go
@@ -68,27 +68,41 @@ func main() {
 				Name:  "no-gpu",
 				Usage: "disable gpu usage for the benchmark run",
 			},
+			&cli.StringFlag{
+				Name:  "benchmark-existing-sectorbuilder",
+				Usage: "pass a directory to run election-post timings on an existing sectorbuilder",
+			},
 		},
 		Action: func(c *cli.Context) error {
 			if c.Bool("no-gpu") {
 				os.Setenv("BELLMAN_NO_GPU", "1")
 			}
-			sdir, err := homedir.Expand(c.String("storage-dir"))
-			if err != nil {
-				return err
-			}
 
-			os.MkdirAll(sdir, 0775)
+			robench := c.String("benchmark-existing-sectorbuilder")
 
-			tsdir, err := ioutil.TempDir(sdir, "bench")
-			if err != nil {
-				return err
-			}
-			defer func() {
-				if err := os.RemoveAll(tsdir); err != nil {
-					log.Warn("remove all: ", err)
+			var sbdir string
+
+			if robench == "" {
+				sdir, err := homedir.Expand(c.String("storage-dir"))
+				if err != nil {
+					return err
 				}
-			}()
+
+				os.MkdirAll(sdir, 0775)
+
+				tsdir, err := ioutil.TempDir(sdir, "bench")
+				if err != nil {
+					return err
+				}
+				defer func() {
+					if err := os.RemoveAll(tsdir); err != nil {
+						log.Warn("remove all: ", err)
+					}
+				}()
+				sbdir = tsdir
+			} else {
+				sbdir = robench
+			}
 
 			maddr, err := address.NewFromString("t0101")
 			if err != nil {
@@ -102,14 +116,17 @@ func main() {
 				Miner:         maddr,
 				SectorSize:    sectorSize,
 				WorkerThreads: 2,
-				CacheDir:      filepath.Join(tsdir, "cache"),
-				SealedDir:     filepath.Join(tsdir, "sealed"),
-				StagedDir:     filepath.Join(tsdir, "staged"),
-				UnsealedDir:   filepath.Join(tsdir, "unsealed"),
+				CacheDir:      filepath.Join(sbdir, "cache"),
+				SealedDir:     filepath.Join(sbdir, "sealed"),
+				StagedDir:     filepath.Join(sbdir, "staged"),
+				UnsealedDir:   filepath.Join(sbdir, "unsealed"),
 			}
-			for _, d := range []string{cfg.CacheDir, cfg.SealedDir, cfg.StagedDir, cfg.UnsealedDir} {
-				if err := os.MkdirAll(d, 0775); err != nil {
-					return err
+
+			if robench == "" {
+				for _, d := range []string{cfg.CacheDir, cfg.SealedDir, cfg.StagedDir, cfg.UnsealedDir} {
+					if err := os.MkdirAll(d, 0775); err != nil {
+						return err
+					}
 				}
 			}
 
@@ -126,7 +143,7 @@ func main() {
 			var sealTimings []SealingResult
 			var sealedSectors []ffi.PublicSectorInfo
 			numSectors := uint64(1)
-			for i := uint64(1); i <= numSectors; i++ {
+			for i := uint64(1); i <= numSectors && robench == ""; i++ {
 				start := time.Now()
 				log.Info("Writing piece into sector...")
 
@@ -267,11 +284,13 @@ func main() {
 			} // TODO: optionally write this as json to a file
 
 			fmt.Println("results")
-			fmt.Printf("seal: addPiece: %s\n", benchout.SealingResults[0].AddPiece) // TODO: average across multiple sealings
-			fmt.Printf("seal: preCommit: %s\n", benchout.SealingResults[0].PreCommit)
-			fmt.Printf("seal: Commit: %s\n", benchout.SealingResults[0].Commit)
-			fmt.Printf("seal: Verify: %s\n", benchout.SealingResults[0].Verify)
-			fmt.Printf("unseal: %s\n", benchout.SealingResults[0].Unseal)
+			if robench == "" {
+				fmt.Printf("seal: addPiece: %s\n", benchout.SealingResults[0].AddPiece) // TODO: average across multiple sealings
+				fmt.Printf("seal: preCommit: %s\n", benchout.SealingResults[0].PreCommit)
+				fmt.Printf("seal: Commit: %s\n", benchout.SealingResults[0].Commit)
+				fmt.Printf("seal: Verify: %s\n", benchout.SealingResults[0].Verify)
+				fmt.Printf("unseal: %s\n", benchout.SealingResults[0].Unseal)
+			}
 			fmt.Printf("generate candidates: %s\n", benchout.PostGenerateCandidates)
 			fmt.Printf("compute epost proof (cold): %s\n", benchout.PostEProofCold)
 			fmt.Printf("compute epost proof (hot): %s\n", benchout.PostEProofHot)
diff --git a/cmd/lotus-seed/main.go b/cmd/lotus-seed/main.go
index 5271936c4..8bb4ace73 100644
--- a/cmd/lotus-seed/main.go
+++ b/cmd/lotus-seed/main.go
@@ -204,6 +204,7 @@ var aggregateSectorDirsCmd = &cli.Command{
 			return err
 		}
 
+		var aggrGenMiner genesis.GenesisMiner
 		var highestSectorID uint64
 		for _, dir := range cctx.Args().Slice() {
 			dir, err := homedir.Expand(dir)
@@ -244,6 +245,8 @@ var aggregateSectorDirsCmd = &cli.Command{
 				}
 			}
 
+			aggrGenMiner = mergeGenMiners(aggrGenMiner, genm)
+
 			opts := badger.DefaultOptions
 			opts.ReadOnly = true
 			mds, err := badger.NewDatastore(filepath.Join(dir, "badger"), &opts)
@@ -274,6 +277,10 @@ var aggregateSectorDirsCmd = &cli.Command{
 			return err
 		}
 
+		if err := seed.WriteGenesisMiner(maddr, destdir, &aggrGenMiner); err != nil {
+			return err
+		}
+
 		return nil
 	},
 }