58 lines
1.9 KiB
Go
58 lines
1.9 KiB
Go
package immoderation
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
|
|
green "github.com/alibabacloud-go/green-20220302/v3/client"
|
|
"github.com/alibabacloud-go/tea/dara"
|
|
)
|
|
|
|
type AliyunConfig struct {
|
|
AccessKeyID string
|
|
AccessKeySecret string
|
|
RegionID string
|
|
Endpoint string
|
|
Service string
|
|
Timeout time.Duration
|
|
}
|
|
|
|
type CheckResult struct {
|
|
RequestID string
|
|
Latency time.Duration
|
|
}
|
|
|
|
func CheckAliyunText(ctx context.Context, config AliyunConfig) (*CheckResult, error) {
|
|
if strings.TrimSpace(config.AccessKeyID) == "" || strings.TrimSpace(config.AccessKeySecret) == "" {
|
|
return nil, errors.New("AccessKey ID或AccessKey Secret未配置")
|
|
}
|
|
clientConfig := &openapi.Config{AccessKeyId: dara.String(config.AccessKeyID), AccessKeySecret: dara.String(config.AccessKeySecret), RegionId: dara.String(config.RegionID), Protocol: dara.String("https")}
|
|
if endpoint := strings.TrimSpace(config.Endpoint); endpoint != "" {
|
|
clientConfig.Endpoint = dara.String(endpoint)
|
|
}
|
|
client, err := green.NewClient(clientConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if config.Timeout <= 0 {
|
|
config.Timeout = 1500 * time.Millisecond
|
|
}
|
|
ctx, cancel := context.WithTimeout(ctx, config.Timeout)
|
|
defer cancel()
|
|
params, _ := json.Marshal(map[string]string{"content": "内容审核连通性测试", "dataId": "admin-config-check"})
|
|
started := time.Now()
|
|
resp, err := client.TextModerationPlusWithContext(ctx, &green.TextModerationPlusRequest{Service: dara.String(config.Service), ServiceParameters: dara.String(string(params))}, &dara.RuntimeOptions{})
|
|
latency := time.Since(started)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp == nil || resp.Body == nil || dara.Int32Value(resp.Body.Code) != 200 {
|
|
return nil, errors.New("阿里云内容审核返回失败")
|
|
}
|
|
return &CheckResult{RequestID: dara.StringValue(resp.Body.RequestId), Latency: latency}, nil
|
|
}
|