add orm function
This commit is contained in:
@@ -241,7 +241,7 @@ func (s *ServerCommon) loadMessage() {
|
||||
s.mu.RLock()
|
||||
for _, v := range s.clientPool {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
go func(v *ClientConn) {
|
||||
defer wg.Done()
|
||||
v.sayGoodByeForTU()
|
||||
v.alive.Store(false)
|
||||
@@ -252,7 +252,7 @@ func (s *ServerCommon) loadMessage() {
|
||||
}
|
||||
v.stopFn()
|
||||
s.removeClient(v)
|
||||
}()
|
||||
}(v)
|
||||
}
|
||||
s.mu.RUnlock()
|
||||
select {
|
||||
@@ -343,8 +343,8 @@ func (s *ServerCommon) dispatchMsg(message Message) {
|
||||
return
|
||||
}
|
||||
//just throw
|
||||
return
|
||||
//fallthrough
|
||||
//return
|
||||
fallthrough
|
||||
default:
|
||||
}
|
||||
callFn := func(fn func(*Message)) {
|
||||
@@ -377,7 +377,9 @@ func (s *ServerCommon) sendTU(c *ClientConn, msg TransferMsg) (WaitMsg, error) {
|
||||
data = c.msgEn(c.SecretKey, data)
|
||||
data = s.queue.BuildMessage(data)
|
||||
if c.maxWriteTimeout.Seconds() != 0 {
|
||||
c.tuConn.SetWriteDeadline(time.Now().Add(c.maxWriteTimeout))
|
||||
if err := c.tuConn.SetWriteDeadline(time.Now().Add(c.maxWriteTimeout)); err != nil {
|
||||
return WaitMsg{}, err
|
||||
}
|
||||
}
|
||||
_, err = c.tuConn.Write(data)
|
||||
//fmt.Println("resend:", float64(time.Now().UnixNano()-nowd)/1000000)
|
||||
@@ -416,7 +418,7 @@ func (s *ServerCommon) sendWait(c *ClientConn, msg TransferMsg, timeout time.Dur
|
||||
s.noFinSyncMsgPool.Delete(data.TransferMsg.ID)
|
||||
return Message{}, os.ErrDeadlineExceeded
|
||||
case <-s.stopCtx.Done():
|
||||
return Message{}, errors.New("Service shutdown")
|
||||
return Message{}, errors.New("service shutdown")
|
||||
case msg, ok := <-data.Reply:
|
||||
if !ok {
|
||||
return msg, os.ErrInvalid
|
||||
@@ -447,7 +449,7 @@ func (s *ServerCommon) sendCtx(c *ClientConn, msg TransferMsg, ctx context.Conte
|
||||
s.noFinSyncMsgPool.Delete(data.TransferMsg.ID)
|
||||
return Message{}, os.ErrClosed
|
||||
case <-s.stopCtx.Done():
|
||||
return Message{}, errors.New("Service shutdown")
|
||||
return Message{}, errors.New("service shutdown")
|
||||
case msg, ok := <-data.Reply:
|
||||
if !ok {
|
||||
return msg, os.ErrInvalid
|
||||
@@ -626,7 +628,7 @@ func (s *ServerCommon) GetClient(id string) *ClientConn {
|
||||
func (s *ServerCommon) GetClientLists() []*ClientConn {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
var list []*ClientConn = make([]*ClientConn, 0, len(s.clientPool))
|
||||
var list = make([]*ClientConn, 0, len(s.clientPool))
|
||||
for _, v := range s.clientPool {
|
||||
list = append(list, v)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user