update deliver

zhangmeng

2019-05-20 622b701e27351e28a6c3df579d4423120afe79fc

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
package main
 
import (
    "context"
    "demo/deliver"
    "fmt"
    "os"
)
 
const dLen = 12 * 1024 * 1024
 
var ctx, cancel = context.WithCancel(context.Background())
 
func modeType(t string) deliver.Mode {
 
    if t == "pushpull" {
        return deliver.PushPull
    } else if t == "pubsub" {
        return deliver.PubSub
    } else if t == "pair" {
        return deliver.Pair
    } else if t == "reqrep" {
        return deliver.ReqRep
    } else if t == "shm" {
        return deliver.Shm
    }
 
    return deliver.NONE
}
 
func senderMode(ipc string, m deliver.Mode) {
    if m == deliver.ReqRep {
        req(ipc, m)
    } else if m == deliver.Shm {
        shmSender(ipc, 2, 32*1024*1024)
    }
    sender(ipc, m)
}
 
func recvMode(ipc string, m deliver.Mode, strCount string) {
    if m == deliver.ReqRep {
        rep(ipc, m)
    } else if m == deliver.Shm {
        shmReciever(ipc, strCount)
    }
    reciever(ipc, m, strCount)
}
 
func main() {
    if len(os.Args) > 3 && os.Args[1] == "producer" {
        m := modeType(os.Args[2])
        if m > deliver.ModeStart {
            senderMode(os.Args[3], m)
        }
        os.Exit(0)
    }
    if len(os.Args) > 3 && os.Args[1] == "consumer" {
        m := modeType(os.Args[2])
        if m > deliver.ModeStart {
            recvMode(os.Args[3], m, os.Args[4])
        }
        os.Exit(0)
    }
    fmt.Fprintf(os.Stderr,
        "Usage: pushpull push|pull <URL> <ARG> ...\n")
    os.Exit(1)
 
}