Go语言开发中常用的一些代码片段

一、通过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)
}
}
}