fix returns

Author: shivaji-kharse

go.mod

@@ -9,7 +9,7 @@ require (
 	github.com/Masterminds/semver/v3 v3.3.1
 	github.com/blevesearch/bleve/v2 v2.4.4
 	github.com/dgraph-io/badger/v4 v4.6.0
-	github.com/dgraph-io/dgo/v250 v250.0.0-20250421044702-8f280c62988a
+	github.com/dgraph-io/dgo/v250 v250.0.0-20250422053155-49e7e4d58e3e
 	github.com/dgraph-io/gqlgen v0.13.2
 	github.com/dgraph-io/gqlparser/v2 v2.2.2
 	github.com/dgraph-io/graphql-transport-ws v0.0.0-20210511143556-2cef522f1f15

go.sum

@@ -122,8 +122,8 @@ github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dgraph-io/badger/v4 v4.6.0 h1:acOwfOOZ4p1dPRnYzvkVm7rUk2Y21TgPVepCy5dJdFQ=
 github.com/dgraph-io/badger/v4 v4.6.0/go.mod h1:KSJ5VTuZNC3Sd+YhvVjk2nYua9UZnnTr/SkXvdtiPgI=
-github.com/dgraph-io/dgo/v250 v250.0.0-20250421044702-8f280c62988a h1:hAb0lbE+vOYuLublMbhgBM5YEpW5p3+kWS5FpGp4IwI=
-github.com/dgraph-io/dgo/v250 v250.0.0-20250421044702-8f280c62988a/go.mod h1:h1cPCTmI0GZWc770q608FkWEdQfC4iiWYAgvQGnvQ9A=
+github.com/dgraph-io/dgo/v250 v250.0.0-20250422053155-49e7e4d58e3e h1:OQPIfYgx57zzqL86EtiF7J0KwyUHzqzwvwDXxMlD8oo=
+github.com/dgraph-io/dgo/v250 v250.0.0-20250422053155-49e7e4d58e3e/go.mod h1:h1cPCTmI0GZWc770q608FkWEdQfC4iiWYAgvQGnvQ9A=
 github.com/dgraph-io/gqlgen v0.13.2 h1:TNhndk+eHKj5qE7BenKKSYdSIdOGhLqxR1rCiMso9KM=
 github.com/dgraph-io/gqlgen v0.13.2/go.mod h1:iCOrOv9lngN7KAo+jMgvUPVDlYHdf7qDwsTkQby2Sis=
 github.com/dgraph-io/gqlparser/v2 v2.1.1/go.mod h1:MYS4jppjyx8b9tuUtjV7jU1UFZK6P9fvO8TsIsQtRKU=

worker/import.go

@@ -100,7 +100,7 @@ func streamToAnotherGroup(in apiv25.Dgraph_StreamSnapshotServer, out pb.Worker_S
 	}
 
 	// Send a close signal to the incoming stream.
-	if err := in.SendAndClose(&apiv25.ReceiveSnapshotKVRequest{Done: true}); err != nil {
+	if err := in.SendAndClose(&apiv25.StreamSnapshotResponse{Done: true}); err != nil {
 		return fmt.Errorf("failed to send: %v", err)
 	}
 
@@ -171,7 +171,7 @@ func FlushKvs(stream apiv25.Dgraph_StreamSnapshotServer) error {
 
 	glog.Infof("P dir writes DONE. Sending ACK")
 	// Send an acknowledgement back to the leader.
-	if err := stream.SendAndClose(&apiv25.ReceiveSnapshotKVRequest{Done: true}); err != nil {
+	if err := stream.SendAndClose(&apiv25.StreamSnapshotResponse{Done: true}); err != nil {
 		return err
 	}