package quic

import "net"

type connection interface {
	write([]byte) error
	setCurrentRemoteAddr(interface{})
	IP() net.IP
}

type udpConn struct {
	conn        *net.UDPConn
	currentAddr *net.UDPAddr
}

var _ connection = &udpConn{}

func (c *udpConn) write(p []byte) error {
	_, err := c.conn.WriteToUDP(p, c.currentAddr)
	return err
}

func (c *udpConn) setCurrentRemoteAddr(addr interface{}) {
	c.currentAddr = addr.(*net.UDPAddr)
}

func (c *udpConn) IP() net.IP {
	return c.currentAddr.IP
}