使用Golang实现Uniqush-Push插件

Uniqush-Push是一个开源的统一推送服务，支持APNs、FCM/GCM等多种推送平台。下面我将介绍如何使用Golang实现一个与Uniqush-Push交互的客户端。

基本概念

Uniqush-Push主要提供以下功能：

• 统一接口支持多种推送服务
• 设备注册管理
• 消息推送
• 支持HTTP REST API

Golang客户端实现

1. 安装必要的依赖

go get github.com/go-resty/resty/v2

2. 基础客户端结构

package uniqush

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/go-resty/resty/v2"
)

type UniqushClient struct {
	baseURL    string
	httpClient *resty.Client
}

func NewUniqushClient(baseURL string) *UniqushClient {
	return &UniqushClient{
		baseURL:    strings.TrimSuffix(baseURL, "/"),
		httpClient: resty.New(),
	}
}

3. 添加推送服务

// AddPushService 添加推送服务
func (c *UniqushClient) AddPushService(service, pushServiceType, projectID, apiKey string) error {
	url := fmt.Sprintf("%s/addpsp", c.baseURL)
	
	resp, err := c.httpClient.R().
		SetQueryParams(map[string]string{
			"service": service,
			"psp":     pushServiceType,
		}).
		SetFormData(map[string]string{
			"projectid": projectID,
			"apikey":    apiKey,
		}).
		Post(url)

	if err != nil {
		return err
	}

	if resp.StatusCode() != 200 {
		return fmt.Errorf("failed to add push service: %s", resp.String())
	}

	return nil
}

4. 注册设备

// Subscribe 注册设备
func (c *UniqushClient) Subscribe(service, subscriber, pushServiceType, regId string) error {
	url := fmt.Sprintf("%s/subscribe", c.baseURL)
	
	resp, err := c.httpClient.R().
		SetQueryParams(map[string]string{
			"service":    service,
			"subscriber": subscriber,
			"psp":        pushServiceType,
			"regid":      regId,
		}).
		Post(url)

	if err != nil {
		return err
	}

	if resp.StatusCode() != 200 {
		return fmt.Errorf("failed to subscribe device: %s", resp.String())
	}

	return nil
}

5. 发送推送通知

// Push 发送推送通知
func (c *UniqushClient) Push(service, subscriber, msg string) error {
	url := fmt.Sprintf("%s/push", c.baseURL)
	
	resp, err := c.httpClient.R().
		SetQueryParams(map[string]string{
			"service":    service,
			"subscriber": subscriber,
		}).
		SetFormData(map[string]string{
			"msg": msg,
		}).
		Post(url)

	if err != nil {
		return err
	}

	if resp.StatusCode() != 200 {
		return fmt.Errorf("failed to push message: %s", resp.String())
	}

	return nil
}

6. 完整示例使用

package main

import (
	"fmt"
	"log"

	"./uniqush" // 替换为你的包路径
)

func main() {
	// 初始化客户端
	client := uniqush.NewUniqushClient("http://localhost:9898")

	// 添加APNs推送服务
	err := client.AddPushService("com.example.app", "apns", "", "path/to/cert.pem")
	if err != nil {
		log.Fatalf("Failed to add APNs service: %v", err)
	}

	// 添加FCM推送服务
	err = client.AddPushService("com.example.app", "gcm", "your-project-id", "your-api-key")
	if err != nil {
		log.Fatalf("Failed to add FCM service: %v", err)
	}

	// 注册设备
	err = client.Subscribe("com.example.app", "user123", "gcm", "device-registration-id")
	if err != nil {
		log.Fatalf("Failed to subscribe device: %v", err)
	}

	// 发送推送通知
	message := `{
		"title": "Hello",
		"body": "This is a test message",
		"sound": "default"
	}`
	err = client.Push("com.example.app", "user123", message)
	if err != nil {
		log.Fatalf("Failed to push message: %v", err)
	}

	fmt.Println("Push notification sent successfully")
}

高级功能

批量推送

// BatchPush 批量推送
func (c *UniqushClient) BatchPush(service string, subscribers []string, msg string) error {
	url := fmt.Sprintf("%s/push", c.baseURL)
	
	resp, err := c.httpClient.R().
		SetQueryParams(map[string]string{
			"service": service,
		}).
		SetFormData(map[string]string{
			"subscriber": strings.Join(subscribers, ","),
			"msg":        msg,
		}).
		Post(url)

	if err != nil {
		return err
	}

	if resp.StatusCode() != 200 {
		return fmt.Errorf("failed to batch push message: %s", resp.String())
	}

	return nil
}

处理推送结果

type PushResult struct {
	Success bool   `json:"success"`
	Error   string `json:"error,omitempty"`
	RegID   string `json:"regid,omitempty"`
}

func (c *UniqushClient) PushWithResult(service, subscriber, msg string) ([]PushResult, error) {
	url := fmt.Sprintf("%s/push", c.baseURL)
	
	resp, err := c.httpClient.R().
		SetQueryParams(map[string]string{
			"service":    service,
			"subscriber": subscriber,
		}).
		SetFormData(map[string]string{
			"msg": msg,
		}).
		Post(url)

	if err != nil {
		return nil, err
	}

	if resp.StatusCode() != 200 {
		return nil, fmt.Errorf("failed to push message: %s", resp.String())
	}

	var results []PushResult
	if err := json.Unmarshal(resp.Body(), &results); err != nil {
		return nil, err
	}

	return results, nil
}

注意事项

1. 确保Uniqush-Push服务已正确安装和配置
2. 根据实际需求调整推送参数
3. 处理网络错误和重试逻辑
4. 在生产环境中添加适当的日志记录
5. 考虑使用连接池优化HTTP客户端性能

这个实现提供了与Uniqush-Push交互的基本功能，你可以根据需要扩展更多功能或优化现有实现。