func (this *DispatcherOutComponent) handleMTRegister(msg *protocol.Message, buffPool *buffpool.BuffPool, respComp *ResponseComponent) error {
if msg.PushType == protocol.PT_PUSHSERVER {
conn := buffPool.QueryService(msg.Number, msg.DestUuid)
if conn == nil {
clientConn := buffPool.GetClient(msg.Number, msg.DestUuid, msg.DeviceId, msg.Token)
if clientConn == nil || clientConn.Conn == nil {
err := errors.New("DispatcherOutComponent.handleMTRegister() Error: can not find the dest conn")
fmt.Println(err.Error())
return err
}
conn = clientConn.Conn
}
if conn == nil {
err := errors.New("DispatcherOutComponent.handleMTRegister() Error: can not find the dest conn")
fmt.Println(err.Error())
return err
}
err := this.sendMessage(msg, conn)
if err != nil {
err = errors.New("DispatcherComponnet.handleMTRegister() Error: can not send message out, " + err.Error())
fmt.Println(err.Error())
return err
}
}
return nil
}
func (this *DispatcherOutComponent) sendToClient2(msg *protocol.Message, forward bool, buffPool *buffpool.BuffPool) error {
if msg == nil || buffPool == nil {
err := errors.New("DispatcherOutComponent.sendToClient2() Error: invalid argument")
fmt.Println(err.Error())
return err
}
clientConnArr := make([]*model.Client, 0)
if msg.Token > 0 {
clientConn := buffPool.GetClient(msg.Number, msg.DestUuid, msg.DeviceId, msg.Token)
if clientConn != nil && clientConn.Conn != nil {
clientConnArr = append(clientConnArr, clientConn)
}
if (clientConnArr == nil || len(clientConnArr) <= 0) && forward && GetRunWithSnode().Flag {
buf, err := msg.Encode()
if err != nil {
fmt.Println("DispatcherOutComponent.sendToClient2() Error: " + err.Error())
fmt.Println("\n", time.Now().String(), "Not find user, not forward due to encode failed.", "\n")
return err
}
snode_component.GetNodeComponent().Forward(stree_define.FORWARD, msg.DestUuid, msg.DeviceId, buf)
return nil
}
} else if msg.Token == 0 && msg.DeviceId > 0 {
clientConnArr = buffPool.QueryForDeviceId(msg.Number, msg.DestUuid, msg.DeviceId)
if (clientConnArr == nil || len(clientConnArr) <= 0) && forward && GetRunWithSnode().Flag {
buf, err := msg.Encode()
if err != nil {
fmt.Println("DispatcherOutComponent.sendToClient2() Error:" + err.Error())
fmt.Println("\n", time.Now().String(), "Not find user, not forward due to encode failed.", "\n")
return err
}
snode_component.GetNodeComponent().Forward(stree_define.FORWARD, msg.DestUuid, msg.DeviceId, buf)
return nil
}
} else if msg.Token == 0 && msg.DeviceId == 0 {
clientConnArr = buffPool.QueryForUuid(msg.Number, msg.DestUuid)
}
err := errors.New("DispatcherOutComponent.sendToClient2() Error: the client's connection not pass authorization yet")
for _, client := range clientConnArr {
if client == nil || client.Conn == nil {
continue
}
if !this.checkConnActivity(client.Time, client, buffPool, msg.Number, msg.DestUuid, msg.DeviceId, msg.Token) {
err := errors.New("DispatcherOutComponent.sendToClient2() Error: client's connection is not alive")
fmt.Println(err.Error())
continue
}
err2 := this.sendMessage(msg, client.Conn)
if err2 != nil {
err2 = errors.New("DispatcherOutComponent.sendToClient2() Error: can not send message to client's connection")
fmt.Println(err.Error())
continue
}
}
return nil
}
func (this *UnregisterComponent) unregisterClient(buffPool *buffpool.BuffPool, msg *protocol.Message) error {
if buffPool == nil || msg == nil {
err := errors.New("UnregisterComponent.unregisterClient() Error: invalid argument")
fmt.Println(err.Error())
return err
}
clientConn := buffPool.GetClient(msg.Number, msg.UserUuid, msg.DeviceId, msg.Token)
if clientConn == nil {
return nil
}
//only remove conn from connPool, but do not disconnect this connection
err := buffPool.RemoveClient(msg.Number, msg.UserUuid, msg.DeviceId, msg.Token)
(*clientConn.Conn).Close()
if err != nil {
err = errors.New("UnregisterComponent.unregisterClient() Error: " + err.Error())
fmt.Println(err.Error())
//send failed response to client
resp, err2 := protocol.NewMessage(protocol.VERSION, protocol.MT_ERROR, protocol.PT_PUSHSERVER, msg.Token, msg.Number, uint16(0),
P_UID, msg.UserUuid, msg.DeviceId, nil)
if err2 != nil {
err2 = errors.New("UnregisterComponent.unregisterClient() Error:" + err2.Error())
fmt.Println(err2.Error())
return err2
}
err3 := sendResponse(clientConn.Conn, resp)
if err3 != nil {
err3 = errors.New("UnregisterComponent.unregisterClient() Error:, " + err3.Error())
fmt.Println(err3.Error())
return err3
}
return err
}
//offline broadcast
err = this.offlineBroadcast(buffPool, msg.UserUuid, msg.DeviceId)
if err != nil {
err = errors.New("UnregisterComponent.unregisterClient() Error: offlineBroadcast send failed, " + err.Error())
fmt.Println(err.Error())
}
resp, err := protocol.NewMessage(protocol.VERSION, protocol.MT_UNREGISTER, protocol.PT_PUSHSERVER, msg.Token, msg.Number, uint16(0),
P_UID, msg.UserUuid, msg.DeviceId, nil)
if err != nil {
err = errors.New("UnregisterComponent.unregisterClient() Error: conn is removed from BuffPool, but can not send response to service" + err.Error())
fmt.Println(err.Error())
return err
}
err2 := sendResponse(clientConn.Conn, resp)
if err2 != nil {
err2 = errors.New("UnregisterComponent.unregisterClient() Error:conn is reomved from BuffPool, but failed when send response to service" + err2.Error())
return err2
}
return nil
}
func (this *DispatcherOutComponent) sendToClient(msg *protocol.Message, buffPool *buffpool.BuffPool) error {
if msg == nil || buffPool == nil {
err := errors.New("DispatcherOutComponent.sendToClient() Error: invalid argument")
fmt.Println(err.Error())
return err
}
clientConnArr := make([]*model.Client, 0)
if msg.Token > 0 {
clientConn := buffPool.GetClient(msg.Number, msg.DestUuid, msg.DeviceId, msg.Token)
if clientConn != nil && clientConn.Conn != nil {
clientConnArr = append(clientConnArr, clientConn)
}
} else if msg.Token == 0 && msg.DeviceId > 0 {
clientConnArr = buffPool.QueryForDeviceId(msg.Number, msg.DestUuid, msg.DeviceId)
} else if msg.Token == 0 && msg.DeviceId == 0 {
clientConnArr = buffPool.QueryForUuid(msg.Number, msg.DestUuid)
}
if clientConnArr != nil && len(clientConnArr) <= 0 {
err := errors.New("DispatcherOutComponent.sendToClient() Error: can not find client's connection")
fmt.Println(err.Error())
return err
}
err := errors.New("DispatcherOutComponent.sendToClient() Error: the client's connection not pass authorization yet")
for _, client := range clientConnArr {
if client == nil || client.Conn == nil {
continue
}
if !this.checkConnActivity(client.Time, client, buffPool, msg.Number, msg.DestId, msg.DeviceId, msg.Token) {
err := errors.New("DispatcherOutComponent.sendToClient() Error: client's connection is not alive")
fmt.Println(err.Error())
continue
}
err2 := this.sendMessage(msg, client.Conn)
if err2 != nil {
err2 = errors.New("DispatcherOutComponent.sendToClient() Error: can not send message to client's connection")
fmt.Println(err.Error())
continue
}
}
return err
}