From c38e1e0cfe9fd95f1cd2ed5f6c6cb10bf820b9b6 Mon Sep 17 00:00:00 2001 From: MHSanaei Date: Tue, 23 May 2023 02:43:15 +0330 Subject: a lot of improvement --- util/sys/sys_windows.go | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) (limited to 'util/sys/sys_windows.go') diff --git a/util/sys/sys_windows.go b/util/sys/sys_windows.go index ef9cdf54..34740ea6 100644 --- a/util/sys/sys_windows.go +++ b/util/sys/sys_windows.go @@ -4,21 +4,27 @@ package sys import ( + "errors" + "github.com/shirou/gopsutil/v3/net" ) -func GetTCPCount() (int, error) { - stats, err := net.Connections("tcp") +func GetConnectionCount(proto string) (int, error) { + if proto != "tcp" && proto != "udp" { + return 0, errors.New("invalid protocol") + } + + stats, err := net.Connections(proto) if err != nil { return 0, err } return len(stats), nil } +func GetTCPCount() (int, error) { + return GetConnectionCount("tcp") +} + func GetUDPCount() (int, error) { - stats, err := net.Connections("udp") - if err != nil { - return 0, err - } - return len(stats), nil + return GetConnectionCount("udp") } -- cgit v1.2.3