You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.1 KiB
101 lines
2.1 KiB
package mq
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"git.oa00.com/go/jcook-sdk/util/http"
|
|
"git.oa00.com/go/jcook-sdk/util/security"
|
|
"net/url"
|
|
"time"
|
|
)
|
|
|
|
const maxSize = "32"
|
|
const autoAck = "false"
|
|
|
|
const (
|
|
PTopic = "topic"
|
|
PConsumerGroupID = "consumerGroupId"
|
|
PSize = "size"
|
|
PAck = "ack"
|
|
PAccessKey = "accessKey"
|
|
PDateTime = "dateTime"
|
|
PSignature = "signature"
|
|
PToken = "token"
|
|
PAckAction = "ackAction"
|
|
PAckIndex = "ackIndex"
|
|
)
|
|
|
|
type Client struct {
|
|
AccessKey string
|
|
SecretKey string
|
|
}
|
|
|
|
func NewClient(ak string, sk string) *Client {
|
|
return &Client{
|
|
AccessKey: ak,
|
|
SecretKey: sk,
|
|
}
|
|
}
|
|
|
|
type UrlBase interface {
|
|
GetTopicName() string
|
|
GetConsumerGroupID() string
|
|
GetEndPoint() string
|
|
}
|
|
|
|
func getUTCStr() string {
|
|
return time.Now().UTC().Format("2006-01-02T15:04:05Z")
|
|
}
|
|
|
|
func (c *Client) Pull(r UrlBase) ([]byte, error) {
|
|
mq := fmt.Sprintf("%s/%s/messages", http.FormatUrl(r.GetEndPoint(), true), "v2")
|
|
value := url.Values{}
|
|
data := map[string]string{
|
|
PTopic: r.GetTopicName(),
|
|
PConsumerGroupID: r.GetConsumerGroupID(),
|
|
PSize: maxSize,
|
|
PAck: autoAck,
|
|
}
|
|
|
|
for k, v := range data {
|
|
value.Set(k, v)
|
|
}
|
|
t := getUTCStr()
|
|
data[PAccessKey] = c.AccessKey
|
|
data[PDateTime] = t
|
|
|
|
sign := security.Sha1OrderlyWithBase64(c.SecretKey, data)
|
|
|
|
header := map[string]string{
|
|
PAccessKey: c.AccessKey,
|
|
PDateTime: t,
|
|
PSignature: sign,
|
|
}
|
|
|
|
return http.JGet(mq, value, header)
|
|
}
|
|
|
|
func (c *Client) Ack(r UrlBase, index string, action string) ([]byte, error) {
|
|
ack := fmt.Sprintf("%s/%s/ack", http.FormatUrl(r.GetEndPoint(), false), "v2")
|
|
t := getUTCStr()
|
|
data := map[string]string{
|
|
PTopic: r.GetTopicName(),
|
|
PAccessKey: c.AccessKey,
|
|
PAckAction: action,
|
|
PAckIndex: index,
|
|
PConsumerGroupID: r.GetConsumerGroupID(),
|
|
PDateTime: t,
|
|
}
|
|
sign := security.Sha1OrderlyWithBase64(c.SecretKey, data)
|
|
|
|
header := map[string]string{
|
|
PAccessKey: c.AccessKey,
|
|
PDateTime: t,
|
|
PSignature: sign,
|
|
}
|
|
|
|
jsonStr, _ := json.Marshal(&data)
|
|
|
|
return http.JPostJson(ack, jsonStr, header)
|
|
}
|