add future

src/controllers/dataController.go

@@ -2,25 +2,40 @@ package controllers
 
 import (
 	"bot/src/models"
+	"database/sql"
+	"fmt"
+	"log"
 	"math/rand"
 	"time"
 
-	"github.com/glebarez/sqlite"
-	"github.com/robfig/cron"
-	"gorm.io/gorm"
+	_ "github.com/go-sql-driver/mysql"
 )
 
-type DataController struct {
-	DB *gorm.DB
+type DataConfig struct {
+	UserName string
+	Password string
+	HostName string
+	DbName   string
 }
 
+func (d *DataConfig) Get() string {
+	return fmt.Sprintf("%s:%s@tcp(%s)/%s", d.UserName, d.Password, d.HostName, d.DbName)
+}
+
+type DataController struct{ DB *sql.DB }
+
 func NewDB() *DataController {
-	db, err := gorm.Open(sqlite.Open("users.db"), &gorm.Config{})
+	config := DataConfig{UserName: "root", Password: "12641264", HostName: "217.12.40.237", DbName: "test"}
+
+	db, err := sql.Open("mysql", config.Get())
 	if err != nil {
-		panic("failed to connect database")
+		panic(err)
 	}
 
-	db.AutoMigrate(models.User{})
+	db.SetConnMaxLifetime(time.Minute * 3)
+	db.SetMaxOpenConns(10)
+	db.SetMaxIdleConns(10)
+
 	return &DataController{DB: db}
 }
 
@@ -29,16 +44,70 @@ func RandRange() int {
 	return rand.Intn(21) - 5
 }
 
+// UpdateUser updates user's dick size and checks if it's blocked
 func (d *DataController) ClearAllBlocked() {
-	d.DB.Model(&models.User{}).Where("is_blocked = ?", true).Update("is_blocked", false)
+	_, err := d.DB.Exec("UPDATE users SET blocked = false")
+	if err != nil {
+		log.Fatal(err)
+	} else {
+		log.Println("Clear block stats complete")
+	}
 }
 
-func (d *DataController) CronRun() {
-	c := cron.New()
-
-	c.AddFunc("0 20 * * *", func() {
-		d.ClearAllBlocked()
-	})
-
-	c.Start()
+// CreateUser creates new user with random dick size and sets it to blocked
+func (d *DataController) CreateUser(id int64, lname string, fname string) {
+	_, err := d.DB.Exec("INSERT INTO users (id, fname, lname, dick_size, admin, blocked, trick) VALUES (?,?,?,?,?,?,?)", id, fname, lname, 0, 0, 0, 1)
+	if err != nil {
+		log.Fatal(err)
+	} else {
+		log.Printf("User %d created", id)
+	}
+}
+
+// GetUser gets user's data by id
+func (d *DataController) GetUser(id int64) *models.User {
+	var user models.User
+	row, err := d.DB.Query("SELECT * FROM users WHERE id =?", id)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for row.Next() {
+		err := row.Scan(&user.ID, &user.FirstName, &user.LastName, &user.DickSize, &user.Admin, &user.Blocked, &user.Trick)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	return &user
+}
+
+// UpdateUser updates user's dick size and blocked status
+func (d *DataController) UpdateDick(id int64, newDickSize int) {
+	_, err := d.DB.Exec("UPDATE users SET dick_size =?, blocked =?, trick =? WHERE id =?", newDickSize, 1, +1, id)
+	if err != nil {
+		log.Fatal(err)
+	} else {
+		log.Printf("User %d dick size updated", id)
+	}
+}
+
+// GetTopUsers gets top 10 users by dick size
+func (d *DataController) GetTopUsers() []models.User {
+	var users []models.User
+	rows, err := d.DB.Query("SELECT * FROM users ORDER BY dick_size DESC LIMIT 10")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for rows.Next() {
+		var user models.User
+		err := rows.Scan(&user.ID, &user.FirstName, &user.LastName, &user.DickSize, &user.Blocked, &user.Admin, &user.Trick)
+		if err != nil {
+			log.Fatal(err)
+		}
+		users = append(users, user)
+	}
+
+	return users
 }