subscribe_runtime_version.go

// Go Substrate RPC Client (GSRPC) provides APIs and types around Polkadot and any Substrate-based chain RPC calls
//
// Copyright 2019 Centrifuge GmbH
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package state //nolint:dupl

import (
	"context"
	"sync"

	"github.com/puneetjindal1997/go-substrate-rpc-client/v4/config"
	gethrpc "github.com/puneetjindal1997/go-substrate-rpc-client/v4/gethrpc"
	"github.com/puneetjindal1997/go-substrate-rpc-client/v4/types"
)

// RuntimeVersionSubscription is a subscription established through one of the Client's subscribe methods.
type RuntimeVersionSubscription struct {
	sub      *gethrpc.ClientSubscription
	channel  chan types.RuntimeVersion
	quitOnce sync.Once // ensures quit is closed once
}

// Chan returns the subscription channel.
//
// The channel is closed when Unsubscribe is called on the subscription.
func (s *RuntimeVersionSubscription) Chan() <-chan types.RuntimeVersion {
	return s.channel
}

// Err returns the subscription error channel. The intended use of Err is to schedule
// resubscription when the client connection is closed unexpectedly.
//
// The error channel receives a value when the subscription has ended due
// to an error. The received error is nil if Close has been called
// on the underlying client and no other error has occurred.
//
// The error channel is closed when Unsubscribe is called on the subscription.
func (s *RuntimeVersionSubscription) Err() <-chan error {
	return s.sub.Err()
}

// Unsubscribe unsubscribes the notification and closes the error channel.
// It can safely be called more than once.
func (s *RuntimeVersionSubscription) Unsubscribe() {
	s.sub.Unsubscribe()
	s.quitOnce.Do(func() {
		close(s.channel)
	})
}

// SubscribeRuntimeVersion subscribes the runtime version, returning a subscription that will
// receive server notifications containing the RuntimeVersion.
func (s *state) SubscribeRuntimeVersion() (
	*RuntimeVersionSubscription, error) {
	ctx, cancel := context.WithTimeout(context.Background(), config.Default().SubscribeTimeout)
	defer cancel()

	c := make(chan types.RuntimeVersion)

	sub, err := s.client.Subscribe(ctx, "state", "subscribeRuntimeVersion", "unsubscribeRuntimeVersion",
		"runtimeVersion", c)
	if err != nil {
		return nil, err
	}

	return &RuntimeVersionSubscription{sub: sub, channel: c}, nil
}