keep node config for state preservation

ava-labs · Dec 12, 2022 · cb70186 · cb70186
1 parent e700f2c
commit cb70186
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/local/blockchain.go b/local/blockchain.go
@@ -755,11 +755,21 @@ func (ln *localNetwork) createBlockchainConfigFiles(
 		// create config, network upgrade and subnet config files
 		if chainSpec.ChainConfig != nil {
 			created = true
-			for nodeName := range ln.nodes {
+			for nodeName, node := range ln.nodes {
+				nodeChainConfig := chainSpec.ChainConfig
+				if conf, ok := chainSpec.PerNodeChainConfig[nodeName]; ok {
+					// keep contents to write file
+					nodeChainConfig = conf
+					// update node config for state preservation
+					if node.config.ChainConfigFiles == nil {
+						node.config.ChainConfigFiles = map[string]string{}
+					}
+					node.config.ChainConfigFiles[chainAlias] = string(conf)
+				}
 				nodeRootDir := getNodeDir(ln.rootDir, nodeName)
 				chainConfigDir := filepath.Join(nodeRootDir, chainConfigSubDir)
 				chainConfigPath := filepath.Join(chainConfigDir, chainAlias, configFileName)
-				if err := createFileAndWrite(chainConfigPath, chainSpec.ChainConfig); err != nil {
+				if err := createFileAndWrite(chainConfigPath, nodeChainConfig); err != nil {
 					return false, fmt.Errorf("couldn't write chain config file at %q: %w", chainConfigPath, err)
 				}
 			}