From 8ba5c95a9cabfec2891817dac0cb650dcf0f7834 Mon Sep 17 00:00:00 2001
From: timerzz <zhhgzhhgzhhg@163.com>
Date: Fri, 14 Jun 2024 17:42:07 +0800
Subject: [PATCH] =?UTF-8?q?feat=20cn=20client=E6=94=AF=E6=8C=81tls?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/coach-client/client_cn.go      | 10 +++++++++-
 pkg/coach-client/client_cn_test.go |  5 ++---
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/pkg/coach-client/client_cn.go b/pkg/coach-client/client_cn.go
index 96d2273..1e1af19 100644
--- a/pkg/coach-client/client_cn.go
+++ b/pkg/coach-client/client_cn.go
@@ -2,9 +2,11 @@ package coach_client
 
 import (
 	"context"
+	"crypto/tls"
 	"fmt"
 	"github.com/bytedance/sonic"
 	"github.com/cloudwego/hertz/pkg/app/client"
+	"github.com/cloudwego/hertz/pkg/network/standard"
 	"github.com/cloudwego/hertz/pkg/protocol"
 	"github.com/corpix/uarand"
 	"time"
@@ -16,7 +18,13 @@ type CN struct {
 
 func CNClient() (cli *CN, err error) {
 	cli = new(CN)
-	cli.c, err = client.NewClient()
+	clientCfg := &tls.Config{
+		InsecureSkipVerify: true,
+	}
+	cli.c, err = client.NewClient(
+		client.WithTLSConfig(clientCfg),
+		client.WithDialer(standard.NewDialer()),
+	)
 	return
 }
 
diff --git a/pkg/coach-client/client_cn_test.go b/pkg/coach-client/client_cn_test.go
index b1bf14e..7a75eed 100644
--- a/pkg/coach-client/client_cn_test.go
+++ b/pkg/coach-client/client_cn_test.go
@@ -10,14 +10,13 @@ func TestCNClient(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	var page, total int
+	var page int
 	var resp ItemListData
 	var items []CNItem
-	for ; page <= total; page++ {
+	for ; page < 2; page++ {
 		if resp, err = cli.ListItems(context.Background(), page, 50); err != nil {
 			t.Fatal(err)
 		}
-		total = resp.TotalPages
 		items = append(items, resp.Items...)
 	}
 	t.Logf("共%d个", len(items))