mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-22 14:56:44 +00:00

* [NOD-857] Add generalized profiler package and use it everwhere * [NOD-857] Dependency-inject log into profiling.Start()
60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/kaspanet/kaspad/version"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
_ "net/http/pprof"
|
|
|
|
"github.com/kaspanet/kaspad/signal"
|
|
"github.com/kaspanet/kaspad/util/panics"
|
|
"github.com/kaspanet/kaspad/util/profiling"
|
|
)
|
|
|
|
func main() {
|
|
defer panics.HandlePanic(log, nil)
|
|
interrupt := signal.InterruptListener()
|
|
|
|
cfg, err := parseConfig()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error parsing command-line arguments: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Show version at startup.
|
|
log.Infof("Version %s", version.Version())
|
|
|
|
if cfg.Verbose {
|
|
enableRPCLogging()
|
|
}
|
|
|
|
// Enable http profiling server if requested.
|
|
if cfg.Profile != "" {
|
|
profiling.Start(cfg.Profile, log)
|
|
}
|
|
|
|
client, err := connectToServer(cfg)
|
|
if err != nil {
|
|
panic(errors.Wrap(err, "Error connecting to the RPC server"))
|
|
}
|
|
defer client.Disconnect()
|
|
|
|
doneChan := make(chan struct{})
|
|
spawn(func() {
|
|
err = mineLoop(client, cfg.NumberOfBlocks, cfg.BlockDelay)
|
|
if err != nil {
|
|
panic(errors.Errorf("Error in mine loop: %s", err))
|
|
}
|
|
doneChan <- struct{}{}
|
|
})
|
|
|
|
select {
|
|
case <-doneChan:
|
|
case <-interrupt:
|
|
}
|
|
}
|