dypid-client/internal/api/api.go

package api

import (
	"context"
	"dypid-client/internal/config"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strings"
	"sync"
	"time"

	"github.com/tidwall/gjson"
)

var httpClient = &http.Client{
	Transport: &http.Transport{
		MaxIdleConns:        500,
		MaxIdleConnsPerHost: 500,
		IdleConnTimeout:     30 * time.Minute,
	},
	Timeout: 30 * time.Second,
}

var limit chan struct{}

func init() {
	//限制同时请求数为500
	limit = make(chan struct{}, 500)
}

// InitConn 创建连接池
func InitConn() {
	wg := &sync.WaitGroup{}

	for range 10 {
		wg.Go(func() {
			for range 50 {
				resp, err := httpClient.Get(config.APPConfig.Url + "/api/test")
				if err != nil {
					fmt.Println(err)
					return
				}
				defer func() {
					io.Copy(io.Discard, resp.Body)
					resp.Body.Close()
				}()
			}
		})
	}

	wg.Wait()
}

func UploadDataToServer(ctx context.Context, data string) error {
	limit <- struct{}{}
	defer func() {
		<-limit
	}()

	//根据账号是否有视频判断要存储的token
	params := url.Values{}
	secUserId := strings.Split(data, "----")[1]
	hasVideo, err := CheckHasVideo(secUserId)
	if err != nil {
		return err
	}
	if hasVideo {
		params.Set("token", config.APPConfig.HasVideoToken)
	} else {
		params.Set("token", config.APPConfig.NoVideoToken)
	}

	params.Set("data", data)

	//http://127.0.0.1:8080/api/data?token=123456&data=123456
	request, err := http.NewRequest(
		"POST",
		config.APPConfig.Url+"/api/data?"+params.Encode(),
		nil,
	)
	if err != nil {
		return err
	}
	request.WithContext(ctx)

	resp, err := httpClient.Do(request)
	if err != nil {
		return err
	}
	defer func() {
		io.Copy(io.Discard, resp.Body)
		resp.Body.Close()
	}()

	return err
}

func CheckHasVideo(secUserId string) (bool, error) {
	client := &http.Client{}
	req, err := http.NewRequest(
		"GET",
		"https://imdesktop.douyin.com/aweme/v1/web/user/profile/other/?sec_user_id="+secUserId,
		nil,
	)

	if err != nil {
		fmt.Println(err)
		return false, err
	}
	req.Header.Add("User-Agent", "Apifox/1.0.0 (https://apifox.com)")
	req.Header.Add("Accept", "*/*")
	req.Header.Add("Host", "imdesktop.douyin.com")
	req.Header.Add("Connection", "keep-alive")

	res, err := client.Do(req)
	if err != nil {
		fmt.Println(err)
		return false, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		fmt.Println(err)
		return false, err
	}

	if gjson.Get(string(body), "user.aweme_count").Int() > 0 {
		return true, nil
	}

	return false, nil
}