62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package nats_client
|
||
|
||
import (
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule-agent/pkg/log"
|
||
"github.com/nats-io/nats.go"
|
||
"time"
|
||
)
|
||
|
||
func Publish(subj string, data []byte) error {
|
||
// Connect Options.
|
||
opts := []nats.Option{nats.Name("NATS Sample Publisher")}
|
||
opts = append(opts, nats.UserInfo(NatsConfig.User, NatsConfig.Password))
|
||
|
||
nc, err := nats.Connect(NatsConfig.Url, opts...)
|
||
if err != nil {
|
||
log.Errorf("连接nats失败：%s", err)
|
||
return err
|
||
}
|
||
defer nc.Close()
|
||
nc.Publish(subj, data)
|
||
nc.Flush()
|
||
|
||
if err := nc.LastError(); err != nil {
|
||
//log.Fatal(err)
|
||
log.Errorf("与nats通信失败：%s", err)
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func Request(subj string, data []byte) (*nats.Msg, error) {
|
||
// Connect Options.
|
||
opts := []nats.Option{nats.Name("NATS Sample Publisher")}
|
||
opts = append(opts, nats.UserInfo(NatsConfig.User, NatsConfig.Password))
|
||
|
||
nc, err := nats.Connect(NatsConfig.Url, opts...)
|
||
if err != nil {
|
||
log.Errorf("连接nats失败：%s", err)
|
||
return nil, err
|
||
}
|
||
defer nc.Close()
|
||
msg, err := nc.Request(subj, data, 5*time.Second)
|
||
if err != nil {
|
||
if nc.LastError() != nil {
|
||
log.Errorf("%v for request", nc.LastError())
|
||
}
|
||
log.Errorf("%v for request", err)
|
||
return nil, err
|
||
}
|
||
|
||
nc.Flush()
|
||
|
||
if err := nc.LastError(); err != nil {
|
||
//log.Fatal(err)
|
||
log.Errorf("与nats通信失败：%s", err)
|
||
return nil, err
|
||
}
|
||
|
||
return msg, nil
|
||
}
|