799 lines
24 KiB
Go
799 lines
24 KiB
Go
package warningnotify
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
"unicode/utf8"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/dao"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/cache"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
)
|
|
|
|
const (
|
|
warningWechatAccessTokenURL = "https://api.weixin.qq.com/cgi-bin/token"
|
|
warningWechatSendURL = "https://api.weixin.qq.com/cgi-bin/message/subscribe/send"
|
|
warningWechatDefaultPage = "pages/message/index"
|
|
warningWechatDefaultTimeout = 5 * time.Second
|
|
warningWechatSentKeyPrefix = "iotlcm:warning:wechat:sent"
|
|
warningWechatMinSentTTL = 24 * time.Hour
|
|
)
|
|
|
|
var warningWechatTokenMu sync.Mutex
|
|
|
|
type warningWechatConfig struct {
|
|
Enabled bool `json:"enabled"`
|
|
AppID string `json:"appId"`
|
|
AppSecret string `json:"appSecret"`
|
|
TemplateID string `json:"templateId"`
|
|
Page string `json:"page"`
|
|
Timeout string `json:"timeout"`
|
|
}
|
|
|
|
type warningWechatBinding struct {
|
|
UserID int64 `orm:"user_id"`
|
|
OpenID string `orm:"open_id"`
|
|
}
|
|
|
|
type warningWechatAccessTokenResponse struct {
|
|
AccessToken string `json:"access_token"`
|
|
ExpiresIn int `json:"expires_in"`
|
|
ErrCode int `json:"errcode"`
|
|
ErrMsg string `json:"errmsg"`
|
|
}
|
|
|
|
type warningWechatSendResponse struct {
|
|
ErrCode int `json:"errcode"`
|
|
ErrMsg string `json:"errmsg"`
|
|
}
|
|
|
|
type warningWechatTemplateValue struct {
|
|
Value string `json:"value"`
|
|
}
|
|
|
|
type warningWechatPayload struct {
|
|
ToUser string `json:"touser"`
|
|
TemplateID string `json:"template_id"`
|
|
Page string `json:"page,omitempty"`
|
|
Data map[string]warningWechatTemplateValue `json:"data"`
|
|
}
|
|
|
|
func sendWarningWechat(ctx context.Context, batch Batch) error {
|
|
cfg, ok := loadWarningWechatConfig(ctx)
|
|
if !ok || !cfg.Enabled {
|
|
g.Log().Infof(
|
|
ctx,
|
|
"WarningNotify wechat subscribe skipped, reason:%s, tenantId:%s, deptId:%d, category:%s, stage:%d, count:%d",
|
|
wechatConfigSkipReason(ok, cfg),
|
|
batch.TenantID,
|
|
batch.DeptID,
|
|
batch.Category,
|
|
batch.WarnStage,
|
|
batchItemCount(batch),
|
|
)
|
|
return nil
|
|
}
|
|
if strings.TrimSpace(cfg.AppID) == "" || strings.TrimSpace(cfg.AppSecret) == "" {
|
|
g.Log().Warningf(ctx, "WarningNotify wechat subscribe skipped, reason:app credentials incomplete, tenantId:%s", batch.TenantID)
|
|
return nil
|
|
}
|
|
|
|
deliveries := resolveWechatDeliveries(ctx, batch)
|
|
if len(deliveries) == 0 {
|
|
return nil
|
|
}
|
|
|
|
var sendErrs []string
|
|
for _, delivery := range deliveries {
|
|
if len(delivery.Group.UserIDs) == 0 || batchItemCount(delivery.Batch) == 0 {
|
|
continue
|
|
}
|
|
userID := delivery.Group.UserIDs[0]
|
|
if batch.Business == BusinessSimWarning {
|
|
enabled, err := warningWechatEnabled(ctx, batch.TenantID, userID)
|
|
if err != nil {
|
|
sendErrs = append(sendErrs, fmt.Sprintf("userId=%d query preference: %v", userID, err))
|
|
continue
|
|
}
|
|
if !enabled {
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe skipped, reason:user preference disabled, tenantId:%s, userId:%d, count:%d", batch.TenantID, userID, batchItemCount(delivery.Batch))
|
|
continue
|
|
}
|
|
}
|
|
bindings, err := queryWechatBindings(ctx, batch.TenantID, userID)
|
|
if err != nil {
|
|
sendErrs = append(sendErrs, fmt.Sprintf("userId=%d query binding: %v", userID, err))
|
|
continue
|
|
}
|
|
if len(bindings) == 0 {
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe skipped, reason:no openid, tenantId:%s, userId:%d, count:%d", batch.TenantID, userID, batchItemCount(delivery.Batch))
|
|
continue
|
|
}
|
|
|
|
acquired, sentKey, err := acquireWechatSendDedupe(ctx, delivery.Batch, userID)
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "WarningNotify wechat subscribe dedupe check failed, tenantId:%s, userId:%d, category:%s, stage:%d, count:%d, err:%+v",
|
|
batch.TenantID, userID, delivery.Batch.Category, delivery.Batch.WarnStage, batchItemCount(delivery.Batch), err)
|
|
} else if !acquired {
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe skipped, reason:duplicated, tenantId:%s, userId:%d, key:%s, category:%s, stage:%d, count:%d",
|
|
batch.TenantID, userID, sentKey, delivery.Batch.Category, delivery.Batch.WarnStage, batchItemCount(delivery.Batch))
|
|
continue
|
|
}
|
|
|
|
successCount := 0
|
|
for _, binding := range bindings {
|
|
if batch.Business == BusinessSimWarning {
|
|
enabled, err := warningWechatEnabled(ctx, batch.TenantID, userID)
|
|
if err != nil {
|
|
sendErrs = append(sendErrs, fmt.Sprintf("userId=%d query preference: %v", userID, err))
|
|
continue
|
|
}
|
|
if !enabled {
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe skipped, reason:user preference disabled before send, tenantId:%s, userId:%d", batch.TenantID, userID)
|
|
continue
|
|
}
|
|
}
|
|
templateID := wechatTemplateID(delivery.Batch.Items, cfg.TemplateID)
|
|
if templateID == "" {
|
|
sendErrs = append(sendErrs, fmt.Sprintf("userId=%d template id empty", userID))
|
|
continue
|
|
}
|
|
payload := buildWechatPayload(delivery.Batch, binding.OpenID, templateID, cfg.Page)
|
|
if err := sendWechatSubscription(ctx, cfg, payload); err != nil {
|
|
sendErrs = append(sendErrs, fmt.Sprintf("userId=%d openid=%s: %v", userID, maskOpenID(binding.OpenID), err))
|
|
g.Log().Errorf(ctx, "WarningNotify wechat subscribe failed, tenantId:%s, userId:%d, openid:%s, category:%s, stage:%d, count:%d, err:%+v",
|
|
batch.TenantID, userID, maskOpenID(binding.OpenID), delivery.Batch.Category, delivery.Batch.WarnStage, batchItemCount(delivery.Batch), err)
|
|
continue
|
|
}
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe sent, tenantId:%s, userId:%d, openid:%s, category:%s, stage:%d, count:%d",
|
|
batch.TenantID, userID, maskOpenID(binding.OpenID), delivery.Batch.Category, delivery.Batch.WarnStage, batchItemCount(delivery.Batch))
|
|
successCount++
|
|
}
|
|
if acquired && successCount == 0 {
|
|
releaseWechatSendDedupe(ctx, sentKey)
|
|
}
|
|
}
|
|
|
|
if len(sendErrs) > 0 {
|
|
return fmt.Errorf("wechat subscribe send completed with failures: %s", strings.Join(sendErrs, "; "))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func resolveWechatDeliveries(ctx context.Context, batch Batch) []platformDelivery {
|
|
if batch.prepared != nil {
|
|
return []platformDelivery{{Group: *batch.prepared, Batch: batch}}
|
|
}
|
|
deliveries := resolveTerminalPlatformDeliveries(ctx, batch)
|
|
appendEnterpriseWechatDeliveries(ctx, batch, &deliveries)
|
|
if len(deliveries) == 0 {
|
|
g.Log().Infof(ctx, "WarningNotify wechat subscribe skipped, reason:no eligible recipients, tenantId:%s, count:%d", batch.TenantID, batchItemCount(batch))
|
|
}
|
|
return deliveries
|
|
}
|
|
|
|
func appendEnterpriseWechatDeliveries(ctx context.Context, batch Batch, deliveries *[]platformDelivery) {
|
|
if batch.Business != BusinessSimWarning {
|
|
return
|
|
}
|
|
itemsByUser := make(map[int64][]Event)
|
|
seenByUser := make(map[int64]map[int]struct{})
|
|
for index, item := range batch.Items {
|
|
if item.ItemType != 1 {
|
|
continue
|
|
}
|
|
createdBy := item.CreatedBy
|
|
if createdBy <= 0 {
|
|
createdBy = batch.CreatedBy
|
|
}
|
|
if createdBy <= 0 || item.ItemID <= 0 {
|
|
continue
|
|
}
|
|
userID, ok, err := queryEnterpriseWarningUser(ctx, batch.TenantID, createdBy, item.ItemID)
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "WarningNotify query enterprise user failed, tenantId:%s, userId:%d, cardId:%d, err:%+v", batch.TenantID, createdBy, item.ItemID, err)
|
|
continue
|
|
}
|
|
if ok {
|
|
appendTerminalEvent(itemsByUser, seenByUser, userID, index, item)
|
|
}
|
|
}
|
|
userIDs := make([]int64, 0, len(itemsByUser))
|
|
for userID := range itemsByUser {
|
|
userIDs = append(userIDs, userID)
|
|
}
|
|
sort.Slice(userIDs, func(i, j int) bool { return userIDs[i] < userIDs[j] })
|
|
for _, userID := range userIDs {
|
|
items := itemsByUser[userID]
|
|
enterpriseBatch := batch
|
|
enterpriseBatch.Items = append([]Event(nil), items...)
|
|
enterpriseBatch.WarnStage = highestStage(enterpriseBatch.Items)
|
|
*deliveries = append(*deliveries, platformDelivery{
|
|
Group: platformRecipientGroup{SendUserType: sendUserTypeAgent, UserIDs: []int64{userID}},
|
|
Batch: enterpriseBatch,
|
|
})
|
|
}
|
|
}
|
|
|
|
func enterpriseWarningRecipientEligible(userType string, userDeptID, cardDeptID int64, ancestors string) bool {
|
|
if userType != consts.SysUserTypeSys || userDeptID <= 0 || cardDeptID <= 0 {
|
|
return false
|
|
}
|
|
if userDeptID == cardDeptID {
|
|
return true
|
|
}
|
|
target := strconv.FormatInt(userDeptID, 10)
|
|
for _, ancestor := range strings.Split(ancestors, ",") {
|
|
if strings.TrimSpace(ancestor) == target {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
func queryEnterpriseWarningUser(ctx context.Context, tenantID string, userID, cardID int64) (int64, bool, error) {
|
|
if strings.TrimSpace(tenantID) == "" || userID <= 0 || cardID <= 0 {
|
|
return 0, false, nil
|
|
}
|
|
var result struct {
|
|
UserID int64 `orm:"user_id"`
|
|
UserType string `orm:"user_type"`
|
|
DeptID int64 `orm:"dept_id"`
|
|
CardDeptID int64 `orm:"card_dept_id"`
|
|
Ancestors string `orm:"ancestors"`
|
|
}
|
|
err := g.DB().Model(dao.TSimCardInfo.Table()+" c").Ctx(ctx).Safe().
|
|
LeftJoin(dao.SysUser.Table()+" u", "u.user_id = "+strconv.FormatInt(userID, 10)).
|
|
InnerJoin(dao.SysDept.Table()+" d", "d.dept_id = c.dept_id AND d.tenant_id = c.tenant_id").
|
|
InnerJoin(dao.SysDept.Table()+" ud", "ud.dept_id = u.dept_id AND ud.tenant_id = u.tenant_id").
|
|
Fields("u.user_id, u.user_type, u.dept_id, c.dept_id AS card_dept_id, d.ancestors").
|
|
Where("c.tenant_id", tenantID).
|
|
Where("c.id", cardID).
|
|
Where("u.tenant_id", tenantID).
|
|
Where("u.user_type", consts.SysUserTypeSys).
|
|
Where("u.status", consts.SysDeptStatusNormal).
|
|
Where("u.deleted_at IS NULL").
|
|
Where("d.deleted_at IS NULL").
|
|
Where("ud.deleted_at IS NULL").
|
|
Scan(&result)
|
|
if err != nil {
|
|
return 0, false, err
|
|
}
|
|
return result.UserID, result.UserID > 0 && enterpriseWarningRecipientEligible(result.UserType, result.DeptID, result.CardDeptID, result.Ancestors), nil
|
|
}
|
|
|
|
func loadWarningWechatConfig(ctx context.Context) (warningWechatConfig, bool) {
|
|
var cfg warningWechatConfig
|
|
loaded := false
|
|
value, err := g.Cfg().Get(ctx, "warning.wechat")
|
|
if err == nil && !value.IsNil() {
|
|
_ = value.Scan(&cfg)
|
|
loaded = true
|
|
}
|
|
|
|
if envValue := strings.TrimSpace(os.Getenv("IOTLCM_WARNING_WECHAT_ENABLED")); envValue != "" {
|
|
if enabled, parseErr := strconv.ParseBool(envValue); parseErr == nil {
|
|
cfg.Enabled = enabled
|
|
loaded = true
|
|
}
|
|
}
|
|
loaded = applyWechatEnv(&cfg.AppID, "IOTLCM_WARNING_WECHAT_APP_ID") || loaded
|
|
loaded = applyWechatEnv(&cfg.AppSecret, "IOTLCM_WARNING_WECHAT_APP_SECRET") || loaded
|
|
loaded = applyWechatEnv(&cfg.TemplateID, "IOTLCM_WARNING_WECHAT_TEMPLATE_ID") || loaded
|
|
loaded = applyWechatEnv(&cfg.Page, "IOTLCM_WARNING_WECHAT_PAGE") || loaded
|
|
loaded = applyWechatEnv(&cfg.Timeout, "IOTLCM_WARNING_WECHAT_TIMEOUT") || loaded
|
|
|
|
cfg.AppID = strings.TrimSpace(cfg.AppID)
|
|
cfg.AppSecret = strings.TrimSpace(cfg.AppSecret)
|
|
cfg.TemplateID = strings.TrimSpace(cfg.TemplateID)
|
|
cfg.Page = strings.TrimSpace(cfg.Page)
|
|
cfg.Timeout = strings.TrimSpace(cfg.Timeout)
|
|
if cfg.TemplateID == "" {
|
|
cfg.TemplateID = consts.WarningWechatTemplateIDDefault
|
|
}
|
|
if cfg.Page == "" {
|
|
cfg.Page = warningWechatDefaultPage
|
|
}
|
|
return cfg, loaded
|
|
}
|
|
|
|
func applyWechatEnv(target *string, key string) bool {
|
|
value := strings.TrimSpace(os.Getenv(key))
|
|
if value == "" {
|
|
return false
|
|
}
|
|
*target = value
|
|
return true
|
|
}
|
|
|
|
func wechatConfigSkipReason(loaded bool, cfg warningWechatConfig) string {
|
|
if !loaded {
|
|
return "warning.wechat config not found"
|
|
}
|
|
if !cfg.Enabled {
|
|
return "warning.wechat disabled"
|
|
}
|
|
if cfg.AppID == "" || cfg.AppSecret == "" {
|
|
return "warning.wechat app credentials incomplete"
|
|
}
|
|
return "unknown"
|
|
}
|
|
|
|
func queryWechatBindings(ctx context.Context, tenantID string, userID int64) ([]warningWechatBinding, error) {
|
|
if strings.TrimSpace(tenantID) == "" || userID <= 0 {
|
|
return nil, nil
|
|
}
|
|
columns := dao.SysSocial.Columns()
|
|
rows := make([]warningWechatBinding, 0)
|
|
err := dao.SysSocial.Ctx(ctx).
|
|
Fields(columns.UserId, columns.OpenId).
|
|
Where(columns.TenantId, strings.TrimSpace(tenantID)).
|
|
Where(columns.UserId, userID).
|
|
Where(columns.Source, consts.SourceWechatMp).
|
|
Where("open_id <> ?", "").
|
|
OrderAsc(columns.Id).
|
|
Scan(&rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
seen := make(map[string]struct{}, len(rows))
|
|
result := make([]warningWechatBinding, 0, len(rows))
|
|
for _, row := range rows {
|
|
row.OpenID = strings.TrimSpace(row.OpenID)
|
|
if row.OpenID == "" {
|
|
continue
|
|
}
|
|
if _, exists := seen[row.OpenID]; exists {
|
|
continue
|
|
}
|
|
seen[row.OpenID] = struct{}{}
|
|
result = append(result, row)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func acquireWechatSendDedupe(ctx context.Context, batch Batch, userID int64) (bool, string, error) {
|
|
key := wechatSendDedupeKey(batch, userID)
|
|
if key == "" {
|
|
return false, "", fmt.Errorf("wechat send dedupe key is empty")
|
|
}
|
|
ttl := wechatSendDedupeTTL(batch)
|
|
ok, err := cache.Instance().SetIfNotExist(ctx, key, "1", ttl)
|
|
return ok, key, err
|
|
}
|
|
|
|
func releaseWechatSendDedupe(ctx context.Context, key string) {
|
|
key = strings.TrimSpace(key)
|
|
if key == "" {
|
|
return
|
|
}
|
|
_, _ = cache.Instance().Remove(ctx, key)
|
|
}
|
|
|
|
func wechatSendDedupeKey(batch Batch, userID int64) string {
|
|
tenantID := strings.TrimSpace(batch.TenantID)
|
|
if tenantID == "" || userID <= 0 {
|
|
return ""
|
|
}
|
|
business := strings.TrimSpace(batch.Business)
|
|
if business == "" {
|
|
business = BusinessSimWarning
|
|
}
|
|
return fmt.Sprintf(
|
|
"%s:%s:%d:%s:%s:%s",
|
|
warningWechatSentKeyPrefix,
|
|
tenantID,
|
|
userID,
|
|
wechatSafeKeyPart(business),
|
|
wechatDedupeCategoryKey(batch),
|
|
wechatDedupePeriodKey(batch),
|
|
)
|
|
}
|
|
|
|
func wechatSendDedupeTTL(batch Batch) time.Duration {
|
|
now := wechatBatchTime(batch)
|
|
if wechatBatchHasTrafficKind(batch) {
|
|
nextMonth := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()).AddDate(0, 1, 0)
|
|
if ttl := nextMonth.Sub(now); ttl > 0 {
|
|
return ttl
|
|
}
|
|
return warningWechatMinSentTTL
|
|
}
|
|
if wechatBatchHasExpireKind(batch) {
|
|
return 36 * time.Hour
|
|
}
|
|
return warningWechatMinSentTTL
|
|
}
|
|
|
|
func wechatDedupePeriodKey(batch Batch) string {
|
|
now := wechatBatchTime(batch)
|
|
if wechatBatchHasTrafficKind(batch) {
|
|
return "m" + now.Format("200601")
|
|
}
|
|
if wechatBatchHasExpireKind(batch) {
|
|
return fmt.Sprintf("d%s:s%d", now.Format("20060102"), batch.WarnStage)
|
|
}
|
|
return "d" + now.Format("20060102")
|
|
}
|
|
|
|
func wechatDedupeCategoryKey(batch Batch) string {
|
|
switch wechatPrimaryKind(batch) {
|
|
case "traffic_mb", "traffic_percent":
|
|
return "card_traffic"
|
|
case "pool_traffic_percent":
|
|
return "flow_pool"
|
|
case "expire_day":
|
|
return "expire_day"
|
|
case "expire_month":
|
|
return "expire_month"
|
|
}
|
|
category := strings.TrimSpace(batch.Category)
|
|
if category == "" {
|
|
category = "warning"
|
|
}
|
|
return "cat_" + sha256Hex(category)
|
|
}
|
|
|
|
func wechatPrimaryKind(batch Batch) string {
|
|
for _, item := range batch.Items {
|
|
kind := strings.TrimSpace(item.Kind)
|
|
if kind != "" {
|
|
return kind
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func wechatBatchHasTrafficKind(batch Batch) bool {
|
|
for _, item := range batch.Items {
|
|
switch strings.TrimSpace(item.Kind) {
|
|
case "traffic_mb", "traffic_percent", "pool_traffic_percent":
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func wechatBatchHasExpireKind(batch Batch) bool {
|
|
for _, item := range batch.Items {
|
|
switch strings.TrimSpace(item.Kind) {
|
|
case "expire_day", "expire_month":
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func wechatBatchTime(batch Batch) time.Time {
|
|
latest := latestEvent(batch.Items)
|
|
if !batch.LatestTime.IsZero() {
|
|
latest.WarnTime = batch.LatestTime
|
|
}
|
|
if latest.WarnTime.IsZero() {
|
|
return time.Now()
|
|
}
|
|
return latest.WarnTime
|
|
}
|
|
|
|
func wechatSafeKeyPart(value string) string {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" {
|
|
return "na"
|
|
}
|
|
var builder strings.Builder
|
|
for _, r := range value {
|
|
if r >= '0' && r <= '9' || r >= 'a' && r <= 'z' || r >= 'A' && r <= 'Z' || r == '_' || r == '-' {
|
|
builder.WriteRune(r)
|
|
}
|
|
}
|
|
if builder.Len() == 0 {
|
|
return sha256Hex(value)
|
|
}
|
|
return strings.ToLower(builder.String())
|
|
}
|
|
|
|
func wechatTemplateID(items []Event, fallback string) string {
|
|
for _, item := range items {
|
|
if value := strings.TrimSpace(item.WxTemplateID); value != "" {
|
|
return value
|
|
}
|
|
}
|
|
return strings.TrimSpace(fallback)
|
|
}
|
|
|
|
func buildWechatPayload(batch Batch, openID string, templateID string, page string) warningWechatPayload {
|
|
latest := latestEvent(batch.Items)
|
|
if !batch.LatestTime.IsZero() {
|
|
latest.WarnTime = batch.LatestTime
|
|
}
|
|
return warningWechatPayload{
|
|
ToUser: strings.TrimSpace(openID),
|
|
TemplateID: strings.TrimSpace(templateID),
|
|
Page: strings.TrimSpace(page),
|
|
Data: map[string]warningWechatTemplateValue{
|
|
"character_string6": {Value: wechatCharacterString(wechatDeviceCharacterText(batch), 32)},
|
|
"thing7": {Value: truncateWechatText(wechatTypeText(batch), 20)},
|
|
"thing9": {Value: truncateWechatText(wechatReasonText(batch), 20)},
|
|
"character_string32": {Value: wechatCharacterString(wechatValueText(batch), 32)},
|
|
"time8": {Value: wechatTimeText(latest)},
|
|
},
|
|
}
|
|
}
|
|
|
|
func latestEvent(items []Event) Event {
|
|
var result Event
|
|
for _, item := range items {
|
|
if result.WarnTime.IsZero() || item.WarnTime.After(result.WarnTime) {
|
|
result = item
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func wechatDeviceText(batch Batch) string {
|
|
if batchItemCount(batch) == 0 {
|
|
return "--"
|
|
}
|
|
first := displayObjectCode(batch.Items[0])
|
|
if batchItemCount(batch) == 1 {
|
|
return first
|
|
}
|
|
return fmt.Sprintf("%s等%d张卡", first, batchItemCount(batch))
|
|
}
|
|
|
|
func wechatDeviceCharacterText(batch Batch) string {
|
|
if batchItemCount(batch) == 0 {
|
|
return "NA"
|
|
}
|
|
first := displayObjectCode(batch.Items[0])
|
|
if batchItemCount(batch) == 1 {
|
|
return first
|
|
}
|
|
return fmt.Sprintf("TOTAL-%d", batchItemCount(batch))
|
|
}
|
|
|
|
func wechatTypeText(batch Batch) string {
|
|
title := warningTitle(batch)
|
|
if title == "" {
|
|
title = "设备预警"
|
|
}
|
|
if batchItemCount(batch) > 1 {
|
|
return fmt.Sprintf("%s,共%d张卡", title, batchItemCount(batch))
|
|
}
|
|
return title
|
|
}
|
|
|
|
func wechatReasonText(batch Batch) string {
|
|
if batchItemCount(batch) == 1 {
|
|
return emailWarningReason(batch.Items[0])
|
|
}
|
|
reason := fmt.Sprintf("%s\uff0c\u5171%d\u5f20\u5361", levelText(batchHighestLevel(batch)), batchItemCount(batch))
|
|
if batchHasAffectOperation(batch) {
|
|
reason += ",涉及停机或断网"
|
|
}
|
|
return reason
|
|
}
|
|
|
|
func wechatValueText(batch Batch) string {
|
|
if batchItemCount(batch) > 1 {
|
|
return fmt.Sprintf("TOTAL-%d", batchItemCount(batch))
|
|
}
|
|
if batchItemCount(batch) == 0 {
|
|
return "NA"
|
|
}
|
|
item := batch.Items[0]
|
|
unit := strings.TrimSpace(item.UnitText)
|
|
switch unit {
|
|
case "天":
|
|
return fmt.Sprintf("%.0fd", item.Actual)
|
|
case "月":
|
|
return fmt.Sprintf("%.0fm", item.Actual)
|
|
case "%":
|
|
return fmt.Sprintf("%.2f%%", item.Actual)
|
|
case "MB":
|
|
if item.TotalFlowMB > 0 {
|
|
return fmt.Sprintf("%.3f/%.3fMB", item.Actual, item.TotalFlowMB)
|
|
}
|
|
return fmt.Sprintf("%.3fMB", item.Actual)
|
|
default:
|
|
if unit == "" {
|
|
return fmt.Sprintf("%.3f", item.Actual)
|
|
}
|
|
return fmt.Sprintf("%.3f%s", item.Actual, unit)
|
|
}
|
|
}
|
|
|
|
func wechatTimeText(item Event) string {
|
|
if item.WarnTime.IsZero() {
|
|
return gtime.Now().Format("2006-01-02 15:04:05")
|
|
}
|
|
return item.WarnTime.Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
func sendWechatSubscription(ctx context.Context, cfg warningWechatConfig, payload warningWechatPayload) error {
|
|
return sendWechatSubscriptionAttempt(ctx, cfg, payload, true)
|
|
}
|
|
|
|
type warningWechatAPIError struct {
|
|
Code int
|
|
Message string
|
|
}
|
|
|
|
func (err *warningWechatAPIError) Error() string {
|
|
return fmt.Sprintf("wechat subscribe code %d: %s", err.Code, err.Message)
|
|
}
|
|
|
|
func warningWechatTokenInvalid(code int) bool {
|
|
return code == 40001 || code == 40014 || code == 42001
|
|
}
|
|
|
|
func sendWechatSubscriptionAttempt(ctx context.Context, cfg warningWechatConfig, payload warningWechatPayload, refreshToken bool) error {
|
|
accessToken, err := warningWechatAccessToken(ctx, cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
query := url.Values{}
|
|
query.Set("access_token", accessToken)
|
|
body, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
timeout := warningWechatTimeout(cfg)
|
|
reqCtx, cancel := context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
req, err := http.NewRequestWithContext(reqCtx, http.MethodPost, warningWechatSendURL+"?"+query.Encode(), bytes.NewReader(body))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Header.Set("Content-Type", "application/json")
|
|
resp, err := (&http.Client{Timeout: timeout}).Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
var result warningWechatSendResponse
|
|
if err = json.NewDecoder(resp.Body).Decode(&result); err != nil {
|
|
return err
|
|
}
|
|
if resp.StatusCode < http.StatusOK || resp.StatusCode >= http.StatusMultipleChoices {
|
|
return fmt.Errorf("wechat subscribe http status %d: %s", resp.StatusCode, result.ErrMsg)
|
|
}
|
|
if result.ErrCode != 0 {
|
|
if refreshToken && warningWechatTokenInvalid(result.ErrCode) {
|
|
cacheKey := "iotlcm:warning:wechat:access_token:" + sha256Hex(cfg.AppID)
|
|
if _, err = g.Redis().Do(ctx, "DEL", cacheKey); err != nil {
|
|
return fmt.Errorf("invalidate wechat access token: %w", err)
|
|
}
|
|
return sendWechatSubscriptionAttempt(ctx, cfg, payload, false)
|
|
}
|
|
return &warningWechatAPIError{Code: result.ErrCode, Message: result.ErrMsg}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func warningWechatAccessToken(ctx context.Context, cfg warningWechatConfig) (string, error) {
|
|
cacheKey := "iotlcm:warning:wechat:access_token:" + sha256Hex(cfg.AppID)
|
|
if value, err := g.Redis().Get(ctx, cacheKey); err == nil && strings.TrimSpace(value.String()) != "" {
|
|
return strings.TrimSpace(value.String()), nil
|
|
}
|
|
|
|
warningWechatTokenMu.Lock()
|
|
defer warningWechatTokenMu.Unlock()
|
|
if value, err := g.Redis().Get(ctx, cacheKey); err == nil && strings.TrimSpace(value.String()) != "" {
|
|
return strings.TrimSpace(value.String()), nil
|
|
}
|
|
|
|
query := url.Values{}
|
|
query.Set("grant_type", "client_credential")
|
|
query.Set("appid", cfg.AppID)
|
|
query.Set("secret", cfg.AppSecret)
|
|
timeout := warningWechatTimeout(cfg)
|
|
reqCtx, cancel := context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
req, err := http.NewRequestWithContext(reqCtx, http.MethodGet, warningWechatAccessTokenURL+"?"+query.Encode(), nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
resp, err := (&http.Client{Timeout: timeout}).Do(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
var result warningWechatAccessTokenResponse
|
|
if err = json.NewDecoder(resp.Body).Decode(&result); err != nil {
|
|
return "", err
|
|
}
|
|
if resp.StatusCode < http.StatusOK || resp.StatusCode >= http.StatusMultipleChoices {
|
|
return "", fmt.Errorf("wechat access token http status %d: %s", resp.StatusCode, result.ErrMsg)
|
|
}
|
|
if result.ErrCode != 0 || strings.TrimSpace(result.AccessToken) == "" {
|
|
return "", fmt.Errorf("wechat access token code %d: %s", result.ErrCode, result.ErrMsg)
|
|
}
|
|
|
|
ttl := result.ExpiresIn - 300
|
|
if ttl <= 0 {
|
|
ttl = 300
|
|
}
|
|
if _, err = g.Redis().Do(ctx, "SETEX", cacheKey, ttl, result.AccessToken); err != nil {
|
|
g.Log().Warningf(ctx, "WarningNotify cache wechat access token failed, err:%+v", err)
|
|
}
|
|
return result.AccessToken, nil
|
|
}
|
|
|
|
func warningWechatTimeout(cfg warningWechatConfig) time.Duration {
|
|
if value := strings.TrimSpace(cfg.Timeout); value != "" {
|
|
if timeout, err := time.ParseDuration(value); err == nil && timeout > 0 {
|
|
return timeout
|
|
}
|
|
}
|
|
return warningWechatDefaultTimeout
|
|
}
|
|
|
|
func maskOpenID(value string) string {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" {
|
|
return ""
|
|
}
|
|
if utf8.RuneCountInString(value) <= 8 {
|
|
return value
|
|
}
|
|
runes := []rune(value)
|
|
return string(runes[:4]) + "..." + string(runes[len(runes)-4:])
|
|
}
|
|
|
|
func sha256Hex(value string) string {
|
|
sum := sha256.Sum256([]byte(strings.TrimSpace(value)))
|
|
return hex.EncodeToString(sum[:])[:16]
|
|
}
|
|
|
|
func truncateWechatText(value string, maxRunes int) string {
|
|
value = strings.TrimSpace(value)
|
|
if maxRunes <= 0 {
|
|
return ""
|
|
}
|
|
runes := []rune(value)
|
|
if len(runes) <= maxRunes {
|
|
return value
|
|
}
|
|
return string(runes[:maxRunes-1]) + "…"
|
|
}
|
|
|
|
func wechatCharacterString(value string, maxRunes int) string {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" || maxRunes <= 0 {
|
|
return "NA"
|
|
}
|
|
var builder strings.Builder
|
|
for _, r := range value {
|
|
if r >= 33 && r <= 126 {
|
|
builder.WriteRune(r)
|
|
}
|
|
}
|
|
result := builder.String()
|
|
if result == "" {
|
|
return "NA"
|
|
}
|
|
runes := []rune(result)
|
|
if len(runes) <= maxRunes {
|
|
return result
|
|
}
|
|
return string(runes[:maxRunes])
|
|
}
|