112 lines
3.1 KiB
Go

// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License-AGPL.txt in the project root for license information.
package log
import (
"fmt"
"github.com/sirupsen/logrus"
log "github.com/sirupsen/logrus"
)
const (
// OwnerField is the log field name of a workspace owner
OwnerField = "userId"
// WorkspaceField is the log field name of a workspace ID (not instance ID)
WorkspaceField = "workspaceId"
// InstanceField is the log field name of a workspace instance ID
InstanceField = "instanceId"
)
// OWI builds a structure meant for logrus which contains the owner, workspace and instance.
// Beware that this refers to the terminology outside of wsman which maps like:
// owner = owner, workspace = metaID, instance = workspaceID
func OWI(owner, workspace, instance string) log.Fields {
return log.Fields{
OwnerField: owner,
WorkspaceField: workspace,
InstanceField: instance,
}
}
// ServiceContext is the shape required for proper error logging in the GCP context.
// See https://cloud.google.com/error-reporting/reference/rest/v1beta1/ServiceContext
// Note that we musn't set resourceType for reporting errors.
type ServiceContext struct {
Service string `json:"service"`
Version string `json:"version"`
}
// Log is the application wide console logger
var Log = log.WithFields(log.Fields{})
// Init initializes/configures the application-wide logger
func Init(service, version string, json, verbose bool) {
Log = log.WithFields(log.Fields{
"serviceContext": ServiceContext{service, version},
})
if json {
log.SetFormatter(&gcpFormatter{
log.JSONFormatter{
FieldMap: log.FieldMap{
log.FieldKeyMsg: "message",
},
},
})
} else {
log.SetFormatter(&logrus.TextFormatter{})
}
if verbose {
log.SetLevel(log.DebugLevel)
} else {
log.SetLevel(log.InfoLevel)
}
}
// gcpFormatter formats errors according to GCP rules, see
type gcpFormatter struct {
log.JSONFormatter
}
func (f *gcpFormatter) Format(entry *log.Entry) ([]byte, error) {
hasError := false
for k, v := range entry.Data {
switch v := v.(type) {
case error:
// Otherwise errors are ignored by `encoding/json`
// https://github.com/sirupsen/logrus/issues/137
//
// Print errors verbosely to get stack traces where available
entry.Data[k] = fmt.Sprintf("%+v", v)
hasError = true
}
}
// map to gcp severity. See https://cloud.google.com/logging/docs/reference/v2/rest/v2/LogEntry#LogSeverity
var severity string = "INFO"
switch entry.Level {
case logrus.TraceLevel:
severity = "DEBUG"
case logrus.DebugLevel:
severity = "DEBUG"
case logrus.InfoLevel:
severity = "INFO"
case logrus.WarnLevel:
severity = "WARNING"
case logrus.ErrorLevel:
severity = "ERROR"
case logrus.FatalLevel:
severity = "CRITICAL"
case logrus.PanicLevel:
severity = "EMERGENCY"
}
entry.Data["severity"] = severity
if entry.Level <= log.WarnLevel && hasError {
entry.Data["@type"] = "type.googleapis.com/google.devtools.clouderrorreporting.v1beta1.ReportedErrorEvent"
}
return f.JSONFormatter.Format(entry)
}