/
z_sub_thr.go
89 lines (73 loc) · 1.95 KB
/
z_sub_thr.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/*
* Copyright (c) 2017, 2020 ADLINK Technology Inc.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
* which is available at https://www.apache.org/licenses/LICENSE-2.0.
*
* SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
*
* Contributors:
* ADLINK zenoh team, <zenoh@adlink-labs.tech>
*/
package main
import (
"fmt"
"time"
"github.com/alexflint/go-arg"
"github.com/eclipse-zenoh/zenoh-go"
)
const n = 100000
var count uint64
var start, stop time.Time
func printStats(start time.Time, stop time.Time) {
t0 := float64(start.UnixNano()) / 1000000000.0
t1 := float64(stop.UnixNano()) / 1000000000
thpt := float64(n) / (t1 - t0)
fmt.Printf("%f msgs/sec\n", thpt)
}
func listener(changes []zenoh.Change) {
if count == 0 {
start = time.Now()
count++
} else if count < n {
count++
} else {
stop = time.Now()
printStats(start, stop)
count = 0
}
}
func main() {
// --- Command line argument parsing --- --- --- --- --- ---
var args struct {
Path string `default:"/zenoh/examples/throughput/data" arg:"-p" help:"The subscriber path"`
Locator string `arg:"-l" help:"The locator to be used to boostrap the zenoh session. By default dynamic discovery is used"`
}
arg.MustParse(&args)
// zenoh-net code --- --- --- --- --- --- --- --- --- --- ---
s, err := zenoh.NewSelector(args.Path)
if err != nil {
panic(err.Error())
}
fmt.Println("Login to Zenoh...")
y, err := zenoh.Login(&args.Locator, nil)
if err != nil {
panic(err.Error())
}
fmt.Println("Use Workspace on '/'")
root, _ := zenoh.NewPath("/")
w := y.Workspace(root)
fmt.Println("Subscribe on " + s.ToString())
subid, err := w.Subscribe(s, listener)
if err != nil {
panic(err.Error())
}
time.Sleep(60 * time.Second)
w.Unsubscribe(subid)
err = y.Logout()
if err != nil {
panic(err.Error())
}
}