Git structure files

app/handlers/sqldb/sqldb.go

@@ -0,0 +1,70 @@
+package sqldb
+
+import (
+	"database/sql"
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+var DB *sql.DB
+var dbUser, dbPass, dbHost, dbPort, dbName string
+
+// check error
+func check(err error) {
+	if err != nil {
+		panic(err.Error())
+	}
+}
+
+// read db cred
+func ReaddbCred() {
+	dbCred := viper.New()
+	dbCred.SetConfigName("db-cred")
+	dbCred.AddConfigPath("../")
+	dbCred.SetConfigType("yaml")
+	err := dbCred.ReadInConfig()
+	check(err)
+	dbUser = dbCred.GetString("dbUser")
+	dbPass = dbCred.GetString("dbPass")
+	dbHost = dbCred.GetString("dbHost")
+	dbPort = dbCred.GetString("dbPort")
+	dbName = dbCred.GetString("dbName")
+}
+
+func Opendb() (db *sql.DB) {
+	// open db
+	sqlopen := fmt.Sprintf(dbUser + ":" + dbPass + "@tcp(" + dbHost + ":" + dbPort + ")/" + dbName)
+	var err error
+	DB, err = sql.Open("mysql", sqlopen)
+	check(err)
+	fmt.Println("Successfully connected to the database")
+	return DB
+}
+
+func Dbinit() {
+	dbinit, err := DB.Query("CREATE TABLE IF NOT EXISTS `media` (`id` INT NOT NULL AUTO_INCREMENT, `ytid` VARCHAR(100), `url` VARCHAR(255), `localpath` TEXT(65535), `size` INT, `sha256sum` VARCHAR(255), `length` INT, `title` VARCHAR(255), `author` VARCHAR(255), PRIMARY KEY (`id`));")
+	if err != nil {
+		fmt.Println("Error while creating table")
+		panic(err.Error())
+	}
+	defer dbinit.Close()
+	fmt.Println("DB-Init successfull")
+}
+
+func Dbentrycount() {
+	var count int
+	dbcount, err := DB.Query("SELECT COUNT(*) FROM testdb.media;")
+	if err != nil {
+		fmt.Println("Error while counting rows from table")
+		panic(err.Error())
+	}
+	defer dbcount.Close()
+	for dbcount.Next() {
+		err = dbcount.Scan(&count)
+		if err != nil {
+			panic(err.Error())
+		}
+	}
+	fmt.Printf("There are %v entries in the database\n", count)
+}