向并行代码添加进度条(Haskell)

import Control.Concurrent
import Control.Concurrent.STM
import System.Console.ANSI

-- 并行任务的示例函数
parallelTask :: Int -> IO ()
parallelTask n = do
  -- 在控制台上显示进度条
  setCursorPosition 0 0
  putStr $ "Progress: [          ]"

  -- 创建共享变量并初始化为0
  progress <- newTVarIO 0

  -- 创建并行任务的线程
  forkIO $ do
    -- 执行并行任务
    performTask n progress

    -- 任务完成后清除进度条
    clearProgressBar

  -- 更新进度条直到任务完成
  updateProgressBar progress

-- 执行并行任务的函数
performTask :: Int -> TVar Int -> IO ()
performTask n progress = do
  -- 并行任务的代码
  -- 在关键点更新进度变量
  atomically $ modifyTVar progress (+1)

-- 更新进度条的函数
updateProgressBar :: TVar Int -> IO ()
updateProgressBar progress = do
  -- 读取进度变量的值
  currentProgress <- readTVarIO progress

  -- 更新进度条的显示
  setCursorPosition 0 11
  putStr $ replicate currentProgress '#'

  -- 如果任务未完成，则继续更新进度条
  if currentProgress < 10
    then updateProgressBar progress
    else return ()

-- 清除进度条的函数
clearProgressBar :: IO ()
clearProgressBar = do
  setCursorPosition 0 0
  clearLine

-- 示例调用
main :: IO ()
main = parallelTask 10