(@SnapshotId uniqueidentifier,@IsPermanent bit,@ChunkName nvarchar(36),@ChunkType int,@ChunkId uniqueidentifier output,@ChunkFlags tinyint output,@MimeType nvarchar(260) output)
if (@IsPermanent = 1) begin		
	select	@ChunkId = ChunkId,
			@ChunkFlags = ChunkFlags,
            @MimeType = MimeType
	from dbo.SegmentedChunk chunk
	where chunk.SnapshotDataId = @SnapshotId and chunk.ChunkName = @ChunkName and chunk.ChunkType = @ChunkType
	
	select	csm.SegmentId, 				
			csm.LogicalByteCount as LogicalSegmentLength, 
			csm.ActualByteCount as ActualSegmentLength		
	from ChunkSegmentMapping csm		
	where csm.ChunkId = @ChunkId
	order by csm.StartByte asc
end
else begin
	select	@ChunkId = ChunkId,
			@ChunkFlags = ChunkFlags,
            @MimeType = MimeType
	from [ReportServerTempDB].dbo.SegmentedChunk chunk
	where chunk.SnapshotDataId = @SnapshotId and chunk.ChunkName = @ChunkName and chunk.ChunkType = @ChunkType
	
	if @ChunkFlags & 0x4 > 0 begin
		-- Shallow copy: read chunk segments from catalog 
		select	csm.SegmentId, 				
				csm.LogicalByteCount as LogicalSegmentLength, 
				csm.ActualByteCount as ActualSegmentLength		
		from ChunkSegmentMapping csm		
		where csm.ChunkId = @ChunkId
		order by csm.StartByte asc
	end
	else begin
		-- Regular copy: read chunk segments from temp db
		select	csm.SegmentId, 				
				csm.LogicalByteCount as LogicalSegmentLength, 
				csm.ActualByteCount as ActualSegmentLength		
		from [ReportServerTempDB].dbo.ChunkSegmentMapping csm		
		where csm.ChunkId = @ChunkId
		order by csm.StartByte asc
	end
end

CREATE PROCEDURE [dbo].[WriteLockSession]
@SessionID as varchar(32),
@Persisted bit,
@CheckLockVersion bit = 0,
@LockVersion int
AS
SET NOCOUNT OFF ;
IF @Persisted = 1
BEGIN
	IF @CheckLockVersion = 0
	BEGIN
		UPDATE [ReportServerTempDB].dbo.SessionLock WITH (ROWLOCK)
		SET SessionID = SessionID
		WHERE SessionID = @SessionID;
	END
	ELSE
	BEGIN
		DECLARE @ActualLockVersion as int

		UPDATE [ReportServerTempDB].dbo.SessionLock WITH (ROWLOCK)
		SET SessionID = SessionID,
		LockVersion = LockVersion + 1
		WHERE SessionID = @SessionID
		AND LockVersion = @LockVersion ;

		IF (@@ROWCOUNT = 0)
		BEGIN
			SELECT @ActualLockVersion = LockVersion
			FROM [ReportServerTempDB].dbo.SessionLock WITH (ROWLOCK)
			WHERE SessionID = @SessionID;

			IF (@ActualLockVersion <> @LockVersion)
				RAISERROR ('Invalid version locked', 16,1)
			END
		END
	END
ELSE
BEGIN
	INSERT INTO [ReportServerTempDB].dbo.SessionLock WITH (ROWLOCK) (SessionID) VALUES (@SessionID)
END