Go基础学习记录 - 编写Web应用程 - 完善Blog Model

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

完善Blog Model

前面的章节 我们分别介绍了MySQL、SQLite、PostgreSQL等数据库的使用
本节简单介绍并完善下我们自己的小博客的Model部分,这部分暂时先分享下Blog Model部分
这部分的完善其实很简单,但是对于像我这样的初学者,还是稍微费力一点

为了保持项目的可学习性,我这里将之前分享的代码积累了下,放在github上,想要尽快入手学习的,可以直接clone我的代码,写代码不上手,都等于白搭,光看的话,对于我来说,我是不行的,没办法学会。

项目地址

https://github.com/durban89/wiki_blog
tag: 1.0.11

有些同学可能看不懂,怎么就只给了这些,完全不懂呀。我把使用的命令打出来,照着操作,就应该可以解决了

git clone https://github.com/durban89/wiki_blog /local/path
cd /local/path
git fetch origin
git checkout 1.0.11

这些我觉得 够清晰了。OK!

继续分享"完善Model"。

在完善Model层之前我们先想一下,再进行博客的数据交互上我们需要哪些基本功能,我列出如下

  • 增加 - 博客空的,是不是我们需要增加的数据
  • 更新 - 如果发布错了,是不是需要进行更新
  • 删除 - 如果发表的文章违法了,是不是需要进行删除
  • 查找 - 如果进行查看,是不是需要把数据对出来,但是这里有读多条和一条的时候

好了功能我们大概知道了,下面我们先写个增加的逻辑,修改blog.go,添加Create()函数

// Create 添加数据
func Create(p *helpers.Page) (int64, error) {
    sql := fmt.Sprintf("INSERT %s SET title=?", tableName)
    stmt, err := Conn.Prepare(sql)
    if err != nil {
        return 0, err
    }

    res, err := stmt.Exec(p.Title)
    if err != nil {
        return 0, err
    }

    id, err := res.LastInsertId()
    if err != nil {
        return 0, err
    }

    return id, nil
}

首先说下参数这里参数的类型是Page,为什么要用Page,其实很好理解,我们前面已经大致列出了一篇文章的主要结构,其实一篇文章的结构的基本上就是我们需要存储的,Page的结构我这里只是简单加了一个ID,
其他项没有修改,后面再继续完善,一下子完善了,是需要花费蛮多时间的。Page目前的结构如下

type Page struct {
    ID     int         // 新加
    Title  string
    Body   []byte
    Script string
    Html   template.HTML
    Token  string
}

Create()函数暂时也存储了标题内容,其余的后期分享
返回的参数,这里(int64, error),也很好理解,就是返回我们创建记录的ID,其余的更新和删除功能都很类似下面列出如下

// Update 更新数据
func Update(p *helpers.Page, id int64) (int64, error) {
    sql := fmt.Sprintf("UPDATE %s SET title=? WHERE autokid=?", tableName)
    stmt, err := Conn.Prepare(sql)
    if err != nil {
        return 0, err
    }

    res, err := stmt.Exec(p.Title, id)
    if err != nil {
        return 0, err
    }

    affect, err := res.RowsAffected()
    if err != nil {
        return 0, err
    }

    return affect, nil
}
// Delete 删除数据
func Delete(id int64) (int64, error) {
    sql := fmt.Sprintf("DELETE FROM %s WHERE autokid=?", tableName)
    stmt, err := Conn.Prepare(sql)
    if err != nil {
        return 0, err
    }

    res, err := stmt.Exec(id)
    if err != nil {
        return 0, err
    }

    affect, err := res.RowsAffected()
    if err != nil {
        return 0, err
    }

    return affect, nil
}
// Query 获取数据
func Query() ([]helpers.Page, error) {
    sql := fmt.Sprintf("SELECT * FROM %s", tableName)
    rows, err := Conn.Query(sql)

    if err != nil {
        return nil, err
    }

    var res = []helpers.Page{}

    for rows.Next() {
        var autokid int
        var title string
        err = rows.Scan(&autokid, &title)

        if err != nil {
            return nil, err
        }

        p := helpers.Page{
            ID:    autokid,
            Title: title,
        }

        res = append(res, p)
    }

    return res, nil
}
// QueryOne 获取一条数据
func QueryOne() ([]helpers.Page, error) {
    sql := fmt.Sprintf("SELECT * FROM %s LIMIT 0, 1", tableName)

    rows, err := Conn.Query(sql)

    if err != nil {
        return nil, err
    }

    var res = []helpers.Page{}

    for rows.Next() {
        var autokid int
        var title string
        err = rows.Scan(&autokid, &title)

        if err != nil {
            return nil, err
        }

        p := helpers.Page{
            ID:    autokid,
            Title: title,
        }

        res = append(res, p)
    }

    return res, nil
}

上面基本上就完成了Blog Model的基础数据交互搭建,下面开始写我们的测试单元,简单的写几个测试,看看功能是否能够运行,完整的测试后面来完善,测试单元代码如下

package models

import (
    "testing"

    "github.com/durban89/wiki/helpers"
)

func TestCreate(t *testing.T) {
    p := helpers.Page{
        Title: "Test Create",
    }
    id, err := Create(&p)
    if err != nil {
        t.Error(err)
    }

    t.Log(id)
}

func TestUpdate(t *testing.T) {
    p := helpers.Page{
        Title: "Test Update",
    }

    effect, err := Update(&p, 2)
    if err != nil {
        t.Error(err)
    }

    t.Log(effect)
}

func TestDelete(t *testing.T) {
    effect, err := Delete(2)
    if err != nil {
        t.Error(err)
    }

    t.Log(effect)
}

func TestQueryOne(t *testing.T) {
    row, err := QueryOne()

    if err != nil {
        t.Error(err)
    } else {
        t.Log(row)
    }

    if len(row) == 1 {
        t.Log("正确")
    } else {
        t.Error("失败")
    }

    for i, k := range row {
        t.Log(i)
        t.Log(k)
    }
}

// TestQuery 测试获取数据
func TestQuery(t *testing.T) {
    row, err := Query()

    if err != nil {
        t.Error(err)
    } else {
        t.Log(row)
    }

    if len(row) > 0 {
        t.Log("正确")
    } else {
        t.Error("失败")
    }

    for i, k := range row {
        t.Log(i)
        t.Log(k)
    }
}

编译并进行测试,测试通过类似如下

$ go test -v
=== RUN   TestCreate
--- PASS: TestCreate (0.01s)
        blog_test.go:18: 6
=== RUN   TestUpdate
--- PASS: TestUpdate (0.00s)
        blog_test.go:31: 1
=== RUN   TestDelete
--- PASS: TestDelete (0.00s)
        blog_test.go:40: 1
=== RUN   TestQueryOne
--- PASS: TestQueryOne (0.00s)
        blog_test.go:49: [{3 title2 []   }]
        blog_test.go:53: 正确
        blog_test.go:59: 0
        blog_test.go:60: {3 title2 []   }
=== RUN   TestQuery
--- PASS: TestQuery (0.00s)
        blog_test.go:71: [{3 title2 []   } {4 Test Create []   } {5 Test Create []   } {6 Test Create []   }]
        blog_test.go:75: 正确
        blog_test.go:81: 0
        blog_test.go:82: {3 title2 []   }
        blog_test.go:81: 1
        blog_test.go:82: {4 Test Create []   }
        blog_test.go:81: 2
        blog_test.go:82: {5 Test Create []   }
        blog_test.go:81: 3
        blog_test.go:82: {6 Test Create []   }
PASS
ok      github.com/durban89/wiki/models 0.030s

今天就分享到这里,如果你有其他疑问请在下方留言或者加群交流

项目更新地址

https://github.com/durban89/typescript_demo.git
tag: 1.0.12

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

本文来自:Segmentfault

感谢作者:DurbanZhang

查看原文:Go基础学习记录 - 编写Web应用程 - 完善Blog Model

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

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