Go sql insert update使用举例

本文结合使用场景简单介绍sql中的insert、update的使用。

以下是代码：

如果记录已经存在，则更新，否则插入新记录。

package main

import (
	"database/sql"
	"fmt"
	"log"
	"time"

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

)

var DB *sql.DB

var dataBase = "root:123456@tcp(127.0.0.1:3306)/web_portal?loc=Local&parseTime=true"

func Init() {
	var err error
	DB, err = sql.Open("mysql", dataBase)
	if err != nil {
		log.Fatalln("open db fail:", err)
	}

	DB.SetMaxOpenConns(20)
	DB.SetMaxIdleConns(15)

	err = DB.Ping()
	if err != nil {
		log.Fatalln("ping db fail:", err)
	}
}

func main() {

	Init()

	entry()

}

// update first, if not exist, then try to insert
func entry() {

	now := time.Now().Unix()
	sql := fmt.Sprintf(
		"update tbl_host set version='%s', timestamp='%d' where ip='%s'",
		"3.20.2",
		now,
		"192.168.11.23",
	)

	log.Println("sql:", sql)

	result, err := DB.Exec(sql)
	if err != nil {
		log.Println("exec failed:", err, ", sql:", sql)
		return
	}

	idAff, err := result.RowsAffected()
	if err != nil {
		log.Println("RowsAffected failed:", err)
		return
	}
	log.Println("id:", idAff)
	if idAff == 0 {
		sql := fmt.Sprintf(
			"insert into tbl_host(ip, version, timestamp) values ('%s', '%s', '%d')",
			"192.168.11.23",
			"3.20.1",
			now,
		)

		log.Println("not exsit, then try to insert")
		tryInsert(sql)

	}

	log.Println("sucess")
}

func tryInsert(sql string) {

	_, err := DB.Exec(sql)
	if err != nil {
		log.Println("exec failed:", err, ", sql:", sql)
	}

}

output:

第一次执行，记录不存在，update不影响任何行，然后尝试insert操作。

2018/04/30 22:15:13 sql: update tbl_host set version='3.20.2', timestamp='1525097713' where ip='192.168.11.23'

2018/04/30 22:15:13 id: 0

2018/04/30 22:15:13 not exsit, then try to insert

2018/04/30 22:15:13 sucess

再次执行，此时已经存在同一个ip的记录，只会执行update更新：

2018/04/30 22:13:30 sql: update tbl_host set version='3.20.2', timestamp='1525097610' where ip='192.168.11.22'

2018/04/30 22:13:30 id: 1

2018/04/30 22:13:30 sucess