YHEchoPackage/ws/helper.go

39 lines
720 B
Go

/*
* 版权所有 (c) 上海元泓软件科技有限公司 2023.
* 严禁通过任何媒介未经授权复制本文件.
*
* 作者:mic
* Email:funui@outlook.com
*/
/*
* 版权所有 (c) 上海元泓软件科技有限公司 2023.
* 严禁通过任何媒介未经授权复制本文件.
*
* 作者:mic
* Email:funui@outlook.com
*/
package wsgnet
import (
"github.com/gobwas/ws"
"github.com/gobwas/ws/wsutil"
"github.com/panjf2000/gnet/v2"
)
func WriteServerMessage(c gnet.Conn, op ws.OpCode, p []byte) error {
if c.Context() != nil {
err := wsutil.WriteServerMessage(c, op, p)
if err != nil {
return err
}
} else {
_, err := c.Write(p)
if err != nil {
return err
}
}
return nil
}