Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/envd/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ start-docker:
-p 8001:8001 \
--rm \
-i envd-debug \
/usr/bin/envd -isnotfc
/usr/bin/envd -isnotfc -verbose

build-and-upload:
make build
Expand Down
22 changes: 1 addition & 21 deletions packages/envd/internal/logs/exporter/exporter.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,8 @@ package exporter
import (
"bytes"
"context"
"fmt"
"log"
"net/http"
"os"
"sync"
"time"

Expand All @@ -18,7 +16,6 @@ const ExporterTimeout = 10 * time.Second
type HTTPExporter struct {
client http.Client
logs [][]byte
isNotFC bool
mmdsOpts *host.MMDSOpts

// Concurrency coordination
Expand All @@ -28,13 +25,12 @@ type HTTPExporter struct {
startOnce sync.Once
}

func NewHTTPLogsExporter(ctx context.Context, isNotFC bool, mmdsChan <-chan *host.MMDSOpts) *HTTPExporter {
func NewHTTPLogsExporter(ctx context.Context, mmdsChan <-chan *host.MMDSOpts) *HTTPExporter {
exporter := &HTTPExporter{
client: http.Client{
Timeout: ExporterTimeout,
},
triggers: make(chan struct{}, 1),
isNotFC: isNotFC,
startOnce: sync.Once{},
mmdsOpts: &host.MMDSOpts{
SandboxID: "unknown",
Expand Down Expand Up @@ -69,10 +65,6 @@ func (w *HTTPExporter) sendInstanceLogs(ctx context.Context, logs []byte, addres
return nil
}

func printLog(logs []byte) {
fmt.Fprintf(os.Stdout, "%v", string(logs))
}

func (w *HTTPExporter) listenForMMDSOptsAndStart(ctx context.Context, mmdsChan <-chan *host.MMDSOpts) {
for {
select {
Expand Down Expand Up @@ -102,32 +94,20 @@ func (w *HTTPExporter) start(ctx context.Context) {
continue
}

if w.isNotFC {
for _, log := range logs {
fmt.Fprintf(os.Stdout, "%v", string(log))
}

continue
}

for _, logLine := range logs {
w.mmdsLock.RLock()
logLineWithOpts, err := w.mmdsOpts.AddOptsToJSON(logLine)
w.mmdsLock.RUnlock()
if err != nil {
log.Printf("error adding instance logging options (%+v) to JSON (%+v) with logs : %v\n", w.mmdsOpts, logLine, err)

printLog(logLine)
Comment thread
cursor[bot] marked this conversation as resolved.

continue
}

err = w.sendInstanceLogs(ctx, logLineWithOpts, w.mmdsOpts.LogsCollectorAddress)
if err != nil {
log.Printf("error sending instance logs: %+v", err)

printLog(logLine)

continue
}
}
Expand Down
9 changes: 5 additions & 4 deletions packages/envd/internal/logs/logger.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,16 +12,17 @@ import (
"github.com/e2b-dev/infra/packages/envd/internal/logs/exporter"
)

func NewLogger(ctx context.Context, isNotFC bool, mmdsChan <-chan *host.MMDSOpts) *zerolog.Logger {
func NewLogger(ctx context.Context, isFC bool, verbose bool, mmdsChan <-chan *host.MMDSOpts) *zerolog.Logger {
zerolog.TimestampFieldName = "timestamp"
zerolog.TimeFieldFormat = time.RFC3339Nano

exporters := []io.Writer{}

if isNotFC {
if isFC {
exporters = append(exporters, exporter.NewHTTPLogsExporter(ctx, mmdsChan))
}
if verbose {
exporters = append(exporters, os.Stdout)
} else {
exporters = append(exporters, exporter.NewHTTPLogsExporter(ctx, isNotFC, mmdsChan), os.Stdout)
}

l := zerolog.
Expand Down
6 changes: 3 additions & 3 deletions packages/envd/internal/services/process/handler/handler.go
Original file line number Diff line number Diff line change
Expand Up @@ -221,7 +221,7 @@ func New(
}

if readErr != nil {
fmt.Fprintf(os.Stderr, "error reading from pty: %s\n", readErr)
logger.Error().Err(readErr).Msg("error reading from pty")

break
}
Expand Down Expand Up @@ -269,7 +269,7 @@ func New(
}

if readErr != nil {
fmt.Fprintf(os.Stderr, "error reading from stdout: %s\n", readErr)
logger.Error().Err(readErr).Msg("error reading from stdout")

break
}
Expand Down Expand Up @@ -313,7 +313,7 @@ func New(
}

if readErr != nil {
fmt.Fprintf(os.Stderr, "error reading from stderr: %s\n", readErr)
logger.Error().Err(readErr).Msg("error reading from stderr")

break
}
Expand Down
12 changes: 10 additions & 2 deletions packages/envd/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,14 +56,15 @@ var (
versionFlag bool
commitFlag bool
cgroupRoot string
verbose bool
)

func parseFlags() {
flag.BoolVar(
&isNotFC,
"isnotfc",
false,
"isNotFCmode prints all logs to stdout",
"run outside of Firecracker (skips MMDS poll and HTTP log exporter)",
)

flag.BoolVar(
Expand Down Expand Up @@ -94,6 +95,13 @@ func parseFlags() {
"cgroup root directory",
)

flag.BoolVar(
&verbose,
"verbose",
false,
"write envd logs to stdout",
)

flag.Parse()
}

Expand Down Expand Up @@ -159,7 +167,7 @@ func main() {
go host.PollForMMDSOpts(ctx, mmdsChan, defaults.EnvVars)
}

l := logs.NewLogger(ctx, isNotFC, mmdsChan)
l := logs.NewLogger(ctx, !isNotFC, verbose, mmdsChan)

m := chi.NewRouter()

Expand Down
2 changes: 1 addition & 1 deletion packages/envd/pkg/version.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
package pkg

const Version = "0.5.20"
const Version = "0.5.21"
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
{{- /*gotype:github.com/e2b-dev/infra/packages/orchestrator/pkg/template/build/core/rootfs.templateModel*/ -}}
{{ .WriteFile "etc/systemd/journald.conf.d/e2b.conf" 0o644 }}

[Journal]
Storage=persistent
SystemMaxUse=8M
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ func TestAdditionalOCILayers(t *testing.T) {

keysIter := maps.Keys(actualFiles)
keys := slices.Collect(keysIter)
assert.Len(t, keys, 13)
assert.Len(t, keys, 14)
assert.Equal(t, "e2b.local", actualFiles["etc/hostname"])
assert.Equal(t, "nameserver 8.8.8.8", actualFiles["etc/resolv.conf"])

Expand Down
Loading