aionet/conn.go at main · pcbuildpluscoding/aionet · GitHub

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
package aionet

import (
	"time"

	"github.com/pcbuildpluscoding/aionet/dtype"
)

// ===========================================================================
type HdpConn struct {
	cid    string
	state  [2]dtype.HDP_STATE2
	statet Statet
	tpt    dtype.MultiCh
}

// ===========================================================================
func (c *HdpConn) Cid() string {
	return c.cid
}

// ===========================================================================
func (c *HdpConn) Close() error {
	c.tpt.SendEvent(W, dtype.HDP_CLOSING).Async()
	return nil
}

// ===========================================================================
func (c *HdpConn) ReadHdr() {
	c.tpt.SendEvent(R, dtype.HDP_READ1, ":data", "bytes", []byte{}).Async()
}

// ===========================================================================
func (c *HdpConn) Read(b []byte) (int, error) {
	res := <-c.tpt.SendEvent(R, dtype.HDP_READ1, ":data", "bytes", b).Sync()
	logger.Debugf("%s got HDP_DATAGRAM read result : %v", c.cid, res)
	return res.Retval()
}

// ===========================================================================
func (c *HdpConn) Write(b []byte) (int, error) {
	logger.Debugf("about to submit a write request")
	res := <-c.tpt.SendEvent(W, dtype.HDP_WRITE1, ":data", "frame", dtype.NewFrame(b)).Sync()
	logger.Debugf("%s got HDP_DATAGRAM write result : %v", c.cid, res)
	return res.Retval()
}

// ===========================================================================
func (c *HdpConn) recycle(ev dtype.HdpEvent) {
	f1 := c.statet
	// reset c.statet so that if a handler does not assign a value
	// then f2 is finally assigned to it.
	c.statet = nil
	var f2 Statet
	for f1 != nil {
		f2 = f1
		f1, ev = f2(ev)
	}
	if c.statet == nil {
		c.statet = f2
	}
	logger.Debugf("%s got final event result : %v", c.cid, ev)
}

// ===========================================================================
func (c *HdpConn) Run() {
	logger.Debugf("%s eventloop is running ...", c.cid)
	// c.statet = c.handleOpen
	for ev := range c.tpt[C] {
		logger.Debugf("############## %s control got another event %v", c.cid, ev)
		c.recycle(ev)
		switch c.state[0] {
		case dtype.HDP_CLOSED:
			logger.Debugf("%s is now closed.", c.cid)
			return
		}
	}
}

// ===========================================================================
func (c *HdpConn) SetReadDeadline(dl time.Time) error {
	res := <-c.tpt.SendEvent(R, dtype.HDP_READ1, ":data", "deadline", dl).Sync()
	logger.Debugf("%s got setReadDeadline result : %v", c.cid, res)
	return res.Err()
}

// ===========================================================================
func (c *HdpConn) SetWriteDeadline(dl time.Time) error {
	res := <-c.tpt.SendEvent(W, dtype.HDP_WRITE1, ":data", "deadline", dl).Sync()
	logger.Debugf("%s got setWriteDeadline result : %v", c.cid, res)
	return res.Err()
}