2
0
mirror of https://github.com/acepanel/panel.git synced 2026-02-04 03:07:20 +08:00

feat: 添加redis驱动

This commit is contained in:
耗子
2024-11-09 02:41:47 +08:00
parent 8e74428023
commit 04d570197a
6 changed files with 172 additions and 1 deletions

1
go.mod
View File

@@ -20,6 +20,7 @@ require (
github.com/go-resty/resty/v2 v2.15.3
github.com/go-sql-driver/mysql v1.8.1
github.com/golang-cz/httplog v0.0.0-20241002114323-98e09d6f537a
github.com/gomodule/redigo v1.9.2
github.com/gorilla/websocket v1.5.3
github.com/hashicorp/go-version v1.7.0
github.com/knadh/koanf/parsers/yaml v0.1.0

2
go.sum
View File

@@ -61,6 +61,8 @@ github.com/gofiber/schema v1.2.0 h1:j+ZRrNnUa/0ZuWrn/6kAtAufEr4jCJ+JuTURAMxNSZg=
github.com/gofiber/schema v1.2.0/go.mod h1:YYwj01w3hVfaNjhtJzaqetymL56VW642YS3qZPhuE6c=
github.com/golang-cz/httplog v0.0.0-20241002114323-98e09d6f537a h1:BAyyIK6rc6Tq9vT2KVlPJXD8EVv/LpaFwn4fgfLwtHg=
github.com/golang-cz/httplog v0.0.0-20241002114323-98e09d6f537a/go.mod h1:bgk4Ij/0OQ89UeoFFAQrSNhbbr4rKJ0fwWfo7wc+TCc=
github.com/gomodule/redigo v1.9.2 h1:HrutZBLhSIU8abiSfW8pj8mPhOyMYjZT/wcA4/L9L9s=
github.com/gomodule/redigo v1.9.2/go.mod h1:KsU3hiK/Ay8U42qpaJk+kuNa3C+spxapWpM+ywhcgtw=
github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
github.com/google/gofuzz v1.2.0 h1:xRy4A+RhZaiKjJ1bPfwQ8sedCA+YS2YcCHW6ec7JMi0=

View File

@@ -1 +1,66 @@
package data
import (
"github.com/TheTNB/panel/internal/app"
"github.com/TheTNB/panel/internal/biz"
"github.com/TheTNB/panel/internal/http/request"
)
type databaseRepo struct{}
func NewDatabaseRepo() biz.DatabaseRepo {
return &databaseRepo{}
}
func (d databaseRepo) Count() (int64, error) {
var count int64
if err := app.Orm.Model(&biz.Database{}).Count(&count).Error; err != nil {
return 0, err
}
return count, nil
}
func (d databaseRepo) List(page, limit uint) ([]*biz.Database, int64, error) {
var database []*biz.Database
var total int64
err := app.Orm.Model(&biz.Database{}).Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&database).Error
return database, total, err
}
func (d databaseRepo) Get(id uint) (*biz.Database, error) {
database := new(biz.Database)
if err := app.Orm.Where("id = ?", id).First(database).Error; err != nil {
return nil, err
}
return database, nil
}
func (d databaseRepo) Create(req *request.DatabaseCreate) error {
database := &biz.Database{
Name: req.Name,
Username: req.Username,
Password: req.Password,
ServerID: req.ServerID,
Status: biz.DatabaseStatusInvalid,
Remark: req.Remark,
}
return app.Orm.Create(database).Error
}
func (d databaseRepo) Update(req *request.DatabaseUpdate) error {
database := &biz.Database{
Name: req.Name,
Username: req.Username,
Password: req.Password,
Remark: req.Remark,
}
return app.Orm.Model(database).Where("id = ?", req.ID).Updates(database).Error
}
func (d databaseRepo) Delete(id uint) error {
return app.Orm.Delete(&biz.Database{}, id).Error
}

View File

@@ -1 +1,70 @@
package data
import (
"github.com/TheTNB/panel/internal/app"
"github.com/TheTNB/panel/internal/biz"
"github.com/TheTNB/panel/internal/http/request"
)
type databaseServerRepo struct{}
func NewDatabaseServerRepo() biz.DatabaseServerRepo {
return &databaseServerRepo{}
}
func (d databaseServerRepo) Count() (int64, error) {
var count int64
if err := app.Orm.Model(&biz.DatabaseServer{}).Count(&count).Error; err != nil {
return 0, err
}
return count, nil
}
func (d databaseServerRepo) List(page, limit uint) ([]*biz.DatabaseServer, int64, error) {
var databaseServer []*biz.DatabaseServer
var total int64
err := app.Orm.Model(&biz.DatabaseServer{}).Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&databaseServer).Error
return databaseServer, total, err
}
func (d databaseServerRepo) Get(id uint) (*biz.DatabaseServer, error) {
databaseServer := new(biz.DatabaseServer)
if err := app.Orm.Where("id = ?", id).First(databaseServer).Error; err != nil {
return nil, err
}
return databaseServer, nil
}
func (d databaseServerRepo) Create(req *request.DatabaseServerCreate) error {
databaseServer := &biz.DatabaseServer{
Name: req.Name,
Type: biz.DatabaseType(req.Type),
Host: req.Host,
Port: req.Port,
Username: req.Username,
Password: req.Password,
Remark: req.Remark,
}
return app.Orm.Create(databaseServer).Error
}
func (d databaseServerRepo) Update(req *request.DatabaseServerCreate) error {
databaseServer := &biz.DatabaseServer{
Name: req.Name,
Type: biz.DatabaseType(req.Type),
Host: req.Host,
Port: req.Port,
Username: req.Username,
Password: req.Password,
Remark: req.Remark,
}
return app.Orm.Save(databaseServer).Error
}
func (d databaseServerRepo) Delete(id uint) error {
return app.Orm.Delete(&biz.DatabaseServer{}, id).Error
}

View File

@@ -1,7 +1,7 @@
package request
type DatabaseCreate struct {
ServerID string `form:"server_id" json:"server_id" validate:"required,exists=database_servers id"`
ServerID uint `form:"server_id" json:"server_id" validate:"required,exists=database_servers id"`
Name string `form:"name" json:"name" validate:"required"`
Username string `form:"username" json:"username"`
Password string `form:"password" json:"password"`

34
pkg/db/redis.go Normal file
View File

@@ -0,0 +1,34 @@
package db
import (
"github.com/gomodule/redigo/redis"
)
type Redis struct {
conn redis.Conn
username string
password string
address string
}
func NewRedis(username, password, address string) (*Redis, error) {
conn, err := redis.Dial("tcp", address, redis.DialUsername(username), redis.DialPassword(password))
if err != nil {
return nil, err
}
return &Redis{
conn: conn,
username: username,
password: password,
address: address,
}, nil
}
func (r *Redis) Close() error {
return r.conn.Close()
}
func (r *Redis) Do(command string, args ...any) (any, error) {
return r.conn.Do(command, args...)
}