116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
|
package privilege
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/gofiber/fiber/v2"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
URL string
|
||
|
|
ServiceName string
|
||
|
|
Privileges []Privilege
|
||
|
|
}
|
||
|
|
|
||
|
|
type Privilege struct {
|
||
|
|
ID string `json:"_id"`
|
||
|
|
PrivilegeID string `json:"privilegeId"`
|
||
|
|
Name string `json:"name"`
|
||
|
|
ServiceKey string `json:"serviceKey"`
|
||
|
|
Description string `json:"description"`
|
||
|
|
Type string `json:"type"`
|
||
|
|
Value string `json:"value"`
|
||
|
|
Price float64 `json:"price"`
|
||
|
|
Amount int `json:"amount"`
|
||
|
|
IsDeleted bool `json:"isDeleted"`
|
||
|
|
CreatedAt time.Time `json:"createdAt"`
|
||
|
|
UpdatedAt time.Time `json:"updatedAt"`
|
||
|
|
DeletedAt time.Time `json:"deletedAt"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Controller struct {
|
||
|
|
client Client
|
||
|
|
fiberClient *fiber.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewPrivilege(client Client, fiberClient *fiber.Client) *Controller {
|
||
|
|
return &Controller{client: client, fiberClient: fiberClient}
|
||
|
|
}
|
||
|
|
|
||
|
|
// PublishPrivileges публикует привилегии в сервисе.
|
||
|
|
func (c *Controller) PublishPrivileges() error {
|
||
|
|
old, err := c.getActualPrivileges()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(old) == 0 {
|
||
|
|
return c.setupActualPrivileges()
|
||
|
|
}
|
||
|
|
|
||
|
|
found := false
|
||
|
|
|
||
|
|
for _, oldPriv := range old {
|
||
|
|
for _, newPriv := range c.client.Privileges {
|
||
|
|
if newPriv.PrivilegeID == oldPriv.PrivilegeID {
|
||
|
|
found = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if !found {
|
||
|
|
if err := c.removePrivilege(oldPriv.PrivilegeID); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
found = false
|
||
|
|
}
|
||
|
|
|
||
|
|
return c.updatePrivileges()
|
||
|
|
}
|
||
|
|
|
||
|
|
// getActualPrivileges возвращает текущие привилегии из сервиса.
|
||
|
|
func (c *Controller) getActualPrivileges() ([]Privilege, error) {
|
||
|
|
res := []Privilege{}
|
||
|
|
_, _, err := c.fiberClient.Get(c.client.URL + "/privilege/service/" + c.client.ServiceName).Struct(&res)
|
||
|
|
if err != nil {
|
||
|
|
return res, err[0]
|
||
|
|
}
|
||
|
|
fmt.Println("str", res)
|
||
|
|
return res, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// updatePrivileges обновляет привилегии в сервисе.
|
||
|
|
func (c *Controller) updatePrivileges() error {
|
||
|
|
_, _, err := c.fiberClient.Put(c.client.URL + "/privilege/many").JSON(map[string][]Privilege{
|
||
|
|
"privilege": c.client.Privileges,
|
||
|
|
}).Bytes()
|
||
|
|
if err != nil {
|
||
|
|
return err[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// setupActualPrivileges настраивает привилегии в сервисе.
|
||
|
|
func (c *Controller) setupActualPrivileges() error {
|
||
|
|
res := []Privilege{}
|
||
|
|
_, _, err := c.fiberClient.Post(c.client.URL + "/privilege/many").JSON(map[string][]Privilege{
|
||
|
|
"privileges": c.client.Privileges,
|
||
|
|
}).Struct(&res)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return err[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// removePrivilege удаляет привилегию из сервиса по ID.
|
||
|
|
func (c *Controller) removePrivilege(id string) error {
|
||
|
|
_, _, err := c.fiberClient.Delete(c.client.URL + "/privilege/").JSON(map[string]string{"privilegeId": id}).Bytes()
|
||
|
|
if err != nil {
|
||
|
|
return err[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|