golang mysql select查询使用分表函数错误判断

Question

func calc_table func(id int64)   string{
if id < 1{
  return ""
}
return "mesage_" + strconv.FormatInt(id%int64(8), 10)
}

sql := "select * from " + calc_table(id) + " where id=10"

我在sql里使用了分表算法 ，我想在calc_table函数返回为空的时候，不执行这个sql，应该怎么做，难道必须先把calc_table(id)的值先赋值给一个变量么

Answer 1

晚上好🌙🌙🌙
本答案参考通义千问

在 Go 语言中，如果你希望在 calc_table(id) 返回空字符串时 不执行 SQL 查询，那么你确实需要先将 calc_table(id) 的结果赋值给一个变量，然后根据该变量的值来决定是否执行 SQL 查询。这是因为直接在字符串拼接中使用函数返回值时，无法直接进行条件判断。

---

✅ 正确做法如下：

1. 调用 calc_table(id) 并将结果赋值给变量
2. 判断变量是否为空
3. 如果非空，再构建并执行 SQL 查询

---

📌 加粗重点说明：

• 必须先调用 calc_table(id) 并赋值给变量，才能进行错误判断。
• 不能直接在 SQL 字符串中判断函数返回值是否为空，因为 Go 不支持这种写法。

---

✅ 详细解决方案（有序列表）：

1. 定义 calc_table 函数，确保返回正确的分表名或空字符串

   func calc_table(id int64) string {
       if id < 1 {
           return ""
       }
       return "message_" + strconv.FormatInt(id%int64(8), 10)
   }
   

2. 调用函数并将结果赋值给变量

   tableName := calc_table(id)
   

3. 判断变量是否为空，若为空则跳过 SQL 执行

   if tableName == "" {
       // 分表名为空，不执行查询
       log.Println("Invalid table name, skipping query.")
       return
   }
   

4. 构建并执行 SQL 查询

   sql := "SELECT * FROM " + tableName + " WHERE id = 10"
   rows, err := db.Query(sql)
   if err != nil {
       log.Fatal(err)
   }
   defer rows.Close()
   // 处理查询结果...
   

---

🔧 修改后的完整代码示例：

package main

import (
	"database/sql"
	"fmt"
	"log"
	"strconv"
)

func calc_table(id int64) string {
	if id < 1 {
		return ""
	}
	return "message_" + strconv.FormatInt(id%int64(8), 10)
}

func main() {
	id := int64(10)
	tableName := calc_table(id)

	if tableName == "" {
		log.Println("Invalid table name, skipping query.")
		return
	}

	sql := fmt.Sprintf("SELECT * FROM %s WHERE id = 10", tableName)
	db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	rows, err := db.Query(sql)
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	// 处理查询结果...
}

---

⚠️ 注意事项：

• SQL 注入风险：虽然在这个例子中是固定字段，但如果你的 id 是用户输入的，建议使用参数化查询，避免 SQL 注入。
• 分表逻辑应尽量合理：例如 id%8 可能会导致某些表没有数据，需根据实际业务调整。

---

✅ 总结：

必须先将 calc_table(id) 的结果赋值给变量，再进行判断，才能实现“当返回为空时不执行 SQL”的需求。