package main import "fmt" import "net" import "sync" func NewClientPeerConn(r *ClientRoute, c net.Conn, id uint32) (*ClientPeerConn) { var cpc ClientPeerConn cpc.route = r cpc.conn = c cpc.conn_id = id cpc.stop_req.Store(false) //cpc.server_peer_status_chan = make(chan bool, 16) //cpc.server_peer_opened_received.Store(false) //cpc.server_peer_closed_received.Store(false) return &cpc } func (cpc *ClientPeerConn) RunTask(wg *sync.WaitGroup) error { //var conn *net.TCPConn //var addr *net.TCPAddr var err error var buf [4096]byte var n int defer wg.Done() fmt.Printf("CONNECTION ESTABLISHED TO PEER... ABOUT TO READ DATA...\n") for { n, err = cpc.conn.Read(buf[:]) if err != nil { fmt.Printf("unable to read from the client-side peer %s - %s\n", cpc.addr, err.Error()) break } // TODO: guarded call.. err = cpc.route.cts.psc.Send(MakePeerDataPacket(cpc.route.id, cpc.conn_id, buf[0:n])) if err != nil { fmt.Printf("unable to write data to server - %s\n", err.Error()) break } } //done: cpc.ReqStop() return nil } func (cpc *ClientPeerConn) ReqStop() { // TODO: because of connect delay in Start, cpc.p may not be yet ready. handle this case... if cpc.stop_req.CompareAndSwap(false, true) { if cpc.conn != nil { cpc.conn.Close() } } }