// auth connection
func connAuth(s *nc.SocketServer, conn net.Conn, line string) error {
if strings.HasPrefix(line, "AUTH:") {
arr := strings.Split(line[5:], "#") // AUTH:API_ID#SECRET#VERSION
if len(arr) == 3 {
var af nc.AuthFunc = func() (int, error) {
merchantId := dps.MerchantService.GetMerchantIdByApiId(arr[0])
apiInfo := dps.MerchantService.GetApiInfo(merchantId)
if apiInfo != nil && apiInfo.ApiSecret == arr[1] {
if apiInfo.Enabled == 0 {
return merchantId, errors.New("api has exipres")
}
}
return merchantId, nil
}
if err := s.Auth(conn, af); err != nil {
return err
}
s.Printf("[ CLIENT] - Version = %s", arr[2])
return nil
}
}
return errors.New("conn reject")
}
// push member summary to tcp client
func pushMemberSummary(s *nc.SocketServer, connList []net.Conn, memberId int) {
s.Printf("[ TCP][ NOTIFY] - notify member update - %d", memberId)
sm := GetMemberSummary(memberId, 0)
if d, err := json.Marshal(sm); err == nil {
d = append([]byte("MSUM:"), d...)
for _, conn := range connList {
conn.Write(append(d, '\n'))
}
}
}
// push member summary to tcp client
func pushMemberAccount(s *nc.SocketServer, connList []net.Conn, memberId int) {
s.Printf("[ TCP][ NOTIFY] - notify account update - %d", memberId)
sm := getMemberAccount(memberId, 0)
if sm != nil {
if d, err := json.Marshal(sm); err == nil {
d = append([]byte("MACC:"), d...)
for _, conn := range connList {
conn.Write(append(d, '\n'))
}
}
}
}
// Handle command of client sending.
func handleCommand(s *nc.SocketServer, ci *nc.Client, cmd string) ([]byte, error) {
if time.Now().Sub(ci.LatestConnectTime) > disconnectDuration { //主动关闭没有活动的连接
//s.Print("--disconnect ---",ci.Addr.String())
ci.Conn.Close()
return nil, nil
}
if !strings.HasPrefix(cmd, "PING") {
s.Printf("[ CLIENT][ MESSAGE] - send by %d ; %s", ci.Source, cmd)
ci.LatestConnectTime = time.Now()
}
i := strings.Index(cmd, ":")
if i != -1 {
plan := cmd[i+1:]
if v, ok := handlers[cmd[:i]]; ok {
return v(ci, plan)
}
}
return nil, errors.New("unknown command:" + cmd)
}
