实名认证接口

ali_cloud_tool/ali_cloud_express_client_test.go

@@ -0,0 +1,130 @@
+package ali_cloud_tool
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/redis/go-redis/v9"
+	"log"
+	"testing"
+	"time"
+)
+
+func TestAliCloudExpressClient_GetLogisticsInfo(t *testing.T) {
+	type fields struct {
+		AppCode string
+		Host    string
+	}
+	type args struct {
+		mobile string
+		number string
+	}
+	tests := []struct {
+		name    string
+		fields  fields
+		args    args
+		wantRes *ExpressRes
+		wantErr bool
+	}{
+		{
+			name: "test1",
+			fields: fields{
+				AppCode: "",
+				Host:    "",
+			},
+			args: args{
+				mobile: "",
+				number: "",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			a := &AliCloudExpressClient{
+				AppCode: tt.fields.AppCode,
+				Host:    tt.fields.Host,
+			}
+			gotRes, err := a.GetLogisticsInfo(tt.args.mobile, tt.args.number)
+			log.Println(gotRes, err)
+		})
+	}
+}
+
+func TestAliCloudExpressClient_GetLogisticsInfoFormCache(t *testing.T) {
+	rdb := redis.NewClient(&redis.Options{
+		Addr:     "",
+		Password: "",
+		DB:       1,
+	})
+
+	// 创建缓存实例
+	cache := NewRedisCache(rdb)
+	a := &AliCloudExpressClient{
+		AppCode: "",
+		Host:    "",
+		cache:   cache,
+	}
+	gotRes, err := a.GetLogisticsInfoFormCache(context.Background(), "", "", "", time.Minute)
+	log.Println(gotRes, err)
+}
+
+func TestAliCloudExpressClient_DeleteLogisticsInfoCache(t *testing.T) {
+	rdb := redis.NewClient(&redis.Options{
+		Addr:     "",
+		Password: "",
+		DB:       1,
+	})
+
+	// 创建缓存实例
+	cache := NewRedisCache(rdb)
+	a := &AliCloudExpressClient{
+		AppCode: "",
+		Host:    "",
+		cache:   cache,
+	}
+	err := a.DeleteLogisticsInfoCache(context.Background(), "", "")
+	log.Println(err)
+}
+
+type RedisCache struct {
+	client *redis.Client
+}
+
+func (r *RedisCache) Del(ctx context.Context, number string) error {
+	return r.client.Del(ctx, number).Err()
+}
+
+func NewRedisCache(client *redis.Client) *RedisCache {
+	return &RedisCache{client: client}
+}
+
+func (r *RedisCache) Set(ctx context.Context, number string, res string, ttl time.Duration) error {
+	if number == "" {
+		return errors.New("number不能为空")
+	}
+
+	return r.client.Set(ctx, number, res, ttl).Err()
+}
+
+// Get 从Redis获取IP信息
+func (r *RedisCache) Get(ctx context.Context, number string) (*ExpressRes, error) {
+	if number == "" {
+		return nil, errors.New("number不能为空")
+	}
+
+	data, err := r.client.Get(ctx, number).Bytes()
+	if err != nil {
+		if errors.Is(err, redis.Nil) {
+			return nil, nil // 键不存在，返回nil而不是错误
+		}
+		return nil, fmt.Errorf("无法获取物流信息: %w", err)
+	}
+
+	var info ExpressRes
+	if err = json.Unmarshal(data, &info); err != nil {
+		return nil, fmt.Errorf("无法解析物流信息: %w", err)
+	}
+
+	return &info, nil
+}