fix

liuxiaolong

2019-11-01 609d3fd50ee75997c05cea21b17c30ec5fc7c367

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
107
108
package main
 
import (
    "flag"
    "fmt"
    "net"
    "strconv"
    "time"
 
    "tcpserver/esutil"
    "tcpserver/config"
    log "tcpserver/log"
)
 
var addr = flag.String("addr", "0.0.0.0", "The address to listen to")
var Eurl = flag.String("eurl", "http://192.168.1.182:9200/", "The port to listen on")
var Picurl = flag.String("picurl", "http://58.118.225.79:41242/", "picture url")
 
var port = flag.Int("port", 6000, "The port to listen on")
var sec = flag.Int("sec", 10, "the second for query data")
 
var Level = flag.String("level","ErrorLevel","log level")
var IsHub = flag.String("hub", "hub", "hub is personIsHub=1")
var Size = flag.Int("size", 100, "size default is 100")
var env = flag.String("env", "config", "env set")
 
func main() {
    flag.Parse()
 
    log.SetLogLevel(*Level)
    config.Init(*env)
    fmt.Println(*port)
    src := *addr + ":" + strconv.Itoa(*port)
    listener, err := net.Listen("tcp", src)
    if err != nil {
        log.Log.Errorln(err)
        return
    }
    log.Log.Infof("Listening on %s.\n", src)
 
    fmt.Println("starting server success.")
    defer listener.Close()
 
    connArr:=make([]net.Conn,0)
 
    for {
        conn, err := listener.Accept()//
 
        connArr = append(connArr,conn)
        if err != nil {
            log.Log.Infoln("some connecion error: ", err)
        }
        go handleConnection(conn,connArr)
    }
}
 
func handleConnection(conn net.Conn, connArr []net.Conn) {
    remoteAddr := conn.RemoteAddr().String()
    log.Log.Infoln("Client connected from ", remoteAddr)
 
    ech := make(chan error)
    go func(conn net.Conn, ech chan error) {
        buf := make([]byte, 10)
        readMsg, err := conn.Read(buf)
        log.Log.Infoln("Read completed,readMsg:",readMsg,",err:",err)
        ech <- err
 
    }(conn, ech)
 
    tick := time.NewTicker(10 * time.Second)
 
    for {
        select {
        case <-tick.C:
            if !handleMessage(conn, connArr){
                conn.Close()
                return
            }
        case err := <-ech:
            log.Log.Infoln(err, "remoteAddr ", remoteAddr, " close")
            conn.Close()
            return
        }
    }
 
    log.Log.Infoln("Client at " + remoteAddr + " disconnected.")
}
 
func handleMessage(conn net.Conn, connArr []net.Conn) bool {
    jsonstring := esutil.PostAction(*sec, *Eurl, *Picurl, *IsHub, *Size)
    if jsonstring == nil {
        log.Log.Infoln("the data is nil,remoteArr:",conn.RemoteAddr())
        if _, err := conn.Write([]byte("\000"));err !=nil {
            log.Log.Infoln("conn.WriteErr:",err)
            return false
        } else {
            return true
        }
    }
    jsonstring = append(jsonstring, []byte("\000")...)
    log.Log.Infoln("jsonstring len: ", len(jsonstring), "\000 data: ", len("\000"))
    _, err := conn.Write(jsonstring)
    if err !=nil{
        log.Log.Infoln("conn.WriteErr:",err)
        return false
    }
    return true
}