golang基础:并行调用翻译接口 | 青训营笔记

344 阅读3分钟

这是我参与「第三届青训营 -后端场」笔记创作活动的的第1篇笔记 主要是并行调用线程 实现调用两个翻译接口,实现结果,熟悉下线程的使用方法


type DictRequest struct {
   TransType string `json:"trans_type"`
   Source    string `json:"source"`
   UserID    string `json:"user_id"`
}

//火山翻译结构体
type HuoshanRequest struct {
   Text     string `json:"text"`
   Language string `json:"language"`
}

//火山翻译响应结构体
type HuoshanResponse struct {
   Words []struct {
      Source  int    `json:"source"`
      Text    string `json:"text"`
      PosList []struct {
         Type      int `json:"type"`
         Phonetics []struct {
            Type int    `json:"type"`
            Text string `json:"text"`
         } `json:"phonetics"`
         Explanations []struct {
            Text     string `json:"text"`
            Examples []struct {
               Type      int `json:"type"`
               Sentences []struct {
                  Text      string `json:"text"`
                  TransText string `json:"trans_text"`
               } `json:"sentences"`
            } `json:"examples"`
            Synonyms []interface{} `json:"synonyms"`
         } `json:"explanations"`
         Relevancys []interface{} `json:"relevancys"`
      } `json:"pos_list"`
   } `json:"words"`
   Phrases  []interface{} `json:"phrases"`
   BaseResp struct {
      StatusCode    int    `json:"status_code"`
      StatusMessage string `json:"status_message"`
   } `json:"base_resp"`
}

type DictResponse struct {
   Rc   int `json:"rc"`
   Wiki struct {
      KnownInLaguages int `json:"known_in_laguages"`
      Description     struct {
         Source string      `json:"source"`
         Target interface{} `json:"target"`
      } `json:"description"`
      ID   string `json:"id"`
      Item struct {
         Source string `json:"source"`
         Target string `json:"target"`
      } `json:"item"`
      ImageURL  string `json:"image_url"`
      IsSubject string `json:"is_subject"`
      Sitelink  string `json:"sitelink"`
   } `json:"wiki"`
   Dictionary struct {
      Prons struct {
         EnUs string `json:"en-us"`
         En   string `json:"en"`
      } `json:"prons"`
      Explanations []string      `json:"explanations"`
      Synonym      []string      `json:"synonym"`
      Antonym      []string      `json:"antonym"`
      WqxExample   [][]string    `json:"wqx_example"`
      Entry        string        `json:"entry"`
      Type         string        `json:"type"`
      Related      []interface{} `json:"related"`
      Source       string        `json:"source"`
   } `json:"dictionary"`
}

//彩虹翻译请求函数
func query(word string, wg *sync.WaitGroup) {
   defer wg.Done() //当执行完函数,调用 把wg的等待组-1
   client := &http.Client{}
   request := DictRequest{TransType: "en2zh", Source: word}
   buf, err := json.Marshal(request)
   if err != nil {
      log.Fatal(err)
   }
   var data = bytes.NewReader(buf)
   req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
   if err != nil {
      log.Fatal(err)
   }
   req.Header.Set("Connection", "keep-alive")
   req.Header.Set("DNT", "1")
   req.Header.Set("os-version", "")
   req.Header.Set("sec-ch-ua-mobile", "?0")
   req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36")
   req.Header.Set("app-name", "xy")
   req.Header.Set("Content-Type", "application/json;charset=UTF-8")
   req.Header.Set("Accept", "application/json, text/plain, */*")
   req.Header.Set("device-id", "")
   req.Header.Set("os-type", "web")
   req.Header.Set("X-Authorization", "token:qgemv4jr1y38jyq6vhvi")
   req.Header.Set("Origin", "https://fanyi.caiyunapp.com")
   req.Header.Set("Sec-Fetch-Site", "cross-site")
   req.Header.Set("Sec-Fetch-Mode", "cors")
   req.Header.Set("Sec-Fetch-Dest", "empty")
   req.Header.Set("Referer", "https://fanyi.caiyunapp.com/")
   req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
   req.Header.Set("Cookie", "_ym_uid=16456948721020430059; _ym_d=1645694872")
   resp, err := client.Do(req)
   if err != nil {
      log.Fatal(err)
   }
   defer resp.Body.Close()
   bodyText, err := ioutil.ReadAll(resp.Body)
   if err != nil {
      log.Fatal(err)
   }
   if resp.StatusCode != 200 {
      log.Fatal("bad StatusCode:", resp.StatusCode, "body", string(bodyText))
   }
   var dictResponse DictResponse
   err = json.Unmarshal(bodyText, &dictResponse)
   if err != nil {
      log.Fatal(err)
   }
   fmt.Println(word, "UK:", dictResponse.Dictionary.Prons.En, "US:", dictResponse.Dictionary.Prons.EnUs)
   for _, item := range dictResponse.Dictionary.Explanations {
      fmt.Println(item)
   }
}
//火山翻译请求函数
func huoshanquery(word string, wg *sync.WaitGroup) {
   defer wg.Done() //当执行完函数,调用 把wg的等待组-1
   client := &http.Client{}
   //var data = strings.NewReader(`{"text":"hello","language":"en"}`)
   huoshandata := HuoshanRequest{word, "en"}
   buf, err := json.Marshal(huoshandata)
   if err != nil {
      log.Fatal(err)
   }
   data := bytes.NewReader(buf)
   req, err := http.NewRequest("POST", "https://translate.volcengine.com/web/dict/match/v1/?msToken=&X-Bogus=DFSzswVYQDcCwXH0SWsIXN7TlqSS&_signature=_02B4Z6wo000018IvwnwAAIDAVvmZc8UdXZfCL8bAAJMGPlRZqUqF2SyJmSvltMSdXl4EdVT.ukCXOr7mj1YHJuqvJkIZRntYrKhi0hVsgOwGRJPps4YHQuvkX01g7fg0OEybhhfHkdOQhbuuc6", data)
   if err != nil {
      log.Fatal(err)
   }
   req.Header.Set("authority", "translate.volcengine.com")
   req.Header.Set("accept", "application/json, text/plain, */*")
   req.Header.Set("accept-language", "zh-CN,zh;q=0.9")
   req.Header.Set("content-type", "application/json")
   req.Header.Set("cookie", "x-jupiter-uuid=16519378244771756; ttcid=7bca6c2893e944b9bbaad76e52ff44aa14; tt_scid=BSAJFpo5loXus7LkHrtgeyGZsEdV6UXGaAeyesTPhPSm00zXF1liTFaUGq1nG8VNee79; i18next=translate")
   req.Header.Set("origin", "https://translate.volcengine.com")
   req.Header.Set("referer", "https://translate.volcengine.com/translate?category=&home_language=zh&source_language=detect&target_language=zh&text=hello")
   req.Header.Set("sec-ch-ua", `" Not A;Brand";v="99", "Chromium";v="101", "Google Chrome";v="101"`)
   req.Header.Set("sec-ch-ua-mobile", "?0")
   req.Header.Set("sec-ch-ua-platform", `"macOS"`)
   req.Header.Set("sec-fetch-dest", "empty")
   req.Header.Set("sec-fetch-mode", "cors")
   req.Header.Set("sec-fetch-site", "same-origin")
   req.Header.Set("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36")
   resp, err := client.Do(req)
   if err != nil {
      log.Fatal(err)
   }
   defer resp.Body.Close()
   bodyText, err := ioutil.ReadAll(resp.Body)
   if err != nil {
      log.Fatal(err)
   }

   var huoshanResponse HuoshanResponse
   err = json.Unmarshal(bodyText, &huoshanResponse)
   if err != nil {
      log.Fatal(err)
   }
   fmt.Printf("%v\n", huoshanResponse)
}


func main() {
   if len(os.Args) != 2 {
      fmt.Fprintf(os.Stderr, `usage: simpleDict WORD
   example: simpleDict hello
         `)
      os.Exit(1)
   }
   word := os.Args[1]
   wg := sync.WaitGroup{}     //创建等待线程组
   wg.Add(2)                  //等待组的计数器 +2
   go query(word, &wg)        //调用线程,维护公共变量 wg
   go huoshanquery(word, &wg) ////调用线程  维护公共变量 wg
   wg.Wait()                  //当等待组计数器不等于 0 时阻塞直到变 0。

}

主要是 main方法中的创建等待线程组,使用等待线程组为公共变量,等待组的计数 没到0,那么线程会一直等待 当翻译函数调用完成 ,defer wg.Done() 会执行,把等待组的计数-1,当到0 的时候 就会停止线程等待