duanhuan5409 2014-06-22 17:10
浏览 414
已采纳

用切片执行SQL准备的语句

I've wrote an function (In Go, of course) that inserting map[string]interface{} to mysql via this library.

Explanation of the code below:

  1. The functions receives string called table and map[string]interface{} called data.
  2. I separate the data to keys (variable called columns) and values (variable called values).
  3. I generate from the columns variable called column_text that will look like this: first_name, last_name, birth_day, date_added
  4. I generate from the values variable called variable_text that will look like this: ?, ?, ?, ?
  5. I open mysql connection: db, err := sql.Open("mysql", "user:pass@/database")
  6. I create prepared statement: stmt, err := db.Prepare("INSERT INTO " + table + " ( " + columns_text + " ) VALUES ( " + values_text + " )")
  7. I execute the prepare statement. but I have a problem. the number of the columns and values changes all the time, and the stmt.Exec() command can't receive an slice (Array) like this: stmt.Exec(values), only the values alone like this: stmt.Exec(values[0], values[1], values[2]...)

The question:

I'm coming from PHP, where PDO::Statement could receive an array when executing. How can I execute the statement with the slice (Array)? (If I can do it with different library, please write the name of the library and how to use it, thank you!)

The code:

func insertToDB(table string, data map[string]interface{}) {
columns := make([]interface{}, 0, len(data))
values := make([]interface{}, 0, len(data))

for  key, _ := range data {
   columns = append(columns, key)
   values = append(values, data[key])
}

columns_text := ""
i := 0
of := len(data)

for i < of {
    column := columns[i].(string)

    if i == 0 {
        columns_text = column
    } else {
        columns_text = columns_text + ", " + column
    }

    i++
}

fmt.Println(columns_text + " = " + table)

values_text := ""

i = 0

for i < of {

    if i == 0 {
        values_text = "?"
    } else {
        values_text = values_text + ", ?"
    }

    i++
}

fmt.Println(values_text)
fmt.Println(values)
fmt.Println(data)

db, err := sql.Open("mysql", "root:root@/bacafe")
if err != nil {
    return -1, err
}
defer db.Close()

stmtIns, err := db.Prepare("INSERT INTO " + table + " ( " + columns_text + " ) VALUES ( " +  values_text + " )")
if err != nil {
    return -1, err
}
defer stmtIns.Close() // Close the statement when we leave main() / the program terminates

result, err := stmtIns.Exec(values...)
if err != nil {
    return -1, err
} else {
    insertedID, err := result.LastInsertId()
    if err != nil {
        return -1, err
    } else {
        return int(insertedID), nil
    }
}
}

EDIT: I've edited the function above and it works perfectly now.

Thank you!

  • 写回答

1条回答 默认 最新

  • douxie3625 2014-06-22 17:26
    关注

    You are on the right track however Stmt.Exec takes args ...interface{}, so for your specific example you need to change 2 things:

    ......
    values := make([]interface{}, 0, len(data))
    ......
    //adding ... expands the values, think of it like func.apply(this, array-of-values) in 
    // javascript, in a way.
    _, err = stmtIns.Exec(values...) 
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 树莓派与pix飞控通信
  • ¥15 自动转发微信群信息到另外一个微信群
  • ¥15 outlook无法配置成功
  • ¥30 这是哪个作者做的宝宝起名网站
  • ¥60 版本过低apk如何修改可以兼容新的安卓系统
  • ¥25 由IPR导致的DRIVER_POWER_STATE_FAILURE蓝屏
  • ¥50 有数据,怎么建立模型求影响全要素生产率的因素
  • ¥50 有数据,怎么用matlab求全要素生产率
  • ¥15 TI的insta-spin例程
  • ¥15 完成下列问题完成下列问题