Remove test code from main.go
This commit is contained in:
parent
5f7f07edea
commit
b25cab0104
|
@ -1,110 +1,9 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
|
||||||
|
|
||||||
"github.com/go-sql-driver/mysql"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Product struct {
|
|
||||||
Name string
|
|
||||||
Price float64
|
|
||||||
Available bool
|
|
||||||
}
|
|
||||||
|
|
||||||
var db *sql.DB
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
connectionConfig := mysql.Config{
|
fmt.Println("Hello, world")
|
||||||
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
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue