Go 打印函数执行时间


package main

import (
    "fmt"
    "time"
)

func timeCost(start time.Time) {
    terminal := time.Since(start)
    fmt.Println(terminal)
}

func main() {
    defer timeCost(time.Now())
    fmt.Println("start program")
    time.Sleep(5 * time.Second)
    fmt.Println("finish program")
}
阅读全文

Go MySQL 库 go-sql-driver


https://github.com/go-sql-driver/mysql

引入库

import (
    "database/sql"

    _ "github.com/go-sql-driver/mysql"
)

查询数据

// 建立 MySQL 链接
db, err := sql.Open("mysql", "user:passwd@tcp(ip:port)/db")
if err != nil {
    fmt.Println(err)
}
defer db.Close()
// 执行 SQL
rows, rerr := db.Query(sqlInfo)
if rerr != nil {
    fmt.Println(rerr)
}
defer rows.Close()
// 获取 SQL 数据
for rows.Next() {
    var returnData string
    err := rows.Scan(&returnData)
    if err != nil {
        fmt.Println(err)
    }
    return returnData
}
return returnData

写入/更新数据

// 建立 MySQL 链接
db, err := sql.Open("mysql", "user:passwd@tcp(ip:port)/db")
if err != nil {
    fmt.Println(err)
}
defer db.Close()
// 执行 SQL
db.Exec(sqlInfo)
阅读全文

Go make 和 new 的区别


new 和 make 都可以用来分配空间,初始化类型,但是它们确有不同。
new(T)返回的是 T 的指针
new(T)为一个 T 类型新值分配空间并将此空间初始化为 T 的零值,返回的是新值的地址,也就是 T 类型的指针 *T,该指针指向 T 的新分配的零值。

package main

import "fmt"

func main() {
    p1 := new(int)
    fmt.Printf("p1 --> %#v \n ", p1)
    fmt.Printf("p1 point to --> %#v \n ", *p1)

    var p2 *int
    i := 0
    p2 = &i
    fmt.Printf("p2 --> %#v \n ", p2)
    fmt.Printf("p2 point to --> %#v \n ", *p2)
}


阅读全文

Go 操作 excel


读 excel

package main
    
import (
    "fmt"
    
    "github.com/tealeg/xlsx"
)
    
func main() {
    excelFileName := "test.xls"
    xlFile, _ := xlsx.OpenFile(excelFileName)
    for _, sheet := range xlFile.Sheets {
        for _, row := range sheet.Rows {
            for _, cell := range row.Cells {
                fmt.Println(cell.String())
            }
        }
    }
}

写 excel

package main
    
import (
    "encoding/csv"
    "os"
)
    
func main() {
    f, err := os.Create("test.xls")
    if err != nil {
        panic(err)
    }
    defer f.Close()
    // 写入 utf-8 编码
    f.WriteString("\xEF\xBB\xBF")
    w := csv.NewWriter(f)
    w.Write([]string{"编号", "姓名", "年龄"})
    w.Write([]string{"1", "张三", "23"})
    w.Write([]string{"2", "李四", "24"})
    w.Write([]string{"3", "王五", "25"})
    w.Write([]string{"4", "赵六", "26"})
    w.Flush()
}
阅读全文

Go 处理时间


package main

import (
    "fmt"
"strconv"
    "time"
)

// TimeUnix is time for unix
func TimeUnix() string {
naiveTime := time.Now().Unix()
naiveTimeString := strconv.FormatInt(naiveTime, 10)
return naiveTimeString
}

func main() {
    timeNow := time.Now()
    fmt.Println("tNow(time format): ", timeNow)
    
    // 时间转化为 string layout 必须为 "2006-01-02 15:04:05"
    timeNowStr := timeNow.Format("2006-01-02 15:04:05")
    fmt.Println("tNow(string format): ", timeNowStr)

    timeNowUnixInt := timeNow.Unix()
    fmt.Println(timeNowUnixInt)

    // string 转化为时间 layout 必须为 "2006-01-02 15:04:05"
    t1, _ := time.Parse("2006-01-02 15:04:05", "2014-06-15 08:37:18")
    fmt.Println("t(time format): ", t1)
    t2 := time.Unix(1389058332, 0).Format("2006-01-02 15:04:05")
    fmt.Println("t(time format): ", t2)
}
阅读全文

Go Error 处理


Go 中关于错误的哲学,应该尽可能优雅地处理错误。

先说结论:

  • 使用"隐藏内部细节的错误处理"
  • 使用 errors.Wrap 封装原始 error
  • 使用 errors.Cause 找出原始 error
  • 为了行为而断言,而不是类型
  • 尽量减少错误值的使用

error 的类型是 interface

type error interface {
    Error() string
}

创建error:

// example 1
func Sqrt(f float64) (float64, error) {
    if f < 0 {
        return 0, errors.New("math: square root of negative number")
    }
    // implementation
}

// example 2
if f < 0 {
    return 0, fmt.Errorf("math: square root of negative number %g", f)
}

阅读全文

Go 使用 OS 包操作文件


判断文件是否存在 存在返回 true 不存在返回 false

func checkFileIsExist(filename string) bool {
    var exist = true
    if _, err := os.Stat(filename); os.IsNotExist(err) {
        exist = false
    }
    return exist
}

打开文件 返回文件指针

file, error := os.Open("/tmp/1.txt")
if error != nil {
    fmt.Println(error)
}
fmt.Println(file)
file.Close()

以读写方式打开文件 返回文件指针 如果不存在则创建

file2, error := os.OpenFile("/tmp/2.txt", os.O_RDWR|os.O_CREATE, 0644)
if error != nil {
    fmt.Println(error)
}
fmt.Println(file2)
file2.Close()

创建文件 Create 函数也是调用的 OpenFile

file3, error := os.Create("/tmp/3.txt")
if error != nil {
    fmt.Println(error)
}
fmt.Println(file3)
file3.Close()

删除文件

del := os.Remove("/tmp/1.txt")
if del != nil {
    fmt.Println(del)
}

删除指定 path 下的所有文件

delDir := os.RemoveAll("/tmp/testdir")
if delDir != nil {
        fmt.Println(delDir)
}

阅读全文

Go 使用 clientv3 连接 etcd


etcd3 和 etcd2 的 api 不太一样。
etcd3 移除了目录概念,只有单纯的 kv。
api 也只有 rpc 模式,http 需要 proxy。

package etcd

import (
    "context"
    "log"
    "time"

    "../conf"

    "go.etcd.io/etcd/clientv3"
)

var (
    endpoints   = make([]string, 1)
    dialTimeout = 3 * time.Second
    username    string
    password    string
)

// Init is init etcd
func Init() {
    endpoints = append(endpoints, conf.RunTimeInfo.EtcdHost)
    username = conf.RunTimeInfo.EtcdUser
    password = conf.RunTimeInfo.EtcdPasswd
}

// connetEtcd is connect etcd
func connetEtcd() *clientv3.Client {
    cli, cliErr := clientv3.New(clientv3.Config{
        Endpoints:   endpoints,
        DialTimeout: dialTimeout,
        Username:    username,
        Password:    password,
    })
    if cliErr != nil {
        log.Println(cliErr)
    }
    return cli
}

// KeepAlive is make etcd keepalive with ttl
func KeepAlive(keyID clientv3.LeaseID) {
    cli := connetEtcd()
    if cli != nil {
        ka, kaErr := cli.KeepAliveOnce(context.TODO(), keyID)
        if kaErr != nil {
            log.Println(kaErr)
        }
        log.Println("ttl:", ka)
        cli.Close()
    }
}

// GetKV is get kv from etcd
func GetKV(dir string) []string {
    cli := connetEtcd()
    var kvList = make([]string, 0)
    if cli != nil {
        resp, respErr := cli.Get(context.TODO(), dir, clientv3.WithPrefix())
        if respErr != nil {
            log.Println(respErr)
        }
        for _, ev := range resp.Kvs {
            kvTemp := string(ev.Key) + ":" + string(ev.Value)
            kvList = append(kvList, kvTemp)
        }
        cli.Close()
    }
    return kvList
}

// PutKV is set kv to etcd
func PutKV(writePath string, writeValue string) {
    cli := connetEtcd()
    if cli != nil {
        resp, respErr := cli.Put(context.TODO(), writePath, writeValue)
        if respErr != nil {
            log.Println(respErr)
        }
        if resp != nil {
            log.Println(resp.Header)
            cli.Close()
        }
    }
}

// PutKVTTL is set kv to etcd with ttl
func PutKVTTL(registKey, registValue string) {
    cli := connetEtcd()
    if cli != nil {
        resp, respErr := cli.Grant(context.TODO(), 10)
        if respErr != nil {
            log.Println(respErr)
        }
        conf.ServID = resp.ID
        putResp, putErr := cli.Put(context.TODO(), registKey, registValue, clientv3.WithLease(resp.ID))
        if putErr != nil {
            log.Println(putErr)
        }
        log.Println(putResp.Header)
        cli.Close()
    }
}

// DelKV is del kv to etcd
func DelKV(writePath string) {
    cli := connetEtcd()
    if cli != nil {
        resp, respErr := cli.Delete(context.TODO(), writePath)
        if respErr != nil {
            log.Println(respErr)
        }
        log.Println(resp.Header)
        cli.Close()
    }
}
阅读全文

Go convert


[]string to string

package main

import (
    "fmt"
    "strings"
)

var naiveList string

func main() {
    naiveArray := []string{"a", "b", "c", "d", "e"}
    for _, v := range naiveArray {
        if naiveList == "" {
        naiveList = v
        } else {
        naiveList = strings.Join([]string{naiveList, v}, ",")
        }
    }
    fmt.Println(naiveList)
}

map to struct

import (
    "testing"
    "github.com/goinggo/mapstructure"
)

func TestMap2Struct(t *testing.T) {
    mapInstance := make(map[string]interface{})
    mapInstance["Name"] = "liang637210"
    mapInstance["Age"] = 28
    var person Person
    // 将 map 转换为指定的结构体
    if err := mapstructure.Decode(mapInstance, &person); err != nil {
        t.Fatal(err)
    }
    t.Logf("map2struct 后得到的 struct 内容为:%v", person)
}

struct to map

import (
    "testing"
    "reflect"
)

type User struct {
    Id        int    `json:"id"`
    Username    string    `json:"username"`
    Password    string    `json:"password"`
}

func Struct2Map(obj interface{}) map[string]interface{} {
    t := reflect.TypeOf(obj)
    v := reflect.ValueOf(obj)
    var data = make(map[string]interface{})
    for i := 0; i < t.NumField(); i++ {
        data[t.Field(i).Name] = v.Field(i).Interface()
    }
    return data
}

func TestStruct2Map(t *testing.T) {
    user := User{5, "zhangsan", "password"}
    data := Struct2Map(user)
    t.Logf("struct2map 得到的 map 内容为:%v", data)
}
阅读全文