acker/acker.go at master · mostafa/acker · GitHub

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package main

import (
	"fmt"
	"os"
	"runtime"
	"time"

	"github.com/urfave/cli"
)

func main() {
	app := &cli.App{
		Name:        "Acker",
		Version:     "v0.0.4",
		HideVersion: true,
		Compiled:    time.Now(),
		HelpName:    "acker",
		Usage:       "A cli to consume/produce messages from/to AMQP servers, e.g. RabbitMQ",
		Commands: []*cli.Command{
			{
				Name:    "consume",
				Aliases: []string{"c"},
				Usage:   "Consume messages from the queue forever",
				Action: func(c *cli.Context) error {
					ConsumeForever(
						c.String("server"),
						c.String("queue"),
						c.Bool("existing_queue"),
						c.Bool("autoack"),
						c.Bool("recover"))
					return nil
				},
				Flags: []cli.Flag{
					&cli.StringFlag{
						Name:    "server",
						Value:   "amqp://guest:guest@localhost:5672/",
						Usage:   "RabbitMQ server address",
						Aliases: []string{"s"},
					},
					&cli.StringFlag{
						Name:     "queue",
						Value:    "",
						Usage:    "Queue name to consume from",
						Aliases:  []string{"q"},
						Required: true,
					},
					&cli.BoolFlag{
						Name:     "existing-queue",
						Value:    true,
						Usage:    "Connect to an existing queue",
						Aliases:  []string{"e"},
						Required: false,
					},
					&cli.BoolFlag{
						Name:     "autoack",
						Value:    true,
						Usage:    "Automatically acknowledges messages upon consumption",
						Aliases:  []string{"a"},
						Required: false,
					},
					&cli.BoolFlag{
						Name:     "recover",
						Value:    false,
						Usage:    "Recover nack messages on the queue before consumption with the current consumer",
						Aliases:  []string{"r"},
						Required: false,
					},
				},
			}, {
				Name:    "produce",
				Aliases: []string{"p"},
				Usage:   "Produce a message to the queue",
				Action: func(c *cli.Context) error {
					Produce(
						c.String("server"),
						c.String("queue"),
						c.Bool("existing_queue"),
						c.String("body"),
						c.Int("count"))
					return nil
				},
				Flags: []cli.Flag{
					&cli.StringFlag{
						Name:    "server",
						Value:   "amqp://guest:guest@localhost:5672/",
						Usage:   "RabbitMQ server address",
						Aliases: []string{"s"},
					},
					&cli.StringFlag{
						Name:     "queue",
						Value:    "",
						Usage:    "Queue name to produce to",
						Aliases:  []string{"q"},
						Required: true,
					},
					&cli.BoolFlag{
						Name:     "existing-queue",
						Value:    true,
						Usage:    "Connect to an existing queue",
						Aliases:  []string{"e"},
						Required: false,
					},
					&cli.StringFlag{
						Name:     "body",
						Value:    "",
						Usage:    "Body of message (as string)",
						Aliases:  []string{"b"},
						Required: true,
					},
					&cli.IntFlag{
						Name:    "count",
						Value:   1,
						Usage:   "Number of messages to produce",
						Aliases: []string{"n"},
					},
				},
			}, {
				Name:    "version",
				Aliases: []string{"v"},
				Usage:   "Shows version info",
				Action: func(c *cli.Context) error {
					fmt.Printf("version=%s runtime=%s\n", c.App.Version, runtime.Version())
					return nil
				},
			},
		},
	}

	err := app.Run(os.Args)
	FailOnError(err, "Failed to initialize app")
}