path: root/simulation/simulation.go

// Copyright 2018 The dexon-consensus-core Authors
// This file is part of the dexon-consensus-core library.
//
// The dexon-consensus-core library is free software: you can redistribute it
// and/or modify it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation, either version 3 of the License,
// or (at your option) any later version.
//
// The dexon-consensus-core library is distributed in the hope that it will be
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
// General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the dexon-consensus-core library. If not, see
// <http://www.gnu.org/licenses/>.

package simulation

import (
    "fmt"
    "sync"

    "github.com/dexon-foundation/dexon-consensus-core/crypto/eth"
    "github.com/dexon-foundation/dexon-consensus-core/simulation/config"
)

// Run starts the simulation.
func Run(configPath string) {
    cfg, err := config.Read(configPath)
    if err != nil {
        panic(err)
    }

    networkType := cfg.Networking.Type

    var (
        vs           []*Validator
        networkModel = &NormalNetwork{
            Sigma:         cfg.Networking.Sigma,
            Mean:          cfg.Networking.Mean,
            LossRateValue: cfg.Networking.LossRateValue,
        }
    )

    if networkType == config.NetworkTypeTCPLocal {
        lock := sync.Mutex{}
        wg := sync.WaitGroup{}
        for i := 0; i < cfg.Validator.Num; i++ {
            prv, err := eth.NewPrivateKey()
            if err != nil {
                panic(err)
            }
            wg.Add(1)
            go func() {
                network := NewTCPNetwork(true, cfg.Networking.PeerServer, networkModel)
                network.Start()
                lock.Lock()
                defer lock.Unlock()
                vs = append(vs, NewValidator(prv, eth.SigToPub, cfg.Validator, network))
                wg.Done()
            }()
        }
        wg.Wait()

        for i := 0; i < cfg.Validator.Num; i++ {
            fmt.Printf("Validator %d: %s\n", i, vs[i].ID)
            go vs[i].Run()
        }
    } else if networkType == config.NetworkTypeTCP {
        prv, err := eth.NewPrivateKey()
        if err != nil {
            panic(err)
        }
        network := NewTCPNetwork(false, cfg.Networking.PeerServer, networkModel)
        network.Start()
        v := NewValidator(prv, eth.SigToPub, cfg.Validator, network)
        go v.Run()
        vs = append(vs, v)
    }

    for _, v := range vs {
        v.Wait()
        fmt.Printf("Validator %s is shutdown\n", v.GetID())
    }

    // Do not exit when we are in TCP node, since k8s will restart the pod and
    // cause confusions.
    if networkType == config.NetworkTypeTCP {
        select {}
    }
}