Create Gorilla Websocket on Subscription

[vyj7]

Hi all,
I have a remote WebSocket server which I am wrapping into a subscription using gqlGen, Does anyone have any ideas on how I can achieve this?
Currently, I am using Gorilla Websocket, created in the subscription Resolver. The snippet is below.

UI is using Apollo Client.
Now when I comment <-ctx.Done() block , it works but when subscription breaks from UI, it wont clear off.
If I don't comment it done, it is getting deleted once the message is received.
Let me know the correct way of doing it.

`func (r *subscriptionResolver) Repo(ctx context.Context, id model.Long, cid []*model.Long) (<-chan *model.Repo, error) {
token := id.String() + ":" + strconv.Itoa(rand.Int())

mc := make(chan *model.Repo, 1)
r.RepoChannel.Store(token, mc)

urlString := "ws://url/subs"
log.Info("connecting to ::", urlString)
c, _, err := websocket.DefaultDialer.Dial(urlString, nil)

if err != nil {
	log.Panic("dial:", err)
	ctx.Done()
	return nil, err
}

input := model2.RepoSummary{
	id: id,
	cid:   cid,
}

reqBodyBytes := new(bytes.Buffer)
json.NewEncoder(reqBodyBytes).Encode(input)

//sending this to ws
c.WriteMessage(1, reqBodyBytes.Bytes())

go func() {
	for {
		mt, message, err := c.ReadMessage()
		log.Info(mt)
		if err != nil {
			log.Println("read:", err)
			ctx.Done()
			return
		}
		stats := gjson.Get(string(message), "stats")

		var metric model.metric
		error := json.Unmarshal([]byte(stats.String()), &metric)

		if error != nil {
			log.Error(error)
			return
		}

		yy := model.Repo{
			stats: &metric,
		}

		mc <- &yy

		log.Printf("recv: %s", message)
		<-ctx.Done()
		{
			c.Close()
			r.RepoChannel.Delete(token)
			log.Println("Subscription Repo Deleted ChannelId: ", token)
			return
		}
	}
}()

log.Info("Subscription Repo Created ChannelId: ", token)
return mc, nil`