parent
7b7c419f03
commit
234a50adb2
GPG Key ID:
D2EF94423C23BF12
@ -0,0 +1,79 @@ |
|||||||
|
package ui |
||||||
|
|
||||||
|
import ( |
||||||
|
"cli-mon/yabl" |
||||||
|
"context" |
||||||
|
json "encoding/json" |
||||||
|
"fmt" |
||||||
|
amqp "github.com/rabbitmq/amqp091-go" |
||||||
|
"os" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
const exchange = "yabl-newproto-in" |
||||||
|
|
||||||
|
var hostname, _ = os.Hostname() |
||||||
|
|
||||||
|
type broker struct { |
||||||
|
uri string |
||||||
|
conn *amqp.Connection |
||||||
|
ch *amqp.Channel |
||||||
|
} |
||||||
|
|
||||||
|
func NewAmqp(uri string) UI { |
||||||
|
return &broker{uri: uri} |
||||||
|
} |
||||||
|
|
||||||
|
func (b *broker) Consume(event *yabl.Event) { |
||||||
|
var bytes []byte |
||||||
|
var err error |
||||||
|
if bytes, err = json.Marshal(event); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
route := fmt.Sprintf( |
||||||
|
"%s.%s.%s.%d", |
||||||
|
hostname, |
||||||
|
event.ActionName, |
||||||
|
event.Field, |
||||||
|
event.GetUnitId()) |
||||||
|
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) |
||||||
|
defer cancel() |
||||||
|
|
||||||
|
if err = b.ch.PublishWithContext(ctx, |
||||||
|
exchange, |
||||||
|
route, |
||||||
|
false, |
||||||
|
false, |
||||||
|
amqp.Publishing{ |
||||||
|
ContentType: "application/json", |
||||||
|
Body: bytes, |
||||||
|
}); err != nil { |
||||||
|
fmt.Printf("Can't publish msg: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (b *broker) Start() { |
||||||
|
var err error |
||||||
|
b.conn, err = amqp.Dial(b.uri) |
||||||
|
if err != nil { |
||||||
|
panic(fmt.Sprintf("Can't connect to AMQP: %v", err)) |
||||||
|
} |
||||||
|
|
||||||
|
b.ch, err = b.conn.Channel() |
||||||
|
if err != nil { |
||||||
|
panic(fmt.Sprintf("Can't create AMQP channel: %v", err)) |
||||||
|
} |
||||||
|
|
||||||
|
fmt.Printf("AMQP client connected to %v, sending data to server..", b.uri) |
||||||
|
} |
||||||
|
|
||||||
|
func (b *broker) Stop() { |
||||||
|
if b.conn != nil { |
||||||
|
b.conn.Close() |
||||||
|
} |
||||||
|
if b.ch != nil { |
||||||
|
b.ch.Close() |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,35 @@ |
|||||||
|
package ui |
||||||
|
|
||||||
|
import ( |
||||||
|
"cli-mon/yabl" |
||||||
|
"testing" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
func TestNewAmqp(t *testing.T) { |
||||||
|
t.Run("Amqp Test", func(t *testing.T) { |
||||||
|
client := NewAmqp("amqp://user:password@localhost:5672/") |
||||||
|
|
||||||
|
if client == nil { |
||||||
|
t.Errorf("Can't create client") |
||||||
|
} |
||||||
|
|
||||||
|
client.Start() |
||||||
|
|
||||||
|
now := time.Now() |
||||||
|
|
||||||
|
event := &yabl.Event{ |
||||||
|
Field: yabl.FBoardReady, |
||||||
|
ActionName: yabl.PPuErrors, |
||||||
|
Object: nil, |
||||||
|
//unit: 3,
|
||||||
|
Updated: &now, |
||||||
|
Value: yabl.ERROR, |
||||||
|
} |
||||||
|
|
||||||
|
client.Consume(event) |
||||||
|
|
||||||
|
client.Stop() |
||||||
|
}) |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue