在之前的小游戏项目中,处理websocket长连接请求的时候,需要根据传递数据包的不同类型,进行不同的处理。为了实现这个场景,比较简单的方法就是使用if-else或者switch-case语句,根据条件进行判断。但是这导致了项目代码复杂混乱,不利用项目后期的维护和扩展。
策略模式和观察者模式都可以解决这个大量条件判断的问题,相比之下,观察者模式比较复杂,适用于更加复杂的场景。而策略模式则比较轻便,简单易上手。
关于观察者模式,感兴趣的朋友可以看这一篇博客:
观察者模式的理解和引用-CSDN博客
旧代码如下
//在信息中枢处根据消息类型进行特定的处理switch requestPkg.Type {case pojo.CertificationType://用户认证client.CertificationProcess(requestPkg)case pojo.CreateRoomType://创建房间号,并将创建者加入房间client.CreateRoomProcess()case pojo.JoinRoomType://1.加入房间的前提,先建立连接//2.完成用户认证//3.发送消息类型和房间号 Type uuid//只有完成上述步骤,才可以加入房间var data map[string]interface{}err = json.Unmarshal([]byte(requestPkg.Data), &data)if err != nil {fmt.Println("解析 JSON 失败:", err)return}uuidValue, ok := data["uuid"].(string)if !ok {fmt.Println("uuid 字段不存在或不是字符串类型")return}client.JoinRoomProcess(uuidValue)case pojo.RefreshScoreType://什么是否进行分数更新,前端判断 type:RefreshScoreType, data:step、step、score//当用户的行为触发前端游戏机制的更新时,前端调用此接口,后端进行分数的转发 不需要做业务处理,直接转发即可fmt.Println("游戏交换中数据", client)client.RefreshScoreProcess(requestPkg)case pojo.DiscontinueQuitType:client.DiscontinueQuitProcess()case pojo.GameOverType://游戏结束类型好像没有太大用,游戏结束的时候的提醒,通过分数更新就可以实现了fmt.Println("GameOverType")case pojo.HeartCheckType://开启一个协程遍历hub中的Client,进行健康检测,生命时间是否会过期,如果过期进行逻辑删除和关闭连接if requestPkg.Data == "PING" {client.HeartCheckProcess()}}
策略模式介绍
策略模式是一种软件设计模式,它允许在运行时选择算法的行为。策略模式定义了一系列算法,并使它们能够相互替换,从而使算法可以独立于其使用者而变化。这意味着在实际使用中,可以在不修改其结构的情况下轻松地切换或替换算法。在我们的场景下,我们通过websocket连接获取到的数据包有很多的类型:CertificationType、CreateRoomType、JoinRoomType、RefreshScoreType、DiscontinueQuitType、GameOverType、HeartCheckType等类型,我们需要根据不同的类型对数据进行不同的处理。即在运行时,根据传入的Type值,选择不同的行为进行处理,使得我们可以在运行的过程中轻松切换行为。【这里可以将算法理解成对同样结构的输入,进行不同的处理。】
Context上下文对象类:该对象持有对策略接口的引用,可以根据需要动态地改变所使用的具体策略。这样,客户端调用上下文对象的方法时,就可以根据所设置的具体策略来执行相应的算法。
//将Strategy策略作为Context上下文对象的一个字段
//封装get、set、execStrategy(执行方法)
type Context struct {Strategy handler.Strategy
}func (c *Context) SetStrategy(strategy handler.Strategy) {c.Strategy = strategy
}func (c *Context) GetStrategy() handler.Strategy {return c.Strategy
}func (c *Context) ExecStrategy(client *pojo.Client, request req.WsReq) {c.Strategy.StrategyMethod(client, request)
}
Strategy策略接口:定义接口,接口里面是StrategyMethod()方法,在Context对象的ExecStrategy()方法中被调用。
type Strategy interface {StrategyMethod(client *pojo.Client, request req.WsReq)
}
ConcreteStrategy具体策略类:该类实现Strategy接口,并且根据自身的需要定义StrategyMethod()方法具体的业务代码。
type LifetimeCheck struct {
}func (c *LifetimeCheck) StrategyMethod(client *pojo.Client, req req.WsReq) {if string(req.Data) == `"PING"` {client.User.Lifetime = client.User.Lifetime.Add(10 * time.Second)}wsResp := resp.WsResp{Type: req.Type, Data: "PONG"}if err := wsResp.WsResponseSuccess(client); err != nil {zap.L().Error("PING 返回出现错误")return}
}
客户端代码:创建上下文对象,根据需求,去调用SetStrategy()方法,并执行所设置策略的行为。我们的选择可以通过Map进行,根据map的key获取对应的具体策略对象。
func WsController(client *pojo.Client) {//map初始化之后就没有写入操作了,所以不存在线程安全的问题!!! 只会进行读取操作context := new(Context)//1.执行认证行为context.SetStrategy(new(specificHandler.CertificationStrategy))context.ExecStrategy(client, request)//2.执行心跳检测context.SetStrategy(new(specificHandler.LifetimeCheck))context.ExecStrategy(client, request)//1.执行聊天信息context.SetStrategy(new(specificHandler.ChatStrategy))context.ExecStrategy(client, request)
}
通过这样改造之后,我们的代码就可以实现:
- 算法可以独立于客户端而变化,客户端可以更灵活地选择算法。
- 可以避免大量的条件语句,将算法的选择和使用分离开来,提高了代码的可维护性和扩展性。
- 可以方便地增加、删除或更改算法,不会影响到客户端的代码。
总的来说,策略模式可以帮助我们更好地组织和管理不同的算法,使系统更加灵活、可扩展和易于维护。
策略选择改进
读到这里,大家可以发现,其实还是没有办法解决我们代码里面出现的大量if - else语句,我们需要通过if - else来做判断,帮助我们选择对应的行为。其实,这里可以引入map帮助我们来做选择,如果命中map中的key,就返回对应的算法对象。
代码如下:
func WsController(client *pojo.Client) {//map初始化之后就没有写入操作了,所以不存在线程安全的问题!!! 只会进行读取操作context := new(Context)handlerMap := map[string]handler.Strategy{}//对客户端发送过来的对应请求类型,进行注册操作handlerMap[CertificationType] = new(specificHandler.CertificationStrategy)handlerMap[ChatType] = new(specificHandler.ChatStrategy)handlerMap[CommentNotificationType] = new(specificHandler.CommentNotification)handlerMap[FansNotificationType] = new(specificHandler.FansNotification)handlerMap[CloseType] = new(specificHandler.CloseConnection)handlerMap[LifetimeCheckType] = new(specificHandler.LifetimeCheck)for {request := req.WsReq{}//循环读取对应的类型,并进行反序列化client.Conn.ReadJSON(&request)//通过定义的map,通过对应的数据类型获取算法对象。如果能够获取到更改上下文,并执行匹配到的策略,处理数据if strategy, ok := handlerMap[request.Type]; ok {context.SetStrategy(strategy)context.ExecStrategy(client, request)} else {zap.L().Error("长连接请求类型不存在")}}
}
更改后的代码整体结构:
总结
代码并不是能跑就可以了,如果可以的话,请尽可能地优化自己的代码,让自己的代码利于维护和扩展,可读性高。