From c51d41ee8a56c9b24ce6beb6e13f4b05236cbff2 Mon Sep 17 00:00:00 2001
From: yahuian <yahuian@126.com>
Date: Wed, 18 Jun 2025 16:14:10 +0800
Subject: [PATCH] =?UTF-8?q?security=20=E6=A8=A1=E5=9D=97=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=20context=20=E8=B0=83=E7=94=A8=E5=87=BD=E6=95=B0=20(#836)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 miniprogram/security/security.go | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/miniprogram/security/security.go b/miniprogram/security/security.go
index 3e979a5..41b4548 100644
--- a/miniprogram/security/security.go
+++ b/miniprogram/security/security.go
@@ -1,6 +1,7 @@
 package security
 
 import (
+	context2 "context"
 	"fmt"
 	"strconv"
 
@@ -64,7 +65,12 @@ type MediaCheckAsyncRequest struct {
 
 // MediaCheckAsync 异步校验图片/音频是否含有违法违规内容
 func (security *Security) MediaCheckAsync(in *MediaCheckAsyncRequest) (traceID string, err error) {
-	accessToken, err := security.GetAccessToken()
+	return security.MediaCheckAsyncContext(context2.Background(), in)
+}
+
+// MediaCheckAsyncContext 异步校验图片/音频是否含有违法违规内容
+func (security *Security) MediaCheckAsyncContext(ctx context2.Context, in *MediaCheckAsyncRequest) (traceID string, err error) {
+	accessToken, err := security.GetAccessTokenContext(ctx)
 	if err != nil {
 		return
 	}
@@ -77,7 +83,7 @@ func (security *Security) MediaCheckAsync(in *MediaCheckAsyncRequest) (traceID s
 	req.Version = 2
 
 	uri := fmt.Sprintf(mediaCheckAsyncURL, accessToken)
-	response, err := util.PostJSON(uri, req)
+	response, err := util.PostJSONContext(ctx, uri, req)
 	if err != nil {
 		return
 	}
@@ -222,7 +228,12 @@ func (security *Security) MsgCheckV1(content string) (res MsgCheckResponse, err
 
 // MsgCheck 检查一段文本是否含有违法违规内容
 func (security *Security) MsgCheck(in *MsgCheckRequest) (res MsgCheckResponse, err error) {
-	accessToken, err := security.GetAccessToken()
+	return security.MsgCheckContext(context2.Background(), in)
+}
+
+// MsgCheckContext 检查一段文本是否含有违法违规内容
+func (security *Security) MsgCheckContext(ctx context2.Context, in *MsgCheckRequest) (res MsgCheckResponse, err error) {
+	accessToken, err := security.GetAccessTokenContext(ctx)
 	if err != nil {
 		return
 	}
@@ -235,7 +246,7 @@ func (security *Security) MsgCheck(in *MsgCheckRequest) (res MsgCheckResponse, e
 	req.Version = 2
 
 	uri := fmt.Sprintf(msgCheckURL, accessToken)
-	response, err := util.PostJSON(uri, req)
+	response, err := util.PostJSONContext(ctx, uri, req)
 	if err != nil {
 		return
 	}