nats-server/server/log.go

// Copyright 2012-2014 Apcera Inc. All rights reserved.

package server

import (
	"sync"
	"sync/atomic"
)

var trace int32
var debug int32
var log = struct {
	logger Logger
	sync.Mutex
}{}

type Logger interface {
	Noticef(format string, v ...interface{})
	Fatalf(format string, v ...interface{})
	Errorf(format string, v ...interface{})
	Debugf(format string, v ...interface{})
	Tracef(format string, v ...interface{})
}

func (s *Server) SetLogger(logger Logger, d, t bool) {
	if d {
		atomic.StoreInt32(&debug, 1)
	}

	if t {
		atomic.StoreInt32(&trace, 1)
	}

	log.Lock()
	defer log.Unlock()
	log.logger = logger
}

func Noticef(format string, v ...interface{}) {
	executeLogCall(func(logger Logger, format string, v ...interface{}) {
		logger.Noticef(format, v...)
	}, format, v...)
}

func Errorf(format string, v ...interface{}) {
	executeLogCall(func(logger Logger, format string, v ...interface{}) {
		logger.Errorf(format, v...)
	}, format, v...)
}

func Fatalf(format string, v ...interface{}) {
	executeLogCall(func(logger Logger, format string, v ...interface{}) {
		logger.Fatalf(format, v...)
	}, format, v...)
}

func Debugf(format string, v ...interface{}) {
	if debug == 0 {
		return
	}

	executeLogCall(func(logger Logger, format string, v ...interface{}) {
		logger.Debugf(format, v...)
	}, format, v...)
}

func Tracef(format string, v ...interface{}) {
	if trace == 0 {
		return
	}

	executeLogCall(func(logger Logger, format string, v ...interface{}) {
		logger.Tracef(format, v...)
	}, format, v...)
}

func executeLogCall(f func(logger Logger, format string, v ...interface{}), format string, args ...interface{}) {
	log.Lock()
	defer log.Unlock()
	if log.logger == nil {
		return
	}

	f(log.logger, format, args...)
}