prysm-pulse/proto/migration/enums.go
2021-07-21 21:34:07 +00:00

23 lines
755 B
Go

package migration
import (
"github.com/pkg/errors"
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1"
ethpb_alpha "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
)
func V1Alpha1ConnectionStateToV1(connState ethpb_alpha.ConnectionState) ethpb.ConnectionState {
alphaString := connState.String()
v1Value := ethpb.ConnectionState_value[alphaString]
return ethpb.ConnectionState(v1Value)
}
func V1Alpha1PeerDirectionToV1(peerDirection ethpb_alpha.PeerDirection) (ethpb.PeerDirection, error) {
alphaString := peerDirection.String()
if alphaString == ethpb_alpha.PeerDirection_UNKNOWN.String() {
return 0, errors.New("peer direction unknown")
}
v1Value := ethpb.PeerDirection_value[alphaString]
return ethpb.PeerDirection(v1Value), nil
}