Remove test code from main.go

This commit is contained in:
The Magician 2024-05-21 12:17:24 +01:00
parent 5f7f07edea
commit b25cab0104
1 changed files with 1 additions and 102 deletions

View File

@ -1,110 +1,9 @@
package main
import (
"database/sql"
"fmt"
"log"
"github.com/go-sql-driver/mysql"
)
type Product struct {
Name string
Price float64
Available bool
}
var db *sql.DB
func main() {
connectionConfig := mysql.Config{
User: "root",
Passwd: "o7MS6CIn660jIApSP",
Net: "tcp",
Addr: "127.0.0.1:3306",
DBName: "goql",
AllowNativePasswords: true,
}
var err error
db, err := sql.Open("mysql", connectionConfig.FormatDSN())
defer db.Close()
if err != nil {
log.Fatal(err)
}
if err = db.Ping(); err != nil {
log.Fatal(err)
}
fmt.Println("Connected!")
createProductTable(db)
insertProduct(db, Product{Name: "Book 1", Price: 15.95, Available: true})
insertProduct(db, Product{Name: "Book 2", Price: 15.95, Available: true})
insertProduct(db, Product{Name: "Book 3", Price: 15.95, Available: true})
insertProduct(db, Product{Name: "Book 4", Price: 15.95, Available: true})
insertProduct(db, Product{Name: "Book 5", Price: 15.95, Available: true})
products := getAllProducts(db)
fmt.Println(products)
}
func createProductTable(db *sql.DB) {
query := `CREATE TABLE IF NOT EXISTS Product (
Id SERIAL PRIMARY KEY,
Name VARCHAR(100) NOT NULL,
Price NUMERIC(6,2) NOT NULL,
Available BOOLEAN,
Created TIMESTAMP DEFAULT NOW()
)`
_, err := db.Exec(query)
if err != nil {
log.Fatal(err)
}
}
func insertProduct(db *sql.DB, product Product) int64 {
query := "INSERT INTO Product (Name, Price, Available) VALUES (?, ?, ?);"
insert, err := db.Prepare(query)
defer insert.Close()
if err != nil {
log.Fatal(err)
}
result, err := insert.Exec(product.Name, product.Price, product.Available)
rowsAffected, _ := result.RowsAffected()
if err != nil || rowsAffected != 1 {
log.Fatal(err)
}
insertedRowId, _ := result.LastInsertId()
return insertedRowId
}
func getAllProducts(db *sql.DB) []Product {
products := []Product{}
rows, err := db.Query("SELECT Name, Price, Available FROM Product;")
if err != nil {
log.Fatal(err)
}
defer rows.Close()
var product Product
for rows.Next() {
err := rows.Scan(&product.Name, &product.Price, &product.Available)
if err != nil {
log.Fatal(err)
}
products = append(products, product)
}
return products
fmt.Println("Hello, world")
}