reflect breaking API changes

hashicorp · Jul 20, 2021 · f2350b1 · f2350b1
1 parent 3cca140
commit f2350b1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/internal/langserver/handlers/service.go b/internal/langserver/handlers/service.go
@@ -416,7 +416,7 @@ func (svc *service) Assigner() (jrpc2.Assigner, error) {
 	return convertMap(m), nil
 }
 
-func (svc *service) Finish(status jrpc2.ServerStatus) {
+func (svc *service) Finish(_ jrpc2.Assigner, status jrpc2.ServerStatus) {
 	if status.Closed() || status.Err != nil {
 		svc.logger.Printf("session stopped unexpectedly (err: %v)", status.Err)
 	}

diff --git a/internal/langserver/langserver.go b/internal/langserver/langserver.go
@@ -160,8 +160,10 @@ func Server(svc server.Service, opts *jrpc2.ServerOptions) (*singleServer, error
 	}
 
 	return &singleServer{
-		srv:        jrpc2.NewServer(assigner, opts),
-		finishFunc: svc.Finish,
+		srv: jrpc2.NewServer(assigner, opts),
+		finishFunc: func(status jrpc2.ServerStatus) {
+			svc.Finish(assigner, status)
+		},
 	}, nil
 }
 

diff --git a/internal/langserver/session/types.go b/internal/langserver/session/types.go
@@ -9,7 +9,7 @@ import (
 
 type Session interface {
 	Assigner() (jrpc2.Assigner, error)
-	Finish(jrpc2.ServerStatus)
+	Finish(jrpc2.Assigner, jrpc2.ServerStatus)
 	SetLogger(*log.Logger)
 }