golang sql根据struct字段来生成记录

疯狂的小企鹅 · · 1539 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
/*
	Author        : tuxpy
	Email         : q8886888@qq.com.com
	Create time   : 2017-11-04 23:13:08
	Filename      : main.go
	Description   :
*/

package main

import (
	"database/sql"
	"errors"
	"fmt"
	"reflect"
	"time"
	"utils"

	_ "github.com/mattn/go-sqlite3"
)

func ToStruct(rows *sql.Rows, to interface{}) error {
	v := reflect.ValueOf(to)
	if v.Elem().Type().Kind() != reflect.Struct {
		return errors.New("Expect a struct")
	}

	scan_dest := []interface{}{}
	column_names, _ := rows.Columns()

	addr_by_column_name := map[string]interface{}{}

	for i := 0; i < v.Elem().NumField(); i++ {
		one_value := v.Elem().Field(i)
		column_name := v.Elem().Type().Field(i).Tag.Get("sql")
		if column_name == "" {
			column_name = one_value.Type().Name()
		}
		addr_by_column_name[column_name] = one_value.Addr().Interface()
	}

	for _, column_name := range column_names {
		scan_dest = append(scan_dest, addr_by_column_name[column_name])
	}

	return rows.Scan(scan_dest...)

}

func main() {
	db, err := sql.Open("sqlite3", "/tmp/firstlink.db")
	utils.CheckErrorPanic(err)
	fmt.Println(db)
	db.Exec(`CREATE TABLE userinfo (
		work_id INTEGER PRIMARY KEY AUTOINCREMENT,
		name VARCHAR(64) NULL,
		departname VARCHAR(64) NULL,
		created DATE NULL
	)`)

	tx, err := db.Begin()

	defer func() {
		tx.Commit()
	}()
	utils.CheckErrorPanic(err)

	//	stmt, err := tx.Prepare("INSERT INTO userinfo(name, departname, created) values(?, ?, ?);")
	//	utils.CheckErrorPanic(err)
	//
	//	_, err = stmt.Exec("我", "爬虫", "2015-12-15")
	//	_, err = stmt.Exec("你", "客服", "2015-12-16")
	//	utils.CheckErrorPanic(err)

	rows, err := tx.Query("SELECT work_id, name, created FROM userinfo")
	utils.CheckErrorPanic(err)

	type Record struct {
		Name    string    `sql:"name"`
		WorkID  int       `sql:"work_id"`
		Created time.Time `sql:"created"`
	}

	for rows.Next() {
		record := Record{}

		utils.CheckErrorPanic(ToStruct(rows, &record))
		fmt.Println(record.WorkID, record.Name, record.Created)
	}
}

输出:

&{0xc42000c3e0 /tmp/firstlink.db 0 {0 0} [] map[] 0 0 0xc4200ee060 false map[] map[] 0 0 0 <nil>}
1 我 2015-12-15 00:00:00 +0000 UTC
2 你 2015-12-16 00:00:00 +0000 UTC

本文来自:开源中国博客

感谢作者:疯狂的小企鹅

查看原文:golang sql根据struct字段来生成记录

入群交流(和以上内容无关):Go中文网 QQ 交流群:798786647 或加微信入微信群:274768166 备注:入群;关注公众号:Go语言中文网

1539 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传