自己封装的golang 操作数据库方法

王岩0001 · · 9261 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

这个是我用Go写的第一个东东,可能还存在些BUG没有测试到。这里主要是提供一个参考。各位可以改写成自己的风格。

      在命令行下输入如下两条命令,进行安装

go get github.com/male110/SimpleDb
        go install github.com/male110/SimpleDb

Go语言的数据库操作,只能用Rows.Scan来一次性读取所有列。感觉很不习惯,我还是习惯按照列名来一列列的取数据。所以我自己封装了一个数据结构MyRows,MyRows实现了一个函数,GetValue(name string, value interface{})可以按列名来取数据。如下所示:

err = rows.GetValue("IsBoy", &isBoy)
        if err != nil {
            fmt.Println(err)
            return
        }

    为了操作方便,还定义了其它的结构体,如MyDb,其Query函数可以直接返回 MyRows。NewDb用来创建MyDb结构,其参数与sql.Open一至,怎么传取决于你所使用的驱动程序。

db, err := SimpleDb.NewDb("mysql", "root:123@tcp(127.0.0.1:3306)/test?charset=utf8")
    if err != nil {
        fmt.Println("打开SQL时出错:", err.Error())
        return
    }
    var rows *SimpleDb.MyRows
    //从数据库中取数据
    rows, err = db.Query("select * from person")
    if err != nil {
        fmt.Println(err)
        return
    }
    //显示数据
    for rows.Next() {
        var id, age int
        var name string
        var isBoy bool
        //按字段名取数据,也可以用rows.Scan(&id,&name,&age),来取
        rows.GetValue("id", &id)
        rows.GetValue("name", &name)

        rows.GetValue("age", &age)
        //可以根据返回值,判断是否成功
        err = rows.GetValue("IsBoy", &isBoy)
        if err != nil {
            fmt.Println(err)
            return
        }
        fmt.Println(id, "\t", name, "\t", age, "\t", isBoy)
    }

        同时还时现了一个简单的ORM,实现了最基本的插入数据,修改数据,删除数据。 我一般使用ORM只用这么几个方法,其它的都是写SQL语句。这里只是一个参考,大家可以根据自己的需要,自己习惯,进行修改。改成自己需要的格式。数据结构的定义格式如下:

type Person struct {
    /*TableName类型只是用来设置表名。如果结构体名跟表名相同可以省略*/
    TableName SimpleDb.TableName "person"

    /*name是表名,PK用来设置是否主键,true主键,false非主键*/
    Id int `name:"id"PK:"true"Auto:"true"`

    Name   string "name" //tag里的name表是对应的字段名
    Age    int    "age"  //tag里的age表是对应的字段名
    IsBoy  bool
    NotUse string "-" //-不会保存到数据库中
}

      上面的说明已经很详细了, SimpleDb.TableName类型的字段,只用来在tag中定义结构体对应的表名,如果没有该字段,认为表名就是结构体名相同。 PK:"true"表示是主键,Auto:"true"表示该字段是自动增长的列,name:"id",来指定该字段对应的数据表中的列名,如不指定认为 跟字段名相同。当只需要指定列名时,可以直接写在tag中,如:"name"、"age"。tag为"-"表示不对应数据表中的任何列。

// 插入数据
p := &Person{Name: "张三丰", Age: 500, IsBoy: true}
db.Insert(p)
//修改数据
db.Update(p)
//删除数据
db.Delete(p)

下面来看一个完整的例子,首先他创表:

CREATE TABLE `person` (
    `id` INT(50) NOT NULL AUTO_INCREMENT,
    `name` VARCHAR(50) NULL DEFAULT NULL,
    `age` INT(50) NULL DEFAULT NULL,
    `IsBoy` SMALLINT(10) NULL,
    PRIMARY KEY (`id`)
)
COLLATE='utf8_general_ci';

insert into `person` (name,age,IsBoy) values('张三',20,0);
insert into `person` (name,age,IsBoy) values('王五',19,1);

下面是完整的代码

package main

import (
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/male110/SimpleDb"
)

type Person struct {
    /*TableName类型只是用来设置表名。如果结构体名跟表名相同可以省略*/
    TableName SimpleDb.TableName "person"

    /*name是表名,PK用来设置是否主键,true主键,false非主键*/
    Id int `name:"id"PK:"true"Auto:"true"`

    Name   string "name" //tag里的name表是对应的字段名
    Age    int    "age"  //tag里的age表是对应的字段名
    IsBoy  bool
    NotUse string "-" //-不会保存到数据库中
}

func main() {
    db, err := SimpleDb.NewDb("mysql", "root:123@tcp(127.0.0.1:3306)/test?charset=utf8")
    if err != nil {
        fmt.Println("打开SQL时出错:", err.Error())
        return
    }
    defer db.Close()
    p := &Person{Name: "张三丰", Age: 500, IsBoy: true}
    //插入一条数据
    err = db.Insert(p)
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println("新插入数据的ID:", p.Id)
    var rows *SimpleDb.MyRows
    //从数据库中取数据
    rows, err = db.Query("select * from person")
    if err != nil {
        fmt.Println(err)
        return
    }
    //显示数据
    for rows.Next() {
        var id, age int
        var name string
        var isBoy bool
        //按字段名取数据,也可以用rows.Scan(&id,&name,&age),来取
        rows.GetValue("id", &id)
        rows.GetValue("name", &name)

        rows.GetValue("age", &age)
        //可以根据返回值,判断是否成功
        err = rows.GetValue("IsBoy", &isBoy)
        if err != nil {
            fmt.Println(err)
            return
        }
        fmt.Println(id, "\t", name, "\t", age, "\t", isBoy)
    }
    //输出分割线
    fmt.Println("==========割割割割割割割割============")
    p.Name = "彭祖"
    p.Age = 800
    //修改数据
    _, err = db.Update(p)
    if err != nil {
        fmt.Println(err, "xxxx")
        return
    }
    //QueryDataRows返回一个DataRow数组,DataRow中有一map来存放行中的数据
    var arrRow []SimpleDb.DataRow
    arrRow, err = db.QueryDataRows("select * from person")
    if err != nil {
        fmt.Println(err, "zzzzz")
        return
    }
    for _, row := range arrRow {
        var id, age int
        var name string
        var isBoy bool
        //只能按字段名取数据
        row.GetValue("id", &id)
        row.GetValue("name", &name)
        row.GetValue("age", &age)
        //可以根据返回值,判断是否成功
        err = rows.GetValue("IsBoy", &isBoy)
        if err != nil {
            fmt.Println(err)
            return
        }
        fmt.Println(id, "\t", name, "\t", age, isBoy)
    }
    var p2 Person
    p2.Id = p.Id
    //根据主键从数据库中取单条数据
    err = db.Load(&p2)
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(p2)
    //根据主键删除一条数据
    db.Delete(p2)
}

有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:王岩0001

查看原文:自己封装的golang 操作数据库方法

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

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