Golang开发中常用的代码片段汇总

2020-01-28 12:23:26王旭

本文主要总结的是我在编写golang应用时使用到的代码片段,提供出来以备以后使用,也给需要的朋友们以参考学习,下面话不多说,来看看详细介绍:

一、通过http package发送user:passwd信息


 发送类似于 curl -u 的请求
 client := &http.Client{}
 req, err := http.NewRequest("GET", <url>, nil)
 req.SetBasicAuth(<username>, <userpasswd>)
 if err != nil {
  log.Fatal(err)
 }

 resp, err := client.Do(req)
 if err != nil {
  log.Fatal(err)
 }
 content, err := ioutil.ReadAll(resp.Body)
 if err != nil {
  log.Fatal(err)
 }

二、golang通过反射将未知对象转换为array


func interfaceSlice(slice interface{}) []interface{} { 
 s := reflect.ValueOf(slice)
 if s.Kind() != reflect.Slice {
  panic("InterfaceSlice() given a non-slice type")
 }

 ret := make([]interface{}, s.Len())

 for i := 0; i < s.Len(); i++ {
  ret[i] = s.Index(i).Interface()
 }

 return ret
}

三、Mux 获取Get Query参数


 vals := r.URL.Query()
 oriDriver, ok := vals["driver"]

四、Mux 跨域访问代码


  methodsOk := handlers.AllowedMethods([]string{"GET", "HEAD", "POST", "PUT", "OPTIONS"})
 headersOk := handlers.AllowedHeaders([]string{"X-Requested-With"})
 originsOk := handlers.AllowedOrigins([]string{"*"})
 log.Println(http.ListenAndServe(":8000", handlers.CORS(headersOk, originsOk, methodsOk)(r)))

五、递归查找文件


 func readAPK(path string, apk map[string]int) {
 // fmt.Println("处理", path)
 files, _ := ioutil.ReadDir(path)
 for _, file := range files {
  // fmt.Println(file.Name(), file.IsDir())
  if file.IsDir() {
   readAPK(path+"/"+file.Name(), apk)
  } else {
   if strings.Compare(file.Name(), "APK.log") == 0 {
    ap := make(map[string]string)
    body, err := ioutil.ReadFile(path + "/" + file.Name())
    if err != nil {
     fmt.Printf("[%s]读取失败[%s]n", file.Name(), err.Error())
     return
    }

    err = json.Unmarshal(body, &ap)
    if err != nil {
     fmt.Printf("[%s]解析失败[%s]n", file.Name(), err.Error())
     return
    }

    for a := range ap {
     at := strings.Split(ap[a], "|+|")
     for _, atemp := range at {
      info := strings.Split(atemp, "|-|")
      if len(info) > 1 {
       header := strings.Split(info[1], " ")
       for _, h := range header {
        if strings.Contains(h, "Referer:") {
         u, err := url.ParseRequestURI(h[8:])
         if err != nil {
          fmt.Println(err.Error())
         } else {
          apk[u.Host]++
         }
        }
       }
      }

     }
    }
   }
  }
 }
}

六、解析Mongo已经存在的数据


type App struct { 
 Id string `json:"id" bson:"_id,omitempty"`
 User_id string `bson:"user_id"`
 Name string `bson:"name"`
 Domain string `bson:"domain"`
 Business_line string `bson:"business_line"`
}