api_logger.go

package apilogger

import (
	"reflect"
	"time"

	"github.com/latolukasz/beeorm"
)

type LogEntity interface {
	beeorm.Entity
	SetID(value uint64)
	SetType(value string)
	SetStatus(value string)
	SetRequest(value interface{})
	SetResponse(value interface{})
	SetMessage(value string)
	SetCreatedAt(value time.Time)
}

type APILogger interface {
	LogStart(ormService *beeorm.Engine, logType string, request interface{})
	LogError(ormService *beeorm.Engine, message string, response interface{})
	LogSuccess(ormService *beeorm.Engine, response interface{})
}

type DBLog struct {
	logEntity  LogEntity
	currentLog LogEntity
}

func NewAPILog(entity LogEntity) APILogger {
	return &DBLog{logEntity: entity}
}

func (l *DBLog) LogStart(ormService *beeorm.Engine, logType string, request interface{}) {
	var logEntity LogEntity

	if l.logEntity.GetID() == 0 {
		logEntity = l.logEntity
	} else {
		logEntity = reflect.New(reflect.ValueOf(l.logEntity).Elem().Type()).Interface().(LogEntity)
	}

	logEntity.SetType(logType)
	logEntity.SetRequest(request)
	logEntity.SetStatus("new")
	logEntity.SetCreatedAt(time.Now())

	ormService.Flush(logEntity)

	l.currentLog = logEntity
}

func (l *DBLog) LogError(ormService *beeorm.Engine, message string, response interface{}) {
	if l.currentLog == nil {
		panic("log is not created")
	}

	currentLog := l.currentLog
	currentLog.SetMessage(message)
	currentLog.SetResponse(response)
	currentLog.SetStatus("failed")

	ormService.Flush(currentLog)
}

func (l *DBLog) LogSuccess(ormService *beeorm.Engine, response interface{}) {
	if l.currentLog == nil {
		panic("log is not created")
	}

	currentLog := l.currentLog

	currentLog.SetStatus("completed")
	currentLog.SetResponse(response)

	ormService.Flush(currentLog)
}