用go封装一下封禁功能

用go封装一下封禁功能

本篇为用go设计开发一个自己的轻量级登录库/框架吧 - 秋玻 - 博客园 (cnblogs.com)的封禁业务篇，会讲讲封禁业务的实现，给库/框架增加新的功能。
源码：https://github.com/weloe/token-go
思路

封禁业务也是在一般项目中比较常见的业务。我们也将它封装在库中作为功能之一。
我们同样使用adapter作为底层的存储结构，将封禁标示作为k-v结构存储。
把id和服务名称service作为key，把封禁的级别level作为value，以此我们能实现一些比较常见的封禁业务。
实现

首先就是封禁方法，首先当然要校验一下参数，然后存储封禁标识。要注意的就是level要大于等于1，在封禁后也要调用我们的logger和watcher。
https://github.com/weloe/token-go/blob/8bf577c0309332cb42d17e33b435f06fb74d8e7b/enforcer.go#L390

1. // Banned ban user, if time == 0,the timeout is not set
2. func (e *Enforcer) Banned(id string, service string, level int, time int64) error {
3.         if id == "" || service == "" {
4.                 return fmt.Errorf("parameter cannot be nil")
5.         }
6.         if level < 1 {
7.                 return fmt.Errorf("unexpected level = %v, level must large or equal 1", level)
8.         }
9.         err := e.adapter.SetStr(e.spliceBannedKey(id, service), strconv.Itoa(level), time)
10.         if err != nil {
11.                 return err
12.         }
14.         // callback
15.         e.logger.Ban(e.loginType, id, service, level, time)
16.         if e.watcher != nil {
17.                 e.watcher.Ban(e.loginType, id, service, level, time)
18.         }
20.         return nil
21. }

复制代码

再来就是解除封禁的实现，很简单，只要删除我们封禁存的那条记录就可以了。
https://github.com/weloe/token-go/blob/8bf577c0309332cb42d17e33b435f06fb74d8e7b/enforcer.go#LL412C6-L412C6

1. // UnBanned Unblock user account
2. func (e *Enforcer) UnBanned(id string, services ...string) error {
3.         if id == "" {
4.                 return fmt.Errorf("parmeter id can not be nil")
5.         }
6.         if len(services) == 0 {
7.                 return fmt.Errorf("parmeter services length can not be 0")
8.         }
10.         for _, service := range services {
11.                 err := e.adapter.DeleteStr(e.spliceBannedKey(id, service))
12.                 if err != nil {
13.                         return err
14.                 }
15.                 e.logger.UnBan(e.loginType, id, service)
16.                 if e.watcher != nil {
17.                         e.watcher.UnBan(e.loginType, id, service)
18.                 }
19.         }
20.         return nil
21. }

复制代码

接着还需要提供几个方便使用的方法
分别用来判断是否被封禁，获得封禁的级别，获得封禁的剩余时间三个方法
https://github.com/weloe/token-go/blob/8bf577c0309332cb42d17e33b435f06fb74d8e7b/enforcer.go#L434
判断是否被封禁

1. // IsBanned if banned return true, else return false
2. func (e *Enforcer) IsBanned(id string, services string) bool {
3.         s := e.adapter.GetStr(e.spliceBannedKey(id, services))
5.         return s != ""
6. }

复制代码

https://github.com/weloe/token-go/blob/8bf577c0309332cb42d17e33b435f06fb74d8e7b/enforcer.go#L441
获得封禁的级别

1. // GetBannedLevel get banned level
2. func (e *Enforcer) GetBannedLevel(id string, service string) (int64, error) {
3.         str := e.adapter.GetStr(e.spliceBannedKey(id, service))
4.         if str == "" {
5.                 return 0, fmt.Errorf("loginId = %v, service = %v is not banned", id, service)
6.         }
7.         time, err := strconv.ParseInt(str, 10, 64)
8.         if err != nil {
9.                 return 0, err
10.         }
11.         return time, nil
12. }

复制代码

https://github.com/weloe/token-go/blob/8bf577c0309332cb42d17e33b435f06fb74d8e7b/enforcer.go#L454
获得封禁的剩余时间

1. // GetBannedTime get banned time
2. func (e *Enforcer) GetBannedTime(id string, service string) int64 {
3.         timeout := e.adapter.GetStrTimeout(e.spliceBannedKey(id, service))
4.         return timeout
5. }

复制代码

有了这几个方法，我们就可以使用token-go更为方便是实现登录业务了。
对于一般的业务，我们其实可以无视封禁级别。只需要使用Banned()进行封禁，在每个需要校验是否被封禁的业务前进行IsBanned()的判断就行了。
测试例子：

1. func TestEnforcer_Banned(t *testing.T) {
2.         err, enforcer, _ := NewTestEnforcer(t)
3.         if err != nil {
4.                 t.Fatalf("NewTestEnforcer() failed: %v", err)
5.         }
6.         err = enforcer.Banned("1", "comment", 1, 100)
7.         if err != nil {
8.                 t.Fatalf("Banned() failed: %v", err)
9.         }
10.         isBanned := enforcer.IsBanned("1", "comment")
11.         if !isBanned {
12.                 t.Errorf("unexpected isBanned is false")
13.         }
15.         err = enforcer.UnBanned("1", "comment")
16.         if err != nil {
17.                 t.Fatalf("UnBanned() failed: %v", err)
18.         }
19.         isBanned = enforcer.IsBanned("1", "comment")
20.         if isBanned {
21.                 t.Errorf("unexpected isBanned is false")
22.         }
23. }

复制代码

当然我们对于一个业务会有一些较为复杂的封禁，比如discuss交流业务，如果级别为1我们会禁止评论，如果级别为2我们会禁止评论和点赞，级别为3我们会禁止评论、点赞和收藏。
这个时候我们只需要再加上GetBannedLevel()，在每个业务前获取封禁级别进行判断就可以简单的实现了。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！