Add changes so far

This commit is contained in:
maru committed 2023-12-05 13:28:08 -05:00
1 parent 4bb225a945
commit f690c53607
6 files changed
+207

No files matched your search

+20
View File
@@ -0,0 +1,20 @@
package db
import (
"database/sql"
"fmt"
)
func GetAccountInfoFromToken(token []byte) (string, error) {
var username string
err := handle.QueryRow("SELECT username FROM accounts WHERE uuid IN (SELECT uuid FROM sessions WHERE token = ? AND expire > UTC_TIMESTAMP())").Scan(&username)
if err != nil {
if err == sql.ErrNoRows {
return "", fmt.Errorf("invalid token")
}
return "", fmt.Errorf("query failed: %s", err)
}
return username, nil
}
+20
View File
@@ -0,0 +1,20 @@
package db
import (
"database/sql"
"fmt"
)
var handle *sql.DB
func Init(username, password, protocol, address, database string) error {
db, err := sql.Open("mysql", username+":"+password+"@"+protocol+"("+address+")/"+database)
if err != nil {
return fmt.Errorf("failed to open database connection: %s", err)
}
handle = db
return nil
}