go-dvote/cmd/pssTest/psstest.go

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/ethereum/go-ethereum/log"
	swarm "github.com/vocdoni/go-dvote/net/swarm"
)

func main() {
	if len(os.Args) < 2 {
		fmt.Println("Use <sym|asym> <key>")
		return
	}

	sn := new(swarm.SwarmNet)
	err := sn.Init()
	if err != nil {
		fmt.Printf("%v\n", err)
		return
	}
	err = sn.SetLog("crit")
	if err != nil {
		fmt.Printf("Cannot set loglevel %v\n", err)
	}

	kind := os.Args[1]
	topic := "vocdoni_test"
	var key string

	if kind == "sym" || kind == "asym" {
		key = os.Args[2]
		sn.PssSub(kind, key, topic, "")
		defer sn.PssTopics[topic].Unregister()
	} else {
		fmt.Println("First parameter must be sym or asym")
		return
	}

	go func() {
		for {
			msg := <-sn.PssTopics[topic].Delivery
			fmt.Printf("<- Pss received: %s\n", msg)
		}
	}()

	hostname, _ := os.Hostname()
	for {
		fmt.Printf("-> Sending %s pss to [%s]\n", kind, key)
		err := sn.PssPub(kind, key, topic, fmt.Sprintf("Hello world from %s", hostname), "")
		log.Info("pss sent", "err", err)
		time.Sleep(10 * time.Second)
	}
}