Compare commits
10 Commits
80352e8601
...
07a5c81fb2
| Author | SHA1 | Date |
|---|---|---|
junleea
|
07a5c81fb2 | |
|
junleea
|
5525fe6f2f | |
|
junleea
|
e4e12324fe | |
|
junleea
|
4cbb9b1201 | |
|
junleea
|
832443e1d2 | |
|
junleea
|
0c5cb83f69 | |
|
junleea
|
1573353f49 | |
|
junleea
|
0bf94a0c2b | |
|
junleea
|
66f17ef709 | |
|
junleea
|
98f9424e77 |
|
|
@ -0,0 +1,6 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="Ask2AgentMigrationStateService">
|
||||
<option name="migrationStatus" value="COMPLETED" />
|
||||
</component>
|
||||
</project>
|
||||
23
dao/cid.go
23
dao/cid.go
|
|
@ -4,6 +4,7 @@ import (
|
|||
"fmt"
|
||||
"gorm.io/gorm"
|
||||
"time"
|
||||
"videoplayer/proto"
|
||||
)
|
||||
|
||||
type CID struct {
|
||||
|
|
@ -28,6 +29,28 @@ type CIDRunLog struct {
|
|||
Error string `gorm:"column:error"`
|
||||
}
|
||||
|
||||
func GetDB() *gorm.DB {
|
||||
var db *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db = DB.Debug()
|
||||
} else {
|
||||
db = DB
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func UpdateCIDLastSuccessTime(id int) error {
|
||||
db := GetDB()
|
||||
res := db.Exec("UPDATE c_ids SET last_success = now() where id = ?", id)
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func UpdateCIDLastFailureTime(id int) error {
|
||||
db := GetDB()
|
||||
res := db.Exec("UPDATE c_ids SET last_fail = now() where id = ?", id)
|
||||
return res.Error
|
||||
}
|
||||
|
||||
// CreateCID 创建持续集成、部署
|
||||
func CreateCID(name, url, script, token string, time, auth_id int) uint {
|
||||
cid := CID{Name: name, Url: url, Script: script, Token: token, Auth_id: auth_id, Time: time}
|
||||
|
|
|
|||
13
dao/db.go
13
dao/db.go
|
|
@ -122,6 +122,19 @@ func Init() error {
|
|||
log.Println("sqlrunhistory table:", err)
|
||||
}
|
||||
|
||||
err = db.AutoMigrate(&proto.DNSServer{})
|
||||
if err != nil {
|
||||
log.Println("dnsserver table:", err)
|
||||
}
|
||||
err = db.AutoMigrate(&proto.DNSZone{})
|
||||
if err != nil {
|
||||
log.Println("dnszone table:", err)
|
||||
}
|
||||
err = db.AutoMigrate(&proto.DNSRecord{})
|
||||
if err != nil {
|
||||
log.Println("dnsrecord table:", err)
|
||||
}
|
||||
|
||||
DB = db
|
||||
return err
|
||||
}
|
||||
|
|
|
|||
13
dao/dbm.go
13
dao/dbm.go
|
|
@ -144,7 +144,7 @@ func FindDBRunHistoryByAuthID(auth_id int) ([]proto.SQLRunHistory, error) {
|
|||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("user_id = ?", auth_id).Find(&histories)
|
||||
res := db2.Where("user_id = ?", auth_id).Order("created_at DESC").Limit(100).Find(&histories)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
|
|
@ -159,7 +159,7 @@ func FindDBRunHistoryByAuthIDAndDbId(auth_id int, db_id uint) ([]proto.SQLRunHis
|
|||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("user_id = ? and db_id = ?", auth_id, db_id).Find(&histories)
|
||||
res := db2.Where("user_id = ? and db_id = ?", auth_id, db_id).Order("created_at DESC").Limit(100).Find(&histories)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
|
|
@ -196,7 +196,7 @@ func FindAllSQLRunHistory() ([]proto.SQLRunHistory, error) {
|
|||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Find(&histories)
|
||||
res := db2.Order("created_at DESC").Limit(1000).Find(&histories)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
|
|
@ -247,7 +247,12 @@ func RunSQLWithOrder(sql string, db_ *gorm.DB) (result proto.SQLResult, err erro
|
|||
// 将当前行数据存入 map（便于按列名访问）
|
||||
rowMap := make(map[string]interface{})
|
||||
for i, col := range columns {
|
||||
rowMap[col] = values[i]
|
||||
val := values[i]
|
||||
if b, ok := val.([]byte); ok {
|
||||
rowMap[col] = string(b)
|
||||
} else {
|
||||
rowMap[col] = values[i]
|
||||
}
|
||||
}
|
||||
result.Rows = append(result.Rows, rowMap)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,289 @@
|
|||
package dao
|
||||
|
||||
import (
|
||||
"videoplayer/proto"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// ==================== DNSServer 相关操作 ====================
|
||||
|
||||
func CreateDNSServer(server proto.DNSServer) (uint, error) {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Create(&server)
|
||||
if res.Error != nil {
|
||||
return 0, res.Error
|
||||
}
|
||||
return server.ID, nil
|
||||
}
|
||||
|
||||
func FindDNSServerByID(id uint) (proto.DNSServer, error) {
|
||||
var server proto.DNSServer
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).First(&server)
|
||||
if res.Error != nil {
|
||||
return proto.DNSServer{}, res.Error
|
||||
}
|
||||
return server, nil
|
||||
}
|
||||
|
||||
func FindDNSServerByUserID(userID uint) ([]proto.DNSServer, error) {
|
||||
var servers []proto.DNSServer
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("user_id = ?", userID).Find(&servers)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func FindAllDNSServer() ([]proto.DNSServer, error) {
|
||||
var servers []proto.DNSServer
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Find(&servers)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func UpdateDNSServer(id uint, server *proto.DNSServer) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Model(&proto.DNSServer{}).Where("id = ?", id).Updates(server)
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSServerByID(id uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).Delete(&proto.DNSServer{})
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSServerByUserID(userID uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("user_id = ?", userID).Delete(&proto.DNSServer{})
|
||||
return res.Error
|
||||
}
|
||||
|
||||
// ==================== DNSZone 相关操作 ====================
|
||||
|
||||
func CreateDNSZone(zone proto.DNSZone) (uint, error) {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Create(&zone)
|
||||
if res.Error != nil {
|
||||
return 0, res.Error
|
||||
}
|
||||
return zone.ID, nil
|
||||
}
|
||||
|
||||
func FindDNSZoneByID(id uint) (proto.DNSZone, error) {
|
||||
var zone proto.DNSZone
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).First(&zone)
|
||||
if res.Error != nil {
|
||||
return proto.DNSZone{}, res.Error
|
||||
}
|
||||
return zone, nil
|
||||
}
|
||||
|
||||
func FindDNSZoneByServerID(serverID uint) ([]proto.DNSZone, error) {
|
||||
var zones []proto.DNSZone
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("server_id = ?", serverID).Find(&zones)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return zones, nil
|
||||
}
|
||||
|
||||
func FindAllDNSZone() ([]proto.DNSZone, error) {
|
||||
var zones []proto.DNSZone
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Find(&zones)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return zones, nil
|
||||
}
|
||||
|
||||
func UpdateDNSZone(id uint, zone *proto.DNSZone) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Model(&proto.DNSZone{}).Where("id = ?", id).Updates(zone)
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSZoneByID(id uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).Delete(&proto.DNSZone{})
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSZoneByServerID(serverID uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("server_id = ?", serverID).Delete(&proto.DNSZone{})
|
||||
return res.Error
|
||||
}
|
||||
|
||||
// ==================== DNSRecord 相关操作 ====================
|
||||
|
||||
func CreateDNSRecord(record proto.DNSRecord) (uint, error) {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Create(&record)
|
||||
if res.Error != nil {
|
||||
return 0, res.Error
|
||||
}
|
||||
return record.ID, nil
|
||||
}
|
||||
|
||||
func FindDNSRecordByID(id uint) (proto.DNSRecord, error) {
|
||||
var record proto.DNSRecord
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).First(&record)
|
||||
if res.Error != nil {
|
||||
return proto.DNSRecord{}, res.Error
|
||||
}
|
||||
return record, nil
|
||||
}
|
||||
|
||||
func FindDNSRecordByZoneID(zoneID uint) ([]proto.DNSRecord, error) {
|
||||
var records []proto.DNSRecord
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("zone_id = ?", zoneID).Find(&records)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return records, nil
|
||||
}
|
||||
|
||||
func FindAllDNSRecord() ([]proto.DNSRecord, error) {
|
||||
var records []proto.DNSRecord
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Find(&records)
|
||||
if res.Error != nil {
|
||||
return nil, res.Error
|
||||
}
|
||||
return records, nil
|
||||
}
|
||||
|
||||
func UpdateDNSRecord(id uint, record *proto.DNSRecord) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Model(&proto.DNSRecord{}).Where("id = ?", id).Updates(record)
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSRecordByID(id uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("id = ?", id).Delete(&proto.DNSRecord{})
|
||||
return res.Error
|
||||
}
|
||||
|
||||
func DeleteDNSRecordByZoneID(zoneID uint) error {
|
||||
var db2 *gorm.DB
|
||||
if proto.Config.SERVER_SQL_LOG {
|
||||
db2 = DB.Debug()
|
||||
} else {
|
||||
db2 = DB
|
||||
}
|
||||
res := db2.Where("zone_id = ?", zoneID).Delete(&proto.DNSRecord{})
|
||||
return res.Error
|
||||
}
|
||||
17
go.mod
17
go.mod
|
|
@ -1,6 +1,6 @@
|
|||
module videoplayer
|
||||
|
||||
go 1.21
|
||||
go 1.24.0
|
||||
|
||||
require (
|
||||
github.com/gin-gonic/gin v1.10.0
|
||||
|
|
@ -8,9 +8,11 @@ require (
|
|||
github.com/golang-jwt/jwt v3.2.2+incompatible
|
||||
github.com/google/uuid v1.6.0
|
||||
github.com/gorilla/websocket v1.5.3
|
||||
github.com/miekg/dns v1.1.72
|
||||
github.com/robfig/cron/v3 v3.0.1
|
||||
gorm.io/driver/mysql v1.5.6
|
||||
gorm.io/driver/postgres v1.5.9
|
||||
gorm.io/driver/sqlite v1.6.0
|
||||
gorm.io/gorm v1.30.0
|
||||
)
|
||||
|
||||
|
|
@ -47,12 +49,13 @@ require (
|
|||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||
github.com/ugorji/go/codec v1.2.12 // indirect
|
||||
golang.org/x/arch v0.8.0 // indirect
|
||||
golang.org/x/crypto v0.23.0 // indirect
|
||||
golang.org/x/net v0.25.0 // indirect
|
||||
golang.org/x/sync v0.9.0 // indirect
|
||||
golang.org/x/sys v0.20.0 // indirect
|
||||
golang.org/x/text v0.20.0 // indirect
|
||||
golang.org/x/crypto v0.46.0 // indirect
|
||||
golang.org/x/mod v0.31.0 // indirect
|
||||
golang.org/x/net v0.48.0 // indirect
|
||||
golang.org/x/sync v0.19.0 // indirect
|
||||
golang.org/x/sys v0.39.0 // indirect
|
||||
golang.org/x/text v0.32.0 // indirect
|
||||
golang.org/x/tools v0.40.0 // indirect
|
||||
google.golang.org/protobuf v1.34.1 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
gorm.io/driver/sqlite v1.6.0 // indirect
|
||||
)
|
||||
|
|
|
|||
16
go.sum
16
go.sum
|
|
@ -73,6 +73,8 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
|
|||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
|
||||
github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||
github.com/miekg/dns v1.1.72 h1:vhmr+TF2A3tuoGNkLDFK9zi36F2LS+hKTRW0Uf8kbzI=
|
||||
github.com/miekg/dns v1.1.72/go.mod h1:+EuEPhdHOsfk6Wk5TT2CzssZdqkmFhf8r+aVyDEToIs=
|
||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
|
|
@ -113,20 +115,34 @@ golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
|
|||
golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
|
||||
golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI=
|
||||
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
|
||||
golang.org/x/crypto v0.46.0 h1:cKRW/pmt1pKAfetfu+RCEvjvZkA9RimPbh7bhFjGVBU=
|
||||
golang.org/x/crypto v0.46.0/go.mod h1:Evb/oLKmMraqjZ2iQTwDwvCtJkczlDuTmdJXoZVzqU0=
|
||||
golang.org/x/mod v0.31.0 h1:HaW9xtz0+kOcWKwli0ZXy79Ix+UW/vOfmWI5QVd2tgI=
|
||||
golang.org/x/mod v0.31.0/go.mod h1:43JraMp9cGx1Rx3AqioxrbrhNsLl2l/iNAvuBkrezpg=
|
||||
golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
|
||||
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
|
||||
golang.org/x/net v0.48.0 h1:zyQRTTrjc33Lhh0fBgT/H3oZq9WuvRR5gPC70xpDiQU=
|
||||
golang.org/x/net v0.48.0/go.mod h1:+ndRgGjkh8FGtu1w1FGbEC31if4VrNVMuKTgcAAnQRY=
|
||||
golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o=
|
||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.9.0 h1:fEo0HyrW1GIgZdpbhCRO0PkJajUS5H9IFUztCgEo2jQ=
|
||||
golang.org/x/sync v0.9.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
|
||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=
|
||||
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.39.0 h1:CvCKL8MeisomCi6qNZ+wbb0DN9E5AATixKsvNtMoMFk=
|
||||
golang.org/x/sys v0.39.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=
|
||||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.20.0 h1:gK/Kv2otX8gz+wn7Rmb3vT96ZwuoxnQlY+HlJVj7Qug=
|
||||
golang.org/x/text v0.20.0/go.mod h1:D4IsuqiFMhST5bX19pQ9ikHC2GsaKyk/oF+pn3ducp4=
|
||||
golang.org/x/text v0.32.0 h1:ZD01bjUt1FQ9WJ0ClOL5vxgxOI/sVCNgX1YtKwcY0mU=
|
||||
golang.org/x/text v0.32.0/go.mod h1:o/rUWzghvpD5TXrTIBuJU77MTaN0ljMWE47kxGJQ7jY=
|
||||
golang.org/x/tools v0.40.0 h1:yLkxfA+Qnul4cs9QA3KnlFu0lVmd8JJfoq+E41uSutA=
|
||||
golang.org/x/tools v0.40.0/go.mod h1:Ik/tzLRlbscWpqqMRjyWYDisX8bG13FrdXp3o4Sr9lc=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
|
||||
|
|
|
|||
|
|
@ -1,14 +1,15 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/gin-gonic/gin"
|
||||
"log"
|
||||
"net/http"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
"videoplayer/dao"
|
||||
"videoplayer/proto"
|
||||
|
|
@ -16,6 +17,19 @@ import (
|
|||
"videoplayer/worker"
|
||||
)
|
||||
|
||||
type CIDTask struct {
|
||||
Name string
|
||||
Url string
|
||||
Script string
|
||||
ID int
|
||||
AuthID int
|
||||
}
|
||||
|
||||
var (
|
||||
cidTaskQueue = make(chan CIDTask, 100)
|
||||
cidQueueOnce sync.Once
|
||||
)
|
||||
|
||||
type CIDCreateReq struct {
|
||||
Name string `json:"name" form:"name"`
|
||||
Url string `json:"url" form:"url"`
|
||||
|
|
@ -47,7 +61,18 @@ type CIDUpdateReq struct {
|
|||
// 全局变量，记录是否进行cron定时任务的刷新
|
||||
var cron_count int
|
||||
|
||||
func initCIDTaskQueue() {
|
||||
cidQueueOnce.Do(func() {
|
||||
go func() {
|
||||
for task := range cidTaskQueue {
|
||||
RunShellCID(task.Name, task.Url, task.Script, task.ID, task.AuthID)
|
||||
}
|
||||
}()
|
||||
})
|
||||
}
|
||||
|
||||
func SetUpCIDGroup(router *gin.Engine) {
|
||||
initCIDTaskQueue()
|
||||
cidGroup := router.Group("/cid") //持续集成、部署
|
||||
cidGroup.POST("/create", CreateCID)
|
||||
cidGroup.POST("/delete", DeleteCID)
|
||||
|
|
@ -99,7 +124,13 @@ func RunCID(c *gin.Context) {
|
|||
c.JSON(200, gin.H{"error": "CID not found", "code": proto.OperationFailed, "message": "failed"})
|
||||
return
|
||||
} else {
|
||||
go RunShellCID(cid.Name, cid.Url, cid.Script, req.ID, authID)
|
||||
cidTaskQueue <- CIDTask{
|
||||
Name: cid.Name,
|
||||
Url: cid.Url,
|
||||
Script: cid.Script,
|
||||
ID: req.ID,
|
||||
AuthID: authID,
|
||||
}
|
||||
c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "data": "success"})
|
||||
}
|
||||
} else {
|
||||
|
|
@ -236,7 +267,13 @@ func CIDCallback(c *gin.Context) {
|
|||
return
|
||||
}
|
||||
if res.ID != 0 {
|
||||
go RunShellCID(res.Name, res.Url, res.Script, int(res.ID), res.Auth_id)
|
||||
cidTaskQueue <- CIDTask{
|
||||
Name: res.Name,
|
||||
Url: res.Url,
|
||||
Script: res.Script,
|
||||
ID: int(res.ID),
|
||||
AuthID: res.Auth_id,
|
||||
}
|
||||
resp.Code, resp.Message, resp.Data = proto.SuccessCode, "success", res.Name
|
||||
c.JSON(http.StatusOK, resp)
|
||||
return
|
||||
|
|
@ -260,17 +297,14 @@ echo "end"`
|
|||
start := time.Now()
|
||||
//执行脚本
|
||||
cmd := exec.Command("/bin/bash", "-c", scriptContent)
|
||||
// 使用bytes.Buffer捕获输出
|
||||
var out bytes.Buffer
|
||||
cmd.Stdout = &out
|
||||
err3 := cmd.Run()
|
||||
output, err3 := cmd.CombinedOutput()
|
||||
err3_info := ""
|
||||
if err3 != nil {
|
||||
err3_info = err3.Error()
|
||||
}
|
||||
elapsed := time.Since(start)
|
||||
//fmt.Println("bash content:", scriptContent)
|
||||
dao.CreateRunLog(id, authID, scriptContent, out.String(), err3_info, elapsed.Seconds()) //添加执行日志
|
||||
dao.CreateRunLog(id, authID, scriptContent, string(output), err3_info, elapsed.Seconds()) //添加执行日志
|
||||
}
|
||||
|
||||
func RunShellCID(cid_name, url, script string, id, authID int) {
|
||||
|
|
@ -299,17 +333,18 @@ echo "end"`
|
|||
start := time.Now()
|
||||
//执行脚本
|
||||
cmd := exec.Command("/bin/bash", "-c", scriptContent)
|
||||
// 使用bytes.Buffer捕获输出
|
||||
var out bytes.Buffer
|
||||
cmd.Stdout = &out
|
||||
err3 := cmd.Run()
|
||||
output, err3 := cmd.CombinedOutput()
|
||||
err3_info := ""
|
||||
if err3 != nil {
|
||||
err3_info = err3.Error()
|
||||
err := dao.UpdateCIDLastFailureTime(id)
|
||||
if err != nil {
|
||||
log.Println("[ERROR] update cid:", id, " last fail time err:", err)
|
||||
}
|
||||
}
|
||||
elapsed := time.Since(start)
|
||||
//fmt.Println("bash content:", scriptContent)
|
||||
dao.CreateRunLog(id, authID, scriptContent, out.String(), err3_info, elapsed.Seconds()) //添加执行日志
|
||||
dao.CreateRunLog(id, authID, scriptContent, string(output), err3_info, elapsed.Seconds()) //添加执行日志
|
||||
if err3 != nil {
|
||||
dao.UpdateLastFailByID(id, time.Now())
|
||||
} else {
|
||||
|
|
@ -325,6 +360,13 @@ echo "end"`
|
|||
break
|
||||
}
|
||||
}
|
||||
//更新运行成功时间
|
||||
err := dao.UpdateCIDLastSuccessTime(id)
|
||||
if err != nil {
|
||||
log.Println("[ERROR] update cid:", id, " last success time err:", err)
|
||||
return
|
||||
}
|
||||
|
||||
proto.CID_Running_Map[authID] = user_running_list
|
||||
proto.CID_RunningMutex.Unlock()
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,406 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"videoplayer/proto"
|
||||
"videoplayer/service"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func SetDNSGroup(router *gin.Engine) {
|
||||
dns := router.Group("/dns")
|
||||
|
||||
dns.POST("/server/create", CreateDNSServerHandler)
|
||||
dns.POST("/server/get", GetDNSServerHandler)
|
||||
dns.POST("/server/update", UpdateDNSServerHandler)
|
||||
dns.POST("/server/delete", DeleteDNSServerHandler)
|
||||
dns.POST("/server/start", StartDNSServerHandler)
|
||||
dns.POST("/server/stop", StopDNSServerHandler)
|
||||
dns.POST("/server/restart", RestartDNSServerHandler)
|
||||
dns.POST("/server/status", GetDNSServerStatusHandler)
|
||||
|
||||
dns.POST("/zone/create", CreateDNSZoneHandler)
|
||||
dns.POST("/zone/get", GetDNSZoneHandler)
|
||||
dns.POST("/zone/update", UpdateDNSZoneHandler)
|
||||
dns.POST("/zone/delete", DeleteDNSZoneHandler)
|
||||
|
||||
dns.POST("/record/create", CreateDNSRecordHandler)
|
||||
dns.POST("/record/get", GetDNSRecordHandler)
|
||||
dns.POST("/record/update", UpdateDNSRecordHandler)
|
||||
dns.POST("/record/delete", DeleteDNSRecordHandler)
|
||||
}
|
||||
|
||||
// ==================== DNSServer 相关 Handler ====================
|
||||
|
||||
func CreateDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.CreateDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
server, err := service.CreateDNSServer(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "创建DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "创建DNS服务器成功"
|
||||
resp.Data = server
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func GetDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.GetDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
servers, err := service.GetDNSServerList(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "获取DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "获取DNS服务器成功"
|
||||
resp.Data = servers
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func UpdateDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.UpdateDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
server, err := service.UpdateDNSServer(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "更新DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "更新DNS服务器成功"
|
||||
resp.Data = server
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func DeleteDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.DeleteDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
err := service.DeleteDNSServer(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "删除DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "删除DNS服务器成功"
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// ==================== DNSZone 相关 Handler ====================
|
||||
|
||||
func CreateDNSZoneHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.CreateDNSZoneReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
zone, err := service.CreateDNSZone(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "创建DNS区域失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "创建DNS区域成功"
|
||||
resp.Data = zone
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func GetDNSZoneHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.GetDNSZoneReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
zones, err := service.GetDNSZoneList(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "获取DNS区域失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "获取DNS区域成功"
|
||||
resp.Data = zones
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func UpdateDNSZoneHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.UpdateDNSZoneReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
zone, err := service.UpdateDNSZone(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "更新DNS区域失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "更新DNS区域成功"
|
||||
resp.Data = zone
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func DeleteDNSZoneHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.DeleteDNSZoneReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
err := service.DeleteDNSZone(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "删除DNS区域失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "删除DNS区域成功"
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// ==================== DNSRecord 相关 Handler ====================
|
||||
|
||||
func CreateDNSRecordHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.CreateDNSRecordReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
record, err := service.CreateDNSRecord(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "创建DNS记录失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "创建DNS记录成功"
|
||||
resp.Data = record
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func GetDNSRecordHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := uint(id.(float64))
|
||||
|
||||
var req proto.GetDNSRecordReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
records, err := service.GetDNSRecordList(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "获取DNS记录失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "获取DNS记录成功"
|
||||
resp.Data = records
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func UpdateDNSRecordHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.UpdateDNSRecordReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
record, err := service.UpdateDNSRecord(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "更新DNS记录失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "更新DNS记录成功"
|
||||
resp.Data = record
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
func DeleteDNSRecordHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.DeleteDNSRecordReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
err := service.DeleteDNSRecord(&req, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "删除DNS记录失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "删除DNS记录成功"
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// 启动DNS服务器
|
||||
func StartDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.StartDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
instance, err := service.StartDNSServer(req.ServerID, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "启动DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "启动DNS服务器成功"
|
||||
resp.Data = instance
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// 停止DNS服务器
|
||||
func StopDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.StopDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
err := service.StopDNSServer(req.ServerID, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "停止DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "停止DNS服务器成功"
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// 重启DNS服务器
|
||||
func RestartDNSServerHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.RestartDNSServerReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
instance, err := service.RestartDNSServer(req.ServerID, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "重启DNS服务器失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "重启DNS服务器成功"
|
||||
resp.Data = instance
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// 获取DNS服务器状态
|
||||
func GetDNSServerStatusHandler(c *gin.Context) {
|
||||
id, _ := c.Get("id")
|
||||
userID := int(id.(float64))
|
||||
|
||||
var req proto.GetDNSServerStatusReq
|
||||
var resp proto.GeneralResp
|
||||
if err := c.ShouldBind(&req); err != nil {
|
||||
resp.Code = proto.ParameterError
|
||||
resp.Message = "请求参数解析错误"
|
||||
} else {
|
||||
status, err := service.GetDNSServerStatus(req.ServerID, userID)
|
||||
if err != nil {
|
||||
resp.Code = proto.ErrorCode
|
||||
resp.Message = "获取DNS服务器状态失败: " + err.Error()
|
||||
} else {
|
||||
resp.Code = proto.SuccessCode
|
||||
resp.Message = "获取DNS服务器状态成功"
|
||||
resp.Data = status
|
||||
}
|
||||
}
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
|
@ -35,9 +35,9 @@ type GetFileListReq struct {
|
|||
}
|
||||
|
||||
type SendMailReq struct {
|
||||
Title string `json:"title" form:"title"`
|
||||
Content string `json:"content" form:"content"`
|
||||
To string `json:"to" form:"to"`
|
||||
Title string `json:"title" form:"title" binding:"required"`
|
||||
Content string `json:"content" form:"content" binding:"required"`
|
||||
To string `json:"to" form:"to" binding:"required"`
|
||||
}
|
||||
|
||||
func SetUpToolGroup(router *gin.Engine) {
|
||||
|
|
@ -457,37 +457,30 @@ func SendMail(title, content string) {
|
|||
}
|
||||
}
|
||||
|
||||
func SendMailTool(c *gin.Context) {
|
||||
func RequestGetUserInfo(c *gin.Context) *dao.User {
|
||||
id, _ := c.Get("id")
|
||||
id1 := int(id.(float64))
|
||||
userId := int(id.(float64))
|
||||
user := service.GetUserByIDWithCache(userId)
|
||||
return &user
|
||||
}
|
||||
|
||||
func SendMailTool(c *gin.Context) {
|
||||
user := RequestGetUserInfo(c)
|
||||
var resp proto.GeneralResp
|
||||
var req SendMailReq
|
||||
if err := c.ShouldBind(&req); err == nil {
|
||||
user := dao.FindUserByUserID(id1)
|
||||
if user.ID == 0 {
|
||||
c.JSON(http.StatusOK, gin.H{"error": "user not found", "code": proto.ParameterError, "message": "failed"})
|
||||
return
|
||||
}
|
||||
//目标邮箱地址是否合法
|
||||
if !service.CheckEmail(req.To) {
|
||||
c.JSON(http.StatusOK, gin.H{"error": "email address is invalid", "code": proto.ParameterError, "message": "failed"})
|
||||
return
|
||||
}
|
||||
if req.Title == "" || req.Content == "" {
|
||||
c.JSON(http.StatusOK, gin.H{"error": "title or content is empty", "code": proto.ParameterError, "message": "failed"})
|
||||
return
|
||||
}
|
||||
//发送邮件
|
||||
if user.Role == "admin" {
|
||||
log.Println("send mail req:", req)
|
||||
go service.SendEmail(req.To, req.Title, req.Content)
|
||||
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": "mail will be sent"})
|
||||
resp.Code, resp.Data, resp.Message = proto.SuccessCode, "mail will be sent", "success"
|
||||
} else {
|
||||
c.JSON(http.StatusOK, gin.H{"error": "no send mail permission", "code": proto.PermissionDenied, "message": "failed"})
|
||||
resp.Code, resp.Data, resp.Message = proto.PermissionDenied, "mail will be sent", "error"
|
||||
}
|
||||
} else {
|
||||
c.JSON(http.StatusOK, gin.H{"error": err.Error(), "code": proto.ParameterError, "message": "failed"})
|
||||
resp.Data, resp.Data, resp.Message = proto.ParameterError, "toEmail or title or content request param error", "success"
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, resp)
|
||||
}
|
||||
|
||||
// DownloadProxyHandle 处理下载代理请求
|
||||
|
|
|
|||
14
main.go
14
main.go
|
|
@ -3,9 +3,6 @@ package main
|
|||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/golang-jwt/jwt"
|
||||
"github.com/robfig/cron/v3"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
|
|
@ -17,6 +14,10 @@ import (
|
|||
"videoplayer/proto"
|
||||
"videoplayer/service"
|
||||
"videoplayer/worker"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/golang-jwt/jwt"
|
||||
"github.com/robfig/cron/v3"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
|
@ -49,6 +50,7 @@ func main() {
|
|||
handler.SetUpFileGroup(r) // File
|
||||
handler.SetUpShellGroup(r) // Shell
|
||||
handler.SetDBManageGroup(r) // DBM
|
||||
handler.SetDNSGroup(r) // DNS
|
||||
defer dao.Close()
|
||||
defer worker.CloseRedis()
|
||||
//定时任务
|
||||
|
|
@ -103,6 +105,10 @@ func initConfig(configPath string) {
|
|||
err := proto.ReadConfig(configPath)
|
||||
if err != nil {
|
||||
panic("failed to read config file:" + err.Error())
|
||||
} else {
|
||||
//输出配置文件
|
||||
config_bytes, _ := json.Marshal(proto.Config)
|
||||
log.Println(string(config_bytes))
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -159,7 +165,7 @@ func JWTAuthMiddleware() gin.HandlerFunc {
|
|||
}
|
||||
if tokenString == "" {
|
||||
//c.AbortWithStatus(200)
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Unauthorized", "error": "token is empty", "code": proto.TokenIsNull})
|
||||
c.AbortWithStatusJSON(http.StatusOK, gin.H{"message": "Unauthorized", "error": "token is empty", "code": proto.TokenIsNull})
|
||||
return
|
||||
}
|
||||
if proto.Config.TOKEN_USE_REDIS {
|
||||
|
|
|
|||
|
|
@ -0,0 +1,200 @@
|
|||
package proto
|
||||
|
||||
import (
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
const (
|
||||
DNS_STATUS_STOPPED = 0 // DNS服务器状态：已停止
|
||||
DNS_STATUS_RUNNING = 1 // DNS服务器状态：运行中
|
||||
DNS_STATUS_ERROR = 2 // DNS服务器状态：错误
|
||||
|
||||
RECORD_TYPE_A = 1 // A记录
|
||||
RECORD_TYPE_AAAA = 28 // AAAA记录
|
||||
RECORD_TYPE_CNAME = 5 // CNAME记录
|
||||
RECORD_TYPE_MX = 15 // MX记录
|
||||
RECORD_TYPE_NS = 2 // NS记录
|
||||
RECORD_TYPE_SOA = 6 // SOA记录
|
||||
RECORD_TYPE_TXT = 16 // TXT记录
|
||||
RECORD_TYPE_SRV = 33 // SRV记录
|
||||
)
|
||||
|
||||
type DNSServer struct {
|
||||
gorm.Model
|
||||
UserID uint `gorm:"column:user_id;index" json:"user_id" form:"user_id"` // 用户ID
|
||||
Name string `gorm:"column:name;type:varchar(255)" json:"name" form:"name"` // DNS服务器名称
|
||||
Port uint `gorm:"column:port;default:53" json:"port" form:"port"` // 监听端口，默认53
|
||||
ListenIP string `gorm:"column:listen_ip;type:varchar(255)" json:"listen_ip" form:"listen_ip"` // 监听IP，默认0.0.0.0
|
||||
UpstreamDNS string `gorm:"column:upstream_dns;type:varchar(255)" json:"upstream_dns" form:"upstream_dns"` // 上游DNS服务器，多个用逗号分隔
|
||||
EnableRecursion bool `gorm:"column:enable_recursion;default:true" json:"enable_recursion" form:"enable_recursion"` // 是否启用递归查询
|
||||
Status uint `gorm:"column:status;default:0" json:"status" form:"status"` // 状态：0停止，1运行中，2错误
|
||||
Description string `gorm:"column:description;type:text" json:"description" form:"description"` // 描述
|
||||
Zones []DNSZone `gorm:"foreignKey:ServerID" json:"zones,omitempty"` // 关联的Zone
|
||||
}
|
||||
|
||||
type DNSZone struct {
|
||||
gorm.Model
|
||||
ServerID uint `gorm:"column:server_id;index" json:"server_id" form:"server_id"` // 所属DNS服务器ID
|
||||
Domain string `gorm:"column:domain;type:varchar(255);index" json:"domain" form:"domain"` // 域名，如 example.com
|
||||
SOA_MName string `gorm:"column:soa_mname;type:varchar(255)" json:"soa_mname" form:"soa_mname"` // SOA记录的主域名服务器
|
||||
SOA_RName string `gorm:"column:soa_rname;type:varchar(255)" json:"soa_rname" form:"soa_rname"` // SOA记录的管理员邮箱
|
||||
SOA_Serial uint `gorm:"column:soa_serial;default:1" json:"soa_serial" form:"soa_serial"` // SOA记录的序列号
|
||||
SOA_Refresh uint `gorm:"column:soa_refresh;default:86400" json:"soa_refresh" form:"soa_refresh"` // SOA记录的刷新时间（秒）
|
||||
SOA_Retry uint `gorm:"column:soa_retry;default:7200" json:"soa_retry" form:"soa_retry"` // SOA记录的重试时间（秒）
|
||||
SOA_Expire uint `gorm:"column:soa_expire;default:3600000" json:"soa_expire" form:"soa_expire"` // SOA记录的过期时间（秒）
|
||||
SOA_Minimum uint `gorm:"column:soa_minimum;default:3600" json:"soa_minimum" form:"soa_minimum"` // SOA记录的最小TTL（秒）
|
||||
TTL uint `gorm:"column:ttl;default:3600" json:"ttl" form:"ttl"` // 默认TTL（秒）
|
||||
Description string `gorm:"column:description;type:text" json:"description" form:"description"` // 描述
|
||||
Server DNSServer `gorm:"foreignKey:ServerID" json:"server,omitempty"` // 关联的DNS服务器
|
||||
Records []DNSRecord `gorm:"foreignKey:ZoneID" json:"records,omitempty"` // 关联的记录
|
||||
}
|
||||
|
||||
type DNSRecord struct {
|
||||
gorm.Model
|
||||
ZoneID uint `gorm:"column:zone_id;index" json:"zone_id" form:"zone_id"` // 所属Zone ID
|
||||
Name string `gorm:"column:name;type:varchar(255);index" json:"name" form:"name"` // 记录名称，如 www 或 @
|
||||
Type uint `gorm:"column:type;index" json:"type" form:"type"` // 记录类型：1=A, 28=AAAA, 5=CNAME, 15=MX等
|
||||
Value string `gorm:"column:value;type:text" json:"value" form:"value"` // 记录值
|
||||
TTL uint `gorm:"column:ttl;default:3600" json:"ttl" form:"ttl"` // TTL（秒），如果为0则使用Zone的默认TTL
|
||||
Priority uint `gorm:"column:priority;default:0" json:"priority" form:"priority"` // 优先级（用于MX、SRV等记录）
|
||||
Weight uint `gorm:"column:weight;default:0" json:"weight" form:"weight"` // 权重（用于SRV记录）
|
||||
Port uint `gorm:"column:port;default:0" json:"port" form:"port"` // 端口（用于SRV记录）
|
||||
Target string `gorm:"column:target;type:varchar(255)" json:"target" form:"target"` // 目标（用于SRV记录）
|
||||
Zone DNSZone `gorm:"foreignKey:ZoneID" json:"zone,omitempty"` // 关联的Zone
|
||||
}
|
||||
|
||||
// DNSServer 创建请求
|
||||
type CreateDNSServerReq struct {
|
||||
Name string `json:"name" form:"name"` // DNS服务器名称
|
||||
Port uint `json:"port" form:"port"` // 监听端口，默认53
|
||||
ListenIP string `json:"listen_ip" form:"listen_ip"` // 监听IP，默认0.0.0.0
|
||||
UpstreamDNS string `json:"upstream_dns" form:"upstream_dns"` // 上游DNS服务器，多个用逗号分隔
|
||||
EnableRecursion bool `json:"enable_recursion" form:"enable_recursion"` // 是否启用递归查询
|
||||
Description string `json:"description" form:"description"` // 描述
|
||||
}
|
||||
|
||||
// DNSServer 更新请求
|
||||
type UpdateDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id"` // DNS服务器ID
|
||||
Name string `json:"name" form:"name"` // DNS服务器名称
|
||||
Port uint `json:"port" form:"port"` // 监听端口
|
||||
ListenIP string `json:"listen_ip" form:"listen_ip"` // 监听IP
|
||||
UpstreamDNS string `json:"upstream_dns" form:"upstream_dns"` // 上游DNS服务器
|
||||
EnableRecursion bool `json:"enable_recursion" form:"enable_recursion"` // 是否启用递归查询
|
||||
Status uint `json:"status" form:"status"` // 状态：0停止，1运行中，2错误
|
||||
Description string `json:"description" form:"description"` // 描述
|
||||
}
|
||||
|
||||
// DNSServer 获取请求
|
||||
type GetDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id"` // DNS服务器ID，0表示获取全部
|
||||
GetType int `json:"get_type" form:"get_type"` // 获取类型：0获取自己的，1获取全部（管理员）
|
||||
}
|
||||
|
||||
// DNSServer 删除请求
|
||||
type DeleteDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id"` // DNS服务器ID
|
||||
DelType uint `json:"del_type" form:"del_type"` // 删除类型：0删除单条，1删除所有
|
||||
}
|
||||
|
||||
// DNSZone 创建请求
|
||||
type CreateDNSZoneReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id"` // 所属DNS服务器ID
|
||||
Domain string `json:"domain" form:"domain"` // 域名，如 example.com
|
||||
SOA_MName string `json:"soa_mname" form:"soa_mname"` // SOA记录的主域名服务器
|
||||
SOA_RName string `json:"soa_rname" form:"soa_rname"` // SOA记录的管理员邮箱
|
||||
SOA_Serial uint `json:"soa_serial" form:"soa_serial"` // SOA记录的序列号
|
||||
SOA_Refresh uint `json:"soa_refresh" form:"soa_refresh"` // SOA记录的刷新时间（秒）
|
||||
SOA_Retry uint `json:"soa_retry" form:"soa_retry"` // SOA记录的重试时间（秒）
|
||||
SOA_Expire uint `json:"soa_expire" form:"soa_expire"` // SOA记录的过期时间（秒）
|
||||
SOA_Minimum uint `json:"soa_minimum" form:"soa_minimum"` // SOA记录的最小TTL（秒）
|
||||
TTL uint `json:"ttl" form:"ttl"` // 默认TTL（秒）
|
||||
Description string `json:"description" form:"description"` // 描述
|
||||
}
|
||||
|
||||
// DNSZone 更新请求
|
||||
type UpdateDNSZoneReq struct {
|
||||
ZoneID uint `json:"zone_id" form:"zone_id"` // Zone ID
|
||||
Domain string `json:"domain" form:"domain"` // 域名
|
||||
SOA_MName string `json:"soa_mname" form:"soa_mname"` // SOA记录的主域名服务器
|
||||
SOA_RName string `json:"soa_rname" form:"soa_rname"` // SOA记录的管理员邮箱
|
||||
SOA_Serial uint `json:"soa_serial" form:"soa_serial"` // SOA记录的序列号
|
||||
SOA_Refresh uint `json:"soa_refresh" form:"soa_refresh"` // SOA记录的刷新时间（秒）
|
||||
SOA_Retry uint `json:"soa_retry" form:"soa_retry"` // SOA记录的重试时间（秒）
|
||||
SOA_Expire uint `json:"soa_expire" form:"soa_expire"` // SOA记录的过期时间（秒）
|
||||
SOA_Minimum uint `json:"soa_minimum" form:"soa_minimum"` // SOA记录的最小TTL（秒）
|
||||
TTL uint `json:"ttl" form:"ttl"` // 默认TTL（秒）
|
||||
Description string `json:"description" form:"description"` // 描述
|
||||
}
|
||||
|
||||
// DNSZone 获取请求
|
||||
type GetDNSZoneReq struct {
|
||||
ZoneID uint `json:"zone_id" form:"zone_id"` // Zone ID，0表示获取全部
|
||||
ServerID uint `json:"server_id" form:"server_id"` // DNS服务器ID，用于过滤
|
||||
GetType int `json:"get_type" form:"get_type"` // 获取类型：0获取自己的，1获取全部（管理员）
|
||||
}
|
||||
|
||||
// DNSZone 删除请求
|
||||
type DeleteDNSZoneReq struct {
|
||||
ZoneID uint `json:"zone_id" form:"zone_id"` // Zone ID
|
||||
DelType uint `json:"del_type" form:"del_type"` // 删除类型：0删除单条，1删除所有
|
||||
}
|
||||
|
||||
// DNSRecord 创建请求
|
||||
type CreateDNSRecordReq struct {
|
||||
ZoneID uint `json:"zone_id" form:"zone_id"` // 所属Zone ID
|
||||
Name string `json:"name" form:"name"` // 记录名称，如 www 或 @
|
||||
Type uint `json:"type" form:"type"` // 记录类型：1=A, 28=AAAA, 5=CNAME, 15=MX等
|
||||
Value string `json:"value" form:"value"` // 记录值
|
||||
TTL uint `json:"ttl" form:"ttl"` // TTL（秒）
|
||||
Priority uint `json:"priority" form:"priority"` // 优先级（用于MX、SRV等记录）
|
||||
Weight uint `json:"weight" form:"weight"` // 权重（用于SRV记录）
|
||||
Port uint `json:"port" form:"port"` // 端口（用于SRV记录）
|
||||
Target string `json:"target" form:"target"` // 目标（用于SRV记录）
|
||||
}
|
||||
|
||||
// DNSRecord 更新请求
|
||||
type UpdateDNSRecordReq struct {
|
||||
RecordID uint `json:"record_id" form:"record_id"` // 记录ID
|
||||
Name string `json:"name" form:"name"` // 记录名称
|
||||
Type uint `json:"type" form:"type"` // 记录类型
|
||||
Value string `json:"value" form:"value"` // 记录值
|
||||
TTL uint `json:"ttl" form:"ttl"` // TTL（秒）
|
||||
Priority uint `json:"priority" form:"priority"` // 优先级
|
||||
Weight uint `json:"weight" form:"weight"` // 权重
|
||||
Port uint `json:"port" form:"port"` // 端口
|
||||
Target string `json:"target" form:"target"` // 目标
|
||||
}
|
||||
|
||||
// DNSRecord 获取请求
|
||||
type GetDNSRecordReq struct {
|
||||
RecordID uint `json:"record_id" form:"record_id"` // 记录ID，0表示获取全部
|
||||
ZoneID uint `json:"zone_id" form:"zone_id"` // Zone ID，用于过滤
|
||||
GetType int `json:"get_type" form:"get_type"` // 获取类型：0获取自己的，1获取全部（管理员）
|
||||
}
|
||||
|
||||
// DNSRecord 删除请求
|
||||
type DeleteDNSRecordReq struct {
|
||||
RecordID uint `json:"record_id" form:"record_id"` // 记录ID
|
||||
DelType uint `json:"del_type" form:"del_type"` // 删除类型：0删除单条，1删除所有
|
||||
}
|
||||
|
||||
// 启动DNS服务器请求
|
||||
type StartDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id" binding:"required"` // DNS服务器ID
|
||||
}
|
||||
|
||||
// 停止DNS服务器请求
|
||||
type StopDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id" binding:"required"` // DNS服务器ID
|
||||
}
|
||||
|
||||
// 重启DNS服务器请求
|
||||
type RestartDNSServerReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id" binding:"required"` // DNS服务器ID
|
||||
}
|
||||
|
||||
// 获取DNS服务器状态请求
|
||||
type GetDNSServerStatusReq struct {
|
||||
ServerID uint `json:"server_id" form:"server_id" binding:"required"` // DNS服务器ID
|
||||
}
|
||||
|
|
@ -0,0 +1,786 @@
|
|||
package service
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"sync"
|
||||
"videoplayer/dao"
|
||||
"videoplayer/proto"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
// DNSServiceInstance 运行中的DNS服务实例
|
||||
type DNSServiceInstance struct {
|
||||
ServerID uint `json:"server_id"`
|
||||
Config *proto.DNSServer `json:"config"`
|
||||
Server *dns.Server `json:"-"`
|
||||
Running bool `json:"running"`
|
||||
mutex sync.Mutex `json:"-"`
|
||||
}
|
||||
|
||||
// DNSServiceManager DNS服务管理器
|
||||
type DNSServiceManager struct {
|
||||
instances map[uint]*DNSServiceInstance
|
||||
rwMutex sync.RWMutex
|
||||
}
|
||||
|
||||
var dnsServiceManager *DNSServiceManager
|
||||
|
||||
func init() {
|
||||
dnsServiceManager = &DNSServiceManager{
|
||||
instances: make(map[uint]*DNSServiceInstance),
|
||||
}
|
||||
}
|
||||
|
||||
// ==================== DNSServer 相关操作 ====================
|
||||
|
||||
func CreateDNSServer(req *proto.CreateDNSServerReq, userID uint) (proto.DNSServer, error) {
|
||||
server := proto.DNSServer{
|
||||
UserID: userID,
|
||||
Name: req.Name,
|
||||
Port: req.Port,
|
||||
ListenIP: req.ListenIP,
|
||||
UpstreamDNS: req.UpstreamDNS,
|
||||
EnableRecursion: req.EnableRecursion,
|
||||
Status: proto.DNS_STATUS_STOPPED,
|
||||
Description: req.Description,
|
||||
}
|
||||
|
||||
if server.Port == 0 {
|
||||
server.Port = 53
|
||||
}
|
||||
if server.ListenIP == "" {
|
||||
server.ListenIP = "0.0.0.0"
|
||||
}
|
||||
|
||||
id, err := dao.CreateDNSServer(server)
|
||||
if err != nil {
|
||||
return proto.DNSServer{}, err
|
||||
}
|
||||
server.ID = id
|
||||
return server, nil
|
||||
}
|
||||
|
||||
func GetDNSServerList(req *proto.GetDNSServerReq, userID uint) ([]proto.DNSServer, error) {
|
||||
var servers []proto.DNSServer
|
||||
var err error
|
||||
|
||||
if req.ServerID > 0 {
|
||||
server, err := dao.FindDNSServerByID(req.ServerID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if server.UserID != userID && GetUserByIDFromUserCenter(int(userID)).Role != "admin" {
|
||||
return nil, errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
servers = append(servers, server)
|
||||
} else {
|
||||
if req.GetType == 0 {
|
||||
servers, err = dao.FindDNSServerByUserID(userID)
|
||||
} else if req.GetType == 1 {
|
||||
user := GetUserByIDFromUserCenter(int(userID))
|
||||
if user.Role != "admin" {
|
||||
return nil, errors.New("未授权访问，仅管理员可获取全部DNS服务器")
|
||||
}
|
||||
servers, err = dao.FindAllDNSServer()
|
||||
} else {
|
||||
return nil, errors.New("无效的获取类型")
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func UpdateDNSServer(req *proto.UpdateDNSServerReq, userID int) (proto.DNSServer, error) {
|
||||
server, err := dao.FindDNSServerByID(req.ServerID)
|
||||
if err != nil {
|
||||
return proto.DNSServer{}, err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return proto.DNSServer{}, errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
|
||||
server.Name = req.Name
|
||||
server.Port = req.Port
|
||||
server.ListenIP = req.ListenIP
|
||||
server.UpstreamDNS = req.UpstreamDNS
|
||||
server.EnableRecursion = req.EnableRecursion
|
||||
server.Status = req.Status
|
||||
server.Description = req.Description
|
||||
|
||||
err = dao.UpdateDNSServer(server.ID, &server)
|
||||
if err != nil {
|
||||
return proto.DNSServer{}, err
|
||||
}
|
||||
|
||||
// 如果服务正在运行，自动重启以应用新配置
|
||||
go RestartDNSServerIfRunning(req.ServerID, userID)
|
||||
|
||||
return server, nil
|
||||
}
|
||||
|
||||
func DeleteDNSServer(req *proto.DeleteDNSServerReq, userID int) error {
|
||||
user := GetUserByIDFromUserCenter(userID)
|
||||
if req.DelType == 0 && req.ServerID > 0 {
|
||||
server, err := dao.FindDNSServerByID(req.ServerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if server.UserID != uint(userID) && user.Role != "admin" {
|
||||
return errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
err = dao.DeleteDNSZoneByServerID(req.ServerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = dao.DeleteDNSServerByID(req.ServerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
} else if req.DelType == 1 {
|
||||
if user.Role != "admin" {
|
||||
return errors.New("未授权访问，仅管理员可删除所有DNS服务器")
|
||||
}
|
||||
servers, err := dao.FindAllDNSServer()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, server := range servers {
|
||||
err = dao.DeleteDNSZoneByServerID(server.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = dao.DeleteDNSServerByID(server.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return errors.New("无效的删除类型或参数")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ==================== DNSZone 相关操作 ====================
|
||||
|
||||
func CreateDNSZone(req *proto.CreateDNSZoneReq, userID uint) (proto.DNSZone, error) {
|
||||
server, err := dao.FindDNSServerByID(req.ServerID)
|
||||
if err != nil {
|
||||
return proto.DNSZone{}, err
|
||||
}
|
||||
if server.UserID != userID && GetUserByIDFromUserCenter(int(userID)).Role != "admin" {
|
||||
return proto.DNSZone{}, errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
|
||||
zone := proto.DNSZone{
|
||||
ServerID: req.ServerID,
|
||||
Domain: req.Domain,
|
||||
SOA_MName: req.SOA_MName,
|
||||
SOA_RName: req.SOA_RName,
|
||||
SOA_Serial: req.SOA_Serial,
|
||||
SOA_Refresh: req.SOA_Refresh,
|
||||
SOA_Retry: req.SOA_Retry,
|
||||
SOA_Expire: req.SOA_Expire,
|
||||
SOA_Minimum: req.SOA_Minimum,
|
||||
TTL: req.TTL,
|
||||
Description: req.Description,
|
||||
}
|
||||
|
||||
if zone.SOA_Serial == 0 {
|
||||
zone.SOA_Serial = 1
|
||||
}
|
||||
if zone.SOA_Refresh == 0 {
|
||||
zone.SOA_Refresh = 86400
|
||||
}
|
||||
if zone.SOA_Retry == 0 {
|
||||
zone.SOA_Retry = 7200
|
||||
}
|
||||
if zone.SOA_Expire == 0 {
|
||||
zone.SOA_Expire = 3600000
|
||||
}
|
||||
if zone.SOA_Minimum == 0 {
|
||||
zone.SOA_Minimum = 3600
|
||||
}
|
||||
if zone.TTL == 0 {
|
||||
zone.TTL = 3600
|
||||
}
|
||||
|
||||
id, err := dao.CreateDNSZone(zone)
|
||||
if err != nil {
|
||||
return proto.DNSZone{}, err
|
||||
}
|
||||
zone.ID = id
|
||||
return zone, nil
|
||||
}
|
||||
|
||||
func GetDNSZoneList(req *proto.GetDNSZoneReq, userID uint) ([]proto.DNSZone, error) {
|
||||
var zones []proto.DNSZone
|
||||
var err error
|
||||
|
||||
if req.ZoneID > 0 {
|
||||
zone, err := dao.FindDNSZoneByID(req.ZoneID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if server.UserID != userID && GetUserByIDFromUserCenter(int(userID)).Role != "admin" {
|
||||
return nil, errors.New("未授权访问DNS区域")
|
||||
}
|
||||
zones = append(zones, zone)
|
||||
} else {
|
||||
if req.GetType == 0 {
|
||||
if req.ServerID > 0 {
|
||||
zones, err = dao.FindDNSZoneByServerID(req.ServerID)
|
||||
} else {
|
||||
servers, err := dao.FindDNSServerByUserID(userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, server := range servers {
|
||||
serverZones, err := dao.FindDNSZoneByServerID(server.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
zones = append(zones, serverZones...)
|
||||
}
|
||||
}
|
||||
} else if req.GetType == 1 {
|
||||
user := GetUserByIDFromUserCenter(int(userID))
|
||||
if user.Role != "admin" {
|
||||
return nil, errors.New("未授权访问，仅管理员可获取全部DNS区域")
|
||||
}
|
||||
if req.ServerID > 0 {
|
||||
zones, err = dao.FindDNSZoneByServerID(req.ServerID)
|
||||
} else {
|
||||
zones, err = dao.FindAllDNSZone()
|
||||
}
|
||||
} else {
|
||||
return nil, errors.New("无效的获取类型")
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return zones, nil
|
||||
}
|
||||
|
||||
func UpdateDNSZone(req *proto.UpdateDNSZoneReq, userID int) (proto.DNSZone, error) {
|
||||
zone, err := dao.FindDNSZoneByID(req.ZoneID)
|
||||
if err != nil {
|
||||
return proto.DNSZone{}, err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return proto.DNSZone{}, err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return proto.DNSZone{}, errors.New("未授权访问DNS区域")
|
||||
}
|
||||
|
||||
zone.Domain = req.Domain
|
||||
zone.SOA_MName = req.SOA_MName
|
||||
zone.SOA_RName = req.SOA_RName
|
||||
zone.SOA_Serial = req.SOA_Serial
|
||||
zone.SOA_Refresh = req.SOA_Refresh
|
||||
zone.SOA_Retry = req.SOA_Retry
|
||||
zone.SOA_Expire = req.SOA_Expire
|
||||
zone.SOA_Minimum = req.SOA_Minimum
|
||||
zone.TTL = req.TTL
|
||||
zone.Description = req.Description
|
||||
|
||||
err = dao.UpdateDNSZone(zone.ID, &zone)
|
||||
if err != nil {
|
||||
return proto.DNSZone{}, err
|
||||
}
|
||||
return zone, nil
|
||||
}
|
||||
|
||||
func DeleteDNSZone(req *proto.DeleteDNSZoneReq, userID int) error {
|
||||
user := GetUserByIDFromUserCenter(userID)
|
||||
if req.DelType == 0 && req.ZoneID > 0 {
|
||||
zone, err := dao.FindDNSZoneByID(req.ZoneID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if server.UserID != uint(userID) && user.Role != "admin" {
|
||||
return errors.New("未授权访问DNS区域")
|
||||
}
|
||||
err = dao.DeleteDNSRecordByZoneID(req.ZoneID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = dao.DeleteDNSZoneByID(req.ZoneID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
} else if req.DelType == 1 {
|
||||
if user.Role != "admin" {
|
||||
return errors.New("未授权访问，仅管理员可删除所有DNS区域")
|
||||
}
|
||||
zones, err := dao.FindAllDNSZone()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, zone := range zones {
|
||||
err = dao.DeleteDNSRecordByZoneID(zone.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = dao.DeleteDNSZoneByID(zone.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return errors.New("无效的删除类型或参数")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ==================== DNSRecord 相关操作 ====================
|
||||
|
||||
func CreateDNSRecord(req *proto.CreateDNSRecordReq, userID uint) (proto.DNSRecord, error) {
|
||||
zone, err := dao.FindDNSZoneByID(req.ZoneID)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
if server.UserID != userID && GetUserByIDFromUserCenter(int(userID)).Role != "admin" {
|
||||
return proto.DNSRecord{}, errors.New("未授权访问DNS区域")
|
||||
}
|
||||
|
||||
record := proto.DNSRecord{
|
||||
ZoneID: req.ZoneID,
|
||||
Name: req.Name,
|
||||
Type: req.Type,
|
||||
Value: req.Value,
|
||||
TTL: req.TTL,
|
||||
Priority: req.Priority,
|
||||
Weight: req.Weight,
|
||||
Port: req.Port,
|
||||
Target: req.Target,
|
||||
}
|
||||
|
||||
if record.TTL == 0 {
|
||||
record.TTL = zone.TTL
|
||||
}
|
||||
|
||||
id, err := dao.CreateDNSRecord(record)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
record.ID = id
|
||||
return record, nil
|
||||
}
|
||||
|
||||
func GetDNSRecordList(req *proto.GetDNSRecordReq, userID uint) ([]proto.DNSRecord, error) {
|
||||
var records []proto.DNSRecord
|
||||
var err error
|
||||
|
||||
if req.RecordID > 0 {
|
||||
record, err := dao.FindDNSRecordByID(req.RecordID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
zone, err := dao.FindDNSZoneByID(record.ZoneID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if server.UserID != userID && GetUserByIDFromUserCenter(int(userID)).Role != "admin" {
|
||||
return nil, errors.New("未授权访问DNS记录")
|
||||
}
|
||||
records = append(records, record)
|
||||
} else {
|
||||
if req.GetType == 0 {
|
||||
if req.ZoneID > 0 {
|
||||
records, err = dao.FindDNSRecordByZoneID(req.ZoneID)
|
||||
} else {
|
||||
servers, err := dao.FindDNSServerByUserID(userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, server := range servers {
|
||||
zones, err := dao.FindDNSZoneByServerID(server.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, zone := range zones {
|
||||
zoneRecords, err := dao.FindDNSRecordByZoneID(zone.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
records = append(records, zoneRecords...)
|
||||
}
|
||||
}
|
||||
}
|
||||
} else if req.GetType == 1 {
|
||||
user := GetUserByIDFromUserCenter(int(userID))
|
||||
if user.Role != "admin" {
|
||||
return nil, errors.New("未授权访问，仅管理员可获取全部DNS记录")
|
||||
}
|
||||
if req.ZoneID > 0 {
|
||||
records, err = dao.FindDNSRecordByZoneID(req.ZoneID)
|
||||
} else {
|
||||
records, err = dao.FindAllDNSRecord()
|
||||
}
|
||||
} else {
|
||||
return nil, errors.New("无效的获取类型")
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return records, nil
|
||||
}
|
||||
|
||||
func UpdateDNSRecord(req *proto.UpdateDNSRecordReq, userID int) (proto.DNSRecord, error) {
|
||||
record, err := dao.FindDNSRecordByID(req.RecordID)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
zone, err := dao.FindDNSZoneByID(record.ZoneID)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return proto.DNSRecord{}, errors.New("未授权访问DNS记录")
|
||||
}
|
||||
|
||||
record.Name = req.Name
|
||||
record.Type = req.Type
|
||||
record.Value = req.Value
|
||||
record.TTL = req.TTL
|
||||
record.Priority = req.Priority
|
||||
record.Weight = req.Weight
|
||||
record.Port = req.Port
|
||||
record.Target = req.Target
|
||||
|
||||
err = dao.UpdateDNSRecord(record.ID, &record)
|
||||
if err != nil {
|
||||
return proto.DNSRecord{}, err
|
||||
}
|
||||
return record, nil
|
||||
}
|
||||
|
||||
func DeleteDNSRecord(req *proto.DeleteDNSRecordReq, userID int) error {
|
||||
user := GetUserByIDFromUserCenter(userID)
|
||||
if req.DelType == 0 && req.RecordID > 0 {
|
||||
record, err := dao.FindDNSRecordByID(req.RecordID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
zone, err := dao.FindDNSZoneByID(record.ZoneID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
server, err := dao.FindDNSServerByID(zone.ServerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if server.UserID != uint(userID) && user.Role != "admin" {
|
||||
return errors.New("未授权访问DNS记录")
|
||||
}
|
||||
err = dao.DeleteDNSRecordByID(req.RecordID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
} else if req.DelType == 1 {
|
||||
if user.Role != "admin" {
|
||||
return errors.New("未授权访问，仅管理员可删除所有DNS记录")
|
||||
}
|
||||
records, err := dao.FindAllDNSRecord()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, record := range records {
|
||||
err = dao.DeleteDNSRecordByID(record.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return errors.New("无效的删除类型或参数")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ==================== DNS 服务运行管理 ====================
|
||||
|
||||
// handleDNSRequest 处理DNS查询请求
|
||||
func (instance *DNSServiceInstance) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
|
||||
m := new(dns.Msg)
|
||||
m.SetReply(r)
|
||||
m.Authoritative = true
|
||||
|
||||
for _, q := range r.Question {
|
||||
// 查找对应的Zone
|
||||
zones, err := dao.FindDNSZoneByServerID(instance.ServerID)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
var matchedZone *proto.DNSZone
|
||||
var qName string = dns.Fqdn(q.Name)
|
||||
|
||||
for _, zone := range zones {
|
||||
zoneFqdn := dns.Fqdn(zone.Domain)
|
||||
if dns.IsSubDomain(zoneFqdn, qName) {
|
||||
matchedZone = &zone
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if matchedZone == nil {
|
||||
// 递归查询
|
||||
if instance.Config.EnableRecursion {
|
||||
c := new(dns.Client)
|
||||
upstreams := instance.Config.UpstreamDNS
|
||||
if upstreams == "" {
|
||||
upstreams = "8.8.8.8:53"
|
||||
}
|
||||
resp, _, err := c.Exchange(r, upstreams)
|
||||
if err == nil {
|
||||
m.Answer = append(m.Answer, resp.Answer...)
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
|
||||
// 查找匹配的记录
|
||||
records, err := dao.FindDNSRecordByZoneID(matchedZone.ID)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
for _, record := range records {
|
||||
recordName := dns.Fqdn(record.Name + "." + matchedZone.Domain)
|
||||
if record.Name == "@" {
|
||||
recordName = dns.Fqdn(matchedZone.Domain)
|
||||
}
|
||||
|
||||
if recordName == qName && record.Type == uint(q.Qtype) {
|
||||
var rr dns.RR
|
||||
var ttl uint32
|
||||
if record.TTL > 0 {
|
||||
ttl = uint32(record.TTL)
|
||||
} else {
|
||||
ttl = uint32(matchedZone.TTL)
|
||||
}
|
||||
|
||||
switch q.Qtype {
|
||||
case dns.TypeA:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN A %s", q.Name, ttl, record.Value))
|
||||
case dns.TypeAAAA:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN AAAA %s", q.Name, ttl, record.Value))
|
||||
case dns.TypeCNAME:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN CNAME %s", q.Name, ttl, record.Value))
|
||||
case dns.TypeMX:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN MX %d %s", q.Name, ttl, record.Priority, record.Value))
|
||||
case dns.TypeNS:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN NS %s", q.Name, ttl, record.Value))
|
||||
case dns.TypeTXT:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN TXT \"%s\"", q.Name, ttl, record.Value))
|
||||
case dns.TypeSRV:
|
||||
rr, _ = dns.NewRR(fmt.Sprintf("%s %d IN SRV %d %d %d %s", q.Name, ttl, record.Priority, record.Weight, record.Port, record.Target))
|
||||
}
|
||||
|
||||
if rr != nil {
|
||||
m.Answer = append(m.Answer, rr)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
w.WriteMsg(m)
|
||||
}
|
||||
|
||||
// StartDNSServer 启动DNS服务
|
||||
func StartDNSServer(serverID uint, userID int) (*DNSServiceInstance, error) {
|
||||
// 检查权限
|
||||
server, err := dao.FindDNSServerByID(serverID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return nil, errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
|
||||
dnsServiceManager.rwMutex.Lock()
|
||||
defer dnsServiceManager.rwMutex.Unlock()
|
||||
|
||||
// 检查是否已经在运行
|
||||
if instance, exists := dnsServiceManager.instances[serverID]; exists && instance.Running {
|
||||
return instance, nil
|
||||
}
|
||||
|
||||
// 创建服务实例
|
||||
instance := &DNSServiceInstance{
|
||||
ServerID: serverID,
|
||||
Config: &server,
|
||||
Running: false,
|
||||
}
|
||||
|
||||
// 创建DNS服务器
|
||||
dns.HandleFunc(".", instance.handleDNSRequest)
|
||||
|
||||
addr := fmt.Sprintf("%s:%d", server.ListenIP, server.Port)
|
||||
udpAddr, err := net.ResolveUDPAddr("udp", addr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
udpConn, err := net.ListenUDP("udp", udpAddr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
tcpAddr, err := net.ResolveTCPAddr("tcp", addr)
|
||||
if err != nil {
|
||||
udpConn.Close()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
tcpListener, err := net.ListenTCP("tcp", tcpAddr)
|
||||
if err != nil {
|
||||
udpConn.Close()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
instance.Server = &dns.Server{
|
||||
Listener: tcpListener,
|
||||
PacketConn: udpConn,
|
||||
}
|
||||
|
||||
// 启动服务
|
||||
go func() {
|
||||
instance.mutex.Lock()
|
||||
instance.Running = true
|
||||
instance.mutex.Unlock()
|
||||
|
||||
err := instance.Server.ActivateAndServe()
|
||||
if err != nil {
|
||||
instance.mutex.Lock()
|
||||
instance.Running = false
|
||||
instance.mutex.Unlock()
|
||||
}
|
||||
}()
|
||||
|
||||
// 更新数据库状态
|
||||
server.Status = proto.DNS_STATUS_RUNNING
|
||||
dao.UpdateDNSServer(serverID, &server)
|
||||
|
||||
dnsServiceManager.instances[serverID] = instance
|
||||
return instance, nil
|
||||
}
|
||||
|
||||
// StopDNSServer 停止DNS服务
|
||||
func StopDNSServer(serverID uint, userID int) error {
|
||||
server, err := dao.FindDNSServerByID(serverID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
|
||||
dnsServiceManager.rwMutex.Lock()
|
||||
defer dnsServiceManager.rwMutex.Unlock()
|
||||
|
||||
instance, exists := dnsServiceManager.instances[serverID]
|
||||
if !exists || !instance.Running {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 停止服务
|
||||
err = instance.Server.Shutdown()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
instance.mutex.Lock()
|
||||
instance.Running = false
|
||||
instance.mutex.Unlock()
|
||||
|
||||
// 更新数据库状态
|
||||
server.Status = proto.DNS_STATUS_STOPPED
|
||||
dao.UpdateDNSServer(serverID, &server)
|
||||
|
||||
delete(dnsServiceManager.instances, serverID)
|
||||
return nil
|
||||
}
|
||||
|
||||
// RestartDNSServer 重启DNS服务
|
||||
func RestartDNSServer(serverID uint, userID int) (*DNSServiceInstance, error) {
|
||||
err := StopDNSServer(serverID, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return StartDNSServer(serverID, userID)
|
||||
}
|
||||
|
||||
// GetDNSServerStatus 获取DNS服务运行状态
|
||||
func GetDNSServerStatus(serverID uint, userID int) (*DNSServiceInstance, error) {
|
||||
server, err := dao.FindDNSServerByID(serverID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if server.UserID != uint(userID) && GetUserByIDFromUserCenter(userID).Role != "admin" {
|
||||
return nil, errors.New("未授权访问DNS服务器")
|
||||
}
|
||||
|
||||
dnsServiceManager.rwMutex.RLock()
|
||||
defer dnsServiceManager.rwMutex.RUnlock()
|
||||
|
||||
instance, exists := dnsServiceManager.instances[serverID]
|
||||
if exists {
|
||||
return instance, nil
|
||||
}
|
||||
|
||||
// 不在运行中，返回配置中的状态
|
||||
return &DNSServiceInstance{
|
||||
ServerID: serverID,
|
||||
Config: &server,
|
||||
Running: server.Status == proto.DNS_STATUS_RUNNING,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// RestartDNSServerIfRunning 如果服务正在运行则重启（配置更新时调用）
|
||||
func RestartDNSServerIfRunning(serverID uint, userID int) error {
|
||||
dnsServiceManager.rwMutex.RLock()
|
||||
instance, exists := dnsServiceManager.instances[serverID]
|
||||
dnsServiceManager.rwMutex.RUnlock()
|
||||
|
||||
if exists && instance.Running {
|
||||
// 获取最新配置
|
||||
server, err := dao.FindDNSServerByID(serverID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// 重启服务（管理员权限）
|
||||
_, err = RestartDNSServer(server.ID, userID) // 假设管理员ID为1
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -1,7 +1,6 @@
|
|||
package service
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"log"
|
||||
"os/exec"
|
||||
|
|
@ -130,13 +129,10 @@ func GetShellWillRunFromMaster(server string) ([]dao.Shell, error) {
|
|||
|
||||
func RunShell(script string) (res, err string) {
|
||||
cmd := exec.Command("/bin/bash", "-c", script)
|
||||
// 使用bytes.Buffer捕获输出
|
||||
var out bytes.Buffer
|
||||
cmd.Stdout = &out
|
||||
err3 := cmd.Run()
|
||||
output, err3 := cmd.CombinedOutput()
|
||||
err3_info := ""
|
||||
if err3 != nil {
|
||||
err3_info = err3.Error()
|
||||
}
|
||||
return strings.TrimSpace(out.String()), err3_info
|
||||
return strings.TrimSpace(string(output)), err3_info
|
||||
}
|
||||
|
|
|
|||
|
|
@ -104,6 +104,10 @@ func SendEmail(email, subject, body string) {
|
|||
em.SmtpUserName = "354425203@qq.com"
|
||||
em.SmtpPort = 587
|
||||
em.ImapPort = 993
|
||||
if email == "" || subject == "" || body == "" {
|
||||
log.Println("tool send email fail: email or subject or body is empty")
|
||||
return
|
||||
}
|
||||
err := em.Send(subject, body, []string{email})
|
||||
if err != nil {
|
||||
fmt.Println("send mail error:", err)
|
||||
|
|
|
|||
Loading…
Reference in New Issue