在 Kubernetes 中运行 Locust 与 Selenium：安装 Chrome 和 ChromeDriver

FROM locustio/locust:2.31.3

# 设置 Chrome 的版本
ENV CHROME_VERSION 130.0.6723.69

USER root 
RUN apt-get update -qq -y && \
    apt-get install -y wget unzip && \
    apt-get install -y \
        libasound2 \
        libatk-bridge2.0-0 \
        libgtk-4-1 \
        libnss3 \
        xdg-utils && \
    wget -q -O chrome-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chrome-linux64.zip && \
    unzip chrome-linux64.zip && \
    rm chrome-linux64.zip && \
    mv chrome-linux64 /opt/chrome/ && \
    ln -s /opt/chrome/chrome /usr/local/bin/ && \
    wget -q -O chromedriver-linux64.zip https://storage.googleapis.com/chrome-for-testing-public/$CHROME_VERSION/linux64/chromedriver-linux64.zip && \
    unzip -j chromedriver-linux64.zip chromedriver-linux64/chromedriver && \
    rm chromedriver-linux64.zip && \
    mv chromedriver /usr/local/bin/

# 设置 Chrome 的配置和缓存目录
ENV XDG_CONFIG_HOME=/tmp/.chromium
ENV XDG_CACHE_HOME=/tmp/.chromium

COPY . .
RUN pip install -r requirements.txt

import platform
from selenium import webdriver

def is_running_in_linux():
    return platform.system() == 'Linux'

def get_chrome_options():
    is_in_linux = is_running_in_linux()
    options_custom = webdriver.ChromeOptions()

    # Linux 下的 Chrome 选项
    if is_in_linux:
        options_custom.add_argument("--headless")  # 无头模式
        options_custom.add_argument('--disable-gpu')  # 禁用 GPU 加速
        options_custom.add_argument("--no-sandbox")  # 禁用沙箱模式
    else:
        options_custom.add_argument("--start-maximized")  # 启动时最大化窗口

    # 其他通用选项
    options_custom.add_argument("--disable-dev-shm-usage")  # 解决资源限制问题
    options_custom.add_argument("--ignore-ssl-errors=yes")  # 忽略 SSL 错误
    options_custom.add_argument("--disable-cache")  # 禁用缓存 

    return options_custom

from locust import User, task

class GetUrl(User):    
    customdriver = None  

    def on_start(self):               
        self.customdriver = webdriver.Chrome(options=get_chrome_options())

    @task
    def load_page(self):
        self.customdriver.get("http://example.com")  # 根据需要替换为实际 URL

apiVersion: apps/v1
kind: Deployment
metadata:
  name: locust
spec:
  replicas: 1
  selector:
    matchLabels:
      app: locust
  template:
    metadata:
      labels:
        app: locust
    spec:
      containers:
        - name: locust
          image: your-docker-image:latest
          ports:
            - containerPort: 8089
          env:
            - name: XDG_CONFIG_HOME
              value: /tmp/.chromium  # 设置 Chrome 的配置目录
            - name: XDG_CACHE_HOME
              value: /tmp/.chromium  # 设置 Chrome 的缓存目录
---
apiVersion: v1
kind: Service
metadata:
  name: locust-service
spec:
  type: NodePort
  ports:
    - port: 8089
      targetPort: 8089
  selector:
    app: locust