码迷,mamicode.com
首页 > 其他好文 > 详细

校验Select和Insert的方法

时间:2018-11-26 22:07:19      阅读:226      评论:0      收藏:0      [点我收藏+]

标签:int   使用   code   工作   front   hello   string   integer   方法   

问题简介:在工作中,表字段多到一定程度,Select 中的字段和结果集中的字段对比很麻烦,还容易出错。于是写了一个检查Select和Insert的方法。(使用的是Scala语言)

  1.先判断语句是select语句还是insert语句

def sqlTest(str: String): Unit = {
    //检测是select语句还是insert语句
    if ( str.contains("select") ) {
      selectTest(str)
    } else {
      insertTest(str)
    }
  }

  2.如果是select语句,使用如下方法

 

def selectTest(selectStr: String): Unit = {
    var selectLists: List[String] = List()
    //1.将select与from中的字段,放入链表中
    selectLists = selectStr.substring(selectStr.indexOf("select") + 6, selectStr.indexOf("from")).replace("\r\n", "").trim.split(",").toList
    //2.获取第一个rs.get中所有的值
    var rsLists: List[String] = List()
    rsLists = nextElement(selectStr, rsLists)
    //3.判断select语句和结果集获取的值中的参数是否重复
    if ( selectLists.distinct.size != selectLists.size ) {
      println("查询语句有重复值")
    }
    if ( rsLists.distinct.size != rsLists.size ) {
      println("设置的结果集有重复值")
    }
    //4.比较select语句和结果集获取的值中的参数是否相等
    val selectSurplus = selectLists.toSet -- rsLists.toSet
    val rsSurplus = rsLists.toSet -- selectLists.toSet
    if ( selectSurplus.nonEmpty ) {
      println("select语句中多结果集多的字段有" + selectSurplus)
    }
    if ( rsSurplus.nonEmpty ) {
      println("结果集语句中比select多的字段有" + rsSurplus)
    }
  }

  @tailrec
  def nextElement(string: String, list: List[String]): List[String] = {
    val rightIndex = string.indexOf("\")")
    val leftIndex = string.indexOf("(\"") + 2
    val lists = list.::(string.substring(leftIndex, rightIndex))
    val subString = string.substring(rightIndex + 2)
    if ( string.lastIndexOf("\")").hashCode() == rightIndex.hashCode() ) {
      lists
    } else {
      nextElement(subString, lists)
    }
  }

 

  3.如果是insert语句,使用如下方法

 def insertTest(insertStr: String): Unit = {
    //1.获取insert和values之间的字符串,计算‘,‘的个数
    val insertSubFront = insertStr.substring(insertStr.indexOf("insert"), insertStr.indexOf("values"))
    val insertSubFrontNums = countNumbers(insertSubFront)
    //2.获取values之后的字符串,计算‘,‘的个数
    val insertSubBack = insertStr.substring(insertStr.indexOf("values"))
    val insertSubBackNums = countNumbers(insertSubBack)
    //3.判断两个字符串中的‘,‘数量差值
    val delNums = insertSubFrontNums - insertSubBackNums
    if ( delNums < 0 ) {
      println(s"values 的括号中多了[$delNums]个数据")
    } else if ( delNums > 0 ) {
      println(s"values 的括号中缺少[$delNums]个数据")
    } else {
      println("INSERT语句中字段都有值对应")
    }
  }

  private def countNumbers(strs: String): Integer = {
    //计算包含‘,‘字符的数量
    var count: Integer = 0
    strs.foreach {
      str =>
        if ( str == ‘,‘ ) {
          count = count + 1
        }
    }
    count
  }

  4.使用的时候直接调用sqlTest方法,放入对应的SQL就成。

//此变量为测试使用,不一定按照此格式使用
val selectStr =
    """
      |select
      |name,
      |password,
      |hello2
      |from
      |aa
      |rs.getString("name")
      |rs.getString("password")
      |rs.getString("hello")
    """.stripMargin
//此变量为测试使用,不一定按照此格式使用
val insertStr =
    """
      |insert
      |into
      |(name,
      |password,
      |hello2)
      | values
      |(?,?,?)
    """.stripMargin

  

 

校验Select和Insert的方法

标签:int   使用   code   工作   front   hello   string   integer   方法   

原文地址:https://www.cnblogs.com/orchid9/p/10022931.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!