forked from andeya/erpc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
session.go
106 lines (91 loc) · 2.2 KB
/
session.go
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
// Copyright 2015-2017 HenryLee. All Rights Reserved.
//
// Licensed 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 teleport
import (
"net"
"github.com/henrylee2cn/goutil/coarsetime"
"github.com/henrylee2cn/teleport/socket"
)
type Session struct {
peer *Peer
apiMap *ApiMap
socket socket.Socket
}
func newSession(peer *Peer, conn net.Conn, id ...string) *Session {
var s = &Session{
peer: peer,
apiMap: peer.ApiMap,
socket: socket.NewSocket(conn, id...),
}
go s.serve()
return s
}
func (s *Session) Id() string {
return s.socket.Id()
}
// TODO
func (s *Session) Pull(uri string, args interface{}, reply interface{}) error {
return nil
}
// TODO
func (s *Session) Push() error {
return nil
}
func (s *Session) Close() error {
return s.socket.Close()
}
// TODO
func (s *Session) serve() {
var ctx = s.peer.getContext(s)
defer func() {
recover()
s.peer.putContext(ctx)
s.Close()
}()
var (
err error
readTimeout = s.peer.readTimeout
writeTimeout = s.peer.writeTimeout
)
for {
// read request, response or push
if readTimeout > 0 {
s.socket.SetReadDeadline(coarsetime.CoarseTimeNow().Add(readTimeout))
}
err = s.socket.ReadPacket(ctx.input)
if err != nil {
Debugf("teleport: ReadPacket: %s", err.Error())
return
}
switch ctx.input.Header.Type {
case TypeRequest:
// handle
go ctx.handle()
ctx.output.Header.Type = TypeResponse
// write response
if writeTimeout > 0 {
s.socket.SetWriteDeadline(coarsetime.CoarseTimeNow().Add(writeTimeout))
}
err = s.socket.WritePacket(ctx.output)
if err != nil {
Debugf("teleport: WritePacket: %s", err.Error())
return
}
}
}
}
func (s *Session) read() {
}
func (s *Session) write() {
}