init xiaomi push golang sdk

施凯伦 · 施凯伦 · commit 77aeacbdaee5 · 2016-09-19T14:02:56.000+08:00
diff --git a/demo.go b/demo.go
@@ -0,0 +1,17 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/yilee/xiaomi-push/xiaomipush"
+)
+
+func main() {
+	msg := xiaomipush.NewMessage("hi baby")
+	client := xiaomipush.NewClient("xxxxxsad", "packagename")
+	result, err := client.Push(msg, []string{"fake_reg_id_1", "fake_reg_id_2"})
+	if err != nil {
+		return
+	}
+	fmt.Printf("result:%#v\n", result)
+}
diff --git a/xiaomipush/client.go b/xiaomipush/client.go
@@ -0,0 +1,145 @@
+package xiaomipush
+
+import (
+	"bytes"
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"runtime"
+	"strconv"
+	"strings"
+)
+
+type MiPush struct {
+	packageName string
+	host        string
+	appSecret   string
+}
+
+func NewClient(appSecret, packageName string) *MiPush {
+	return &MiPush{
+		packageName: packageName,
+		host:        ProductionHost,
+		appSecret:   appSecret,
+	}
+}
+
+func (m *MiPush) Push(msg *Message, regIDList []string) (*Result, error) {
+	params := m.assemblePushParams(msg, regIDList)
+	return m.doPost(m.host+RegURL, params)
+}
+
+func (m *MiPush) Stats(start, end string) (*Result, error) {
+	params := m.assembleStatsParams(start, end)
+	return m.doGet(m.host+StatsURL, params)
+}
+
+func (m *MiPush) Status(msgID string) (*Result, error) {
+	params := m.assembleStatusParams(msgID)
+	return m.doGet(m.host+MessageStatusURL, params)
+}
+
+func (m *MiPush) assemblePushParams(msg *Message, regIDList []string) map[string]string {
+	params := make(map[string]string)
+	params["registration_id"] = strings.Join(regIDList, ",")
+	params["restricted_package_name"] = m.packageName
+
+	if msg.timeToLive > 0 {
+		params["time_to_live"] = strconv.FormatInt(msg.timeToLive, 10)
+	}
+	if len(msg.payload) > 0 {
+		params["payload"] = msg.payload
+	}
+	if len(msg.title) > 0 {
+		params["title"] = msg.title
+	}
+	if len(msg.description) > 0 {
+		params["description"] = msg.description
+	}
+	params["notify_type"] = strconv.FormatInt(int64(msg.notifyType), 10)
+	params["pass_through"] = strconv.FormatInt(int64(msg.passThrough), 10)
+	if msg.notifyID > 0 {
+		params["notify_id"] = strconv.FormatInt(msg.notifyID, 10)
+	}
+	if msg.timeToSend > 0 {
+		params["time_to_send"] = strconv.FormatInt(msg.timeToSend, 10)
+	}
+	if msg.extra != nil && len(msg.extra) > 0 {
+		for k, v := range msg.extra {
+			params["extra."+k] = v
+		}
+	}
+	return params
+}
+
+func (m *MiPush) assembleStatsParams(start, end string) string {
+	form := url.Values{}
+	form.Add("start_date", start)
+	form.Add("end_date", end)
+	form.Add("restricted_package_name", m.packageName)
+	return "?" + form.Encode()
+}
+
+func (m *MiPush) assembleStatusParams(msgID string) string {
+	form := url.Values{}
+	form.Add("msg_id", msgID)
+	return "?" + form.Encode()
+}
+
+func (m *MiPush) handlePushResponse(response *http.Response) (*Result, error) {
+	defer func() {
+		_ = response.Body.Close()
+	}()
+	data, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		return nil, err
+	}
+	return getResultFromJSON(data)
+}
+
+func (m *MiPush) doPost(url string, params map[string]string) (*Result, error) {
+	var result *Result
+	var req *http.Request
+	var resp *http.Response
+	var err error
+	data, _ := json.Marshal(params)
+	req, err = http.NewRequest("POST", url, bytes.NewBuffer(data))
+	req.Header.Set("X-PUSH-OS", runtime.GOOS)
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded;charset=UTF-8")
+	req.Header.Set("Authorizatione", "key="+m.appSecret)
+
+	client := &http.Client{}
+	resp, err = client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func() { _ = resp.Body.Close() }()
+	result, err = m.handlePushResponse(resp)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
+
+func (m *MiPush) doGet(url string, params string) (*Result, error) {
+	var result *Result
+	var req *http.Request
+	var resp *http.Response
+	var err error
+	req, err = http.NewRequest("GET", url+params, nil)
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded;charset=UTF-8")
+	req.Header.Set("Authorizatione", "key="+m.appSecret)
+
+	client := &http.Client{}
+	resp, err = client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func() { _ = resp.Body.Close() }()
+	result, err = m.handlePushResponse(resp)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
diff --git a/xiaomipush/constants.go b/xiaomipush/constants.go
@@ -0,0 +1,122 @@
+package xiaomipush
+
+const (
+	ProductionHost = "https://api.xmpush.xiaomi.com"
+	SandboxHost    = "https://sandbox.xmpush.xiaomi.com" // iOS supported only
+)
+
+const (
+	RegURL           = "/v3/message/regid"          // 向某个regid或一组regid列表推送某条消息
+	MessageAllURL    = "/v3/message/all"            // 向所有设备推送某条消息
+	MultiMessagesURL = "/v2/multi_messages/regids"  // 针对不同的regid推送不同的消息
+	StatsURL         = "/v1/stats/message/counters" // 统计push
+	MessageStatusURL = "/v1/trace/message/status"   // 获取指定ID的消息状态
+)
+
+var (
+	ErrorCodeMap = map[string]string{
+		"-1":    "未知错误",
+		"0":     "成功",
+		"100":   "参数不能为空",
+		"10001": "系统错误",
+		"10002": "服务暂停",
+		"10003": "服务暂停",
+		"10004": "服务暂停",
+		"10005": "服务暂停",
+		"10006": "服务暂停",
+		"10007": "服务暂停",
+		"10008": "服务暂停",
+		"10009": "服务暂停",
+		"10010": "服务暂停",
+		"10011": "服务暂停",
+		"10012": "服务暂停",
+		"10013": "服务暂停",
+		"10014": "服务暂停",
+		"10016": "服务暂停",
+		"10017": "服务暂停",
+		"10018": "服务暂停",
+		"10020": "服务暂停",
+		"10021": "服务暂停",
+		"10022": "服务暂停",
+		"10023": "服务暂停",
+		"10024": "服务暂停",
+		"10025": "服务暂停",
+		"10026": "服务暂停",
+		"10027": "服务暂停",
+		"10028": "服务暂停",
+		"10029": "服务暂停",
+
+		"22000": "服务暂停",
+		"22001": "服务暂停",
+		"22002": "服务暂停",
+		"22003": "服务暂停",
+		"22004": "服务暂停",
+		"22005": "服务暂停",
+		"22006": "服务暂停",
+		"22007": "服务暂停",
+		"22008": "服务暂停",
+		"22020": "服务暂停",
+		"22021": "服务暂停",
+		"22022": "服务暂停",
+		"22100": "服务暂停",
+		"22101": "服务暂停",
+		"22102": "服务暂停",
+		"22103": "服务暂停",
+	}
+)
+
+// for targeted push
+var (
+	BrandsMap = map[string]string{
+		"品牌":    "MODEL",
+		"小米":    "xiaomi",
+		"三星":    "samsung",
+		"华为":    "huawei",
+		"中兴":    "zte",
+		"中兴努比亚": "nubia",
+		"酷派":    "coolpad",
+		"联想":    "lenovo",
+		"魅族":    "meizu",
+		"HTC":   "htc",
+		"OPPO":  "oppo",
+		"VIVO":  "vivo",
+		"摩托罗拉":  "motorola",
+		"索尼":    "sony",
+		"LG":    "lg",
+		"金立":    "jinli",
+		"天语":    "tianyu",
+		"诺基亚":   "nokia",
+		"美图秀秀":  "meitu",
+		"谷歌":    "google",
+		"TCL":   "tcl",
+		"锤子手机":  "chuizi",
+		"一加手机":  "1+",
+		"中国移动":  "chinamobile",
+		"昂达":    "angda",
+		"邦华":    "banghua",
+		"波导":    "bird",
+		"长虹":    "changhong",
+		"大可乐":   "dakele",
+		"朵唯":    "doov",
+		"海尔":    "haier",
+		"海信":    "hisense",
+		"康佳":    "konka",
+		"酷比魔方":  "kubimofang",
+		"米歌":    "mige",
+		"欧博信":   "ouboxin",
+		"欧新":    "ouxin",
+		"飞利浦":   "philip",
+		"维图":    "voto",
+		"小辣椒":   "xiaolajiao",
+		"夏新":    "xiaxin",
+		"亿通":    "yitong",
+		"语信":    "yuxin",
+	}
+
+	PriceMap = map[string]string{
+		"0-999":     "0-999",
+		"1000-1999": "1000-1999",
+		"2000-3999": "2000-3999",
+		"4000+":     "4000+",
+	}
+)
diff --git a/xiaomipush/message.go b/xiaomipush/message.go
@@ -0,0 +1,108 @@
+package xiaomipush
+
+import "time"
+
+type Message struct {
+	uniqueID    string
+	collapseKey string
+	payload     string
+	title       string
+	description string
+	passThrough int32
+	notifyType  int32
+	notifyID    int64
+	timeToLive  int64
+	timeToSend  int64
+	extra       map[string]string
+}
+
+const (
+	MaxTimeToSend = time.Hour * 24 * 7
+	MaxTimeToLive = time.Hour * 24 * 7 * 2
+)
+
+type TargetType int32
+
+const (
+	TargetTypeRegID   TargetType = 1
+	TargetTypeReAlias TargetType = 2
+	TargetTypeAccount TargetType = 3
+)
+
+func NewMessage(payload string) *Message {
+	return &Message{
+		uniqueID:    "",
+		payload:     payload,
+		title:       "", // TBD
+		description: "", // TBD
+		passThrough: 1,
+		notifyType:  -1, // default notify type
+		extra:       make(map[string]string),
+	}
+}
+
+func (m *Message) SetUniqueID(uniqueID string) *Message {
+	m.uniqueID = uniqueID
+	return m
+}
+
+func (m *Message) SetTimeToSend(tts int64) *Message {
+	if time.Since(time.Unix(0, tts*int64(time.Millisecond))) > MaxTimeToSend {
+		m.timeToSend = time.Now().Add(MaxTimeToSend).UnixNano() / 1e6
+	} else {
+		m.timeToSend = tts
+	}
+	return m
+}
+
+func (m *Message) SetTimeToLive(ttl int64) *Message {
+	if time.Since(time.Unix(0, ttl*int64(time.Millisecond))) > MaxTimeToLive {
+		m.timeToLive = time.Now().Add(MaxTimeToLive).UnixNano() / 1e6
+	} else {
+		m.timeToLive = ttl
+	}
+	return m
+}
+
+func (m *Message) EnableFlowControl() *Message {
+	m.extra["flow_control"] = "1"
+	return m
+}
+
+func (m *Message) DisableFlowControl() *Message {
+	delete(m.extra, "flow_control")
+	return m
+}
+
+// 小米推送服务器每隔1s将已送达或已点击的消息ID和对应设备的regid或alias通过调用第三方http接口传给开发者。
+func (m *Message) SetCallback(callbackURL string) *Message {
+	m.extra["callback"] = callbackURL
+	m.extra["callback.param"] = m.uniqueID
+	m.extra["callback.type"] = "3" // 1:送达回执, 2:点击回执, 3:送达和点击回执
+	return m
+}
+
+// TargetedMessage封装了MiPush推送服务系统中的消息Message对象，和该Message对象所要发送到的目标。
+type TargetedMessage struct {
+	message    *Message
+	targetType TargetType
+	target     string
+}
+
+func NewTargetedMessage(m *Message, target string, targetType TargetType) *TargetedMessage {
+	return &TargetedMessage{
+		message:    m,
+		targetType: targetType,
+		target:     target,
+	}
+}
+
+func (tm *TargetedMessage) SetTargetType(targetType TargetType) *TargetedMessage {
+	tm.targetType = targetType
+	return tm
+}
+
+func (tm *TargetedMessage) SetTarget(target string) *TargetedMessage {
+	tm.target = target
+	return tm
+}
diff --git a/xiaomipush/result.go b/xiaomipush/result.go
