arnaucube
arnaucube/go-dvotePublic Self-hosted repository
go-dvote/cmd/pssTest/psstest.go
cmd/pssTest/psstest.goRaw
 1package main
 2
 3import (
 4	"fmt"
 5	"os"
 6	"time"
 7
 8	"github.com/ethereum/go-ethereum/log"
 9	swarm "github.com/vocdoni/go-dvote/net/swarm"
10)
11
12func main() {
13	if len(os.Args) < 2 {
14		fmt.Println("Use <sym|asym> <key>")
15		return
16	}
17
18	sn := new(swarm.SwarmNet)
19	err := sn.Init()
20	if err != nil {
21		fmt.Printf("%v\n", err)
22		return
23	}
24	err = sn.SetLog("crit")
25	if err != nil {
26		fmt.Printf("Cannot set loglevel %v\n", err)
27	}
28
29	kind := os.Args[1]
30	topic := "vocdoni_test"
31	var key string
32
33	if kind == "sym" || kind == "asym" {
34		key = os.Args[2]
35		sn.PssSub(kind, key, topic, "")
36		defer sn.PssTopics[topic].Unregister()
37	} else {
38		fmt.Println("First parameter must be sym or asym")
39		return
40	}
41
42	go func() {
43		for {
44			msg := <-sn.PssTopics[topic].Delivery
45			fmt.Printf("<- Pss received: %s\n", msg)
46		}
47	}()
48
49	hostname, _ := os.Hostname()
50	for {
51		fmt.Printf("-> Sending %s pss to [%s]\n", kind, key)
52		err := sn.PssPub(kind, key, topic, fmt.Sprintf("Hello world from %s", hostname), "")
53		log.Info("pss sent", "err", err)
54		time.Sleep(10 * time.Second)
55	}
56}