113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func simulate(cCtx *cli.Context) error {
|
|
ctx, cancel := context.WithCancel(cCtx.Context)
|
|
|
|
go func() {
|
|
sig := make(chan os.Signal, 1)
|
|
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
|
|
<-sig
|
|
cancel()
|
|
time.Sleep(1 * time.Second)
|
|
os.Exit(1)
|
|
}()
|
|
|
|
logger, err := getSLogger(cCtx.Bool(DebugLevel))
|
|
if err != nil {
|
|
zap.S().Fatalf("Error initializing logger: %v", err)
|
|
}
|
|
|
|
logger.Infof("Running %v command, with:\n%v", cCtx.Command, flagsUsed(cCtx))
|
|
|
|
// Start messengers
|
|
apiModules := cCtx.String(APIModulesFlag)
|
|
telemetryUrl := cCtx.String(TelemetryServerURLFlag)
|
|
failMessages := cCtx.Bool(MessageFailureFlag)
|
|
fleet := cCtx.String(FleetFlag)
|
|
|
|
alice, err := start(StartParams{
|
|
Name: "Alice",
|
|
APIModules: apiModules,
|
|
TelemetryURL: telemetryUrl,
|
|
Fleet: fleet,
|
|
}, logger.Named("alice"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer alice.stop()
|
|
|
|
charlie, err := start(StartParams{
|
|
Name: "Charlie",
|
|
APIModules: apiModules,
|
|
TelemetryURL: telemetryUrl,
|
|
Fleet: fleet,
|
|
}, logger.Named("charlie"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer charlie.stop()
|
|
|
|
// Retrieve for messages
|
|
msgCh := make(chan string)
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(1)
|
|
go alice.retrieveMessagesLoop(ctx, RetrieveInterval, nil, &wg)
|
|
wg.Add(1)
|
|
go charlie.retrieveMessagesLoop(ctx, RetrieveInterval, msgCh, &wg)
|
|
|
|
// Send contact request from Alice to Charlie, charlie accept the request
|
|
time.Sleep(WaitingInterval)
|
|
destID := charlie.messenger.GetSelfContact().ID
|
|
err = alice.sendContactRequest(ctx, destID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msgID := <-msgCh
|
|
err = charlie.sendContactRequestAcceptance(ctx, msgID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
time.Sleep(WaitingInterval)
|
|
|
|
// Send DM between alice to charlie
|
|
interactive := cCtx.Bool(InteractiveFlag)
|
|
if interactive {
|
|
interactiveSendMessageLoop(ctx, alice, charlie)
|
|
} else {
|
|
for i := 0; i < cCtx.Int(CountFlag); i++ {
|
|
err = alice.sendDirectMessage(ctx, fmt.Sprintf("message from alice, number: %d", i+1), failMessages)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
time.Sleep(WaitingInterval)
|
|
|
|
err = charlie.sendDirectMessage(ctx, fmt.Sprintf("message from charlie, number: %d", i+1), failMessages)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
time.Sleep(WaitingInterval)
|
|
}
|
|
cancel()
|
|
}
|
|
|
|
wg.Wait()
|
|
logger.Info("Exiting")
|
|
|
|
return nil
|
|
}
|