同学们好!成功登录《我家租房网》系统的用户,可以点击位于搜索页面右上角的用户名,进入用户页面。点击用户页面中的“我的房源”,会打开我的房源页面。点击我的房源页面中的“发布新房源”,会打开发布新房源页面。用户在填写完房源信息表单后,点击“发布房源信息”,会打开上传房屋图片页面。点击上传房屋图片页面中的“选择文件”,选择事先准备好的房屋图片后,点击“上传”,系统后台会将房屋图片保存到文件服务器,同时将文件服务器返回的文件凭证存入数据库。这节课我们将实现上传房屋图片的功能。
我们将首先实现支持上传图片的后端微服务,再在前端服务器中添加路由处理函数并远程调用后端微服务,最后再将前后端连在一起做完整的功能测试。
首先我们来实现后端服务器。
第一步是创建特定的微服务。
创建微服务的过程还是和先前一样,直接在back-end工程目录下执行go-micro命令,子命令为new,参数为service,微服务名为UploadImage。
接着,需要修改一下go-micro为我们自动生成的接口描述。在back-end工程目录下的UploadImage微服务目录中,有一个名为proto的子目录,这就是存放接口描述脚本及其Go代码文件的地方。现阶段里面只有一个名为UploadImage.proto的ProtoBuf脚本文件。我们需要对该文件做一些修改:
x1...
2message CallRequest {
3 string houseId = 1;
4 bytes image = 2;
5 string suffix = 3;
6}
7
8message Image {
9 string url = 1;
10}
11
12message CallResponse {
13 string errno = 1;
14 string errmsg = 2;
15 Image data = 3;
16}
17...
这段代码描述的是,前端服务器和后端微服务之间的数据交换格式。CallRequest表示前端提供给后端的请求数据,其中包含字符串形式的房屋ID、字节序列形式的图片数据,和字符串形式的图片文件名后缀。CallResponse表示后端返回给前端的响应数据,其中包含错误代码、错误描述和图片信息。图片信息中只含有一个字段,即房屋图片的URL。
有了用ProtoBuf语言描述的远程调用接口,我们还需要把它编译成基于Go语言的程序代码。在UploadImage微服务目录中执行make命令,并携带四个目标参数,它们是init、proto、update和tidy。
在UploadImage微服务目录下,创建一个名为model的子目录,表示该微服务的模型层。在该子目录中创建mysql.go文件,封装所有与操作MySQL数据库有关的代码。在实际编写数据库访问代码之前,我们可以先把之前为前端服务器编写的mysql.go文件中的内容原封不动地复制到这里:
xxxxxxxxxx
1011package model
2
3import (
4 _ "github.com/go-sql-driver/mysql"
5 "github.com/jinzhu/gorm"
6 "time"
7)
8
9// 用户表
10type User struct {
11 ID uint `json:"id"` // 用户ID
12 Name string `json:"name" gorm:"size:32;unique"` // 用户名
13 Password_hash string `json:"password_hash" gorm:"size:128"` // 密码哈希
14 Mobile string `json:"mobile" gorm:"size:11;unique"` // 手机号
15 Real_name string `json:"real_name" gorm:"size:32"` // 真实姓名
16 Id_card string `json:"id_card" gorm:"size:20"` // 身份证号
17 Avatar_url string `json:"avatar_url" gorm:"size:256"` // 头像URL
18 Houses []*House `json:"houses"` // 房屋集
19 Orders []*Order `json:"orders"` // 订单集
20}
21
22// 房屋表
23type House struct {
24 gorm.Model
25 User_id uint `json:"user_id"` // 房主用户ID
26 Area_id uint `json:"area_id"` // 归属地区ID
27 Title string `json:"title" gorm:"size:64"` // 标题
28 Address string `json:"address" gorm:"size:512"` // 地址
29 Room_count int `json:"room_count" gorm:"default:1"` // 房间数
30 Acreage int `json:"acreage" gorm:"default:0"` // 面积
31 Price int `json:"price"` // 价格
32 Unit string `json:"unit" gorm:"size:32;default:''"` // 房间配置
33 Capacity int `json:"capacity" gorm:"default:1"` // 容纳人数
34 Beds string `json:"beds" gorm:"size:64;default:''"` // 床铺配置
35 Deposit int `json:"deposit" gorm:"default:0"` // 押金
36 Min_days int `json:"min_days" gorm:"default:1"` // 最少入住天数
37 Max_days int `json:"max_days" gorm:"default:0"` // 最多入住天数
38 Order_count int `json:"order_count" gorm:"default:0"` // 预定订单数
39 Index_image_url string `json:"index_image_url" gorm:"size:256;default:''"` // 主图片URL
40 Facilities []*Facility `json:"facilities" gorm:"many2many:houses_facilities"` // 设施集
41 Images []*Image `json:"images"` // 图片集
42 Orders []*Order `json:"orders"` // 订单集
43}
44
45// 城区表
46type Area struct {
47 ID uint `json:"id"` // 城区ID
48 Name string `json:"name" gorm:"size:32"` // 城区名
49 Houses []*House `json:"houses"` // 房屋集
50}
51
52// 设施表
53type Facility struct {
54 ID uint `json:"id"` // 设施ID
55 Name string `json:"name" gorm:"size:32"` // 设施名
56 Houses []*House `json:"houses"` // 房屋集
57}
58
59// 图片表
60type Image struct {
61 ID uint `json:"id"` // 图片ID
62 Url string `json:"url" gorm:"size:256"` // 图片URL
63 House_id uint `json:"house_id"` // 所属房屋ID
64}
65
66// 订单表
67type Order struct {
68 gorm.Model
69 User_id uint `json:"user_id"` // 下单用户ID
70 House_id uint `json:"house_id"` // 预定房屋ID
71 Begin_date time.Time `json:"begin_date" gorm:"type:datetime"` // 预定起始日
72 End_date time.Time `json:"end_date" gorm:"type:datetime"` // 预定终止日
73 Days int `json:"days"` // 预定天数
74 House_price int `json:"house_price"` // 房屋价格
75 Amount int `json:"amount"` // 总金额
76 Status string `json:"status" gorm:"default:'WAIT_ACCEPT'"` // 状态
77 Comment string `json:"comment" gorm:"size:512"` // 评论
78 Credit bool `json:"credit"` // 征信良好
79}
80
81var db *gorm.DB // 数据库连接池
82
83// 初始化数据库
84func InitDB() error {
85 var err error
86
87 // 连接数据库
88 if db, err = gorm.Open("mysql",
89 "root:123456@tcp(127.0.0.1:3306)/ihomedb?"+
90 "charset=utf8&parseTime=True&loc=Local"); err == nil {
91 // 设置最大连接数和最大空闲连接数
92 db.DB().SetMaxOpenConns(100)
93 db.DB().SetMaxIdleConns(10)
94
95 // 创建表
96 return db.AutoMigrate(new(User), new(House),
97 new(Area), new(Facility), new(Image), new(Order)).Error
98 }
99
100 return err
101}
修改UploadImage微服务目录中的main.go文件,添加对InitDB函数的调用,将服务发现改为Consul,绑定IP地址和端口:
xxxxxxxxxx
161...
2func main() {
3 // 初始化数据库
4 if err := model.InitDB(); err != nil {
5 logger.Fatal(err)
6 }
7 ...
8 srv.Init(
9 micro.Name(service),
10 micro.Version(version),
11 micro.Registry(consul.NewRegistry()),
12 micro.Address("127.0.0.1:9011"),
13 )
14 ...
15}
16...
程序启动伊始,即完成对数据库的初始化。其中包括两个动作,其一是连接ihomedb数据库并获得连接池对象,其二是在该数据库中创建表。当然,如果这些表已经存在,则不再创建。接着,在服务器对象的初始化部分,指示其注册到Consul服务器,同时绑定本机的9011端口。
接下来,我们为UploadImage微服务编写与数据持久化有关的代码。
打开UploadImage微服务目录下,model子目录中的mysql.go文件,在其中添加有关更新MySQL数据库中的图片的代码:
xxxxxxxxxx
221...
2// 更新MySQL数据库中的图片
3func UpdateImage(houseId uint, fileid string) error {
4 // 先根据房屋ID在房屋表中查询
5 var house House
6 if err := db.Where("id=?", houseId).Find(&house).Error; err != nil {
7 return err
8 }
9
10 // 若主图片URL为空则将此图片作为房屋主图存入房屋表
11 if house.Index_image_url == "" {
12 return db.Model(new(House)).Where("id=?", houseId).
13 Update("Index_image_url", fileid).Error
14 }
15
16 // 否则将此图片作为房屋副图存入图片表
17 var image Image
18 image.Url = fileid
19 image.House_id = houseId
20 return db.Create(&image).Error
21}
22...
定义名为UpdateImage的函数,用于更新MySQL数据库中的图片。该函数接收房屋ID和该房屋的图片文件凭证两个参数。以房屋ID为条件,查询房屋表中特定记录,获得表示该记录的房屋结构体变量。检查该结构体变量中表示主图片URL的成员,若其值为空,则以从参数传入的房屋图片文件凭证,更新该房屋记录的主图片URL字段。否则创建并初始化一个表示图片表记录的结构体变量,将房屋ID和该房屋的图片文件凭证填入其中,通过数据库连接池对象的Create方法,将该结构体变量所表示的图片记录插入到图片表中,返回错误对象。这里需要强调的是,一座房屋可能会有多张图片,其中一张为主图,其文件凭证保存在房屋表的主图片URL字段中,其它为副图,保存在图片表的图片URL字段中,并通过该表的所属房屋ID字段,与房屋表中的特定房屋建立关联。
在完成UploadImage微服务模型层的所有开发工作之后,我们将着手编写用于处理上传图片业务的代码。
打开UploadImage微服务目录下,handler子目录中的UploadImage.go文件,在其中的Call方法里,添加与上传图片有关的操作:
xxxxxxxxxx
401...
2// 上传图片
3func (e *UploadImage) Call(ctx context.Context, req *pb.CallRequest, rsp *pb.CallResponse) error {
4 logger.Infof("Received UploadImage.Call request: %v", req)
5
6 // 创建FastDFS客户机
7 client, err := fdfs_client.NewClientWithConfig("./conf/fdfs.conf")
8 defer client.Destory()
9 if err != nil {
10 logger.Error(err)
11 rsp.Errno = utils.ERROR_IO
12 rsp.Errmsg = utils.StrError(rsp.Errno)
13 return nil
14 }
15
16 // 上传数据到FastDFS
17 fileid, err := client.UploadByBuffer(req.Image, req.Suffix[1:])
18 if err != nil {
19 logger.Error(err)
20 rsp.Errno = utils.ERROR_IO
21 rsp.Errmsg = utils.StrError(rsp.Errno)
22 return nil
23 }
24
25 // 更新MySQL数据库中的图片
26 houseId, _ := strconv.Atoi(req.HouseId)
27 if err := model.UpdateImage(uint(houseId), fileid); err != nil {
28 logger.Error(err)
29 rsp.Errno = utils.ERROR_DATABASE
30 rsp.Errmsg = utils.StrError(rsp.Errno)
31 return nil
32 }
33
34 rsp.Data = &pb.Image{Url: "http://192.168.0.111:8888/" + fileid}
35
36 rsp.Errno = utils.ERROR_OK
37 rsp.Errmsg = utils.StrError(rsp.Errno)
38 return nil
39}
40...
这里首先以FastDFS客户机配置文件的路径为参数,调用FastDFS客户端SDK的NewClientWithConfig函数,创建一个FastDFS客户机对象。接着将请求对象中的图片数据和图片文件名后缀,传给FastDFS客户机对象的UploadByBuffer方法。注意调用该方法的第二个参数不包括文件名后缀中的“.”字符。如果不发生错误的话,房屋的图片文件已经上传到FastDFS服务器中了。FastDFS客户机对象的UploadByBuffer方法成功返回字符串形式的图片文件凭证。我们以请求对象中的房屋ID和上一步得到的图片文件凭证为参数,调用模型层的UpdateImage函数,更新MySQL数据库中的图片。若成功,则将图片文件凭证拼接在Nginx服务器的IP地址和侦听端口之后,形成供前端下载房屋图片的URL,填入响应对象的特定字段中。
在完成UploadImage后端服务器的开发后,我们需要为前端服务器添加一条路由,并在路由处理函数中完成对后端微服务远程方法的调用。
前端和后端共用同一套接口描述。
因此,这里我们将UploadImage微服务目录下proto子目录中的所有文件,原封不动地复制到front-end工程目录proto子目录下的UploadImage目录中。
这里我们要为上传图片添加一条路由,POST方法结合/api/v1.0/houses/:id/images路径,处理函数名为UploadImage。
在front-end工程目录下的main.go文件中添加一条路由:
xxxxxxxxxx
91...
2func main() {
3 ...
4 // 路由匹配
5 ...
6 apiv10.POST("/houses/:id/images", controller.UploadImage)
7 ...
8}
9...
这里我们调用了路由对象的POST方法,为发布房源添加了一条路由,将POST方法结合/houses/:id/images路径,路由到controller包的UploadImage函数。将UploadImage函数定义在controller包里是因为该函数的主要任务是执行业务逻辑,属于MVC中的C,即控制器层的部分。
当然,在controller包里真的得有UploadImage函数。为此,我们打开front-end工程目录下controller子目录中的house.go文件。在该文件的import部分增加一行,同时定义UploadImage函数:
xxxxxxxxxx
591...
2import (
3 ...
4 pbUploadImage "iHome/front-end/proto/UploadImage"
5 ...
6)
7...
8// 上传图片
9func UploadImage(ctx *gin.Context) {
10 response := make(map[string]string)
11
12 // 获取房屋ID
13 houseId := ctx.Param("id")
14
15 fileHeader, err := ctx.FormFile("house_image")
16 if err != nil {
17 fmt.Println(err)
18 response["errno"] = utils.ERROR_IO
19 response["errmsg"] = utils.StrError(response["errno"])
20 ctx.JSON(http.StatusOK, response)
21 return
22 }
23
24 file, err := fileHeader.Open()
25 if err != nil {
26 fmt.Println(err)
27 response["errno"] = utils.ERROR_IO
28 response["errmsg"] = utils.StrError(response["errno"])
29 ctx.JSON(http.StatusOK, response)
30 return
31 }
32
33 image := make([]byte, fileHeader.Size)
34 _, err = file.Read(image)
35 if err != nil {
36 fmt.Println(err)
37 response["errno"] = utils.ERROR_IO
38 response["errmsg"] = utils.StrError(response["errno"])
39 ctx.JSON(http.StatusOK, response)
40 return
41 }
42
43 // 调用微服务
44 srv := micro.NewService(micro.Registry(consul.NewRegistry()))
45 srv.Init()
46 c := pbUploadImage.NewUploadImageService("uploadimage", srv.Client())
47 rsp, err := c.Call(context.Background(), &pbUploadImage.CallRequest{
48 HouseId: houseId,
49 Image: image,
50 Suffix: path.Ext(fileHeader.Filename),
51 })
52 if err != nil {
53 fmt.Println(err)
54 return
55 }
56
57 ctx.JSON(http.StatusOK, rsp)
58}
59...
这段代码首先调用了Gin框架调用此函数时传入的上下文对象的Param方法,取添加路由时放在请求路径中的占位符“id”为参数,接收其返回的房屋ID。接着从请求包体中根据键获取图片文件头对象,并通过该对象的Open方法获得图片文件对象。根据图片文件的大小分配足量的字节切片,将图片文件中的数据读到该切片中。通过Consul服务发现,获取有关名为uploadimage的微服务的信息,远程调用其中的UploadImage对象的Call方法。该方法以包含房屋ID、图片数据和图片文件名后缀的请求对象为参数,并返回包含错误代码、错误描述和图片信息的响应对象。图片信息中包含房屋图片的URL,浏览器将据此显示房屋图片。最后将该响应对象序列化为一个JSON字符串,编码到HTTP响应中,回传给浏览器。
至此,我们已经完成上传图片的全部开发工作。下面我们将对这部分功能进行测试。
在虚拟机中启动FastDFS和Nginx。启动Consul服务器、GetAreas、UserLogin、GetUser、PublishHouse、UploadImage后端微服务和前端服务器。通过登录页面登录系统,点击位于搜索页面右上角的用户名,进入用户页面,点击“我的房源”,进入我的房源页面,点击“发布新房源”,进入发布新房源页面,填写房源信息表单,点击“发布房源信息”,进入上传房屋图片页面,点击“选择文件”,选择图片文件并点击“上传”。可上传多张房屋图片。浏览器会显示所上传的全部房屋图片。
进入MySQL控制台,检查ihomedb数据库houses表和images表中与房屋图片有关的字段。
谢谢大家,我们下节课再见!