Updated blake3pow and progpow to allow setting thread count

consensus/blake3pow/blake3pow.go

@@ -43,6 +43,8 @@ type Config struct {
 	NotifyFull bool
 
 	Log *log.Logger `toml:"-"`
+	// Number of threads to mine on if mining
+	NumThreads int
 }
 
 // Blake3pow is a proof-of-work consensus engine using the blake3 hash algorithm
@@ -70,9 +72,10 @@ type Blake3pow struct {
 // packages.
 func New(config Config, notify []string, noverify bool, logger *log.Logger) *Blake3pow {
 	blake3pow := &Blake3pow{
-		config: config,
-		update: make(chan struct{}),
-		logger: logger,
+		config:  config,
+		update:  make(chan struct{}),
+		logger:  logger,
+		threads: config.NumThreads,
 	}
 	if config.PowMode == ModeShared {
 		blake3pow.shared = sharedBlake3pow

consensus/consensus.go

@@ -175,6 +175,8 @@ type Engine interface {
 	// VerifySeal computes the PowHash and checks if work meets the difficulty
 	// requirement specified in header
 	VerifySeal(header *types.WorkObjectHeader) (common.Hash, error)
+
+	SetThreads(threads int)
 }
 
 func TargetToDifficulty(target *big.Int) *big.Int {

consensus/progpow/progpow.go

@@ -157,6 +157,8 @@ type Config struct {
 	NotifyFull bool
 
 	Log *log.Logger `toml:"-"`
+	// Number of threads to mine on if mining
+	NumThreads int
 }
 
 // Progpow is a proof-of-work consensus engine using the blake3 hash algorithm
@@ -196,10 +198,11 @@ func New(config Config, notify []string, noverify bool, logger *log.Logger) *Pro
 		}).Info("Disk storage enabled for ethash caches")
 	}
 	progpow := &Progpow{
-		config: config,
-		caches: newlru("cache", config.CachesInMem, newCache, logger),
-		update: make(chan struct{}),
-		logger: logger,
+		config:  config,
+		caches:  newlru("cache", config.CachesInMem, newCache, logger),
+		update:  make(chan struct{}),
+		logger:  logger,
+		threads: config.NumThreads,
 	}
 	if config.PowMode == ModeShared {
 		progpow.shared = sharedProgpow