Add WithUpstreamOptions server option (#10)

cmd/temporalite/main.go

@@ -12,6 +12,7 @@ import (
 	"github.com/urfave/cli/v2"
 	"go.temporal.io/server/common/headers"
 	"go.temporal.io/server/common/log"
+	"go.temporal.io/server/temporal"
 	"go.uber.org/zap"
 	"go.uber.org/zap/zapcore"
 
@@ -106,6 +107,9 @@ func buildCLI() *cli.App {
 					temporalite.WithFrontendPort(c.Int(portFlag)),
 					temporalite.WithDatabaseFilePath(c.String(dbPathFlag)),
 					temporalite.WithNamespaces(c.StringSlice(namespaceFlag)...),
+					temporalite.WithUpstreamOptions(
+						temporal.InterruptOn(temporal.InterruptCh()),
+					),
 				}
 				if c.Bool(ephemeralFlag) {
 					opts = append(opts, temporalite.WithPersistenceDisabled())

internal/liteconfig/config.go

@@ -15,6 +15,7 @@ import (
 	"go.temporal.io/server/common/log"
 	"go.temporal.io/server/common/metrics"
 	"go.temporal.io/server/common/persistence/sql/sqlplugin/sqlite"
+	"go.temporal.io/server/temporal"
 )
 
 const (
@@ -30,6 +31,7 @@ type Config struct {
 	DynamicPorts     bool
 	Namespaces       []string
 	Logger           log.Logger
+	UpstreamOptions  []temporal.ServerOption
 	portProvider     *portProvider
 }
 

options.go

@@ -6,6 +6,7 @@ package temporalite
 
 import (
 	"go.temporal.io/server/common/log"
+	"go.temporal.io/server/temporal"
 
 	"github.com/DataDog/temporalite/internal/liteconfig"
 )
@@ -54,6 +55,13 @@ func WithNamespaces(namespaces ...string) ServerOption {
 	})
 }
 
+// WithUpstreamOptions registers Temporal server options.
+func WithUpstreamOptions(options ...temporal.ServerOption) ServerOption {
+	return newApplyFuncContainer(func(cfg *liteconfig.Config) {
+		cfg.UpstreamOptions = append(cfg.UpstreamOptions, options...)
+	})
+}
+
 type applyFuncContainer struct {
 	applyInternal func(*liteconfig.Config)
 }

server.go

@@ -72,18 +72,23 @@ func NewServer(opts ...ServerOption) (*Server, error) {
 		return nil, fmt.Errorf("unable to instantiate claim mapper: %w", err)
 	}
 
+	serverOpts := []temporal.ServerOption{
+		temporal.WithConfig(cfg),
+		temporal.ForServices(temporal.Services),
+		temporal.WithLogger(c.Logger),
+		temporal.WithAuthorizer(authorizer),
+		temporal.WithClaimMapper(func(cfg *config.Config) authorization.ClaimMapper {
+			return claimMapper
+		}),
+		temporal.WithDynamicConfigClient(dynamicconfig.NewNoopClient()),
+	}
+
+	if len(c.UpstreamOptions) > 0 {
+		serverOpts = append(serverOpts, c.UpstreamOptions...)
+	}
+
 	s := &Server{
-		internal: temporal.NewServer(
-			temporal.WithConfig(cfg),
-			temporal.ForServices(temporal.Services),
-			temporal.WithLogger(c.Logger),
-			temporal.InterruptOn(temporal.InterruptCh()),
-			temporal.WithAuthorizer(authorizer),
-			temporal.WithClaimMapper(func(cfg *config.Config) authorization.ClaimMapper {
-				return claimMapper
-			}),
-			temporal.WithDynamicConfigClient(dynamicconfig.NewNoopClient()),
-		),
+		internal:         temporal.NewServer(serverOpts...),
 		frontendHostPort: cfg.PublicClient.HostPort,
 		config:           c,
 	}