接上一篇笔记,这里我们又做了些许优化,具体我直接标注在了代码的后面,可以根据前一篇笔记比对
package main
import (
"fmt"
"net"
"sort"
)
//处理工作的worker函数
func worker(ports, results chan int) {
for p := range ports { //range是go语言系统定义的一个函数。 函数的含义是在一个数组中遍历每一个值,返回该值的下标值和此处的实际值。 假如说a[0]=10,则遍历到a[0]的时候返回值为0,10两个值
address := fmt.Sprintf("wangehacker.cn:%d", p)
conn, err := net.Dial("tcp", address)
if err != nil { //这里如果端口关闭则传递一个0数值给results
results <- 0
continue
}
conn.Close()
results <- p //端口开启则将端口号传递给results
}
}
func main() {
ports := make(chan int, 100)
results := make(chan int) //创建一个单独的通道
var openports []int
for i := 0; i < cap(ports); i++ { //进行切片储存,方便排序
go worker(ports, results)
}
go func() {
for i := 1; i < 1024; i++ { //1024个端口号依次放入ports参数中
ports <- 1
}
}()
for i := 1; i <= 1024; i++ { //收集循环,结果不为0,也就是端口开启,则追加到切片等待排序
port := <-results
if port != 0 {
openports = append(openports, port)
}
}
close(ports)
close(results)
sort.Ints(openports) // sort.Ints排序,openports为int类型,也就是传入为int类型用ints
for _, port := range openports { //对结果进行一个排序
fmt.Printf("%d open\n", port)
}
}
有后续继续完善还会更新
版权声明:本文为qq_41132792原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。