spark练习--由IP得到所在地

package cn.wj.spark.day06

import java.io.{BufferedReader, FileInputStream, InputStreamReader}

import scala.collection.mutable.ArrayBuffer

/**
  * Created by WJ on 2017/1/4.
  */
object IPLocationDemo_2 {

  // 将IP转换成为Long值
  def ip2Long(ip:String):Long ={
    val fragments = ip.split("[.]")
    var ipNum = 0L
    for(i <- 0 until fragments.length){
      ipNum = fragments(i).toLong | ipNum << 8L       //8进制变为10进制
    }
    ipNum
  }
   //进行二分法的查找，这个的前提是顺序已经是排序过的了
  def binarySearch(lines: ArrayBuffer[String], ip: Long) : Int = {
    var low = 0
    var high = lines.length - 1
    while (low <= high) {
      val middle = (low + high) / 2
      if ((ip >= lines(middle).split("\\|")(2).toLong) && (ip <= lines(middle).split("\\|")(3).toLong))
        return middle
      if (ip < lines(middle).split("\\|")(2).toLong)
        high = middle - 1
      else {
        low = middle + 1
      }
    }
    -1
  }

  //从一个文件里面读取数据
  def readData(path:String) = {
    val br = new BufferedReader(new InputStreamReader(new FileInputStream(path)))
    var s:String  = null
    var flag = true
    var lines = ArrayBuffer[String]()
    while(flag){
      s = br.readLine()
      if(s != null)
        lines += s
      else
        flag = false
    }
    lines
  }

  def main(args: Array[String]): Unit = {
    val ip ="118.144.130.10"
    val ipNum = ip2Long(ip)
    println(ipNum)
    val lines = readData("e://Test/ip.txt")
    val index = binarySearch(lines,ipNum)
    println(lines(index))
  }
}