Go设计模式10-门面模式

笔记

代码实现

假设现在我有一个网站,以前有登录和注册的流程,登录的时候调用用户的查询接口,注册时调用用户的创建接口。为了简化用户的使用流程,我们现在提供直接验证码登录/注册的功能,如果该手机号已注册那么我们就走登录流程,如果该手机号未注册,那么我们就创建一个新的用户。

code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package facade

// IUser 用户接口
type IUser interface {
Login(phone int, code int) (*User, error)
Register(phone int, code int) (*User, error)
}

// IUserFacade 门面模式
type IUserFacade interface {
LoginOrRegister(phone int, code int) error
}

// User 用户
type User struct {
Name string
}

// UserService UserService
type UserService struct {}

// Login 登录
func (u UserService) Login(phone int, code int) (*User, error) {
// 校验操作 ...
return &User{Name: "test login"}, nil
}

// Register 注册
func (u UserService) Register(phone int, code int) (*User, error) {
// 校验操作 ...
// 创建用户
return &User{Name: "test register"}, nil
}

// LoginOrRegister 登录或注册
func (u UserService)LoginOrRegister(phone int, code int) (*User, error) {
user, err := u.Login(phone, code)
if err != nil {
return nil, err
}

if user != nil {
return user, nil
}

return u.Register(phone, code)
}

单元测试

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
package facade

import (
"github.com/stretchr/testify/assert"
"testing"
)

func TestUserService_Login(t *testing.T) {
service := UserService{}
user, err := service.Login(13001010101, 1234)
assert.NoError(t, err)
assert.Equal(t, &User{Name: "test login"}, user)
}

func TestUserService_LoginOrRegister(t *testing.T) {
service := UserService{}
user, err := service.LoginOrRegister(13001010101, 1234)
assert.NoError(t, err)
assert.Equal(t, &User{Name: "test login"}, user)
}

func TestUserService_Register(t *testing.T) {
service := UserService{}
user, err := service.Register(13001010101, 1234)
assert.NoError(t, err)
assert.Equal(t, &User{Name: "test register"}, user)
}

关注我获取更新

wechat
知乎
开发者头条
github

猜你喜欢