mirror of
https://gitee.com/goploy/goploy.git
synced 2024-12-05 13:38:47 +08:00
123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
package model
|
|
|
|
import (
|
|
sq "github.com/Masterminds/squirrel"
|
|
)
|
|
|
|
const namespaceTable = "`namespace`"
|
|
|
|
// Namespace -
|
|
type Namespace struct {
|
|
ID int64 `json:"id"`
|
|
Name string `json:"name"`
|
|
UserID int64 `json:"-"`
|
|
RoleID int64 `json:"role_id"`
|
|
PermissionIDs []int64 `json:"permissionIds"`
|
|
InsertTime string `json:"insertTime,omitempty"`
|
|
UpdateTime string `json:"updateTime,omitempty"`
|
|
}
|
|
|
|
// Namespaces -
|
|
type Namespaces []Namespace
|
|
|
|
// AddRow return LastInsertId
|
|
func (ns Namespace) AddRow() (int64, error) {
|
|
result, err := sq.
|
|
Insert(namespaceTable).
|
|
Columns("name").
|
|
Values(ns.Name).
|
|
RunWith(DB).
|
|
Exec()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, err := result.LastInsertId()
|
|
return id, err
|
|
}
|
|
|
|
// EditRow -
|
|
func (ns Namespace) EditRow() error {
|
|
_, err := sq.
|
|
Update(namespaceTable).
|
|
Set("name", ns.Name).
|
|
Where(sq.Eq{"id": ns.ID}).
|
|
RunWith(DB).
|
|
Exec()
|
|
return err
|
|
}
|
|
|
|
func (ns Namespace) GetAllByUserID() (Namespaces, error) {
|
|
rows, err := sq.
|
|
Select("namespace.id, namespace.name, role_id").
|
|
From(namespaceTable).
|
|
Join(namespaceUserTable + " ON namespace_user.namespace_id = namespace.id").
|
|
Where(sq.Eq{"user_id": ns.UserID}).
|
|
RunWith(DB).
|
|
Query()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
namespaces := Namespaces{}
|
|
for rows.Next() {
|
|
var namespace Namespace
|
|
if err := rows.Scan(&namespace.ID, &namespace.Name, &namespace.RoleID); err != nil {
|
|
return nil, err
|
|
}
|
|
namespaces = append(namespaces, namespace)
|
|
}
|
|
return namespaces, nil
|
|
}
|
|
|
|
func (ns Namespace) GetList(pagination Pagination) (Namespaces, error) {
|
|
rows, err := sq.
|
|
Select("id, name, insert_time, update_time").
|
|
From(namespaceTable).
|
|
OrderBy("id DESC").
|
|
Limit(pagination.Rows).
|
|
Offset((pagination.Page - 1) * pagination.Rows).
|
|
RunWith(DB).
|
|
Query()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
namespaces := Namespaces{}
|
|
for rows.Next() {
|
|
var namespace Namespace
|
|
|
|
if err := rows.Scan(&namespace.ID, &namespace.Name, &namespace.InsertTime, &namespace.UpdateTime); err != nil {
|
|
return nil, err
|
|
}
|
|
namespaces = append(namespaces, namespace)
|
|
}
|
|
|
|
return namespaces, nil
|
|
}
|
|
|
|
func (ns Namespace) GetTotal() (int64, error) {
|
|
var total int64
|
|
err := sq.
|
|
Select("COUNT(*) AS count").
|
|
From(namespaceTable).
|
|
RunWith(DB).
|
|
QueryRow().
|
|
Scan(&total)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return total, nil
|
|
}
|
|
|
|
// GetData -
|
|
func (ns Namespace) GetData() (Namespace, error) {
|
|
var namespace Namespace
|
|
err := sq.
|
|
Select("name, insert_time, update_time").
|
|
From(namespaceTable).
|
|
Where(sq.Eq{"id": ns.ID}).
|
|
RunWith(DB).
|
|
QueryRow().
|
|
Scan(&namespace.Name, &namespace.InsertTime, &namespace.UpdateTime)
|
|
return namespace, err
|
|
}
|