From 6c49d1ec3153852f0e5d4cff334a1bc0476f9dce Mon Sep 17 00:00:00 2001 From: Vidhu Kant Sharma Date: Tue, 6 May 2025 12:38:32 +0530 Subject: first commit --- db/db.go | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 db/db.go (limited to 'db/db.go') diff --git a/db/db.go b/db/db.go new file mode 100644 index 0000000..11b6fd4 --- /dev/null +++ b/db/db.go @@ -0,0 +1,53 @@ +/* guestbook - Standalone guestbook server for static websites + * Copyright (C) 2025 Vidhu Kant Sharma + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package db + +import ( + "fmt" + "log" + "database/sql" + + "github.com/spf13/viper" + _ "github.com/go-sql-driver/mysql" +) + +type GuestbookEntry struct { + Name string `json:"name"` + Website string `json:"website"` + Message string `json:"message"` + Validation string `json:"validation"` +} + +func Connect() *sql.DB { + conn, err := sql.Open( + "mysql", + fmt.Sprintf( + "%s:%s@%s/%s?%s", + viper.GetString("database.username"), + viper.GetString("database.password"), + viper.GetString("database.location"), + viper.GetString("database.db_name"), + viper.GetString("database.params"), + ), + ) + if err != nil { + log.Fatal(err) + } + + return conn +} -- cgit v1.2.3