feat(Multi-tenancy): Add support for multi-tenancy

* Add support for multi-tenancy
---------
Co-authored-by: Chengjun Li <>

Author: licjun

cmd/aws-lambda-rie/main.go

@@ -4,151 +4,9 @@
 package main
 
 import (
-	"context"
-	"fmt"
-	"net"
-	"os"
-	"runtime/debug"
-
-	"github.com/jessevdk/go-flags"
-	"go.amzn.com/lambda/interop"
-	"go.amzn.com/lambda/rapidcore"
-
-	log "github.com/sirupsen/logrus"
+	"github.com/aws/aws-lambda-runtime-interface-emulator/internal/lambda/rie"
 )
 
-const (
-	optBootstrap     = "/opt/bootstrap"
-	runtimeBootstrap = "/var/runtime/bootstrap"
-)
-
-type options struct {
-	LogLevel           string `long:"log-level" description:"The level of AWS Lambda Runtime Interface Emulator logs to display. Can also be set by the environment variable 'LOG_LEVEL'. Defaults to the value 'info'."`
-	InitCachingEnabled bool   `long:"enable-init-caching" description:"Enable support for Init Caching"`
-	// Do not have a default value so we do not need to keep it in sync with the default value in lambda/rapidcore/sandbox_builder.go
-	RuntimeAPIAddress               string `long:"runtime-api-address" description:"The address of the AWS Lambda Runtime API to communicate with the Lambda execution environment."`
-	RuntimeInterfaceEmulatorAddress string `long:"runtime-interface-emulator-address" default:"0.0.0.0:8080" description:"The address for the AWS Lambda Runtime Interface Emulator to accept HTTP request upon."`
-}
-
 func main() {
-	// More frequent GC reduces the tail latencies, equivalent to export GOGC=33
-	debug.SetGCPercent(33)
-
-	opts, args := getCLIArgs()
-
-	logLevel := "info"
-
-	// If you specify an option by using a parameter on the CLI command line, it overrides any value from either the corresponding environment variable.
-	//
-	// https://docs.aws.amazon.com/cli/latest/userguide/cli-configure-envvars.html
-	if opts.LogLevel != "" {
-		logLevel = opts.LogLevel
-	} else if envLogLevel, envLogLevelSet := os.LookupEnv("LOG_LEVEL"); envLogLevelSet {
-		logLevel = envLogLevel
-	}
-
-	rapidcore.SetLogLevel(logLevel)
-
-	if opts.RuntimeAPIAddress != "" {
-		_, _, err := net.SplitHostPort(opts.RuntimeAPIAddress)
-
-		if err != nil {
-			log.WithError(err).Fatalf("The command line value for \"--runtime-api-address\" is not a valid network address %q.", opts.RuntimeAPIAddress)
-		}
-	}
-
-	_, _, err := net.SplitHostPort(opts.RuntimeInterfaceEmulatorAddress)
-
-	if err != nil {
-		log.WithError(err).Fatalf("The command line value for \"--runtime-interface-emulator-address\" is not a valid network address %q.", opts.RuntimeInterfaceEmulatorAddress)
-	}
-
-	bootstrap, handler := getBootstrap(args, opts)
-	sandbox := rapidcore.
-		NewSandboxBuilder().
-		AddShutdownFunc(context.CancelFunc(func() { os.Exit(0) })).
-		SetExtensionsFlag(true).
-		SetInitCachingFlag(opts.InitCachingEnabled)
-
-	if len(handler) > 0 {
-		sandbox.SetHandler(handler)
-	}
-
-	if opts.RuntimeAPIAddress != "" {
-		sandbox.SetRuntimeAPIAddress(opts.RuntimeAPIAddress)
-	}
-
-	sandboxContext, internalStateFn := sandbox.Create()
-	// Since we have not specified a custom interop server for standalone, we can
-	// directly reference the default interop server, which is a concrete type
-	sandbox.DefaultInteropServer().SetSandboxContext(sandboxContext)
-	sandbox.DefaultInteropServer().SetInternalStateGetter(internalStateFn)
-
-	startHTTPServer(opts.RuntimeInterfaceEmulatorAddress, sandbox, bootstrap)
-}
-
-func getCLIArgs() (options, []string) {
-	var opts options
-	parser := flags.NewParser(&opts, flags.IgnoreUnknown)
-	args, err := parser.ParseArgs(os.Args)
-
-	if err != nil {
-		log.WithError(err).Fatal("Failed to parse command line arguments:", os.Args)
-	}
-
-	return opts, args
-}
-
-func isBootstrapFileExist(filePath string) bool {
-	file, err := os.Stat(filePath)
-	return !os.IsNotExist(err) && !file.IsDir()
-}
-
-func getBootstrap(args []string, opts options) (interop.Bootstrap, string) {
-	var bootstrapLookupCmd []string
-	var handler string
-	currentWorkingDir := "/var/task" // default value
-
-	if len(args) <= 1 {
-		// set default value to /var/task/bootstrap, but switch to the other options if it doesn't exist
-		bootstrapLookupCmd = []string{
-			fmt.Sprintf("%s/bootstrap", currentWorkingDir),
-		}
-
-		if !isBootstrapFileExist(bootstrapLookupCmd[0]) {
-			var bootstrapCmdCandidates = []string{
-				optBootstrap,
-				runtimeBootstrap,
-			}
-
-			for i, bootstrapCandidate := range bootstrapCmdCandidates {
-				if isBootstrapFileExist(bootstrapCandidate) {
-					bootstrapLookupCmd = []string{bootstrapCmdCandidates[i]}
-					break
-				}
-			}
-		}
-
-		// handler is used later to set an env var for Lambda Image support
-		handler = ""
-	} else if len(args) > 1 {
-
-		bootstrapLookupCmd = args[1:]
-
-		if cwd, err := os.Getwd(); err == nil {
-			currentWorkingDir = cwd
-		}
-
-		if len(args) > 2 {
-			// Assume last arg is the handler
-			handler = args[len(args)-1]
-		}
-
-		log.Infof("exec '%s' (cwd=%s, handler=%s)", args[1], currentWorkingDir, handler)
-
-	} else {
-		log.Panic("insufficient arguments: bootstrap not provided")
-	}
-
-	return NewSimpleBootstrap(bootstrapLookupCmd, currentWorkingDir), handler
+	rie.Run()
 }

go.mod

@@ -1,4 +1,4 @@
-module go.amzn.com
+module github.com/aws/aws-lambda-runtime-interface-emulator
 
 go 1.25
 
@@ -13,6 +13,13 @@ require (
 	github.com/google/uuid v1.6.0
 	github.com/jessevdk/go-flags v1.5.0
 	github.com/shirou/gopsutil v2.19.10+incompatible
+	github.com/aws/aws-lambda-go v1.46.0
+	github.com/go-chi/chi v1.5.5
+	github.com/go-chi/chi/v5 v5.2.2
+	github.com/google/uuid v1.6.0
+	github.com/jessevdk/go-flags v1.5.0
+	github.com/orcaman/concurrent-map v1.0.0
+	github.com/santhosh-tekuri/jsonschema/v5 v5.3.1
 	github.com/sirupsen/logrus v1.9.3
 	github.com/stretchr/testify v1.9.0
 	golang.org/x/sync v0.12.0

lambda/agents/agent.go internal/lambda/agents/agent.golambda/agents/agent.go renamed to internal/lambda/agents/agent.go

@@ -8,8 +8,8 @@ import (
 	"net/http"
 	"strings"
 
-	"go.amzn.com/lambda/fatalerror"
-	"go.amzn.com/lambda/interop"
+	"github.com/aws/aws-lambda-runtime-interface-emulator/internal/lambda/fatalerror"
+	"github.com/aws/aws-lambda-runtime-interface-emulator/internal/lambda/interop"
 
 	log "github.com/sirupsen/logrus"
 )

lambda/agents/agent_test.go internal/lambda/agents/agent_test.golambda/agents/agent_test.go renamed to internal/lambda/agents/agent_test.go

@@ -8,11 +8,11 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/aws/aws-lambda-runtime-interface-emulator/internal/lambda/fatalerror"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"go.amzn.com/lambda/fatalerror"
 
-	"go.amzn.com/lambda/interop"
+	"github.com/aws/aws-lambda-runtime-interface-emulator/internal/lambda/interop"
 )
 
 func runTestRequestWithUserAgent(t *testing.T, userAgent string, expectedRuntimeRelease string) {

lambda/appctx/appctx.go internal/lambda/appctx/appctx.golambda/appctx/appctx.go renamed to internal/lambda/appctx/appctx.go

@@ -5,6 +5,7 @@ package core
 
 import (
 	"errors"
+
 	"github.com/google/uuid"
 )
 

lambda/appctx/appctxutil.go internal/lambda/appctx/appctxutil.golambda/appctx/appctxutil.go renamed to internal/lambda/appctx/appctxutil.go

@@ -4,9 +4,10 @@
 package core
 
 import (
+	"testing"
+
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/require"
-	"testing"
 )
 
 func TestExternalAgentsMapLookupByName(t *testing.T) {