mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-27 00:35:18 +00:00
1a223bdce2
* add KavaNodeRunner comments * update kvtool * refactor pingEvm() * refactor pingKava() * refactor EvmRpcPort -> EvmRpcUrl * refactor ChainDetails (g)rpcPort -> (g)rpcUrl * fixup chain details * extract kvtool options to separate config * refactor waitForChainStart() * pull out KavaNodeRunner into kvtool file * rename runner to KvtoolRunner * rename runner.Config to KvtoolRunnerConfig * prefix hardcoded chain details with "kvtool"
105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package runner
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
)
|
|
|
|
type KvtoolRunnerConfig struct {
|
|
KavaConfigTemplate string
|
|
|
|
ImageTag string
|
|
IncludeIBC bool
|
|
|
|
EnableAutomatedUpgrade bool
|
|
KavaUpgradeName string
|
|
KavaUpgradeHeight int64
|
|
KavaUpgradeBaseImageTag string
|
|
|
|
SkipShutdown bool
|
|
}
|
|
|
|
// KvtoolRunner implements a NodeRunner that spins up local chains with kvtool.
|
|
// It has support for the following:
|
|
// - running a Kava node
|
|
// - optionally, running an IBC node with a channel opened to the Kava node
|
|
// - optionally, start the Kava node on one version and upgrade to another
|
|
type KvtoolRunner struct {
|
|
config KvtoolRunnerConfig
|
|
}
|
|
|
|
var _ NodeRunner = &KvtoolRunner{}
|
|
|
|
func NewKvtoolRunner(config KvtoolRunnerConfig) *KvtoolRunner {
|
|
return &KvtoolRunner{
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
func (k *KvtoolRunner) StartChains() Chains {
|
|
// install kvtool if not already installed
|
|
installKvtoolCmd := exec.Command("./scripts/install-kvtool.sh")
|
|
installKvtoolCmd.Stdout = os.Stdout
|
|
installKvtoolCmd.Stderr = os.Stderr
|
|
if err := installKvtoolCmd.Run(); err != nil {
|
|
panic(fmt.Sprintf("failed to install kvtool: %s", err.Error()))
|
|
}
|
|
|
|
// start local test network with kvtool
|
|
log.Println("starting kava node")
|
|
kvtoolArgs := []string{"testnet", "bootstrap", "--kava.configTemplate", k.config.KavaConfigTemplate}
|
|
// include an ibc chain if desired
|
|
if k.config.IncludeIBC {
|
|
kvtoolArgs = append(kvtoolArgs, "--ibc")
|
|
}
|
|
// handle automated upgrade functionality, if defined
|
|
if k.config.EnableAutomatedUpgrade {
|
|
kvtoolArgs = append(kvtoolArgs,
|
|
"--upgrade-name", k.config.KavaUpgradeName,
|
|
"--upgrade-height", fmt.Sprint(k.config.KavaUpgradeHeight),
|
|
"--upgrade-base-image-tag", k.config.KavaUpgradeBaseImageTag,
|
|
)
|
|
}
|
|
// start the chain
|
|
startKavaCmd := exec.Command("kvtool", kvtoolArgs...)
|
|
startKavaCmd.Env = os.Environ()
|
|
startKavaCmd.Env = append(startKavaCmd.Env, fmt.Sprintf("KAVA_TAG=%s", k.config.ImageTag))
|
|
startKavaCmd.Stdout = os.Stdout
|
|
startKavaCmd.Stderr = os.Stderr
|
|
log.Println(startKavaCmd.String())
|
|
if err := startKavaCmd.Run(); err != nil {
|
|
panic(fmt.Sprintf("failed to start kava: %s", err.Error()))
|
|
}
|
|
|
|
// wait for chain to be live.
|
|
// if an upgrade is defined, this waits for the upgrade to be completed.
|
|
if err := waitForChainStart(kvtoolKavaChain); err != nil {
|
|
k.Shutdown()
|
|
panic(err)
|
|
}
|
|
log.Println("kava is started!")
|
|
|
|
chains := NewChains()
|
|
chains.Register("kava", &kvtoolKavaChain)
|
|
if k.config.IncludeIBC {
|
|
chains.Register("ibc", &kvtoolIbcChain)
|
|
}
|
|
return chains
|
|
}
|
|
|
|
func (k *KvtoolRunner) Shutdown() {
|
|
if k.config.SkipShutdown {
|
|
log.Printf("would shut down but SkipShutdown is true")
|
|
return
|
|
}
|
|
log.Println("shutting down kava node")
|
|
shutdownKavaCmd := exec.Command("kvtool", "testnet", "down")
|
|
shutdownKavaCmd.Stdout = os.Stdout
|
|
shutdownKavaCmd.Stderr = os.Stderr
|
|
if err := shutdownKavaCmd.Run(); err != nil {
|
|
panic(fmt.Sprintf("failed to shutdown kvtool: %s", err.Error()))
|
|
}
|
|
}
|