meh

server/options.go

@@ -1,6 +1,8 @@
 package server
 
 import (
+	"time"
+
 	"github.com/micro/go-micro/broker"
 	"github.com/micro/go-micro/codec"
 	"github.com/micro/go-micro/registry"
@@ -32,6 +34,10 @@ type Options struct {
 	Context context.Context
 }
 
+type RegisterOptions struct {
+	TTL time.Duration
+}
+
 func newOptions(opt ...Option) Options {
 	opts := Options{
 		Codecs:   make(map[string]codec.NewCodec),
@@ -167,3 +173,10 @@ func WrapSubscriber(w SubscriberWrapper) Option {
 		o.SubWrappers = append(o.SubWrappers, w)
 	}
 }
+
+// Register the service with a TTL
+func RegisterTTL(t time.Duration) RegisterOption {
+	return func(o *RegisterOptions) {
+		o.TTL = t
+	}
+}

server/rpc_server.go

@@ -155,7 +155,15 @@ func (s *rpcServer) Subscribe(sb Subscriber) error {
 	return nil
 }
 
-func (s *rpcServer) Register() error {
+func (s *rpcServer) Register(opts ...RegisterOption) error {
+	var options RegisterOptions
+	for _, o := range opts {
+		o(&options)
+	}
+
+	// create registry options
+	rOpts := []registry.RegisterOption{registry.WithTTL(options.TTL)}
+
 	// parse address for host, port
 	config := s.Options()
 	var advt, host string
@@ -220,7 +228,7 @@ func (s *rpcServer) Register() error {
 	}
 
 	log.Infof("Registering node: %s", node.Id)
-	if err := config.Registry.Register(service); err != nil {
+	if err := config.Registry.Register(service, rOpts...); err != nil {
 		return err
 	}
 

server/server.go

@@ -45,7 +45,7 @@ type Server interface {
 	NewHandler(interface{}, ...HandlerOption) Handler
 	NewSubscriber(string, interface{}, ...SubscriberOption) Subscriber
 	Subscribe(Subscriber) error
-	Register() error
+	Register(...RegisterOption) error
 	Deregister() error
 	Start() error
 	Stop() error
@@ -86,6 +86,8 @@ type HandlerOption func(*HandlerOptions)
 
 type SubscriberOption func(*SubscriberOptions)
 
+type RegisterOption func(*RegisterOptions)
+
 var (
 	DefaultAddress        = ":0"
 	DefaultName           = "go-server"