当前位置 博文首页 > 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)
}
}
}
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流，谢谢大家对站长博客的支持。
