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

本文主要总结的是我在编写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"`
}

重点:

1.使用bson修饰符

2.各个flag直接使用空格分割

3._id和omitempty使用,分割。同时不能存在空格

七、定时器

 for {
  now := time.Now()
  next := now.Add(time.Minute * 10)
  next = time.Date(next.Year(), next.Month(), next.Day(), next.Hour(), next.Minute(), 0, 0, next.Location())
  t := time.NewTimer(next.Sub(now))
  log.Printf("下次采集时间为[%s]\n", next.Format("200601021504"))

  select {
  case <-t.C:
   err := sync.Gather()
   if err != nil {
    log.Println(err)
   }
  }
 }

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流，谢谢大家对脚本之家的支持。