Forwarding grpc stream messages over websocket - go

Question

I want to continuously feed the websocket via grpc stream(server stream).
I need to terminate the stream when the websocket connection down. But I couldn’t find any way out, unfortunately. I tried to use chan, but it didn't go the way I wanted.
if anyone has any ideas, I’d love to hear them. Thank you.
SERVER:
done := make(chan bool)
go func() {
for {
response, err := req.RequestToServiceWithTrace()
if err != nil {
fmt.Println("Error getting")
}
if err := srv.Send(response); err != nil {
log.Printf("send error %v", err)
done <- true
}
log.Printf("finishing request number : %d", 1)
time.Sleep(1 * time.Second)
}
}()
<-done
CLIENT:
wsupgrader.CheckOrigin = func(r *http.Request) bool { return true }
connWebSocket, err := wsupgrader.Upgrade(w, r, nil)
if err != nil {
fmt.Printf("Failed to set websocket upgrade: %s", err.Error())
return
}
var stream trace.StreamService_GetResponseTimeClient
for {
done := make(chan bool)
t, msg, errWebsocket := connWebSocket.ReadMessage()
if errWebsocket != nil {
fmt.Println(errWebsocket.Error(), "ssssssssssss")
done <- true
}
stream = sada(msg)
fmt.Println("vvvvvvvvv")
go func() {
for {
select {
case _, ok := <-done:
if ok {
fmt.Println("ssscccsssccscscscscscss")
return
} else {
fmt.Println("Channel closed!")
}
default:
resp, err := stream.Recv()
if err != nil {
return
}
log.Printf("Resp received: %s", resp)
jsonBytes, _ := protojson.Marshal(resp)
connWebSocket.WriteMessage(t, jsonBytes)
time.Sleep(1 * time.Second)
}
}
}()
}