node-api/apps/main.go

188 lines
4.1 KiB
Go
Raw Normal View History

2020-07-08 22:09:19 +00:00
package apps
2020-07-11 11:04:37 +00:00
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/sqlite"
2022-02-03 00:31:47 +00:00
"github.com/rosti-cz/node-api/detector"
2020-07-11 11:04:37 +00:00
)
2020-07-08 22:09:19 +00:00
// AppsProcessor encapsulates functions for apps manipulation
// This handles only the database part but not containers
type AppsProcessor struct {
DB *gorm.DB
}
func (a *AppsProcessor) Init() {
a.DB.AutoMigrate(Label{})
a.DB.AutoMigrate(App{})
2020-07-08 22:09:19 +00:00
}
// Get returns one app
func (a *AppsProcessor) Get(name string) (App, error) {
2020-07-08 22:09:19 +00:00
var app App
err := a.DB.Preload("Labels").Where("name = ?", name).First(&app).Error
2020-07-08 22:09:19 +00:00
if err != nil {
return app, err
2020-07-08 22:09:19 +00:00
}
return app, nil
2020-07-08 22:09:19 +00:00
}
// List returns all apps located on this node
func (a *AppsProcessor) List() (Apps, error) {
2020-07-30 20:46:09 +00:00
var apps Apps
2020-07-08 22:09:19 +00:00
err := a.DB.Preload("Labels").Find(&apps).Error
2020-07-08 22:09:19 +00:00
if err != nil {
return nil, err
}
return apps, nil
2020-07-08 22:09:19 +00:00
}
// New creates new record about application in the database
func (a *AppsProcessor) New(name string, SSHPort int, HTTPPort int, image string, CPU int, memory int) error {
2020-07-08 22:09:19 +00:00
app := App{
Name: name,
SSHPort: SSHPort,
HTTPPort: HTTPPort,
Image: image,
CPU: CPU,
Memory: memory,
}
2020-07-09 22:27:23 +00:00
validationErrors := app.Validate()
if len(validationErrors) != 0 {
return ValidationError{
Errors: validationErrors,
}
}
if err := a.DB.Create(&app).Error; err != nil {
2020-07-08 22:09:19 +00:00
return err
}
return nil
}
// Update changes value about app in the database
func (a *AppsProcessor) Update(name string, SSHPort int, HTTPPort int, image string, CPU int, memory int) (*App, error) {
2020-07-08 22:09:19 +00:00
var app App
err := a.DB.Where("name = ?", name).First(&app).Error
2020-07-08 22:09:19 +00:00
if err != nil {
2020-07-11 11:04:37 +00:00
return &app, err
}
// Update affected fields
if image != "" {
app.Image = image
2020-07-08 22:09:19 +00:00
}
2020-07-11 11:04:37 +00:00
if CPU != 0 {
app.CPU = CPU
}
if memory != 0 {
app.Memory = memory
}
// SSH port and HTTP port cannot be turned off when they are once set
if SSHPort != 0 {
app.SSHPort = SSHPort
}
// SSH port and HTTP port cannot be turned off when they are once set
if HTTPPort != 0 {
app.HTTPPort = HTTPPort
}
2020-07-09 22:27:23 +00:00
validationErrors := app.Validate()
if len(validationErrors) != 0 {
2020-07-11 11:04:37 +00:00
return &app, ValidationError{
2020-07-09 22:27:23 +00:00
Errors: validationErrors,
}
}
2020-07-11 11:04:37 +00:00
// Apply the changes
err = a.DB.Save(&app).Error
2020-07-11 11:04:37 +00:00
return &app, err
2020-07-08 22:09:19 +00:00
}
// UpdateResources updates various metrics saved in the database
func (a *AppsProcessor) UpdateResources(name string, state string, OOMKilled bool, CPUUsage float64, memory int, diskUsageBytes int, diskUsageInodes int, flags detector.Flags) error {
err := a.DB.Model(&App{}).Where("name = ?", name).Updates(App{
2020-07-16 17:05:38 +00:00
State: state,
OOMKilled: OOMKilled,
2020-07-16 17:05:38 +00:00
CPUUsage: CPUUsage,
MemoryUsage: memory,
DiskUsageBytes: diskUsageBytes,
DiskUsageInodes: diskUsageInodes,
Flags: flags.String(),
}).Error
return err
}
// UpdateState sets container's state
func (a *AppsProcessor) UpdateState(name string, state string, OOMKilled bool) error {
err := a.DB.Model(&App{}).Where("name = ?", name).Updates(App{
State: state,
OOMKilled: OOMKilled,
}).Error
2020-07-16 17:05:38 +00:00
return err
}
2020-07-08 22:09:19 +00:00
// Delete removes records about one app from the database
func (a *AppsProcessor) Delete(name string) error {
app, err := a.Get(name)
if err != nil {
return err
}
err = a.DB.Where("app_id = ?", app.ID).Delete(&Label{}).Error
if err != nil {
return err
}
err = a.DB.Where("id = ?", app.ID).Delete(App{}).Error
2020-07-08 22:09:19 +00:00
return err
}
2020-07-16 21:24:09 +00:00
// AddLabel adds label to existing application
func (a *AppsProcessor) AddLabel(appName string, label string) error {
2020-07-16 21:24:09 +00:00
var app App
err := a.DB.Where("name = ?", appName).First(&app).Error
2020-07-16 21:24:09 +00:00
if err != nil {
return err
}
// Check if there is such label already
var count int
err = a.DB.Model(&Label{}).Where("value = ? AND app_id = ?", label, app.ID).Count(&count).Error
2020-07-16 21:24:09 +00:00
if err != nil {
return err
}
if count > 0 {
return nil
}
// And create the label
return a.DB.Create(&Label{AppID: app.ID, Value: label}).Error
2020-07-16 21:24:09 +00:00
}
// RemoveLabel removes label from existing application
func (a *AppsProcessor) RemoveLabel(appName string, label string) error {
2020-07-16 21:24:09 +00:00
var app App
err := a.DB.Where("name = ?", appName).First(&app).Error
2020-07-16 21:24:09 +00:00
if err != nil {
return err
}
return a.DB.Where("value = ? AND app_id = ?", label, app.ID).Delete(&Label{}).Error
2020-07-16 21:24:09 +00:00
}