/**
 * @Author: gan
 * @Description:
 * @File:  main
 * @Version: 1.0.0
 * @Date: 2023/8/25 2:37 PM
 */

package main

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"runtime"
	"time"
)

var workers = runtime.NumCPU()

type result struct {
	jobName    string
	resultCode int
	resultInfo string
}

type job struct {
	jobName string
	results chan<- result
}

func main() {
	num := 10
	var jobNames []string
	for i := 0; i < num; i++ {
		jobNames = append(jobNames, fmt.Sprintf("%v", i))
	}
	exec(jobNames)
}

func exec(jobNames []string) {
	jobs := make(chan job, workers)
	results := make(chan result, len(jobNames))
	done := make(chan struct{}, workers)

	/**
	 * 把任务写入 JobCh通道
	 **/
	go func(jobs chan<- job, jobNames []string, results chan<- result) {
		for _, jobName := range jobNames {
			jobs <- job{jobName, results}
		}
		close(jobs)
	}(jobs, jobNames, results)

	/**
	 * 开启n个协程处理任务
	 **/
	for i := 0; i < workers; i++ {
		go func(done chan<- struct{}, jobs <-chan job) {
			for job := range jobs {
				job.do()
			}
			done <- struct{}{}
		}(done, jobs)
	}

	/**
	 * 查看任务是否完成
	 **/
	go func(done <-chan struct{}, results chan result) {
		for i := 0; i < workers; i++ {
			<-done
		}
		close(results)
	}(done, results)

	/**
	 * 取出结果
	 **/
	for result := range results {
		fmt.Printf("done: %s,%d,%s\n", result.jobName, result.resultCode, result.resultInfo)
	}
}

func (job job) do() {
	// 模拟处理结果
	demoRes := demoHandle()
	if demoRes != "" {
		job.results <- result{job.jobName, 0, "ok"}
	} else {
		job.results <- result{job.jobName, -1, "error"}
	}
}

func demoHandle() (result string) {
	url := "https://www.baidu.com"
	data := map[string]interface{}{
		"timestamp": 1692944539,
	}
	jsonData, err := json.Marshal(data)
	if err != nil {
		panic(err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
	defer cancel()
	req, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewBuffer(jsonData))
	if err != nil {
		panic(err)
	}
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		panic(err)
	}
	result = string(body)
	return
}