同学们好!成功登录《我家租房网》系统的用户,可以点击位于搜索页面右上角的用户名,进入用户页面。点击用户页面个人信息栏中的“修改”,会打开修改个人信息的页面,在这里可以上传头像也可以更新用户名。这节课我们将实现更新用户名的功能。
我们将首先实现支持更新用户名的后端微服务,再在前端服务器中添加路由处理函数并远程调用后端微服务,最后再将前后端连在一起做完整的功能测试。
首先我们来实现后端服务器。
第一步是创建特定的微服务。
创建微服务的过程还是和先前一样,直接在back-end工程目录下执行go-micro命令,子命令为new,参数为service,微服务名为SetUsername。
接着,需要修改一下go-micro为我们自动生成的接口描述。在back-end工程目录下的SetUsername微服务目录中,有一个名为proto的子目录,这就是存放接口描述脚本及其Go代码文件的地方。现阶段里面只有一个名为SetUsername.proto的ProtoBuf脚本文件。我们需要对该文件做一些修改:
x1...
2message CallRequest {
3 string oldUsername = 1;
4 string newUsername = 2;
5}
6
7message User {
8 string name = 1;
9}
10
11message CallResponse {
12 string errno = 1;
13 string errmsg = 2;
14 User data = 3;
15}
16...
这段代码描述的是,前端服务器和后端微服务之间的数据交换格式。CallRequest表示前端提供给后端的请求数据,其中包含两个字符串,原用户名和新用户名。CallResponse表示后端返回给前端的响应数据,其中包含错误代码、错误描述和用户信息。用户信息中只含有一个字段,即更新后的新用户名。
有了用ProtoBuf语言描述的远程调用接口,我们还需要把它编译成基于Go语言的程序代码。在SetUsername微服务目录中执行make命令,并携带四个目标参数,它们是init、proto、update和tidy。
在SetUsername微服务目录下,创建一个名为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}
修改SetUsername微服务目录中的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:9007"),
13 )
14 ...
15}
16...
程序启动伊始,即完成对数据库的初始化。其中包括两个动作,其一是连接ihomedb数据库并获得连接池对象,其二是在该数据库中创建表。当然,如果这些表已经存在,则不再创建。接着,在服务器对象的初始化部分,指示其注册到Consul服务器,同时绑定本机的9007端口。
接下来,我们为SetUsername微服务编写与数据持久化有关的代码。
打开SetUsername微服务目录下,model子目录中的mysql.go文件,在其中添加有关更新MySQL数据库中的用户名的代码:
xxxxxxxxxx
71...
2// 更新MySQL数据库中的用户名
3func UpdateUsername(oldUsername, newUsername string) error {
4 return db.Model(new(User)).Where("name=?", oldUsername).
5 Update("name", newUsername).Error
6}
7...
定义名为UpdateUsername的函数,用于更新MySQL数据库中的用户名。该函数接收原用户名和新用户名两个参数,以原用户名为条件,更新用户表中特定记录的用户名字段为新用户名,并将错误对象返回给函数的调用者。
在完成SetUsername微服务模型层的所有开发工作之后,我们将着手编写用于处理更新用户名业务的代码。
打开SetUsername微服务目录下,handler子目录中的SetUsername.go文件,在其中的Call方法里,添加与更新用户名有关的操作:
xxxxxxxxxx
221...
2// 更新用户名
3func (e *SetUsername) Call(ctx context.Context, req *pb.CallRequest, rsp *pb.CallResponse) error {
4 logger.Infof("Received SetUsername.Call request: %v", req)
5
6 // 更新MySQL数据库中的用户名
7 if err := model.UpdateUsername(req.OldUsername, req.NewUsername); err != nil {
8 logger.Error(err)
9 rsp.Errno = utils.ERROR_DATABASE
10 rsp.Errmsg = utils.StrError(rsp.Errno)
11 return nil
12 }
13
14 rsp.Data = &pb.User{
15 Name: req.NewUsername,
16 }
17
18 rsp.Errno = utils.ERROR_OK
19 rsp.Errmsg = utils.StrError(rsp.Errno)
20 return nil
21}
22...
这里首先调用模型层的UpdateUsername函数,更新MySQL数据库中的用户名。若成功,则将更新后的新用户名填入响应对象的特定字段中,否则,设置错误响应并返回。
在完成SetUsername后端服务器的开发后,我们需要为前端服务器添加一条路由,并在路由处理函数中完成对后端微服务远程方法的调用。
前端和后端共用同一套接口描述。
因此,这里我们将SetUsername微服务目录下proto子目录中的所有文件,原封不动地复制到front-end工程目录proto子目录下的SetUsername目录中。
这里我们要为更新用户名添加一条路由,PUT方法结合/api/v1.0/user/name路径,处理函数名为SetUsername。
在front-end工程目录下的main.go文件中添加一条路由:
xxxxxxxxxx
91...
2func main() {
3 ...
4 // 路由匹配
5 ...
6 apiv10.PUT("/user/name", controller.SetUsername)
7 ...
8}
9...
这里我们调用了路由对象的PUT方法,为更新用户名添加了一条路由,将PUT方法结合/user/name路径,路由到controller包的SetUsername函数。将SetUsername函数定义在controller包里是因为该函数的主要任务是执行业务逻辑,属于MVC中的C,即控制器层的部分。
当然,在controller包里真的得有SetUsername函数。为此,我们打开front-end工程目录下controller子目录中的user.go文件。在该文件的import部分增加一行,同时定义SetUsername函数:
xxxxxxxxxx
511...
2import (
3 ...
4 pbSetUsername "iHome/front-end/proto/SetUsername"
5 ...
6)
7...
8// 更新用户名
9func SetUsername(ctx *gin.Context) {
10 // 从Session中读取原用户名
11 oldUsername := model.ReadUsername(ctx)
12 if oldUsername == "" {
13 rsp := make(map[string]string)
14 rsp["errno"] = utils.ERROR_USER_NOT_LOGIN
15 rsp["errmsg"] = utils.StrError(rsp["errno"])
16 ctx.JSON(http.StatusOK, rsp)
17 return
18 }
19
20 // 获取新用户名
21 req := struct {
22 NewUsername string `json:"name"`
23 }{}
24 ctx.Bind(&req)
25
26 // 调用微服务
27 srv := micro.NewService(micro.Registry(consul.NewRegistry()))
28 srv.Init()
29 c := pbSetUsername.NewSetUsernameService("setusername", srv.Client())
30 rsp, err := c.Call(context.Background(), &pbSetUsername.CallRequest{
31 OldUsername: oldUsername,
32 NewUsername: req.NewUsername,
33 })
34 if err != nil {
35 fmt.Println(err)
36 return
37 }
38
39 // 如果更新成功
40 if rsp.Errno == utils.ERROR_OK {
41 // 则将新用户名保存到Session中
42 if err = model.SaveUsername(ctx, rsp.Data.Name); err != nil {
43 fmt.Println(err)
44 rsp.Errno = utils.ERROR_USER_LOGIN
45 rsp.Errmsg = utils.StrError(rsp.Errno)
46 }
47 }
48
49 ctx.JSON(http.StatusOK, rsp)
50}
51...
这段代码首先从Session中读取原用户名,接着对请求包体中的JSON字符串做反序列化,得到新用户名。通过Consul服务发现,获取有关名为setusername的微服务的信息,远程调用其中的SetUsername对象的Call方法。该方法以包含原用户名和新用户名的请求对象为参数,并返回包含错误代码、错误描述和新用户名的响应对象。如果后端微服务的远程方法执行成功,则将新用户名保存到Session中。最后将该响应对象序列化为一个JSON字符串,编码到HTTP响应中,回传给浏览器。
至此,我们已经完成更新用户名的全部开发工作。下面我们将对这部分功能进行测试。
启动Consul服务器、UserLogin、GetUser、SetUsername后端微服务和前端服务器。通过登录页面登录系统,点击位于搜索页面右上角的用户名,进入用户页面,点击个人信息栏中的“修改”,进入修改个人信息的页面,填入新用户名,点击“保存”,提示保存成功。返回用户页面,可以看到个人信息栏中的用户名显示为修改后的新用户名。返回搜索页面,位于右上角的用户名同样显示为修改后的新用户名。
谢谢大家,我们下节课再见!