Verschiedene Typen in package types ausgelagert und Channels zum asynchronen Datentransport eingesetzt

This commit is contained in:
2023-10-05 19:52:11 +02:00
parent fcb509c9fe
commit 1bcfbfd325
4 changed files with 66 additions and 33 deletions

View File

@ -8,10 +8,17 @@ import (
"strings"
"syscall"
"streifling.com/jason/sicherheitsunterweisung/types"
"github.com/go-sql-driver/mysql"
"golang.org/x/term"
)
type DB struct {
*sql.DB
Name string
}
func getCredentials() (string, string, error) {
fmt.Printf("DB Benutzer: ")
user, err := bufio.NewReader(os.Stdin).ReadString('\n')
@ -30,7 +37,8 @@ func getCredentials() (string, string, error) {
return strings.TrimSpace(user), strings.TrimSpace(pass), nil
}
func OpenDB(dbName string) (*sql.DB, error) {
func OpenDB(dbName string) (*DB, error) {
var db *DB
var err error
cfg := mysql.NewConfig()
@ -40,7 +48,8 @@ func OpenDB(dbName string) (*sql.DB, error) {
return nil, fmt.Errorf("OpenDB: getCredentials(): %v\n", err)
}
db, err := sql.Open("mysql", cfg.FormatDSN())
db.Name = dbName
db.DB, err = sql.Open("mysql", cfg.FormatDSN())
if err != nil {
return nil, fmt.Errorf("newDB: sql.Open(\"mysql\", cfg.FormatDSN()): %v\n", err)
}
@ -50,3 +59,14 @@ func OpenDB(dbName string) (*sql.DB, error) {
return db, nil
}
func (db *DB) WriteBriefing(b *types.Briefing) error {
for i := 1; i < len(b.Participants); i++ {
_, err := db.Exec("INSERT INTO ? (instructor_first, instructor_last, date, time, state, location, participant_first, participant_last, company) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", db.Name, b.FirstName, b.LastName, b.Date, b.Time, b.State, b.Location, b.Participants[i].FirstName, b.Participants[i].LastName, b.Participants[i].Company)
if err != nil {
return fmt.Errorf("*DB.WriteBriefing: db.Exec(\"INSERT INTO ? (instructor_first, instructor_last, date, time, state, location, participant_first, participant_last, company) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)\", db.Name, b.FirstName, b.LastName, b.Date, b.Time, b.State, b.Location, b.Participants[i].FirstName, b.Participants[i].LastName, b.Participants[i].Company): %v\n", err)
}
}
return nil
}