87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
package sql_driver
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
_ "github.com/mattn/go-sqlite3" // Import the SQLite3 driver
|
|
"log"
|
|
)
|
|
|
|
func InitTables() {
|
|
db, err := sql.Open("sqlite3", "./example.db")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// Create a table
|
|
createTableSQL := `
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT,
|
|
age INTEGER
|
|
);
|
|
`
|
|
_, err = db.Exec(createTableSQL)
|
|
if err != nil {
|
|
log.Fatalf("Error creating table: %s", err)
|
|
}
|
|
|
|
// Insert data into the table
|
|
insertUserSQL := `INSERT INTO users (name, age) VALUES (?, ?)`
|
|
_, err = db.Exec(insertUserSQL, "Alice", 30)
|
|
if err != nil {
|
|
log.Fatalf("Error inserting data: %s", err)
|
|
}
|
|
|
|
_, err = db.Exec(insertUserSQL, "Bob", 25)
|
|
if err != nil {
|
|
log.Fatalf("Error inserting data: %s", err)
|
|
}
|
|
|
|
// Query the data
|
|
queryUserSQL := `SELECT id, name, age FROM users`
|
|
rows, err := db.Query(queryUserSQL)
|
|
if err != nil {
|
|
log.Fatalf("Error querying data: %s", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var id int
|
|
var name string
|
|
var age int
|
|
err = rows.Scan(&id, &name, &age)
|
|
if err != nil {
|
|
log.Fatalf("Error scanning data: %s", err)
|
|
}
|
|
fmt.Printf("User: ID=%d, Name=%s, Age=%d\n", id, name, age)
|
|
}
|
|
|
|
// Update data
|
|
updateUserSQL := `UPDATE users SET age = ? WHERE name = ?`
|
|
_, err = db.Exec(updateUserSQL, 35, "Alice")
|
|
if err != nil {
|
|
log.Fatalf("Error updating data: %s", err)
|
|
}
|
|
|
|
// Verify the update
|
|
fmt.Println("After update:")
|
|
rows, err = db.Query(queryUserSQL)
|
|
if err != nil {
|
|
log.Fatalf("Error querying data: %s", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var id int
|
|
var name string
|
|
var age int
|
|
err = rows.Scan(&id, &name, &age)
|
|
if err != nil {
|
|
log.Fatalf("Error scanning data: %s", err)
|
|
}
|
|
fmt.Printf("User: ID=%d, Name=%s, Age=%d\n", id, name, age)
|
|
}
|
|
}
|