在使用GORM(Go語言的ORM庫)時,避免常見的SQL錯誤可以通過以下方法來實現:
檢查數據庫連接:確保你已經正確配置了數據庫連接信息,并且數據庫服務器正在運行。
初始化GORM:在使用GORM之前,確保你已經正確初始化了GORM并創建了數據庫連接。例如:
import (
"gorm.io/gorm"
"gorm.io/driver/mysql"
)
func main() {
dsn := "username:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
if err != nil {
panic("failed to connect database")
}
}
type User struct {
ID uint `gorm:"primaryKey"`
Name string `gorm:"size:255"`
Email string `gorm:"unique;not null"`
CreatedAt time.Time
UpdatedAt time.Time
}
db.AutoMigrate(&User{})
// 錯誤的做法:直接拼接SQL語句
sql := fmt.Sprintf("SELECT * FROM users WHERE name = '%s'", name)
db.Raw(sql).Find(&users)
// 正確的做法:使用參數化查詢
db.Where("name = ?", name).Find(&users)
result := db.Create(&user)
if result.Error != nil {
// 處理錯誤
}
tx := db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
}
}()
// 執行數據庫操作
tx.Create(&user)
tx.Model(&user).Update("name", "new_name")
// 提交事務
tx.Commit()
優化查詢:避免使用SELECT *,而是只查詢所需的字段。此外,使用索引、分頁和排序來提高查詢性能。
測試:編寫單元測試和集成測試來確保你的代碼在不同情況下都能正確工作。
遵循以上建議,可以幫助你在使用GORM時避免常見的SQL錯誤。