node-api/tools.go

package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"log"
	"time"

	"github.com/nats-io/nats.go"
	"github.com/rosti-cz/node-api/apps"
	"github.com/rosti-cz/node-api/common"
	docker "github.com/rosti-cz/node-api/containers"
)

func errorReplyFormater(m *nats.Msg, message string, err error) error {
	reply := ReplyMessage{
		Error:   true,
		Payload: message,
	}

	log.Println("ERROR:", err.Error())

	data, err := json.Marshal(reply)
	if err != nil {
		log.Println("ERROR:", err.Error())
		return err
	}

	err = m.Respond(data)
	if err != nil {
		log.Println("ERROR:", err.Error())
		return err
	}

	return err
}

func publish(appName string, state string, isErr bool) {
	stateMessage := StateMessage{
		AppName: appName,
		Error:   isErr,
		Message: state,
	}

	data, err := stateMessage.JSON()
	if err != nil {
		log.Println("ERROR: publish:", err.Error())
	}

	subjectEvents := fmt.Sprintf("admin.apps.%s.states", config.NATSAlias)
	err = nc.Publish(subjectEvents, data)
	if err != nil {
		log.Println("ERROR: publish:", err.Error())
	}
}

// waitForApp waits until app is ready or timeout is reached.
// It's used in some async calls that need at least part of the
// environment prepared.
func waitForApp(appName string) error {
	processor := apps.AppsProcessor{
		DB: common.GetDBConnection(),
	}

	sleepFor := 5 * time.Second
	loops := 6

	for i := 0; i < loops; i++ {
		err := updateState(appName)
		if err != nil {
			time.Sleep(sleepFor)
			continue
		}

		app, err := processor.Get(appName)
		if err != nil {
			return err
		}

		container := docker.Container{
			App: app,
		}

		status, err := container.Status()

		if status == "running" {
			return nil
		}

		time.Sleep(sleepFor)
		continue
	}

	return errors.New("timeout reached")
}