1 Star 0 Fork 0

0x43/dubbo-getty

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
readwriter.go 1.98 KB
一键复制 编辑 原始数据 按行查看 历史
0x43 提交于 2个月前 . module
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package main
import (
"bytes"
"errors"
"time"
)
import (
getty "gitee.com/git4chen/dubbo-getty"
)
var echoPkgHandler = NewEchoPackageHandler()
type EchoPackageHandler struct{}
func NewEchoPackageHandler() *EchoPackageHandler {
return &EchoPackageHandler{}
}
func (h *EchoPackageHandler) Read(ss getty.Session, data []byte) (interface{}, int, error) {
var (
err error
len int
pkg EchoPackage
buf *bytes.Buffer
)
buf = bytes.NewBuffer(data)
len, err = pkg.Unmarshal(buf)
if err != nil {
if err == ErrNotEnoughStream {
return nil, 0, nil
}
return nil, 0, err
}
return &pkg, len, nil
}
func (h *EchoPackageHandler) Write(ss getty.Session, pkg interface{}) ([]byte, error) {
var (
ok bool
err error
startTime time.Time
echoPkg *EchoPackage
buf *bytes.Buffer
)
startTime = time.Now()
if echoPkg, ok = pkg.(*EchoPackage); !ok {
log.Errorf("illegal pkg:%+v", pkg)
return nil, errors.New("invalid echo package!")
}
buf, err = echoPkg.Marshal()
if err != nil {
log.Warnf("binary.Write(echoPkg{%#v}) = err{%#v}", echoPkg, err)
return nil, err
}
log.Debugf("WriteEchoPkgTimeMs = %s", time.Since(startTime).String())
return buf.Bytes(), nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/git4chen/dubbo-getty.git
git@gitee.com:git4chen/dubbo-getty.git
git4chen
dubbo-getty
dubbo-getty
v1.4.11

搜索帮助