226 lines
5.0 KiB
Go
226 lines
5.0 KiB
Go
package docker
|
|
|
|
import (
|
|
"log"
|
|
"path"
|
|
|
|
"github.com/rosti-cz/node-api/apps"
|
|
)
|
|
|
|
// username in the containers under which all containers run
|
|
const appUsername = "app"
|
|
const passwordFile = "/srv/.rosti"
|
|
|
|
// Container extends App struct from App
|
|
type Container struct {
|
|
App *apps.App `json:"app"`
|
|
}
|
|
|
|
func (c *Container) getDriver() *Driver {
|
|
driver := &Driver{}
|
|
return driver
|
|
}
|
|
|
|
// volumeHostPath each container has one volume mounted into it,
|
|
func (c *Container) volumeHostPath() string {
|
|
return path.Join("/srv", c.App.Name)
|
|
}
|
|
|
|
// GetRawResourceStats returns RAW CPU and memory usage directly from Docker API
|
|
func (c *Container) GetRawResourceStats() (int64, int, error) {
|
|
driver := c.getDriver()
|
|
cpu, memory, err := driver.RawStats(c.App.Name)
|
|
return cpu, memory, err
|
|
}
|
|
|
|
// GetState returns app state object with populated state fields
|
|
func (c *Container) GetState() (*apps.AppState, error) {
|
|
status, err := c.Status()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cpu, memory, err := c.ResourceUsage()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
bytes, inodes, err := c.DiskUsage()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
state := apps.AppState{
|
|
State: status,
|
|
CPUUsage: cpu,
|
|
MemoryUsage: memory,
|
|
DiskUsageBytes: bytes,
|
|
DiskUsageInodes: inodes,
|
|
}
|
|
|
|
return &state, nil
|
|
}
|
|
|
|
// Status returns state of the container
|
|
// Possible values: running, stopped, no-container, unknown
|
|
func (c *Container) Status() (string, error) {
|
|
status := "unknown"
|
|
|
|
// if _, err := os.Stat(path.Join("/srv", c.App.Name)); !os.IsNotExist(err) {
|
|
// status = "data-only"
|
|
// }
|
|
|
|
driver := c.getDriver()
|
|
containerStatus, err := driver.Status(c.App.Name)
|
|
if err != nil && err.Error() == "no container found" {
|
|
return "no-container", nil
|
|
}
|
|
if err != nil {
|
|
return status, err
|
|
}
|
|
|
|
status = containerStatus
|
|
|
|
return status, nil
|
|
}
|
|
|
|
// DiskUsage returns number of bytes and inodes used by the container in it's mounted volume
|
|
func (c *Container) DiskUsage() (int, int, error) {
|
|
return du(c.volumeHostPath())
|
|
}
|
|
|
|
// ResourceUsage returns amount of memory in B and CPU in % that the app occupies
|
|
func (c *Container) ResourceUsage() (float64, int, error) {
|
|
driver := c.getDriver()
|
|
|
|
cpu, memory, err := driver.Stats(c.App.Name)
|
|
if err != nil {
|
|
return 0.0, 0, err
|
|
}
|
|
|
|
return cpu, memory, nil
|
|
}
|
|
|
|
// Create creates the container
|
|
func (c *Container) Create() error {
|
|
driver := c.getDriver()
|
|
_, err := driver.Create(
|
|
c.App.Name,
|
|
c.App.Image,
|
|
c.volumeHostPath(),
|
|
c.App.HTTPPort,
|
|
c.App.SSHPort,
|
|
c.App.CPU,
|
|
c.App.Memory,
|
|
[]string{},
|
|
)
|
|
|
|
return err
|
|
}
|
|
|
|
// Start starts the container
|
|
func (c *Container) Start() error {
|
|
driver := c.getDriver()
|
|
|
|
return driver.Start(c.App.Name)
|
|
}
|
|
|
|
// Stop stops the container
|
|
func (c *Container) Stop() error {
|
|
driver := c.getDriver()
|
|
|
|
return driver.Stop(c.App.Name)
|
|
}
|
|
|
|
// Restart restarts the container
|
|
func (c *Container) Restart() error {
|
|
driver := c.getDriver()
|
|
|
|
err := driver.Stop(c.App.Name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return driver.Start(c.App.Name)
|
|
}
|
|
|
|
// Destroy removes the container but keeps the data so it can be created again
|
|
func (c *Container) Destroy() error {
|
|
driver := c.getDriver()
|
|
|
|
return driver.Remove(c.App.Name)
|
|
}
|
|
|
|
// Delete removes both data and the container
|
|
func (c *Container) Delete() error {
|
|
err := c.Destroy()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
volumePath := path.Join("/srv", c.App.Name)
|
|
err = removeDirectory(volumePath)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SetPassword configures password for system user app in the container
|
|
func (c *Container) SetPassword(password string) error {
|
|
driver := c.getDriver()
|
|
|
|
err := driver.Exec(c.App.Name, []string{"chpasswd"}, appUsername+":"+password, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = driver.Exec(c.App.Name, []string{"tee", passwordFile}, password, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// SetFileContent uploads text into a file inside the container. It's greate for uploading SSH keys.
|
|
// The method creates the diretory where the file is located and sets mode of the final file
|
|
func (c *Container) SetFileContent(filename string, text string, mode string) error {
|
|
driver := c.getDriver()
|
|
|
|
directory := path.Dir(filename)
|
|
|
|
err := driver.Exec(c.App.Name, []string{"mkdir", "-p", directory}, "", []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = driver.Exec(c.App.Name, []string{"tee", filename}, text, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = driver.Exec(c.App.Name, []string{"chown", directory, "app:app"}, "", []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = driver.Exec(c.App.Name, []string{"chown", filename, "app:app"}, "", []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = driver.Exec(c.App.Name, []string{"chmod", mode, filename}, "", []string{})
|
|
return err
|
|
}
|
|
|
|
// SetTechnology prepares container for given technology (Python, PHP, Node.js, ...)
|
|
// Where tech can be php, python or node and latest available version is used.
|
|
func (c *Container) SetTechnology(tech string) error {
|
|
driver := c.getDriver()
|
|
|
|
err := driver.Exec(c.App.Name, []string{"su", "app", "-c", "rosti " + tech}, "", []string{})
|
|
return err
|
|
}
|