Files
iotlcm/admin-svr/server/internal/logic/warningnotify/wechat_sender.go
T

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])
}