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)
            }
        }
    }