当前位置：首页 > 文章列表 > 数据库 > MySQL > Go学习之抓数据并存入MySQL和返回json数据（三）

Go学习之抓数据并存入MySQL和返回json数据（三）

来源：SegmentFault 2023-01-23 20:01:18 0浏览收藏

怎么入门数据库编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《Go抓取网页数据并存入MySQL和返回json数据》，涉及到MySQL、JSON、go，有需要的可以收藏一下

上一节主要实现了使用 goquery 从图片网站 http://www.gratisography.com/ 抓取数据。主要抓取图片的

CREATE TABLE `gratisography` (
  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
  `img_url` varchar(255) DEFAULT NULL,
  `type_name` varchar(50) DEFAULT NULL,
  `title` varchar(255) DEFAULT NULL,
  `width` int(11) DEFAULT NULL,
  `height` int(11) DEFAULT NULL,
  `create_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=388 DEFAULT CHARSET=utf8;

创建完数据库之后，就开始使用GO来实现连接数据库等操作了。在GO中使用Go-MySQL-Driver is a lightweight and fast MySQL-Driver for Go's (golang) database/sql package

文档：http://godoc.org/github.com/go-sql-driver/mysql

在使用之前需要先使用以下命令获取该包：

package crawldata

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

然后写一个打开数据库的方法

package crawldata

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

func OpenDatabase() (*sql.DB, error) {
    // 连接数据库
    db, err := sql.Open("mysql", "root:mysql@tcp(xxx.xx.xx.xxx:3306)/databaseName?charset=utf8")
    if err != nil {
        return nil, err
    }
    return db, nil
}

上一节已在

package crawldata

import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "strconv"
    s "strings"
)

func OpenDatabase() (*sql.DB, error) {
    // 连接数据库
    db, err := sql.Open("mysql", "root:mysql@tcp(xxx.xx.xx.xxx:3306)/databaseName?charset=utf8")
    if err != nil {
        return nil, err
    }
    return db, nil
}

/*
   该函数将获取的数据存储到数据库
*/
func InsertData(datas *ImageDatas) {
    imageDatas := *datas
    // 连接数据库
    db, err := OpenDatabase()
    if err != nil {
        fmt.Printf(s.Join([]string{"连接数据库失败", err.Error()}, "-->"))
    }
    defer db.Close()

    for i := 0; i "))
        }
        defer imgIns.Close() // Close the statement when we leave main()

        img, err := imgIns.Exec(s.Join([]string{"http://www.gratisography.com", imageData.Src}, "/"), imageData.Tp, imageData.Title, imageData.Width, imageData.Height)
        if err != nil {
            fmt.Println(s.Join([]string{"插入数据失败", err.Error()}, "-->"))
        } else {
            success, _ := img.LastInsertId()
            // 数字变成字符串,success是int64型的值，需要转为int，网上说的Itoa64()在strconv包里不存在
            insertId := strconv.Itoa(int(success))
            fmt.Println(s.Join([]string{"成功插入数据：", insertId}, "t-->t"))
        }
    }
}

到此已经完成了数据抓取并存入数据库，在命令行中切换到