hezhizheng / go-ip2region Goto Github PK

package main

import (
	"bufio"
	"fmt"
	"os"

	"github.com/lionsoul2014/ip2region/binding/golang/ip2region"
)

func main() {
	result := make(chan ip2region.IpInfo)
	region, err := ip2region.New("ip2region.db")

	if err != nil {
		fmt.Println(err)
		return
	}
	defer region.Close()
	ips := findLines()
	for _, ip := range ips {
		go func(ip string) {
			r, _ := region.MemorySearch(ip)
			fmt.Println(r)
			result <- r
		}(ip)
	}
	for i := range result {
		fmt.Println(i)
	}
}

func findLines() []string {
	filePath := "./ips.txt"  //里面有1.7W个IP

	file, err := os.Open(filePath)
	if err != nil {
		fmt.Println(err)
	}
	defer file.Close()

	fd := bufio.NewReader(file)
	var tempArray []string
	for {
		lData, err := fd.ReadString('\n')
		if err != nil {
			break
		}
		tempArray = append(tempArray, lData)
	}

	return tempArray
}