|
@@ -1,74 +1,83 @@
|
|
|
-
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
+ "database/sql"
|
|
|
+ "encoding/json"
|
|
|
"fmt"
|
|
|
"io/ioutil"
|
|
|
- "encoding/json"
|
|
|
- "database/sql"
|
|
|
+ "log"
|
|
|
+
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
|
)
|
|
|
|
|
|
type ConnectionProperties struct {
|
|
|
- User string
|
|
|
- Pass string
|
|
|
- Host string
|
|
|
- Db string
|
|
|
+ User string
|
|
|
+ Pass string
|
|
|
+ Host string
|
|
|
+ Db string
|
|
|
}
|
|
|
|
|
|
type connection struct {
|
|
|
- dBOpenStr string
|
|
|
+ dBOpenStr string
|
|
|
connProperties ConnectionProperties
|
|
|
}
|
|
|
|
|
|
-func (c *connection) Init(filepath string){
|
|
|
- if (NOMYSQL) {
|
|
|
+type Torrent struct {
|
|
|
+ ID int64
|
|
|
+ Hash string
|
|
|
+ Name string
|
|
|
+ OwnerID string
|
|
|
+ Time string
|
|
|
+ Deleted bool
|
|
|
+}
|
|
|
+
|
|
|
+func (c *connection) Init(filepath string) {
|
|
|
+ if NOMYSQL {
|
|
|
return
|
|
|
}
|
|
|
b, err := ioutil.ReadFile(filepath)
|
|
|
- if err != nil {
|
|
|
- fmt.Print(err)
|
|
|
- }
|
|
|
- propJson := string(b)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Print(err)
|
|
|
+ }
|
|
|
+ propJson := string(b)
|
|
|
|
|
|
json.Unmarshal([]byte(propJson), &c.connProperties)
|
|
|
-
|
|
|
+
|
|
|
//c.connProperties.Db = "nosite"
|
|
|
fmt.Printf("Connection data:\n%s\n%s\n%s\n%s\n", c.connProperties.User, c.connProperties.Pass, c.connProperties.Host, c.connProperties.Db)
|
|
|
- c.dBOpenStr = fmt.Sprintf ("%s:%s@tcp(%s)/%s", c.connProperties.User, c.connProperties.Pass, c.connProperties.Host, c.connProperties.Db)
|
|
|
+ c.dBOpenStr = fmt.Sprintf("%s:%s@tcp(%s)/%s", c.connProperties.User, c.connProperties.Pass, c.connProperties.Host, c.connProperties.Db)
|
|
|
fmt.Printf("Connecting with:\n%s\n", c.dBOpenStr)
|
|
|
db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
if err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
if err = db.Ping(); err != nil {
|
|
|
- db.Close()
|
|
|
- logger.Print("Fatal : Error with connection to database!")
|
|
|
- } else {
|
|
|
- fmt.Print("Connection succesfull!")
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
-
|
|
|
- logger.Print("Trying to connect to DB-server...")
|
|
|
- c.dBOpenStr = fmt.Sprintf ("%s:%s@tcp(%s)/", c.connProperties.User, c.connProperties.Pass, c.connProperties.Host)
|
|
|
- fmt.Printf("Connecting with:\n%s\n", c.dBOpenStr)
|
|
|
+ db.Close()
|
|
|
+ logger.Println("Fatal : Error with connection to database!")
|
|
|
+ } else {
|
|
|
+ fmt.Println("Connection succesfull!")
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ logger.Print("Trying to connect to DB-server...")
|
|
|
+ c.dBOpenStr = fmt.Sprintf("%s:%s@tcp(%s)/", c.connProperties.User, c.connProperties.Pass, c.connProperties.Host)
|
|
|
+ fmt.Printf("Connecting with:\n%s\n", c.dBOpenStr)
|
|
|
db, err = sql.Open("mysql", c.dBOpenStr)
|
|
|
if err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
if err = db.Ping(); err != nil {
|
|
|
- db.Close()
|
|
|
- logger.Print("Fatal : Error with connection to database server!")
|
|
|
- return
|
|
|
- } else {
|
|
|
-
|
|
|
- }
|
|
|
- c.databaseInitialization();
|
|
|
+ db.Close()
|
|
|
+ logger.Print("Fatal : Error with connection to database server!")
|
|
|
+ return
|
|
|
+ } else {
|
|
|
+
|
|
|
+ }
|
|
|
+ c.databaseInitialization()
|
|
|
}
|
|
|
|
|
|
/* создает на сервере необходимую бд и таблицы */
|
|
|
-func (c *connection) databaseInitialization() {
|
|
|
+func (c *connection) databaseInitialization() {
|
|
|
db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
|
|
|
if err != nil {
|
|
@@ -83,46 +92,60 @@ func (c *connection) databaseInitialization() {
|
|
|
act_query = "SELECT count(*) FROM information_schema.tables WHERE TABLE_SCHEMA = '" + c.connProperties.Db + "';"
|
|
|
db.QueryRow(act_query).Scan(&counter)
|
|
|
fmt.Print(counter)
|
|
|
- if (counter != 0) {
|
|
|
+ if counter != 0 {
|
|
|
logger.Print("Server already has the specified database")
|
|
|
} else {
|
|
|
logger.Print("The server does not have the specified database")
|
|
|
logger.Printf("Creating database '%s'...", c.connProperties.Db)
|
|
|
act_query = "CREATE SCHEMA " + c.connProperties.Db + " DEFAULT CHARACTER SET utf8 ;"
|
|
|
result, err := db.Exec(act_query)
|
|
|
- if err != nil{
|
|
|
- panic(err)
|
|
|
- }
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
|
|
|
- rowsCount, _ := result.RowsAffected()
|
|
|
+ rowsCount, _ := result.RowsAffected()
|
|
|
fmt.Printf("Lines changed: %d\n", rowsCount)
|
|
|
- if (rowsCount == 1) {
|
|
|
- logger.Print("Succesfull!")
|
|
|
- }
|
|
|
+ if rowsCount == 1 {
|
|
|
+ logger.Print("Succesfull!")
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
logger.Print("Checking for existence of table 'users' on server...")
|
|
|
- act_query = "SELECT count(*) FROM information_schema.tables WHERE TABLE_NAME = 'users' AND TABLE_SCHEMA = '"+ c.connProperties.Db +"';"
|
|
|
+ act_query = "SELECT count(*) FROM information_schema.tables WHERE TABLE_NAME = 'users' AND TABLE_SCHEMA = '" + c.connProperties.Db + "';"
|
|
|
db.QueryRow(act_query).Scan(&counter)
|
|
|
fmt.Print(counter)
|
|
|
- if (counter != 0) {
|
|
|
+ if counter != 0 {
|
|
|
logger.Print("Server already has the specified table!")
|
|
|
} else {
|
|
|
logger.Print("The server does not have the specified table")
|
|
|
- logger.Printf("Creating table '%s'.'users'...",c.connProperties.Db)
|
|
|
+ logger.Printf("Creating table '%s'.'users'...", c.connProperties.Db)
|
|
|
act_query = "CREATE TABLE `" + c.connProperties.Db + "`.`users` ( `idusers` INT NOT NULL AUTO_INCREMENT,`username` VARCHAR(45) NOT NULL, `password` VARCHAR(45) NOT NULL, PRIMARY KEY(`idusers`), UNIQUE INDEX `idusers_UNIQUE` (`idusers` ASC), UNIQUE INDEX `username_UNIQUE` (`username` ASC)) ENGINE = InnoDB DEFAULT CHARACTER SET utf8 ;"
|
|
|
fmt.Print(act_query)
|
|
|
_, err := db.Exec(act_query)
|
|
|
- if err != nil{
|
|
|
- panic(err)
|
|
|
- }
|
|
|
- logger.Print("Succesfull!")
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ logger.Print("Succesfull!")
|
|
|
}
|
|
|
+ /*
|
|
|
+
|
|
|
+ CREATE TABLE `gosite`.`torrents` (
|
|
|
+ `id` INT NOT NULL AUTO_INCREMENT,
|
|
|
+ `hash` VARCHAR(45) NOT NULL,
|
|
|
+ `name` VARCHAR(45) NOT NULL,
|
|
|
+ `ownerid` VARCHAR(45) NOT NULL,
|
|
|
+ `time` TIMESTAMP NOT NULL,
|
|
|
+ `deleted` TINYINT NULL,
|
|
|
+ UNIQUE INDEX `idtorrents_UNIQUE` (`id` ASC) VISIBLE,
|
|
|
+ PRIMARY KEY (`hash`),
|
|
|
+ UNIQUE INDEX `hash_UNIQUE` (`hash` ASC) VISIBLE);
|
|
|
|
|
|
|
|
|
+ */
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-func (c connection) LogIn(username string, password string) bool{
|
|
|
+func (c connection) LogIn(username string, password string) bool {
|
|
|
fmt.Printf("\n\nLogIn\nConnecting with:\n%s\n", c.dBOpenStr)
|
|
|
db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
|
|
@@ -137,12 +160,11 @@ func (c connection) LogIn(username string, password string) bool{
|
|
|
act_query := fmt.Sprintf("SELECT count(*) FROM %s.users WHERE username='%s' AND password=SHA('%s');", c.connProperties.Db, username, password)
|
|
|
db.QueryRow(act_query).Scan(&counter)
|
|
|
fmt.Println("we have", counter, "rows")
|
|
|
-
|
|
|
- if (counter == 0) {
|
|
|
- return false
|
|
|
- }
|
|
|
- return true
|
|
|
|
|
|
+ if counter == 0 {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ return true
|
|
|
|
|
|
}
|
|
|
|
|
@@ -157,40 +179,75 @@ func (c connection) IsNameUsed(username string) bool {
|
|
|
var counter int
|
|
|
act_query := fmt.Sprintf("SELECT count(*) FROM %s.users WHERE username='%s';", c.connProperties.Db, username)
|
|
|
db.QueryRow(act_query).Scan(&counter)
|
|
|
-
|
|
|
- if (counter == 0) {
|
|
|
+
|
|
|
+ if counter == 0 {
|
|
|
fmt.Printf("Username unused\n")
|
|
|
- return false
|
|
|
- }
|
|
|
+ return false
|
|
|
+ }
|
|
|
fmt.Printf("Username used\n")
|
|
|
return true
|
|
|
}
|
|
|
|
|
|
func (c connection) SigInUser(username string, password string) bool {
|
|
|
- db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
- if err != nil {
|
|
|
- panic(err)
|
|
|
- }
|
|
|
- defer db.Close()
|
|
|
- act_query := fmt.Sprintf("INSERT INTO %s.users (username, password) VALUES ('%s', SHA('%s'))", c.connProperties.Db, username, password)
|
|
|
- result, err := db.Exec(act_query)
|
|
|
- if err != nil{
|
|
|
- panic(err)
|
|
|
- }
|
|
|
-
|
|
|
- rowsCount, _ := result.RowsAffected()
|
|
|
- if (rowsCount == 1) {
|
|
|
- fmt.Printf("Lines changed: %d\n", rowsCount)
|
|
|
- return true
|
|
|
- } else {
|
|
|
- return false
|
|
|
- }
|
|
|
+ db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ defer db.Close()
|
|
|
+ act_query := fmt.Sprintf("INSERT INTO %s.users (username, password) VALUES ('%s', SHA('%s'))", c.connProperties.Db, username, password)
|
|
|
+ result, err := db.Exec(act_query)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ rowsCount, _ := result.RowsAffected()
|
|
|
+ if rowsCount == 1 {
|
|
|
+ fmt.Printf("Lines changed: %d\n", rowsCount)
|
|
|
+ return true
|
|
|
+ } else {
|
|
|
+ return false
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
-func (c connection) SubmitScore(score int){
|
|
|
+func (c connection) SubmitScore(score int) {
|
|
|
fmt.Printf("Submiting score %d", score)
|
|
|
}
|
|
|
|
|
|
+func (c connection) TakeTorrent(hash string) (Torrent, error) {
|
|
|
+ db, err := sql.Open("mysql", c.dBOpenStr)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ defer db.Close()
|
|
|
+ act_query := fmt.Sprintf("SELECT * FROM `%s`.`torrents` WHERE `hash`='%s'", c.connProperties.Db, hash)
|
|
|
+ fmt.Println(act_query)
|
|
|
+ rows, err := db.Query(act_query)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ defer rows.Close()
|
|
|
+
|
|
|
+ var torrent Torrent
|
|
|
+ for rows.Next() {
|
|
|
+
|
|
|
+ if err := rows.Scan(
|
|
|
+ &torrent.ID,
|
|
|
+ &torrent.Hash,
|
|
|
+ &torrent.Name,
|
|
|
+ &torrent.OwnerID,
|
|
|
+ &torrent.Time,
|
|
|
+ &torrent.Deleted,
|
|
|
+ ); err != nil {
|
|
|
+ log.Fatal(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ fmt.Println(torrent)
|
|
|
+ return torrent, nil
|
|
|
+ }
|
|
|
+
|
|
|
+ return torrent, fmt.Errorf("Torrent not found")
|
|
|
+}
|
|
|
+
|
|
|
/*
|
|
|
func main() {
|
|
|
|
|
@@ -209,4 +266,4 @@ func main() {
|
|
|
fmt.Printf("logIn error\n")
|
|
|
}
|
|
|
}
|
|
|
-*/
|
|
|
+*/
|