package utils

import (
	"fmt"
	"git.hpds.cc/Component/network/log"
	"net"
)

// GetAvailablePort 获取可用端口
func GetAvailablePort() (int, error) {
	address, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:0", "0.0.0.0"))
	if err != nil {
		return 0, err
	}

	listener, err := net.ListenTCP("tcp", address)
	if err != nil {
		return 0, err
	}

	defer func(listener *net.TCPListener) {
		_ = listener.Close()
	}(listener)
	return listener.Addr().(*net.TCPAddr).Port, nil
}

// IsPortAvailable 判断端口是否可以（未被占用）
func IsPortAvailable(port int) bool {
	address := fmt.Sprintf("%s:%d", "0.0.0.0", port)
	listener, err := net.Listen("tcp", address)
	if err != nil {
		log.Infof("port %s is taken: %s", address, err)
		return false
	}

	defer func(listener net.Listener) {
		_ = listener.Close()
	}(listener)
	return true
}