gofmt
This commit is contained in:
haibo.chen
@ -1,152 +1,152 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"sync"
|
||||
|
||||
"github.com/ossrs/srs-sip/pkg/models"
|
||||
_ "modernc.org/sqlite"
|
||||
)
|
||||
|
||||
var (
|
||||
instance *MediaServerDB
|
||||
once sync.Once
|
||||
)
|
||||
|
||||
type MediaServerDB struct {
|
||||
models.MediaServerResponse
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// GetInstance 返回 MediaServerDB 的单例实例
|
||||
func GetInstance(dbPath string) (*MediaServerDB, error) {
|
||||
var err error
|
||||
once.Do(func() {
|
||||
instance, err = NewMediaServerDB(dbPath)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return instance, nil
|
||||
}
|
||||
|
||||
func NewMediaServerDB(dbPath string) (*MediaServerDB, error) {
|
||||
db, err := sql.Open("sqlite", dbPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 创建媒体服务器表
|
||||
_, err = db.Exec(`
|
||||
CREATE TABLE IF NOT EXISTS media_servers (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
type TEXT NOT NULL,
|
||||
name TEXT NOT NULL,
|
||||
ip TEXT NOT NULL,
|
||||
port INTEGER NOT NULL,
|
||||
username TEXT,
|
||||
password TEXT,
|
||||
secret TEXT,
|
||||
is_default INTEGER NOT NULL DEFAULT 0,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &MediaServerDB{db: db}, nil
|
||||
}
|
||||
|
||||
// GetMediaServerByNameAndIP 根据名称和IP查询媒体服务器
|
||||
func (m *MediaServerDB) GetMediaServerByNameAndIP(name, ip string) (*models.MediaServerResponse, error) {
|
||||
var ms models.MediaServerResponse
|
||||
err := m.db.QueryRow(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers WHERE name = ? AND ip = ?
|
||||
`, name, ip).Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ms, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) AddMediaServer(name, serverType, ip string, port int, username, password, secret string, isDefault int) error {
|
||||
_, err := m.db.Exec(`
|
||||
INSERT INTO media_servers (name, type, ip, port, username, password, secret, is_default)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`, name, serverType, ip, port, username, password, secret, isDefault)
|
||||
return err
|
||||
}
|
||||
|
||||
// AddOrUpdateMediaServer 添加或更新媒体服务器（如果已存在则更新）
|
||||
func (m *MediaServerDB) AddOrUpdateMediaServer(name, serverType, ip string, port int, username, password, secret string, isDefault int) error {
|
||||
// 检查是否已存在
|
||||
existing, err := m.GetMediaServerByNameAndIP(name, ip)
|
||||
if err == nil && existing != nil {
|
||||
// 已存在，更新记录
|
||||
_, err = m.db.Exec(`
|
||||
UPDATE media_servers
|
||||
SET type = ?, port = ?, username = ?, password = ?, secret = ?, is_default = ?
|
||||
WHERE name = ? AND ip = ?
|
||||
`, serverType, port, username, password, secret, isDefault, name, ip)
|
||||
return err
|
||||
}
|
||||
|
||||
// 不存在，插入新记录
|
||||
return m.AddMediaServer(name, serverType, ip, port, username, password, secret, isDefault)
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) DeleteMediaServer(id int) error {
|
||||
_, err := m.db.Exec("DELETE FROM media_servers WHERE id = ?", id)
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) GetMediaServer(id int) (*models.MediaServerResponse, error) {
|
||||
var ms models.MediaServerResponse
|
||||
err := m.db.QueryRow(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers WHERE id = ?
|
||||
`, id).Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ms, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) ListMediaServers() ([]models.MediaServerResponse, error) {
|
||||
rows, err := m.db.Query(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers ORDER BY created_at DESC
|
||||
`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var servers []models.MediaServerResponse
|
||||
for rows.Next() {
|
||||
var ms models.MediaServerResponse
|
||||
err := rows.Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
servers = append(servers, ms)
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) SetDefaultMediaServer(id int) error {
|
||||
// 先将所有服务器设置为非默认
|
||||
if _, err := m.db.Exec("UPDATE media_servers SET is_default = 0"); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 将指定ID的服务器设置为默认
|
||||
_, err := m.db.Exec("UPDATE media_servers SET is_default = 1 WHERE id = ?", id)
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) Close() error {
|
||||
return m.db.Close()
|
||||
}
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"sync"
|
||||
|
||||
"github.com/ossrs/srs-sip/pkg/models"
|
||||
_ "modernc.org/sqlite"
|
||||
)
|
||||
|
||||
var (
|
||||
instance *MediaServerDB
|
||||
once sync.Once
|
||||
)
|
||||
|
||||
type MediaServerDB struct {
|
||||
models.MediaServerResponse
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// GetInstance 返回 MediaServerDB 的单例实例
|
||||
func GetInstance(dbPath string) (*MediaServerDB, error) {
|
||||
var err error
|
||||
once.Do(func() {
|
||||
instance, err = NewMediaServerDB(dbPath)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return instance, nil
|
||||
}
|
||||
|
||||
func NewMediaServerDB(dbPath string) (*MediaServerDB, error) {
|
||||
db, err := sql.Open("sqlite", dbPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 创建媒体服务器表
|
||||
_, err = db.Exec(`
|
||||
CREATE TABLE IF NOT EXISTS media_servers (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
type TEXT NOT NULL,
|
||||
name TEXT NOT NULL,
|
||||
ip TEXT NOT NULL,
|
||||
port INTEGER NOT NULL,
|
||||
username TEXT,
|
||||
password TEXT,
|
||||
secret TEXT,
|
||||
is_default INTEGER NOT NULL DEFAULT 0,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &MediaServerDB{db: db}, nil
|
||||
}
|
||||
|
||||
// GetMediaServerByNameAndIP 根据名称和IP查询媒体服务器
|
||||
func (m *MediaServerDB) GetMediaServerByNameAndIP(name, ip string) (*models.MediaServerResponse, error) {
|
||||
var ms models.MediaServerResponse
|
||||
err := m.db.QueryRow(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers WHERE name = ? AND ip = ?
|
||||
`, name, ip).Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ms, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) AddMediaServer(name, serverType, ip string, port int, username, password, secret string, isDefault int) error {
|
||||
_, err := m.db.Exec(`
|
||||
INSERT INTO media_servers (name, type, ip, port, username, password, secret, is_default)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`, name, serverType, ip, port, username, password, secret, isDefault)
|
||||
return err
|
||||
}
|
||||
|
||||
// AddOrUpdateMediaServer 添加或更新媒体服务器（如果已存在则更新）
|
||||
func (m *MediaServerDB) AddOrUpdateMediaServer(name, serverType, ip string, port int, username, password, secret string, isDefault int) error {
|
||||
// 检查是否已存在
|
||||
existing, err := m.GetMediaServerByNameAndIP(name, ip)
|
||||
if err == nil && existing != nil {
|
||||
// 已存在，更新记录
|
||||
_, err = m.db.Exec(`
|
||||
UPDATE media_servers
|
||||
SET type = ?, port = ?, username = ?, password = ?, secret = ?, is_default = ?
|
||||
WHERE name = ? AND ip = ?
|
||||
`, serverType, port, username, password, secret, isDefault, name, ip)
|
||||
return err
|
||||
}
|
||||
|
||||
// 不存在，插入新记录
|
||||
return m.AddMediaServer(name, serverType, ip, port, username, password, secret, isDefault)
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) DeleteMediaServer(id int) error {
|
||||
_, err := m.db.Exec("DELETE FROM media_servers WHERE id = ?", id)
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) GetMediaServer(id int) (*models.MediaServerResponse, error) {
|
||||
var ms models.MediaServerResponse
|
||||
err := m.db.QueryRow(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers WHERE id = ?
|
||||
`, id).Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ms, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) ListMediaServers() ([]models.MediaServerResponse, error) {
|
||||
rows, err := m.db.Query(`
|
||||
SELECT id, name, type, ip, port, username, password, secret, is_default, created_at
|
||||
FROM media_servers ORDER BY created_at DESC
|
||||
`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var servers []models.MediaServerResponse
|
||||
for rows.Next() {
|
||||
var ms models.MediaServerResponse
|
||||
err := rows.Scan(&ms.ID, &ms.Name, &ms.Type, &ms.IP, &ms.Port, &ms.Username, &ms.Password, &ms.Secret, &ms.IsDefault, &ms.CreatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
servers = append(servers, ms)
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) SetDefaultMediaServer(id int) error {
|
||||
// 先将所有服务器设置为非默认
|
||||
if _, err := m.db.Exec("UPDATE media_servers SET is_default = 0"); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 将指定ID的服务器设置为默认
|
||||
_, err := m.db.Exec("UPDATE media_servers SET is_default = 1 WHERE id = ?", id)
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *MediaServerDB) Close() error {
|
||||
return m.db.Close()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user