package gosql

import (

"errors"

"strings"

)

func formatSql(cmd string, args ...interface{}) (string, error) {

// 先转为为小写

lowercmd := strings.ToLower(cmd)

// 找到关键字 values

tmp_index := strings.Index(lowercmd, " values")

if tmp_index < 0 {

return "", errors.New("insert sql error")

}

// 找到关键字 后面的第一个 (

start_index := strings.Index(cmd[tmp_index:], "(")

if start_index < 0 {

return "", errors.New("sql error: eg: insert into table(name) values(?)")

}

end_index := strings.LastIndex(cmd, ")")

if start_index < 0 {

return "", errors.New("sql error: eg: insert into table(name) values(?)")

}

value := cmd[tmp_index+start_index : end_index+1]

//查看一行数据有多少列

column := 0

for _, v := range strings.Split(value[1:len(value)-1], ",") {

opt := strings.Trim(v, " ")

if opt == "?" {

column++

}

}

// 总共多少参数

count := len(args)

if count%column != 0 {

return "", errors.New("args error")

}

addcmd := "," + value

for i := 1; i < count/column; i++ {

cmd = cmd + addcmd

}

return cmd, nil

}