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

发布时间：2020-12-16 19:07:40 所属栏目：大数据来源：网络整理

导读：本文主要总结的是我在编写golang应用时使用到的代码片段，提供出来以备以后使用，也给需要的朋友们以参考学习，下面话不多说，来看看详细介绍：一、通过http package发送user:passwd信息发送类似于 curl -u 的请求 client := lt;url,nil) req.SetBasicAuth

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

总结

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!