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

时间:2022-09-05 17:11:44

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

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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
发送类似于 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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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参数

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

四、Mux 跨域访问代码

?
1
2
3
4
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)))

五、递归查找文件

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
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已经存在的数据

?
1
2
3
4
5
6
7
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使用,分割。同时不能存在空格 

七、定时器

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)
  }
 }
}

总结

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

原文链接:https://chinazt.cc/2017/05/27/golangjing-chang-yong-dao-de-dai-ma-pian-duan/