Method SetNodeInfo

data/key/grpc/handler.go:183–270

(stream encryption.NodeKeyManager_SetNodeInfoServer)

}

func (km *NodeKeyManagerHandler) SetNodeInfo(stream encryption.NodeKeyManager_SetNodeInfoServer) error {

	ctx := context.WithoutCancel(stream.Context())

	dao, err := manager.Resolve[key.DAO](ctx)
	if err != nil {
		return err
	}

	var rangedBlocks *encryption.RangedBlock
	var nodeUuid string
	received := make(map[string]int64, 4)

	for {

		req, rErr := stream.Recv()
		if rErr != nil {
			if rErr != io.EOF {
				log.Logger(ctx).Error("data.key.handler.SetNodeInfo: failed to read SetInfoRequest", zap.Error(rErr))
			}
			break
		}
		log.Logger(ctx).Debug("Key Stream Received " + req.GetAction().String())
		received[req.Action.String()]++

		switch req.Action {

		case encryption.SetNodeInfoActionType_KEY:

			if err = km.saveNodeKey(ctx, req.SetNodeKey.NodeKey); err != nil {
				log.Logger(ctx).Error("failed to save key", zap.Error(err))
			}

		case encryption.SetNodeInfoActionType_CLEAR:

			if err = dao.ClearNodeEncryptedBlockInfo(ctx, req.SetBlock.NodeUuid); err != nil {
				log.Logger(ctx).Error("failed to clear old blocks", zap.Error(err))
				return err
			}

			if err = dao.UpgradeNodeVersion(ctx, req.SetBlock.NodeUuid); err != nil {
				log.Logger(ctx).Error("failed to upgrade node version", zap.Error(err))
				return err
			}

		case encryption.SetNodeInfoActionType_BLOCK:
			if nodeUuid == "" {
				nodeUuid = req.SetBlock.NodeUuid
			}

			tmpRangeBlock := &encryption.RangedBlock{
				BlockSize:  req.SetBlock.Block.BlockSize,
				OwnerId:    req.SetBlock.Block.OwnerId,
				HeaderSize: req.SetBlock.Block.HeaderSize,
				PartId:     req.SetBlock.Block.PartId,
			}

			if rangedBlocks == nil {