shutdown.go

package tcp

import "errors"

func (conn *TcpConn) VShutdown(sdType int) error {
	sock := state.sockets[*conn]

	if sdType == SHUTDOWN_BOTH {
		if !sock.canRead && !sock.canWrite {
			return errors.New("socket read and write already shutdown")
		}
		if !sock.canRead {
			return errors.New("socket read already shutdown")
		}
		if !sock.canWrite {
			return errors.New("socket write already shutdown")
		}
		sock.canRead = false
		sock.canWrite = false
	} else if sdType == SHUTDOWN_READ {
		if sock.canRead {
			sock.canRead = false
		} else {
			return errors.New("socket read already shutdown")
		}
	} else if sdType == SHUTDOWN_WRITE {
		if sock.canWrite {
			sock.canWrite = false
		} else {
			return errors.New("socket write already shutdown")
		}
		// signal handlewriter
	} else {
		return errors.New("invalid shutdown type")
	}

	if sdType&SHUTDOWN_WRITE != 0 {
		sock.writeBufferLock.Lock()
		sock.writeBufferIsNotEmpty.Broadcast()
		sock.writeBufferLock.Unlock()
	}
	if sdType&SHUTDOWN_READ != 0 {
		sock.readBufferLock.Lock()
		sock.readBufferIsNotEmpty.Broadcast()
		sock.readBufferLock.Unlock()
	}

	return nil
}