Repository: gocronx-team/gocron Branch: master Commit: 03e18f086ea5 Files: 230 Total size: 854.8 KB Directory structure: gitextract_vqi6y207/ ├── .air.toml ├── .dockerignore ├── .gitattributes ├── .github/ │ ├── dependabot.yml │ └── workflows/ │ ├── ci.yml │ ├── helm-release.yml │ └── release.yml ├── .gitignore ├── .goreleaser.yml ├── .husky/ │ ├── commit-msg │ └── pre-commit ├── .prettierignore ├── .prettierrc ├── CLAUDE.md ├── Dockerfile.gocron ├── LICENSE ├── README.md ├── README_ZH.md ├── app.ini.sqlite.example ├── cmd/ │ ├── gocron/ │ │ └── gocron.go │ └── node/ │ └── node.go ├── commitlint.config.cjs ├── docker-compose.yml ├── embed.go ├── go.mod ├── go.sum ├── helm/ │ └── gocron/ │ ├── Chart.yaml │ ├── templates/ │ │ ├── NOTES.txt │ │ ├── _helpers.tpl │ │ ├── configmap.yaml │ │ ├── deployment.yaml │ │ ├── ingress.yaml │ │ ├── pvc.yaml │ │ ├── service.yaml │ │ └── serviceaccount.yaml │ └── values.yaml ├── internal/ │ ├── models/ │ │ ├── agent_token.go │ │ ├── audit_log.go │ │ ├── audit_log_test.go │ │ ├── cleanup_verify_test.go │ │ ├── host.go │ │ ├── login_log.go │ │ ├── migration.go │ │ ├── model.go │ │ ├── scheduler_lock.go │ │ ├── scheduler_lock_test.go │ │ ├── setting.go │ │ ├── setting_init.go │ │ ├── setting_refactor_test.go │ │ ├── task.go │ │ ├── task_host.go │ │ ├── task_log.go │ │ ├── task_log_test.go │ │ ├── task_optimization_test.go │ │ ├── task_retention_test.go │ │ ├── task_script_version.go │ │ ├── task_script_version_test.go │ │ ├── task_tag_test.go │ │ ├── task_template.go │ │ ├── task_template_test.go │ │ ├── user.go │ │ └── webhook_test.go │ ├── modules/ │ │ ├── app/ │ │ │ ├── app.go │ │ │ └── app_test.go │ │ ├── httpclient/ │ │ │ ├── http_client.go │ │ │ ├── http_client_benchmark_test.go │ │ │ └── http_client_test.go │ │ ├── i18n/ │ │ │ ├── en_us.go │ │ │ ├── i18n.go │ │ │ └── zh_cn.go │ │ ├── leader/ │ │ │ ├── election.go │ │ │ └── election_test.go │ │ ├── logger/ │ │ │ ├── async_logger.go │ │ │ ├── async_logger_test.go │ │ │ ├── compatibility_test.go │ │ │ ├── logger.go │ │ │ ├── logger_test.go │ │ │ ├── performance_report_test.go │ │ │ └── performance_test.go │ │ ├── notify/ │ │ │ ├── mail.go │ │ │ ├── notify.go │ │ │ ├── notify_test.go │ │ │ ├── slack.go │ │ │ ├── webhook.go │ │ │ └── webhook_test.go │ │ ├── rpc/ │ │ │ ├── auth/ │ │ │ │ └── Certification.go │ │ │ ├── client/ │ │ │ │ └── client.go │ │ │ ├── grpcpool/ │ │ │ │ └── grpc_pool.go │ │ │ ├── proto/ │ │ │ │ ├── task.pb.go │ │ │ │ ├── task.proto │ │ │ │ └── task_grpc.pb.go │ │ │ └── server/ │ │ │ └── server.go │ │ ├── setting/ │ │ │ ├── setting.go │ │ │ └── setting_test.go │ │ └── utils/ │ │ ├── execshell_integration_test.go │ │ ├── execshell_test.go │ │ ├── html_entity.go │ │ ├── html_entity_test.go │ │ ├── json.go │ │ ├── login_limiter.go │ │ ├── login_limiter_test.go │ │ ├── password.go │ │ ├── password_test.go │ │ ├── utils.go │ │ ├── utils_test.go │ │ ├── utils_unix.go │ │ ├── utils_unix_test.go │ │ ├── utils_windows.go │ │ └── utils_windows_test.go │ ├── routers/ │ │ ├── agent/ │ │ │ └── agent.go │ │ ├── audit/ │ │ │ ├── audit.go │ │ │ └── audit_test.go │ │ ├── base/ │ │ │ ├── base.go │ │ │ └── response.go │ │ ├── host/ │ │ │ └── host.go │ │ ├── install/ │ │ │ └── install.go │ │ ├── loginlog/ │ │ │ └── login_log.go │ │ ├── manage/ │ │ │ └── manage.go │ │ ├── routers.go │ │ ├── statistics/ │ │ │ └── statistics.go │ │ ├── task/ │ │ │ ├── cron_preview.go │ │ │ ├── task.go │ │ │ ├── task_tag_test.go │ │ │ └── task_version.go │ │ ├── tasklog/ │ │ │ ├── task_log.go │ │ │ └── task_log_test.go │ │ ├── template/ │ │ │ └── template.go │ │ └── user/ │ │ ├── twofa.go │ │ └── user.go │ └── service/ │ ├── cron_preview.go │ ├── cron_preview_test.go │ ├── issue66_test.go │ ├── single_instance_test.go │ ├── task.go │ ├── task_cleanup_test.go │ ├── task_partial_output_test.go │ └── task_test.go ├── makefile ├── package.json ├── package.sh ├── release.sh ├── test_windows_cmd.go ├── web/ │ └── vue/ │ ├── .editorconfig │ ├── .gitattributes │ ├── .gitignore │ ├── .prettierrc.json │ ├── README.md │ ├── eslint.config.js │ ├── index.html │ ├── jsconfig.json │ ├── package.json │ ├── src/ │ │ ├── App.vue │ │ ├── api/ │ │ │ ├── agent.js │ │ │ ├── audit.js │ │ │ ├── host.js │ │ │ ├── install.js │ │ │ ├── notification.js │ │ │ ├── statistics.js │ │ │ ├── system.js │ │ │ ├── task.js │ │ │ ├── taskLog.js │ │ │ ├── template.js │ │ │ └── user.js │ │ ├── components/ │ │ │ └── common/ │ │ │ ├── CronInput.vue │ │ │ ├── CronPreview.vue │ │ │ ├── HeatmapSvg.vue │ │ │ ├── LanguageSwitcher.vue │ │ │ ├── MonacoEditor.vue │ │ │ ├── footer.vue │ │ │ ├── header.vue │ │ │ ├── navMenu.vue │ │ │ ├── notFound.vue │ │ │ └── sidebar.vue │ │ ├── composables/ │ │ │ ├── __tests__/ │ │ │ │ ├── useDebounce.spec.js │ │ │ │ ├── useLoading.spec.js │ │ │ │ └── useMessage.spec.js │ │ │ ├── useDebounce.js │ │ │ ├── useLoading.js │ │ │ └── useMessage.js │ │ ├── const/ │ │ │ ├── index.js │ │ │ └── lang.js │ │ ├── locales/ │ │ │ ├── en-US.js │ │ │ ├── index.js │ │ │ └── zh-CN.js │ │ ├── main.js │ │ ├── pages/ │ │ │ ├── host/ │ │ │ │ ├── edit.vue │ │ │ │ └── list.vue │ │ │ ├── install/ │ │ │ │ └── index.vue │ │ │ ├── statistics/ │ │ │ │ └── index.vue │ │ │ ├── system/ │ │ │ │ ├── auditLog.vue │ │ │ │ ├── logRetention.vue │ │ │ │ ├── loginLog.vue │ │ │ │ ├── notification/ │ │ │ │ │ ├── email.vue │ │ │ │ │ ├── slack.vue │ │ │ │ │ ├── tab.vue │ │ │ │ │ └── webhook.vue │ │ │ │ └── sidebar.vue │ │ │ ├── task/ │ │ │ │ ├── edit.vue │ │ │ │ ├── list.vue │ │ │ │ └── sidebar.vue │ │ │ ├── taskLog/ │ │ │ │ └── list.vue │ │ │ ├── template/ │ │ │ │ ├── edit.vue │ │ │ │ └── list.vue │ │ │ └── user/ │ │ │ ├── edit.vue │ │ │ ├── editMyPassword.vue │ │ │ ├── editPassword.vue │ │ │ ├── list.vue │ │ │ ├── login.vue │ │ │ └── twoFactor.vue │ │ ├── router/ │ │ │ └── index.js │ │ ├── storage/ │ │ │ └── user.js │ │ ├── stores/ │ │ │ └── user.js │ │ └── utils/ │ │ ├── __tests__/ │ │ │ ├── cronValidator.spec.js │ │ │ └── env.spec.js │ │ ├── cronValidator.js │ │ ├── env.js │ │ ├── httpClient.js │ │ ├── performance.js │ │ ├── progress/ │ │ │ └── index.js │ │ └── request.js │ ├── static/ │ │ ├── .gitkeep │ │ └── robots.txt │ ├── verify.sh │ ├── vite.config.js │ └── vitest.config.js └── webhook-test/ ├── go.mod ├── go.sum ├── start-webhook-server.sh ├── test-webhook.sh └── webhook-test-server.go ================================================ FILE CONTENTS ================================================ ================================================ FILE: .air.toml ================================================ root = "." testdata_dir = "testdata" tmp_dir = "tmp" [build] args_bin = ["web", "-e", "dev"] bin = "./tmp/gocron" cmd = "go build -o ./tmp/gocron ./cmd/gocron" delay = 2000 exclude_dir = ["assets", "tmp", "vendor", "testdata", "web/vue/node_modules", "web/public", "web/vue/dist"] exclude_file = [] exclude_regex = ["_test.go"] exclude_unchanged = false follow_symlink = false full_bin = "" include_dir = [] include_ext = ["go", "tpl", "tmpl", "html"] include_file = [] kill_delay = "5s" log = "build-errors.log" poll = false poll_interval = 0 rerun = false rerun_delay = 2000 send_interrupt = true stop_on_error = true [color] app = "" build = "yellow" main = "magenta" runner = "green" watcher = "cyan" [log] main_only = false time = false [misc] clean_on_exit = false [screen] clear_on_rebuild = false keep_scroll = true ================================================ FILE: .dockerignore ================================================ .git .github web/vue/node_modules web/vue/dist bin gocron-package gocron-node-package *.md .gitignore .gitattributes .dockerignore ================================================ FILE: .gitattributes ================================================ *.js linguist-language=go *.css linguist-language=go *.html linguist-language=go ================================================ FILE: .github/dependabot.yml ================================================ version: 2 updates: - package-ecosystem: gomod directory: / schedule: interval: weekly groups: go-deps: patterns: ["*"] - package-ecosystem: npm directory: /web/vue schedule: interval: weekly groups: npm-deps: patterns: ["*"] - package-ecosystem: npm directory: / schedule: interval: weekly groups: npm-dev-deps: patterns: ["*"] - package-ecosystem: docker directory: / schedule: interval: monthly - package-ecosystem: github-actions directory: / schedule: interval: monthly ignore: - dependency-name: "actions/*" update-types: ["version-update:semver-major"] - dependency-name: "pnpm/*" update-types: ["version-update:semver-major"] ================================================ FILE: .github/workflows/ci.yml ================================================ name: CI on: push: branches: [master] pull_request: branches: [master] permissions: contents: read jobs: go-test: name: Go Test runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 - uses: actions/setup-go@v5 with: go-version-file: go.mod - name: Create frontend dist placeholder run: mkdir -p web/vue/dist && touch web/vue/dist/.gitkeep - name: Format check run: | unformatted=$(gofmt -l .) if [ -n "$unformatted" ]; then echo "::error::Unformatted files:" && echo "$unformatted" && exit 1 fi - name: Vet run: go vet ./... - name: Test run: go test -race -coverprofile=coverage.out ./... - name: Upload coverage if: github.event_name == 'push' uses: actions/upload-artifact@v4 with: name: coverage path: coverage.out frontend-build: name: Frontend Build runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 - uses: pnpm/action-setup@v4 with: version: 10 - uses: actions/setup-node@v4 with: node-version: 20 cache: pnpm cache-dependency-path: web/vue/pnpm-lock.yaml - name: Install & Build working-directory: web/vue run: | pnpm install --frozen-lockfile pnpm build - name: Lint working-directory: web/vue run: pnpm lint:check docker-build: name: Docker Build runs-on: ubuntu-latest needs: [go-test, frontend-build] steps: - uses: actions/checkout@v4 - name: Build image run: docker build -f Dockerfile.gocron -t gocron:ci . ================================================ FILE: .github/workflows/helm-release.yml ================================================ name: Release Helm Chart on: push: branches: - master paths: - "helm/**" jobs: release: runs-on: ubuntu-latest permissions: contents: write steps: - uses: actions/checkout@v4 with: fetch-depth: 0 - name: Configure Git run: | git config user.name "$GITHUB_ACTOR" git config user.email "$GITHUB_ACTOR@users.noreply.github.com" - uses: azure/setup-helm@v4 - uses: helm/chart-releaser-action@v1.7.0 with: charts_dir: helm env: CR_TOKEN: "${{ secrets.GITHUB_TOKEN }}" ================================================ FILE: .github/workflows/release.yml ================================================ name: Release on: push: tags: - "v*" permissions: contents: write jobs: frontend: name: Build Frontend runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 - uses: pnpm/action-setup@v4 with: version: 10 - uses: actions/setup-node@v4 with: node-version: 20 cache: pnpm cache-dependency-path: web/vue/pnpm-lock.yaml - name: Install & Build working-directory: web/vue run: | pnpm install --frozen-lockfile pnpm build - uses: actions/upload-artifact@v4 with: name: frontend-dist path: web/vue/dist/ release: name: GoReleaser needs: frontend runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 with: fetch-depth: 0 - uses: actions/setup-go@v5 with: go-version-file: go.mod - uses: actions/download-artifact@v4 with: name: frontend-dist path: web/vue/dist/ - uses: goreleaser/goreleaser-action@v6 with: version: latest args: release --clean env: GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} ================================================ FILE: .gitignore ================================================ # Compiled Object files, Static and Dynamic libs (Shared Objects) *.o *.a *.so # Folders _obj _test # Architecture specific extensions/prefixes *.[568vq] [568vq].out *.cgo1.go *.cgo2.c _cgo_defun.c _cgo_gotypes.go _cgo_export.* _testmain.go *.exe *.test *.prof .DS_Store .idea log data conf profile/* /gocron /gocron-node /bin /web/public/static /web/public/index.html /gocron-package /gocron-node-package /dist node_modules tmp/ build-errors.log .gocache .gstack/ .design-review/ ================================================ FILE: .goreleaser.yml ================================================ version: 2 before: hooks: - go mod tidy builds: - id: gocron main: ./cmd/gocron binary: gocron env: - CGO_ENABLED=0 goos: - linux - darwin - windows goarch: - amd64 - arm64 ignore: - goos: windows goarch: arm64 ldflags: - -s -w - -X 'main.AppVersion={{ .Version }}' - -X 'main.BuildDate={{ .Date }}' - -X 'main.GitCommit={{ .ShortCommit }}' - id: gocron-node main: ./cmd/node binary: gocron-node env: - CGO_ENABLED=0 goos: - linux - darwin - windows goarch: - amd64 - arm64 ignore: - goos: windows goarch: arm64 ldflags: - -s -w - -X 'main.AppVersion={{ .Version }}' - -X 'main.BuildDate={{ .Date }}' - -X 'main.GitCommit={{ .ShortCommit }}' archives: - id: gocron ids: - gocron name_template: "gocron-{{ .Version }}-{{ .Os }}-{{ .Arch }}" formats: - tar.gz format_overrides: - goos: windows formats: - zip - id: gocron-node ids: - gocron-node name_template: "gocron-node-{{ .Os }}-{{ .Arch }}" formats: - tar.gz format_overrides: - goos: windows formats: - zip checksum: name_template: "checksums.txt" changelog: sort: asc filters: exclude: - "^docs:" - "^style:" - "^chore\\(deps\\):" groups: - title: "New Features" regexp: '^feat' - title: "Bug Fixes" regexp: '^fix' - title: "Performance" regexp: '^perf' - title: "Others" order: 999 release: github: owner: gocronx-team name: gocron draft: false prerelease: auto ================================================ FILE: .husky/commit-msg ================================================ npx --no -- commitlint --edit $1 ================================================ FILE: .husky/pre-commit ================================================ pnpm run lint:lint-staged ================================================ FILE: .prettierignore ================================================ node_modules dist build .git .gocache .gocron bin data log tmp *.min.js *.min.css vendor ================================================ FILE: .prettierrc ================================================ { "semi": false, "singleQuote": true, "printWidth": 100, "trailingComma": "none", "arrowParens": "avoid", "endOfLine": "lf" } ================================================ FILE: CLAUDE.md ================================================ # gocron ## Project Overview A lightweight, distributed scheduled task management system written in Go with a Vue.js web interface. ## Tech Stack - **Backend:** Go 1.26, Gin, GORM (MySQL/PostgreSQL/SQLite) - **Frontend:** Vue 3 (Options API), Element Plus, vue-i18n, Vite, pnpm - **RPC:** gRPC + Protocol Buffers - **Auth:** JWT + TOTP 2FA ## Development ```bash # Backend (with hot reload) air # Frontend dev server cd web/vue && pnpm dev # Build frontend cd web/vue && pnpm build # Run tests go test ./... # Build go build ./... ``` ## Project Structure ``` cmd/gocron/ - Main entry point internal/ models/ - GORM data models routers/ - Gin HTTP handlers (grouped by domain) service/ - Business logic (scheduler, execution) modules/ - Utilities (logger, i18n, notify, RPC) web/vue/ - Vue.js frontend src/api/ - API client services src/pages/ - Page components src/components/ - Shared components src/locales/ - i18n (zh-CN, en-US) src/router/ - Vue Router config src/stores/ - Pinia stores ``` ## Conventions - Commit messages follow Conventional Commits: `feat:`, `fix:`, `chore:`, `refactor:`, `style:`, `test:` - Do not add `Co-Authored-By` lines in commit messages - Backend i18n: `internal/modules/i18n/zh_cn.go` and `en_us.go` - Frontend i18n: `web/vue/src/locales/zh-CN.js` and `en-US.js` - Database migrations: `internal/models/migration.go` (sequential version IDs) ================================================ FILE: Dockerfile.gocron ================================================ # Frontend build stage FROM node:20-alpine AS frontend RUN corepack enable && corepack prepare pnpm@latest --activate WORKDIR /app/web/vue COPY web/vue/package.json web/vue/pnpm-lock.yaml ./ RUN pnpm install --frozen-lockfile COPY web/vue ./ RUN pnpm build # Backend build stage FROM golang:1.26-alpine AS builder RUN apk add --no-cache git WORKDIR /app COPY go.mod go.sum ./ RUN go mod download # Copy frontend build files first COPY --from=frontend /app/web/vue/dist ./web/vue/dist # Then copy the rest of the project COPY . . # Build with pure Go SQLite (no CGO required) RUN CGO_ENABLED=0 go build -ldflags="-s -w" -trimpath -o gocron ./cmd/gocron FROM alpine:latest RUN apk add --no-cache ca-certificates tzdata COPY --from=builder /app/gocron /gocron EXPOSE 5920 ENTRYPOINT ["/gocron", "web"] ================================================ FILE: LICENSE ================================================ MIT License Copyright (c) 2025 gocronx team Copyright (c) 2017 qiang.ou Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. ================================================ FILE: README.md ================================================ # gocron - Distributed scheduled Task Scheduler [![Release](https://img.shields.io/github/release/gocronx-team/gocron.svg?label=Release)](https://github.com/gocronx-team/gocron/releases) [![Downloads](https://img.shields.io/github/downloads/gocronx-team/gocron/total.svg)](https://github.com/gocronx-team/gocron/releases) [![License](https://img.shields.io/github/license/gocronx-team/gocron.svg)](https://github.com/gocronx-team/gocron/blob/master/LICENSE) English | [简体中文](README_ZH.md) A lightweight distributed scheduled task management system developed in Go, designed to replace Linux-crontab. ## 📖 Documentation Full documentation is available at: **[document](https://gocron-docs.pages.dev/en/)** - 🚀 [Quick Start](https://gocron-docs.pages.dev/en/guide/quick-start) - Installation and deployment guide - 🤖 [Agent Auto-Registration](https://gocron-docs.pages.dev/en/guide/agent-registration) - One-click task node deployment - ⚙️ [Configuration](https://gocron-docs.pages.dev/en/guide/configuration) - Detailed configuration guide - 🔌 [API Documentation](https://gocron-docs.pages.dev/en/guide/api) - API reference ## ✨ Features - **Web Interface**: Intuitive task management interface - **Second-level Precision**: Supports Crontab expressions with second precision - **High Availability**: Database-lock-based leader election, automatic failover in seconds - **Task Retry**: Configurable retry policies for failed tasks - **Task Dependency**: Supports task dependency configuration - **Access Control**: Comprehensive user and permission management - **2FA Security**: Two-Factor Authentication support - **Agent Auto-Registration**: One-click installation for Linux/macOS - **Multi-Database**: MySQL / PostgreSQL / SQLite support - **Log Management**: Complete execution logs with auto-cleanup - **Notifications**: Email, Slack, Webhook support ## 🚀 Quick Start (Docker) The easiest way to deploy is using Docker Compose: ```bash # 1. Clone the project git clone https://github.com/gocronx-team/gocron.git cd gocron # 2. Start services docker-compose up -d # 3. Access Web Interface # http://localhost:5920 ``` For more deployment methods (Binary, Development), please refer to the [Installation Guide](https://gocron-docs.pages.dev/en/guide/quick-start). ## 🔷 High Availability (Optional) Deploy multiple gocron instances pointing to the same **MySQL/PostgreSQL** database. Leader election is automatic — no extra configuration needed. SQLite runs in single-node mode. ```bash # Node 1 ./gocron web --port 5920 # Node 2 (same database) ./gocron web --port 5921 ``` See the [High Availability Guide](https://gocron-docs.pages.dev/en/guide/high-availability) for setup details, K8s deployment, and environment variable overrides. ## 📸 Screenshots

Scheduled Tasks
Scheduled Tasks

Agent Auto-Registration Task Management
Agent Auto-Registration Task Management
Statistics Notifications
Statistics Notifications
## 🤝 Contributing We warmly welcome community contributions! ### How to Contribute 1. **Fork the repository** 2. **Clone your fork** ```bash git clone https://github.com/YOUR_USERNAME/gocron.git cd gocron ``` 3. **Install dependencies** ```bash pnpm install pnpm run prepare ``` 4. **Create a feature branch** ```bash git checkout -b feature/your-feature-name ``` 5. **Make your changes and commit** ```bash git add . pnpm run commit # Use interactive commit tool ``` 6. **Push and create a Pull Request** ```bash git push origin feature/your-feature-name ``` ### Commit Message Guidelines This project uses [commitizen](https://github.com/commitizen/cz-cli) and [cz-git](https://cz-git.qbb.sh/) for standardized commit messages. Instead of `git commit`, use: ```bash pnpm run commit ``` This will guide you through an interactive prompt to create properly formatted commit messages like: - `feat(task): add task dependency configuration` - `fix(api): fix task status update issue` - `docs: update API documentation` ### Other Ways to Contribute - 🐛 **Report Bugs**: Please submit via GitHub Issues - 💡 **Feature Requests**: Share your ideas through Issues - 📝 **Documentation**: Help improve our documentation ## 📄 License This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details. ## Star History [![Star History Chart](https://api.star-history.com/svg?repos=gocronx-team/gocron&type=Date)](https://www.star-history.com/#gocronx-team/gocron&Date) ================================================ FILE: README_ZH.md ================================================ # gocron - 分布式定时任务调度系统 [![Release](https://img.shields.io/github/release/gocronx-team/gocron.svg?label=Release)](https://github.com/gocronx-team/gocron/releases) [![Downloads](https://img.shields.io/github/downloads/gocronx-team/gocron/total.svg)](https://github.com/gocronx-team/gocron/releases) [![License](https://img.shields.io/github/license/gocronx-team/gocron.svg)](https://github.com/gocronx-team/gocron/blob/master/LICENSE) [English](README.md) | 简体中文 使用 Go 语言开发的轻量级分布式定时任务集中调度和管理系统，用于替代 Linux-crontab。 ## 📖 文档 访问完整文档请跳转：[文档](https://gocron-docs.pages.dev/zh/) - 🚀 [快速开始](https://gocron-docs.pages.dev/zh/guide/quick-start) - 安装部署指南 - 🤖 [Agent 自动注册](https://gocron-docs.pages.dev/zh/guide/agent-registration) - 一键部署任务节点 - ⚙️ [配置文件](https://gocron-docs.pages.dev/zh/guide/configuration) - 详细配置说明 - 🔌 [API 文档](https://gocron-docs.pages.dev/zh/guide/api) - API 接口说明 ## ✨ 功能特性 - **Web 界面管理**：直观的定时任务管理界面 - **秒级定时**：支持 Crontab 时间表达式，精确到秒 - **高可用**：基于数据库锁的 Leader 选举，秒级自动故障转移 - **任务重试**：支持任务执行失败重试设置 - **任务依赖**：支持配置任务依赖关系 - **多用户权限**：完善的用户和权限控制 - **双因素认证**：支持 2FA，提升系统安全性 - **Agent 自动注册**：支持 Linux/macOS 一键安装注册 - **多数据库支持**：MySQL / PostgreSQL / SQLite - **日志管理**：完整的任务执行日志，支持自动清理 - **消息通知**：支持邮件、Slack、Webhook 等多种通知方式 ## 🚀 快速开始 (Docker) 最简单的部署方式是使用 Docker Compose： ```bash # 1. 克隆项目 git clone https://github.com/gocronx-team/gocron.git cd gocron # 2. 启动服务 docker-compose up -d # 3. 访问 Web 界面 # http://localhost:5920 ``` 更多部署方式（二进制部署、开发环境）请查看 [安装部署指南](https://gocron-docs.pages.dev/zh/guide/quick-start)。 ## 🔷 高可用部署（可选） 多个 gocron 实例连接同一个 **MySQL/PostgreSQL** 数据库即可实现高可用，Leader 选举自动完成，无需额外配置。SQLite 以单节点模式运行。 ```bash # 节点 1 ./gocron web --port 5920 # 节点 2（连接同一数据库） ./gocron web --port 5921 ``` 详细部署步骤、K8s 配置和环境变量覆盖请参考 [高可用部署指南](https://gocron-docs.pages.dev/zh/guide/high-availability)。 ## 📸 界面截图

任务调度
任务调度

Agent自动注册 任务管理
Agent自动注册 任务管理
数据统计 消息通知
数据统计 消息通知
## 🤝 贡献 我们非常欢迎社区的贡献！ ### 如何贡献 1. **Fork 仓库** 2. **克隆你的 fork** ```bash git clone https://github.com/YOUR_USERNAME/gocron.git cd gocron ``` 3. **安装依赖** ```bash pnpm install pnpm run prepare ``` 4. **创建功能分支** ```bash git checkout -b feature/your-feature-name ``` 5. **修改代码并提交** ```bash git add . pnpm run commit # 使用交互式提交工具 ``` 6. **推送并创建 Pull Request** ```bash git push origin feature/your-feature-name ``` ### 提交信息规范 本项目使用 [commitizen](https://github.com/commitizen/cz-cli) 和 [cz-git](https://cz-git.qbb.sh/) 来规范化提交信息。 请使用以下命令代替 `git commit`： ```bash pnpm run commit ``` 这将引导你通过交互式提示创建格式正确的提交信息，例如： - `feat(task): 添加任务依赖配置` - `fix(api): 修复任务状态更新问题` - `docs: 更新 API 文档` ### 其他贡献方式 - 🐛 **提交 Bug**：请在 GitHub Issues 中提交 - 💡 **功能建议**：通过 Issues 分享你的想法 - 📝 **文档改进**：帮助我们完善文档 ## 📄 许可证 本项目遵循 MIT 许可证。详情请见 [LICENSE](LICENSE) 文件。 ## Star History [![Star History Chart](https://api.star-history.com/svg?repos=gocronx-team/gocron&type=Date)](https://www.star-history.com/#gocronx-team/gocron&Date) ================================================ FILE: app.ini.sqlite.example ================================================ # gocron SQLite 配置示例 # 将此文件复制到 ~/.gocron/conf/app.ini 并修改相应配置 [default] # 数据库配置 - SQLite db.engine=sqlite # SQLite 数据库文件路径（相对路径或绝对路径） # 相对路径：相对于程序运行目录 # 绝对路径：/path/to/gocron.db 或 ~/.gocron/data/gocron.db db.database=./data/gocron.db # SQLite 不需要以下配置，但保留以兼容配置文件格式 db.host= db.port= db.user= db.password= db.charset=utf8 db.prefix= db.max.idle.conns=30 db.max.open.conns=100 # 应用配置 app.name=定时任务管理系统 # API配置 api.key= api.secret= api.sign.enable=true # 允许访问的IP列表，多个IP用逗号分隔，为空表示不限制 allow_ips= # 并发队列大小 concurrency.queue=500 # 认证密钥（自动生成，无需手动配置） auth_secret= # TLS配置 enable_tls=false ca_file= cert_file= key_file= ================================================ FILE: cmd/gocron/gocron.go ================================================ // Command gocron package main import ( "context" "errors" "fmt" "net/http" "os" "os/signal" "syscall" "time" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/leader" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/setting" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers" "github.com/gocronx-team/gocron/internal/service" "github.com/urfave/cli/v2" ) var ( AppVersion = "1.6.0" BuildDate, GitCommit string // leaderElection 全局选举实例，用于 graceful shutdown 时释放锁 leaderElection *leader.Election ) // Default port for web server const DefaultPort = 5920 // Graceful shutdown timeout const shutdownTimeout = 30 * time.Second func main() { cliApp := cli.NewApp() cliApp.Name = "gocron" cliApp.Usage = "gocron service" cliApp.Version, _ = utils.FormatAppVersion(AppVersion, GitCommit, BuildDate) cliApp.Commands = getCommands() cliApp.Flags = append(cliApp.Flags, []cli.Flag{}...) // Auto-append "web" command when double-clicking on Windows if len(os.Args) == 1 && utils.IsWindows() { os.Args = append(os.Args, "web") } err := cliApp.Run(os.Args) if err != nil { logger.Fatal(err) } } // getCommands func getCommands() []*cli.Command { command := &cli.Command{ Name: "web", Usage: "run web server", Action: runWeb, Flags: []cli.Flag{ &cli.StringFlag{ Name: "host", Value: "0.0.0.0", Usage: "bind host", }, &cli.IntFlag{ Name: "port", Aliases: []string{"p"}, Value: DefaultPort, Usage: "bind port", }, &cli.StringFlag{ Name: "env", Aliases: []string{"e"}, Value: "prod", Usage: "runtime environment, dev|test|prod", }, }, } return []*cli.Command{command} } func runWeb(ctx *cli.Context) error { // Set runtime environment setEnvironment(ctx) fmt.Printf("Starting gocron web server...\n") // Initialize application app.InitEnv(AppVersion) fmt.Printf("Application initialized\n") // Initialize modules: DB, scheduled tasks, etc. initModule() fmt.Printf("Modules initialized\n") // Security warning: agent gRPC channel unencrypted when TLS is off if app.Installed && app.Setting != nil && !app.Setting.EnableTLS { logger.Warn("SECURITY: agent gRPC TLS is disabled (enable_tls=false); agent port is reachable without authentication") } // 屏蔽 Gin 启动时打印完整路由表（保留 debug 模式的其他提示） gin.DebugPrintRouteFunc = func(httpMethod, absolutePath, handlerName string, nuHandlers int) {} r := gin.Default() // Register middleware routers.RegisterMiddleware(r) // Register routes routers.Register(r) host := parseHost(ctx) port := parsePort(ctx) addr := fmt.Sprintf("%s:%d", host, port) // Use http.Server to support graceful shutdown srv := &http.Server{ Addr: addr, Handler: r, } // Start HTTP server in a goroutine go func() { fmt.Printf("Server listening on %s\n", addr) if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) { logger.Fatalf("Failed to start server: %v", err) } }() // Wait for shutdown signal, blocks the main goroutine waitForShutdown(srv) return nil } func initModule() { if !app.Installed { return } config, err := setting.Read(app.AppConfig) if err != nil { logger.Fatal("Failed to read application config", err) } app.Setting = config // Initialize DB models.Db = models.CreateDb() // Version upgrade upgradeIfNeed() // Auto-create missing tables ensureTables() // Repair missing settings records if err := models.RepairSettings(); err != nil { logger.Error("Failed to repair settings records", err) } // Initialize scheduler infrastructure service.ServiceTask.Initialize() // SQLite: single-node only, skip leader election if models.Db.Dialector.Name() == "sqlite" { logger.Info("SQLite detected, skipping leader election (single-node mode)") service.ServiceTask.StartScheduler() return } // Ensure scheduler_lock table exists if !models.Db.Migrator().HasTable(&models.SchedulerLock{}) { logger.Info("scheduler_lock table not found, creating...") if err := models.Db.AutoMigrate(&models.SchedulerLock{}); err != nil { logger.Error("Failed to create scheduler_lock table", err) } } // Start leader election — scheduler only runs on the leader node leaderElection = leader.New( models.Db, func() { service.ServiceTask.StartScheduler() }, // onElected func() { service.ServiceTask.StopScheduler() }, // onEvicted ) leaderElection.Start() } // parsePort parses the port from CLI flags func parsePort(ctx *cli.Context) int { port := DefaultPort if ctx.IsSet("port") { port = ctx.Int("port") } if port <= 0 || port >= 65535 { port = DefaultPort } return port } func parseHost(ctx *cli.Context) string { if ctx.IsSet("host") { return ctx.String("host") } return "0.0.0.0" } func setEnvironment(ctx *cli.Context) { env := "prod" if ctx.IsSet("env") { env = ctx.String("env") } switch env { case "test": gin.SetMode(gin.TestMode) case "dev": gin.SetMode(gin.DebugMode) default: gin.SetMode(gin.ReleaseMode) } } // waitForShutdown waits for OS signals and performs graceful shutdown func waitForShutdown(srv *http.Server) { quit := make(chan os.Signal, 1) signal.Notify(quit, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM) for { s := <-quit logger.Info("Received signal -- ", s) switch s { case syscall.SIGHUP: logger.Info("Received terminal disconnect signal, ignoring") continue case syscall.SIGINT, syscall.SIGTERM: // Proceed to graceful shutdown } break } logger.Info("Shutting down gracefully, press Ctrl+C again to force exit...") // Allow forced exit: immediately exit on receiving another signal go func() { forceQuit := make(chan os.Signal, 1) signal.Notify(forceQuit, syscall.SIGINT, syscall.SIGTERM) <-forceQuit logger.Warn("Forced shutdown") os.Exit(1) }() // Step 1: Stop HTTP server, reject new requests, wait for in-flight requests to complete logger.Info("Step 1/3: Stopping HTTP server (waiting for in-flight requests)...") shutdownCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout) defer cancel() if err := srv.Shutdown(shutdownCtx); err != nil { logger.Errorf("HTTP server shutdown error: %v", err) } else { logger.Info("HTTP server stopped successfully") } // Step 2: Stop leader election and release lock if app.Installed { if leaderElection != nil { logger.Info("Step 2/4: Stopping leader election (releasing lock)...") leaderElection.Stop() logger.Info("Leader election stopped") } // Step 3: Stop scheduled task scheduler, wait for running tasks to complete logger.Info("Step 3/4: Stopping scheduled task scheduler (waiting for running tasks)...") service.ServiceTask.WaitAndExit() logger.Info("Scheduled task scheduler stopped") // Step 4: Close database connections logger.Info("Step 4/4: Closing database connections...") closeDatabase() logger.Info("Database connections closed") } logger.Info("Graceful shutdown completed") logger.Close() } // closeDatabase closes the database connection pool func closeDatabase() { if models.Db == nil { return } // Stop the keep-alive goroutine before closing the connection models.StopKeepAlive() sqlDB, err := models.Db.DB() if err != nil { logger.Errorf("Failed to get database connection for closing: %v", err) return } if err := sqlDB.Close(); err != nil { logger.Errorf("Failed to close database connection: %v", err) } } // upgradeIfNeed checks if the app needs upgrading when version file exists and version < app.VersionId func upgradeIfNeed() { currentVersionId := app.GetCurrentVersionId() // No version file found if currentVersionId == 0 { return } if currentVersionId >= app.VersionId { return } migration := new(models.Migration) logger.Infof("Starting version upgrade, current version: %d", currentVersionId) migration.Upgrade(currentVersionId) app.UpdateVersionFile() logger.Infof("Upgraded to latest version: %d", app.VersionId) } // ensureTables ensures all required tables exist func ensureTables() { if !models.Db.Migrator().HasTable(&models.AgentToken{}) { logger.Info("agent_token table not found, creating...") if err := models.Db.AutoMigrate(&models.AgentToken{}); err != nil { logger.Error("Failed to create agent_token table", err) } else { logger.Info("agent_token table created successfully") } } if !models.Db.Migrator().HasTable(&models.AuditLog{}) { logger.Info("audit_log table not found, creating...") if err := models.Db.AutoMigrate(&models.AuditLog{}); err != nil { logger.Error("Failed to create audit_log table", err) } else { logger.Info("audit_log table created successfully") } } // 始终 AutoMigrate 新表，确保字段同步（AutoMigrate 幂等，只加列不删列） if err := models.Db.AutoMigrate(&models.TaskScriptVersion{}); err != nil { logger.Error("Failed to migrate task_script_version table", err) } if err := models.Db.AutoMigrate(&models.TaskTemplate{}); err != nil { logger.Error("Failed to migrate task_template table", err) } } ================================================ FILE: cmd/node/node.go ================================================ // Command gocron-node package main import ( "flag" "os" "runtime" "strings" "github.com/gocronx-team/gocron/internal/modules/rpc/auth" "github.com/gocronx-team/gocron/internal/modules/rpc/server" "github.com/gocronx-team/gocron/internal/modules/utils" log "github.com/sirupsen/logrus" ) var ( AppVersion, BuildDate, GitCommit string ) func main() { var serverAddr string var allowRoot bool var version bool var CAFile string var certFile string var keyFile string var enableTLS bool var logLevel string flag.BoolVar(&allowRoot, "allow-root", false, "./gocron-node -allow-root") flag.StringVar(&serverAddr, "s", "0.0.0.0:5921", "./gocron-node -s ip:port") flag.BoolVar(&version, "v", false, "./gocron-node -v") flag.BoolVar(&enableTLS, "enable-tls", false, "./gocron-node -enable-tls") flag.StringVar(&CAFile, "ca-file", "", "./gocron-node -ca-file path") flag.StringVar(&certFile, "cert-file", "", "./gocron-node -cert-file path") flag.StringVar(&keyFile, "key-file", "", "./gocron-node -key-file path") flag.StringVar(&logLevel, "log-level", "info", "-log-level error") flag.Parse() level, err := log.ParseLevel(logLevel) if err != nil { log.Fatal(err) } log.SetLevel(level) if version { utils.PrintAppVersion(AppVersion, GitCommit, BuildDate) return } if enableTLS { if !utils.FileExist(CAFile) { log.Fatalf("failed to read ca cert file: %s", CAFile) } if !utils.FileExist(certFile) { log.Fatalf("failed to read server cert file: %s", certFile) return } if !utils.FileExist(keyFile) { log.Fatalf("failed to read server key file: %s", keyFile) return } } certificate := auth.Certificate{ CAFile: strings.TrimSpace(CAFile), CertFile: strings.TrimSpace(certFile), KeyFile: strings.TrimSpace(keyFile), } if runtime.GOOS != "windows" && os.Getuid() == 0 && !allowRoot { log.Fatal("Do not run gocron-node as root user") return } server.Start(serverAddr, enableTLS, certificate) } ================================================ FILE: commitlint.config.cjs ================================================ /** * commitlint configuration file * Documentation * https://commitlint.js.org/#/reference-rules * https://cz-git.qbb.sh/guide/ */ module.exports = { // Extends rules extends: ['@commitlint/config-conventional'], // Custom rules rules: { // Type enum, git commit type must be one of the following types 'type-enum': [ 2, 'always', [ 'feat', // A new feature 'fix', // A bug fix 'docs', // Documentation only changes 'style', // Changes that do not affect the meaning of the code 'refactor', // A code change that neither fixes a bug nor adds a feature 'perf', // A code change that improves performance 'test', // Adding missing tests or correcting existing tests 'build', // Changes that affect the build system or external dependencies 'ci', // Changes to our CI configuration files and scripts 'revert', // Reverts a previous commit 'chore', // Other changes that don't modify src or test files 'wip' // Work in progress ] ], 'subject-case': [0], // No validation for subject case 'type-case': [0], // No validation for type case 'type-empty': [0], // Allow empty type 'subject-empty': [0] // Allow empty subject }, // Standard conventional commit parser parserPreset: { parserOpts: { headerPattern: /^([\w-]+)(?:\(([\w-]+)\))?:\s(.+)$/, headerCorrespondence: ['type', 'scope', 'subject'] } }, prompt: { messages: { type: 'Select the type of change that you\'re committing:', scope: 'Denote the SCOPE of this change (optional):', customScope: 'Denote the SCOPE of this change:', subject: 'Write a SHORT, IMPERATIVE tense description of the change:\n', body: 'Provide a LONGER description of the change (optional). Use "|" to break new line:\n', breaking: 'List any BREAKING CHANGES (optional). Use "|" to break new line:\n', footerPrefixesSelect: 'Select the ISSUES type of changeList by this change (optional):', customFooterPrefix: 'Input ISSUES prefix:', footer: 'List any ISSUES by this change. E.g.: #31, #34:\n', confirmCommit: 'Are you sure you want to proceed with the commit above?' }, // prettier-ignore types: [ { value: "feat", name: "feat: A new feature" }, { value: "fix", name: "fix: A bug fix" }, { value: "docs", name: "docs: Documentation only changes" }, { value: "style", name: "style: Changes that do not affect the meaning of the code" }, { value: "refactor", name: "refactor: A code change that neither fixes a bug nor adds a feature" }, { value: "perf", name: "perf: A code change that improves performance" }, { value: "test", name: "test: Adding missing tests or correcting existing tests" }, { value: "build", name: "build: Changes that affect the build system or external dependencies" }, { value: "ci", name: "ci: Changes to our CI configuration files and scripts" }, { value: "revert", name: "revert: Reverts a previous commit" }, { value: "chore", name: "chore: Other changes that don't modify src or test files" }, { value: "wip", name: "wip: Work in progress" }, ], useEmoji: false, emojiAlign: 'center', themeColorCode: '', scopes: [ { value: 'web', name: 'web: Frontend related' }, { value: 'api', name: 'api: API interface' }, { value: 'task', name: 'task: Task scheduling' }, { value: 'node', name: 'node: Node management' }, { value: 'auth', name: 'auth: Authentication' }, { value: 'db', name: 'db: Database' }, { value: 'config', name: 'config: Configuration' }, { value: 'deps', name: 'deps: Dependencies update' } ], allowCustomScopes: true, allowEmptyScopes: true, customScopesAlign: 'bottom', customScopesAlias: 'custom', emptyScopesAlias: 'empty', upperCaseSubject: false, markBreakingChangeMode: false, allowBreakingChanges: ['feat', 'fix'], breaklineNumber: 100, breaklineChar: '|', skipQuestions: ['breaking', 'footerPrefix', 'footer'], // Skip these steps issuePrefixes: [{ value: 'closed', name: 'closed: ISSUES has been processed' }], customIssuePrefixAlign: 'top', emptyIssuePrefixAlias: 'skip', customIssuePrefixAlias: 'custom', allowCustomIssuePrefix: true, allowEmptyIssuePrefix: true, confirmColorize: true, maxHeaderLength: Infinity, maxSubjectLength: Infinity, minSubjectLength: 0, scopeOverrides: undefined, defaultBody: '', defaultIssues: '', defaultScope: '', defaultSubject: '' } } ================================================ FILE: docker-compose.yml ================================================ services: gocron: build: context: . dockerfile: Dockerfile.gocron image: gocron:latest container_name: gocron ports: - "5920:5920" volumes: - gocron-data:/.gocron environment: - TZ=Asia/Shanghai restart: unless-stopped volumes: gocron-data: driver: local ================================================ FILE: embed.go ================================================ package embed import ( "embed" "io/fs" ) //go:embed all:web/vue/dist var files embed.FS func StaticFS() (fs.FS, error) { return fs.Sub(files, "web/vue/dist") } ================================================ FILE: go.mod ================================================ module github.com/gocronx-team/gocron go 1.26.2 require ( github.com/gin-gonic/gin v1.12.0 github.com/go-gomail/gomail v0.0.0-20160411212932-81ebce5c23df github.com/go-sql-driver/mysql v1.9.3 github.com/gocronx-team/cron v0.1.3 github.com/golang-jwt/jwt/v5 v5.3.1 github.com/lib/pq v1.12.0 github.com/ncruces/go-sqlite3/gormlite v0.33.3 github.com/pquerna/otp v1.5.0 github.com/sirupsen/logrus v1.9.4 github.com/urfave/cli/v2 v2.27.7 golang.org/x/crypto v0.50.0 golang.org/x/text v0.36.0 google.golang.org/grpc v1.79.3 google.golang.org/protobuf v1.36.11 gopkg.in/ini.v1 v1.67.1 gorm.io/driver/mysql v1.6.0 gorm.io/driver/postgres v1.6.0 gorm.io/gorm v1.31.1 ) require ( filippo.io/edwards25519 v1.1.1 // indirect github.com/boombuler/barcode v1.0.1-0.20190219062509-6c824513bacc // indirect github.com/bytedance/gopkg v0.1.3 // indirect github.com/bytedance/sonic v1.15.0 // indirect github.com/bytedance/sonic/loader v0.5.0 // indirect github.com/cloudwego/base64x v0.1.6 // indirect github.com/cpuguy83/go-md2man/v2 v2.0.7 // indirect github.com/gabriel-vasile/mimetype v1.4.12 // indirect github.com/gin-contrib/sse v1.1.0 // indirect github.com/go-playground/locales v0.14.1 // indirect github.com/go-playground/universal-translator v0.18.1 // indirect github.com/go-playground/validator/v10 v10.30.1 // indirect github.com/goccy/go-json v0.10.5 // indirect github.com/goccy/go-yaml v1.19.2 // indirect github.com/jackc/pgpassfile v1.0.0 // indirect github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect github.com/jackc/pgx/v5 v5.9.1 // indirect github.com/jackc/puddle/v2 v2.2.2 // indirect github.com/jinzhu/inflection v1.0.0 // indirect github.com/jinzhu/now v1.1.5 // indirect github.com/json-iterator/go v1.1.12 // indirect github.com/klauspost/cpuid/v2 v2.3.0 // indirect github.com/leodido/go-urn v1.4.0 // indirect github.com/mattn/go-isatty v0.0.20 // indirect github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect github.com/modern-go/reflect2 v1.0.2 // indirect github.com/ncruces/go-sqlite3 v0.33.3 // indirect github.com/ncruces/go-sqlite3-wasm v1.1.1-0.20260409221933-87e4b35a38d0 // indirect github.com/ncruces/julianday v1.0.0 // indirect github.com/pelletier/go-toml/v2 v2.2.4 // indirect github.com/quic-go/qpack v0.6.0 // indirect github.com/quic-go/quic-go v0.59.0 // indirect github.com/russross/blackfriday/v2 v2.1.0 // indirect github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/ugorji/go/codec v1.3.1 // indirect github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 // indirect go.mongodb.org/mongo-driver/v2 v2.5.0 // indirect golang.org/x/arch v0.22.0 // indirect golang.org/x/net v0.52.0 // indirect golang.org/x/sync v0.20.0 // indirect golang.org/x/sys v0.43.0 // indirect google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 // indirect gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc // indirect gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df // indirect ) ================================================ FILE: go.sum ================================================ filippo.io/edwards25519 v1.1.1 h1:YpjwWWlNmGIDyXOn8zLzqiD+9TyIlPhGFG96P39uBpw= filippo.io/edwards25519 v1.1.1/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4= github.com/boombuler/barcode v1.0.1-0.20190219062509-6c824513bacc h1:biVzkmvwrH8WK8raXaxBx6fRVTlJILwEwQGL1I/ByEI= github.com/boombuler/barcode v1.0.1-0.20190219062509-6c824513bacc/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8= github.com/bytedance/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M= github.com/bytedance/gopkg v0.1.3/go.mod h1:576VvJ+eJgyCzdjS+c4+77QF3p7ubbtiKARP3TxducM= github.com/bytedance/sonic v1.15.0 h1:/PXeWFaR5ElNcVE84U0dOHjiMHQOwNIx3K4ymzh/uSE= github.com/bytedance/sonic v1.15.0/go.mod h1:tFkWrPz0/CUCLEF4ri4UkHekCIcdnkqXw9VduqpJh0k= github.com/bytedance/sonic/loader v0.5.0 h1:gXH3KVnatgY7loH5/TkeVyXPfESoqSBSBEiDd5VjlgE= github.com/bytedance/sonic/loader v0.5.0/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCcdf5x77udYiSJRo= github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs= github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= github.com/cloudwego/base64x v0.1.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M= github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU= github.com/cpuguy83/go-md2man/v2 v2.0.7 h1:zbFlGlXEAKlwXpmvle3d8Oe3YnkKIK4xSRTd3sHPnBo= github.com/cpuguy83/go-md2man/v2 v2.0.7/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/gabriel-vasile/mimetype v1.4.12 h1:e9hWvmLYvtp846tLHam2o++qitpguFiYCKbn0w9jyqw= github.com/gabriel-vasile/mimetype v1.4.12/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s= github.com/gin-contrib/sse v1.1.0 h1:n0w2GMuUpWDVp7qSpvze6fAu9iRxJY4Hmj6AmBOU05w= github.com/gin-contrib/sse v1.1.0/go.mod h1:hxRZ5gVpWMT7Z0B0gSNYqqsSCNIJMjzvm6fqCz9vjwM= github.com/gin-gonic/gin v1.12.0 h1:b3YAbrZtnf8N//yjKeU2+MQsh2mY5htkZidOM7O0wG8= github.com/gin-gonic/gin v1.12.0/go.mod h1:VxccKfsSllpKshkBWgVgRniFFAzFb9csfngsqANjnLc= github.com/go-gomail/gomail v0.0.0-20160411212932-81ebce5c23df h1:Bao6dhmbTA1KFVxmJ6nBoMuOJit2yjEgLJpIMYpop0E= github.com/go-gomail/gomail v0.0.0-20160411212932-81ebce5c23df/go.mod h1:GJr+FCSXshIwgHBtLglIg9M2l2kQSi6QjVAngtzI08Y= github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI= github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY= github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= github.com/go-playground/validator/v10 v10.30.1 h1:f3zDSN/zOma+w6+1Wswgd9fLkdwy06ntQJp0BBvFG0w= github.com/go-playground/validator/v10 v10.30.1/go.mod h1:oSuBIQzuJxL//3MelwSLD5hc2Tu889bF0Idm9Dg26cM= github.com/go-sql-driver/mysql v1.9.3 h1:U/N249h2WzJ3Ukj8SowVFjdtZKfu9vlLZxjPXV1aweo= github.com/go-sql-driver/mysql v1.9.3/go.mod h1:qn46aNg1333BRMNU69Lq93t8du/dwxI64Gl8i5p1WMU= github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4= github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M= github.com/goccy/go-yaml v1.19.2 h1:PmFC1S6h8ljIz6gMRBopkjP1TVT7xuwrButHID66PoM= github.com/goccy/go-yaml v1.19.2/go.mod h1:XBurs7gK8ATbW4ZPGKgcbrY1Br56PdM69F7LkFRi1kA= github.com/gocronx-team/cron v0.1.3 h1:flJFseOHRkDWnV8vQyOH4V8HXX8per4wmfO5nygcORY= github.com/gocronx-team/cron v0.1.3/go.mod h1:HqYzaybMPpKERqcHeuVBO3IGyA+n7fVmRlowQ/uDfyw= github.com/golang-jwt/jwt/v5 v5.3.1 h1:kYf81DTWFe7t+1VvL7eS+jKFVWaUnK9cB1qbwn63YCY= github.com/golang-jwt/jwt/v5 v5.3.1/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE= github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8= github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU= github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM= github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg= github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo= github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM= github.com/jackc/pgx/v5 v5.9.1 h1:uwrxJXBnx76nyISkhr33kQLlUqjv7et7b9FjCen/tdc= github.com/jackc/pgx/v5 v5.9.1/go.mod h1:mal1tBGAFfLHvZzaYh77YS/eC6IX9OWbRV1QIIM0Jn4= github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo= github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4= github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y= github.com/klauspost/cpuid/v2 v2.3.0/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0= github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ= github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI= github.com/lib/pq v1.12.0 h1:mC1zeiNamwKBecjHarAr26c/+d8V5w/u4J0I/yASbJo= github.com/lib/pq v1.12.0/go.mod h1:/p+8NSbOcwzAEI7wiMXFlgydTwcgTr3OSKMsD2BitpA= github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= github.com/ncruces/go-sqlite3 v0.33.3 h1:6jCR3KuGvJSEwhaQrkHDGeIe2qCQ6nOUDNsPz7ZIotw= github.com/ncruces/go-sqlite3 v0.33.3/go.mod h1:t2Osfw0wcKzJTgv2EvrkTtVLqlbKTA5Yvwb2ypAlBcY= github.com/ncruces/go-sqlite3-wasm v1.1.1-0.20260409221933-87e4b35a38d0 h1:ymE9H30x1AyW5VfMNkJC9teuI2W1jjMsQS7kc6zl6Tg= github.com/ncruces/go-sqlite3-wasm v1.1.1-0.20260409221933-87e4b35a38d0/go.mod h1:/H3+JykPsfSlvKbOxNSx9kKwm3ecqQGzyCs1e9KkNsU= github.com/ncruces/go-sqlite3/gormlite v0.33.3 h1:JzLk8XymgvHvy60ib5MtNmd0fIYwGi7FUj2DpRFmnWQ= github.com/ncruces/go-sqlite3/gormlite v0.33.3/go.mod h1:qDjzlaffXDGg5bhZs2VaaSY0Qb3rsiKq0O4pXkmQfHI= github.com/ncruces/julianday v1.0.0 h1:fH0OKwa7NWvniGQtxdJRxAgkBMolni2BjDHaWTxqt7M= github.com/ncruces/julianday v1.0.0/go.mod h1:Dusn2KvZrrovOMJuOt0TNXL6tB7U2E8kvza5fFc9G7g= github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4= github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/pquerna/otp v1.5.0 h1:NMMR+WrmaqXU4EzdGJEE1aUUI0AMRzsp96fFFWNPwxs= github.com/pquerna/otp v1.5.0/go.mod h1:dkJfzwRKNiegxyNb54X/3fLwhCynbMspSyWKnvi1AEg= github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8= github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII= github.com/quic-go/quic-go v0.59.0 h1:OLJkp1Mlm/aS7dpKgTc6cnpynnD2Xg7C1pwL6vy/SAw= github.com/quic-go/quic-go v0.59.0/go.mod h1:upnsH4Ju1YkqpLXC305eW3yDZ4NfnNbmQRCMWS58IKU= github.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk= github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= github.com/sirupsen/logrus v1.9.4 h1:TsZE7l11zFCLZnZ+teH4Umoq5BhEIfIzfRDZ1Uzql2w= github.com/sirupsen/logrus v1.9.4/go.mod h1:ftWc9WdOfJ0a92nsE2jF5u5ZwH8Bv2zdeOC42RjbV2g= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY= github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4= github.com/urfave/cli/v2 v2.27.7 h1:bH59vdhbjLv3LAvIu6gd0usJHgoTTPhCFib8qqOwXYU= github.com/urfave/cli/v2 v2.27.7/go.mod h1:CyNAG/xg+iAOg0N4MPGZqVmv2rCoP267496AOXUZjA4= github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 h1:gEOO8jv9F4OT7lGCjxCBTO/36wtF6j2nSip77qHd4x4= github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1/go.mod h1:Ohn+xnUBiLI6FVj/9LpzZWtj1/D6lUovWYBkxHVV3aM= go.mongodb.org/mongo-driver/v2 v2.5.0 h1:yXUhImUjjAInNcpTcAlPHiT7bIXhshCTL3jVBkF3xaE= go.mongodb.org/mongo-driver/v2 v2.5.0/go.mod h1:yOI9kBsufol30iFsl1slpdq1I0eHPzybRWdyYUs8K/0= go.opentelemetry.io/auto/sdk v1.2.1 h1:jXsnJ4Lmnqd11kwkBV2LgLoFMZKizbCi5fNZ/ipaZ64= go.opentelemetry.io/auto/sdk v1.2.1/go.mod h1:KRTj+aOaElaLi+wW1kO/DZRXwkF4C5xPbEe3ZiIhN7Y= go.opentelemetry.io/otel v1.39.0 h1:8yPrr/S0ND9QEfTfdP9V+SiwT4E0G7Y5MO7p85nis48= go.opentelemetry.io/otel v1.39.0/go.mod h1:kLlFTywNWrFyEdH0oj2xK0bFYZtHRYUdv1NklR/tgc8= go.opentelemetry.io/otel/metric v1.39.0 h1:d1UzonvEZriVfpNKEVmHXbdf909uGTOQjA0HF0Ls5Q0= go.opentelemetry.io/otel/metric v1.39.0/go.mod h1:jrZSWL33sD7bBxg1xjrqyDjnuzTUB0x1nBERXd7Ftcs= go.opentelemetry.io/otel/sdk v1.39.0 h1:nMLYcjVsvdui1B/4FRkwjzoRVsMK8uL/cj0OyhKzt18= go.opentelemetry.io/otel/sdk v1.39.0/go.mod h1:vDojkC4/jsTJsE+kh+LXYQlbL8CgrEcwmt1ENZszdJE= go.opentelemetry.io/otel/sdk/metric v1.39.0 h1:cXMVVFVgsIf2YL6QkRF4Urbr/aMInf+2WKg+sEJTtB8= go.opentelemetry.io/otel/sdk/metric v1.39.0/go.mod h1:xq9HEVH7qeX69/JnwEfp6fVq5wosJsY1mt4lLfYdVew= go.opentelemetry.io/otel/trace v1.39.0 h1:2d2vfpEDmCJ5zVYz7ijaJdOF59xLomrvj7bjt6/qCJI= go.opentelemetry.io/otel/trace v1.39.0/go.mod h1:88w4/PnZSazkGzz/w84VHpQafiU4EtqqlVdxWy+rNOA= go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y= go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU= golang.org/x/arch v0.22.0 h1:c/Zle32i5ttqRXjdLyyHZESLD/bB90DCU1g9l/0YBDI= golang.org/x/arch v0.22.0/go.mod h1:dNHoOeKiyja7GTvF9NJS1l3Z2yntpQNzgrjh1cU103A= golang.org/x/crypto v0.50.0 h1:zO47/JPrL6vsNkINmLoo/PH1gcxpls50DNogFvB5ZGI= golang.org/x/crypto v0.50.0/go.mod h1:3muZ7vA7PBCE6xgPX7nkzzjiUq87kRItoJQM1Yo8S+Q= golang.org/x/net v0.52.0 h1:He/TN1l0e4mmR3QqHMT2Xab3Aj3L9qjbhRm78/6jrW0= golang.org/x/net v0.52.0/go.mod h1:R1MAz7uMZxVMualyPXb+VaqGSa3LIaUqk0eEt3w36Sw= golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4= golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0= golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.43.0 h1:Rlag2XtaFTxp19wS8MXlJwTvoh8ArU6ezoyFsMyCTNI= golang.org/x/sys v0.43.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw= golang.org/x/text v0.36.0 h1:JfKh3XmcRPqZPKevfXVpI1wXPTqbkE5f7JA92a55Yxg= golang.org/x/text v0.36.0/go.mod h1:NIdBknypM8iqVmPiuco0Dh6P5Jcdk8lJL0CUebqK164= gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk= gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E= google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 h1:gRkg/vSppuSQoDjxyiGfN4Upv/h/DQmIR10ZU8dh4Ww= google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217/go.mod h1:7i2o+ce6H/6BluujYR+kqX3GKH+dChPTQU19wjRPiGk= google.golang.org/grpc v1.79.3 h1:sybAEdRIEtvcD68Gx7dmnwjZKlyfuc61Dyo9pGXXkKE= google.golang.org/grpc v1.79.3/go.mod h1:KmT0Kjez+0dde/v2j9vzwoAScgEPx/Bw1CYChhHLrHQ= google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE= google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco= gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc h1:2gGKlE2+asNV9m7xrywl36YYNnBG5ZQ0r/BOOxqPpmk= gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc/go.mod h1:m7x9LTH6d71AHyAX77c9yqWCCa3UKHcVEj9y7hAtKDk= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df h1:n7WqCuqOuCbNr617RXOY0AWRXxgwEyPp2z+p0+hgMuE= gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df/go.mod h1:LRQQ+SO6ZHR7tOkpBDuZnXENFzX8qRjMDMyPD6BRkCw= gopkg.in/ini.v1 v1.67.1 h1:tVBILHy0R6e4wkYOn3XmiITt/hEVH4TFMYvAX2Ytz6k= gopkg.in/ini.v1 v1.67.1/go.mod h1:x/cyOwCgZqOkJoDIJ3c1KNHMo10+nLGAhh+kn3Zizss= gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= gorm.io/driver/mysql v1.6.0 h1:eNbLmNTpPpTOVZi8MMxCi2aaIm0ZpInbORNXDwyLGvg= gorm.io/driver/mysql v1.6.0/go.mod h1:D/oCC2GWK3M/dqoLxnOlaNKmXz8WNTfcS9y5ovaSqKo= gorm.io/driver/postgres v1.6.0 h1:2dxzU8xJ+ivvqTRph34QX+WrRaJlmfyPqXmoGVjMBa4= gorm.io/driver/postgres v1.6.0/go.mod h1:vUw0mrGgrTK+uPHEhAdV4sfFELrByKVGnaVRkXDhtWo= gorm.io/gorm v1.31.1 h1:7CA8FTFz/gRfgqgpeKIBcervUn3xSyPUmr6B2WXJ7kg= gorm.io/gorm v1.31.1/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs= ================================================ FILE: helm/gocron/Chart.yaml ================================================ apiVersion: v2 name: gocron description: A Helm chart for gocron - cron job management system type: application version: 0.1.0 appVersion: "1.5.9" keywords: - cron - scheduler - task home: https://github.com/gocronx-team/gocron icon: https://raw.githubusercontent.com/gocronx-team/gocron/master/web/vue/public/favicon.ico sources: - https://github.com/gocronx-team/gocron maintainers: - name: gocronx-team ================================================ FILE: helm/gocron/templates/NOTES.txt ================================================ gocron has been deployed successfully! Get the application URL: {{- if .Values.ingress.enabled }} {{- range $host := .Values.ingress.hosts }} http{{ if $.Values.ingress.tls }}s{{ end }}://{{ $host.host }} {{- end }} {{- else if contains "NodePort" .Values.service.type }} export NODE_PORT=$(kubectl get --namespace {{ .Release.Namespace }} -o jsonpath="{.spec.ports[0].nodePort}" services {{ include "gocron.fullname" . }}) export NODE_IP=$(kubectl get nodes --namespace {{ .Release.Namespace }} -o jsonpath="{.items[0].status.addresses[0].address}") echo http://$NODE_IP:$NODE_PORT {{- else if contains "LoadBalancer" .Values.service.type }} export SERVICE_IP=$(kubectl get svc --namespace {{ .Release.Namespace }} {{ include "gocron.fullname" . }} --template "{{"{{ range (index .status.loadBalancer.ingress 0) }}{{.}}{{ end }}"}}") echo http://$SERVICE_IP:{{ .Values.service.port }} {{- else }} kubectl --namespace {{ .Release.Namespace }} port-forward svc/{{ include "gocron.fullname" . }} {{ .Values.service.port }}:{{ .Values.service.port }} echo http://127.0.0.1:{{ .Values.service.port }} {{- end }} Database: {{ .Values.db.engine }} {{- if eq .Values.db.engine "sqlite" }} Note: SQLite mode supports single replica only. Data is stored in PVC. {{- end }} ================================================ FILE: helm/gocron/templates/_helpers.tpl ================================================ {{/* Expand the name of the chart. */}} {{- define "gocron.name" -}} {{- default .Chart.Name .Values.nameOverride | trunc 63 | trimSuffix "-" }} {{- end }} {{/* Create a default fully qualified app name. */}} {{- define "gocron.fullname" -}} {{- if .Values.fullnameOverride }} {{- .Values.fullnameOverride | trunc 63 | trimSuffix "-" }} {{- else }} {{- $name := default .Chart.Name .Values.nameOverride }} {{- if contains $name .Release.Name }} {{- .Release.Name | trunc 63 | trimSuffix "-" }} {{- else }} {{- printf "%s-%s" .Release.Name $name | trunc 63 | trimSuffix "-" }} {{- end }} {{- end }} {{- end }} {{/* Create chart name and version as used by the chart label. */}} {{- define "gocron.chart" -}} {{- printf "%s-%s" .Chart.Name .Chart.Version | replace "+" "_" | trunc 63 | trimSuffix "-" }} {{- end }} {{/* Common labels */}} {{- define "gocron.labels" -}} helm.sh/chart: {{ include "gocron.chart" . }} {{ include "gocron.selectorLabels" . }} app.kubernetes.io/version: {{ .Values.image.tag | default .Chart.AppVersion | quote }} app.kubernetes.io/managed-by: {{ .Release.Service }} {{- end }} {{/* Selector labels */}} {{- define "gocron.selectorLabels" -}} app.kubernetes.io/name: {{ include "gocron.name" . }} app.kubernetes.io/instance: {{ .Release.Name }} {{- end }} {{/* Create the name of the service account to use */}} {{- define "gocron.serviceAccountName" -}} {{- if .Values.serviceAccount.create }} {{- default (include "gocron.fullname" .) .Values.serviceAccount.name }} {{- else }} {{- default "default" .Values.serviceAccount.name }} {{- end }} {{- end }} {{/* Image tag */}} {{- define "gocron.imageTag" -}} {{- .Values.image.tag | default .Chart.AppVersion }} {{- end }} ================================================ FILE: helm/gocron/templates/configmap.yaml ================================================ apiVersion: v1 kind: ConfigMap metadata: name: {{ include "gocron.fullname" . }} labels: {{- include "gocron.labels" . | nindent 4 }} data: app.ini: | [default] db.engine={{ .Values.db.engine }} db.host={{ .Values.db.host }} db.port={{ .Values.db.port }} db.user={{ .Values.db.user }} db.password={{ .Values.db.password }} db.database={{ .Values.db.database }} db.prefix={{ .Values.db.prefix }} db.charset={{ .Values.db.charset }} db.max.idle.conns={{ .Values.db.maxIdleConns }} db.max.open.conns={{ .Values.db.maxOpenConns }} app.name={{ .Values.app.name }} api.key={{ .Values.app.apiKey }} api.secret={{ .Values.app.apiSecret }} allow_ips={{ .Values.app.allowIps }} concurrency.queue={{ .Values.app.concurrencyQueue }} enable_tls={{ .Values.app.enableTls }} ================================================ FILE: helm/gocron/templates/deployment.yaml ================================================ apiVersion: apps/v1 kind: Deployment metadata: name: {{ include "gocron.fullname" . }} labels: {{- include "gocron.labels" . | nindent 4 }} spec: replicas: {{ .Values.replicaCount }} {{- if eq .Values.db.engine "sqlite" }} strategy: type: Recreate {{- end }} selector: matchLabels: {{- include "gocron.selectorLabels" . | nindent 6 }} template: metadata: annotations: checksum/config: {{ include (print $.Template.BasePath "/configmap.yaml") . | sha256sum }} labels: {{- include "gocron.selectorLabels" . | nindent 8 }} spec: {{- with .Values.imagePullSecrets }} imagePullSecrets: {{- toYaml . | nindent 8 }} {{- end }} serviceAccountName: {{ include "gocron.serviceAccountName" . }} containers: - name: {{ .Chart.Name }} image: "{{ .Values.image.repository }}:{{ include "gocron.imageTag" . }}" imagePullPolicy: {{ .Values.image.pullPolicy }} ports: - name: http containerPort: 5920 protocol: TCP env: - name: TZ value: {{ .Values.timezone | quote }} livenessProbe: httpGet: path: / port: http initialDelaySeconds: 10 periodSeconds: 30 readinessProbe: httpGet: path: / port: http initialDelaySeconds: 5 periodSeconds: 10 volumeMounts: - name: data mountPath: /.gocron - name: config mountPath: /.gocron/conf/app.ini subPath: app.ini {{- with .Values.resources }} resources: {{- toYaml . | nindent 12 }} {{- end }} volumes: - name: config configMap: name: {{ include "gocron.fullname" . }} - name: data {{- if .Values.persistence.enabled }} persistentVolumeClaim: claimName: {{ include "gocron.fullname" . }} {{- else }} emptyDir: {} {{- end }} {{- with .Values.nodeSelector }} nodeSelector: {{- toYaml . | nindent 8 }} {{- end }} {{- with .Values.tolerations }} tolerations: {{- toYaml . | nindent 8 }} {{- end }} {{- with .Values.affinity }} affinity: {{- toYaml . | nindent 8 }} {{- end }} ================================================ FILE: helm/gocron/templates/ingress.yaml ================================================ {{- if .Values.ingress.enabled }} apiVersion: networking.k8s.io/v1 kind: Ingress metadata: name: {{ include "gocron.fullname" . }} labels: {{- include "gocron.labels" . | nindent 4 }} {{- with .Values.ingress.annotations }} annotations: {{- toYaml . | nindent 4 }} {{- end }} spec: {{- if .Values.ingress.className }} ingressClassName: {{ .Values.ingress.className }} {{- end }} {{- if .Values.ingress.tls }} tls: {{- range .Values.ingress.tls }} - hosts: {{- range .hosts }} - {{ . | quote }} {{- end }} secretName: {{ .secretName }} {{- end }} {{- end }} rules: {{- range .Values.ingress.hosts }} - host: {{ .host | quote }} http: paths: {{- range .paths }} - path: {{ .path }} pathType: {{ .pathType }} backend: service: name: {{ include "gocron.fullname" $ }} port: name: http {{- end }} {{- end }} {{- end }} ================================================ FILE: helm/gocron/templates/pvc.yaml ================================================ {{- if .Values.persistence.enabled }} apiVersion: v1 kind: PersistentVolumeClaim metadata: name: {{ include "gocron.fullname" . }} labels: {{- include "gocron.labels" . | nindent 4 }} spec: accessModes: - {{ .Values.persistence.accessMode }} {{- if .Values.persistence.storageClass }} storageClassName: {{ .Values.persistence.storageClass | quote }} {{- end }} resources: requests: storage: {{ .Values.persistence.size }} {{- end }} ================================================ FILE: helm/gocron/templates/service.yaml ================================================ apiVersion: v1 kind: Service metadata: name: {{ include "gocron.fullname" . }} labels: {{- include "gocron.labels" . | nindent 4 }} spec: type: {{ .Values.service.type }} ports: - port: {{ .Values.service.port }} targetPort: http protocol: TCP name: http selector: {{- include "gocron.selectorLabels" . | nindent 4 }} ================================================ FILE: helm/gocron/templates/serviceaccount.yaml ================================================ {{- if .Values.serviceAccount.create }} apiVersion: v1 kind: ServiceAccount metadata: name: {{ include "gocron.serviceAccountName" . }} labels: {{- include "gocron.labels" . | nindent 4 }} {{- with .Values.serviceAccount.annotations }} annotations: {{- toYaml . | nindent 4 }} {{- end }} {{- end }} ================================================ FILE: helm/gocron/values.yaml ================================================ replicaCount: 1 image: repository: gocronx/gocron tag: "" # defaults to Chart appVersion pullPolicy: IfNotPresent imagePullSecrets: [] nameOverride: "" fullnameOverride: "" # Database configuration db: # sqlite, mysql, postgres engine: sqlite host: "" port: 0 user: "" password: "" database: ./data/gocron.db prefix: "" charset: utf8 maxIdleConns: 5 maxOpenConns: 100 # Application configuration app: name: "gocron" apiKey: "" apiSecret: "" allowIps: "" concurrencyQueue: 500 enableTls: false timezone: Asia/Shanghai service: type: ClusterIP port: 5920 ingress: enabled: false className: "" annotations: {} hosts: - host: gocron.local paths: - path: / pathType: Prefix tls: [] persistence: enabled: true storageClass: "" accessMode: ReadWriteOnce size: 1Gi resources: {} # limits: # cpu: 500m # memory: 256Mi # requests: # cpu: 100m # memory: 128Mi nodeSelector: {} tolerations: [] affinity: {} serviceAccount: create: true name: "" annotations: {} ================================================ FILE: internal/models/agent_token.go ================================================ package models import ( "time" ) // AgentToken agent注册token type AgentToken struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Token string `json:"token" gorm:"type:varchar(64);uniqueIndex;not null"` ExpiresAt time.Time `json:"expires_at" gorm:"not null"` Used bool `json:"used" gorm:"default:false"` UsedAt *time.Time `json:"used_at" gorm:"default:null"` CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"` } func (t *AgentToken) Create() error { return Db.Create(t).Error } func (t *AgentToken) FindByToken(token string) error { return Db.Where("token = ?", token).First(t).Error } func (t *AgentToken) MarkAsUsed() error { if !t.Used { t.Used = true now := time.Now() t.UsedAt = &now return Db.Save(t).Error } return nil } func (t *AgentToken) IsValid() bool { return time.Now().Before(t.ExpiresAt) } // CleanExpired 清理过期token func (t *AgentToken) CleanExpired() error { return Db.Where("expires_at < ?", time.Now()).Delete(&AgentToken{}).Error } ================================================ FILE: internal/models/audit_log.go ================================================ package models import ( "time" "gorm.io/gorm" ) // AuditLog records who did what and when type AuditLog struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Username string `json:"username" gorm:"type:varchar(32);not null;index"` Ip string `json:"ip" gorm:"type:varchar(45);not null"` Module string `json:"module" gorm:"type:varchar(32);not null;index"` // task, host, user, system Action string `json:"action" gorm:"type:varchar(32);not null"` // create, update, delete, enable, disable, run TargetId int `json:"target_id" gorm:"default:0"` TargetName string `json:"target_name" gorm:"type:varchar(128)"` Detail string `json:"detail" gorm:"type:text"` CreatedAt time.Time `json:"created" gorm:"column:created;autoCreateTime;index"` BaseModel `json:"-" gorm:"-"` } func (log *AuditLog) Create() (insertId int, err error) { result := Db.Create(log) if result.Error == nil { insertId = log.Id } return insertId, result.Error } func (log *AuditLog) List(params CommonMap) ([]AuditLog, error) { log.parsePageAndPageSize(params) list := make([]AuditLog, 0) err := log.buildQuery(params).Order("id DESC").Limit(log.PageSize).Offset(log.pageLimitOffset()).Find(&list).Error return list, err } func (log *AuditLog) Total(params CommonMap) (int64, error) { var count int64 err := log.buildQuery(params).Model(&AuditLog{}).Count(&count).Error return count, err } func (log *AuditLog) buildQuery(params CommonMap) *gorm.DB { db := Db if module, ok := params["Module"]; ok && module != "" { db = db.Where("module = ?", module) } if action, ok := params["Action"]; ok && action != "" { db = db.Where("action = ?", action) } if username, ok := params["Username"]; ok && username != "" { db = db.Where("username LIKE ?", "%"+username.(string)+"%") } if startDate, ok := params["StartDate"]; ok && startDate != "" { db = db.Where("created >= ?", startDate) } if endDate, ok := params["EndDate"]; ok && endDate != "" { db = db.Where("created <= ?", endDate) } return db } ================================================ FILE: internal/models/audit_log_test.go ================================================ package models import ( "testing" "time" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) func setupAuditLogTestDB(t *testing.T) func() { t.Helper() originalDb := Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } if err := db.AutoMigrate(&AuditLog{}); err != nil { t.Fatalf("failed to migrate test database: %v", err) } Db = db return func() { Db = originalDb } } func TestAuditLog_Create(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() log := &AuditLog{ Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create", TargetId: 1, TargetName: "my-task", Detail: "created task my-task", } insertId, err := log.Create() if err != nil { t.Fatalf("Create returned error: %v", err) } if insertId <= 0 { t.Errorf("expected insertId > 0, got %d", insertId) } } func TestAuditLog_List_Empty(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() auditLog := new(AuditLog) params := CommonMap{"Page": 1, "PageSize": 20} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 0 { t.Errorf("expected empty list, got %d items", len(list)) } } func TestAuditLog_List_Pagination(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() // Insert 5 records for i := 0; i < 5; i++ { log := &AuditLog{ Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create", } if _, err := log.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) params := CommonMap{"Page": 1, "PageSize": 3} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 3 { t.Errorf("expected 3 items (page size), got %d", len(list)) } // Page 2 should have the remaining 2 params["Page"] = 2 list2, err := auditLog.List(params) if err != nil { t.Fatalf("List page 2 returned error: %v", err) } if len(list2) != 2 { t.Errorf("expected 2 items on page 2, got %d", len(list2)) } } func TestAuditLog_List_FilterByModule(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() entries := []AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "delete"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) params := CommonMap{"Page": 1, "PageSize": 20, "Module": "task"} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 2 { t.Errorf("expected 2 task entries, got %d", len(list)) } for _, item := range list { if item.Module != "task" { t.Errorf("expected module 'task', got '%s'", item.Module) } } } func TestAuditLog_List_FilterByAction(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() entries := []AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "delete"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "create"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) params := CommonMap{"Page": 1, "PageSize": 20, "Action": "delete"} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 1 { t.Errorf("expected 1 delete entry, got %d", len(list)) } } func TestAuditLog_List_FilterByUsername(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() entries := []AuditLog{ {Username: "alice", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "bob", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "alice_admin", Ip: "127.0.0.1", Module: "host", Action: "delete"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) // LIKE match: "alice" matches "alice" and "alice_admin" params := CommonMap{"Page": 1, "PageSize": 20, "Username": "alice"} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 2 { t.Errorf("expected 2 alice entries, got %d", len(list)) } } func TestAuditLog_List_FilterByDateRange(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() // Insert records with specific timestamps via raw insert now := time.Now() yesterday := now.AddDate(0, 0, -1).Format("2006-01-02 15:04:05") tomorrow := now.AddDate(0, 0, 1).Format("2006-01-02 15:04:05") twoDaysAgo := now.AddDate(0, 0, -2).Format("2006-01-02 15:04:05") Db.Exec("INSERT INTO audit_log (username, ip, module, action, target_id, target_name, detail, created) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", "admin", "127.0.0.1", "task", "create", 0, "", "", yesterday) Db.Exec("INSERT INTO audit_log (username, ip, module, action, target_id, target_name, detail, created) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", "admin", "127.0.0.1", "task", "delete", 0, "", "", twoDaysAgo) auditLog := new(AuditLog) // Filter to only yesterday startDate := now.AddDate(0, 0, -1).Format("2006-01-02") + " 00:00:00" endDate := tomorrow params := CommonMap{"Page": 1, "PageSize": 20, "StartDate": startDate, "EndDate": endDate} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 1 { t.Errorf("expected 1 entry in date range, got %d", len(list)) } } func TestAuditLog_Total(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() // Empty DB auditLog := new(AuditLog) params := CommonMap{} total, err := auditLog.Total(params) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 0 { t.Errorf("expected 0, got %d", total) } // Insert 3 records for i := 0; i < 3; i++ { log := &AuditLog{ Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create", } if _, err := log.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } total, err = auditLog.Total(params) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 3 { t.Errorf("expected 3, got %d", total) } } func TestAuditLog_Total_WithFilter(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() entries := []AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "delete"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) params := CommonMap{"Module": "task"} total, err := auditLog.Total(params) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 2 { t.Errorf("expected 2 task entries, got %d", total) } } func TestAuditLog_List_OrderByIdDesc(t *testing.T) { cleanup := setupAuditLogTestDB(t) defer cleanup() for i := 0; i < 3; i++ { log := &AuditLog{ Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create", } if _, err := log.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } auditLog := new(AuditLog) params := CommonMap{"Page": 1, "PageSize": 10} list, err := auditLog.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) < 2 { t.Fatalf("expected at least 2 entries, got %d", len(list)) } // Verify descending order for i := 1; i < len(list); i++ { if list[i].Id > list[i-1].Id { t.Errorf("expected descending order by id, but list[%d].Id=%d > list[%d].Id=%d", i, list[i].Id, i-1, list[i-1].Id) } } } ================================================ FILE: internal/models/cleanup_verify_test.go ================================================ package models import ( "fmt" "testing" "time" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) // TestCleanupIntegration 端到端验证任务级日志清理 func TestCleanupIntegration(t *testing.T) { db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } oldDb := Db Db = db defer func() { Db = oldDb }() Db.AutoMigrate(&Task{}, &TaskLog{}) // 创建任务: task 10 保留2天, task 20 保留7天, task 30 无自定义(0) Db.Exec("INSERT INTO tasks (id, name, log_retention_days, level, protocol, spec, command, status, tag) VALUES (10, 'task-2day', 2, 1, 2, '@daily', 'ls', 0, '')") Db.Exec("INSERT INTO tasks (id, name, log_retention_days, level, protocol, spec, command, status, tag) VALUES (20, 'task-7day', 7, 1, 2, '@daily', 'ls', 0, '')") Db.Exec("INSERT INTO tasks (id, name, log_retention_days, level, protocol, spec, command, status, tag) VALUES (30, 'task-global', 0, 1, 2, '@daily', 'ls', 0, '')") now := time.Now() // 插入日志 // Task 10: 5条1天前(应保留), 5条3天前(应删除), 5条10天前(应删除) // Task 20: 5条3天前(应保留), 5条10天前(应删除) // Task 30: 5条10天前(无自定义策略，不由任务级清理处理) insertLogs := func(taskId int, name string, age time.Duration, count int) { for i := 0; i < count; i++ { Db.Create(&TaskLog{ TaskId: taskId, Name: name, Spec: "@daily", Protocol: 2, Command: "ls", StartTime: LocalTime(now.Add(-age)), EndTime: LocalTime(now.Add(-age).Add(time.Second)), Status: Finish, Result: "ok", }) } } insertLogs(10, "task-2day", 1*24*time.Hour, 5) // 1天前 → 保留 insertLogs(10, "task-2day", 3*24*time.Hour, 5) // 3天前 → 删除 insertLogs(10, "task-2day", 10*24*time.Hour, 5) // 10天前 → 删除 insertLogs(20, "task-7day", 3*24*time.Hour, 5) // 3天前 → 保留 insertLogs(20, "task-7day", 10*24*time.Hour, 5) // 10天前 → 删除 insertLogs(30, "task-global", 10*24*time.Hour, 5) // 不由任务级策略处理 // 验证初始状态 var count10, count20, count30 int64 Db.Model(&TaskLog{}).Where("task_id = 10").Count(&count10) Db.Model(&TaskLog{}).Where("task_id = 20").Count(&count20) Db.Model(&TaskLog{}).Where("task_id = 30").Count(&count30) fmt.Printf("Before cleanup - Task10: %d, Task20: %d, Task30: %d\n", count10, count20, count30) if count10 != 15 || count20 != 10 || count30 != 5 { t.Fatalf("Initial state wrong: %d, %d, %d", count10, count20, count30) } // 模拟 cron 清理逻辑: 查找自定义保留天数的任务并清理 taskLogModel := new(TaskLog) var tasks []Task Db.Where("log_retention_days > 0").Find(&tasks) if len(tasks) != 2 { t.Fatalf("Expected 2 tasks with custom retention, got %d", len(tasks)) } for _, task := range tasks { count, err := taskLogModel.RemoveByTaskIdAndDays(task.Id, task.LogRetentionDays) if err != nil { t.Fatalf("RemoveByTaskIdAndDays failed for task %d: %v", task.Id, err) } fmt.Printf("Task %d (%s, retention=%d days): deleted %d logs\n", task.Id, task.Name, task.LogRetentionDays, count) } // 验证清理后状态 Db.Model(&TaskLog{}).Where("task_id = 10").Count(&count10) Db.Model(&TaskLog{}).Where("task_id = 20").Count(&count20) Db.Model(&TaskLog{}).Where("task_id = 30").Count(&count30) fmt.Printf("After cleanup - Task10: %d, Task20: %d, Task30: %d\n", count10, count20, count30) // Task 10 (保留2天): 应只剩1天前的5条 if count10 != 5 { t.Errorf("Task 10: expected 5 logs remaining (1-day-old), got %d", count10) } // Task 20 (保留7天): 应只剩3天前的5条 if count20 != 5 { t.Errorf("Task 20: expected 5 logs remaining (3-day-old), got %d", count20) } // Task 30 (无自定义): 应该不受影响，仍有5条 if count30 != 5 { t.Errorf("Task 30: expected 5 logs untouched, got %d", count30) } fmt.Println("\n✅ 任务级日志清理验证通过!") fmt.Println(" - Task 10 (2天保留): 删除了3天前和10天前的日志，保留了1天前的") fmt.Println(" - Task 20 (7天保留): 删除了10天前的日志，保留了3天前的") fmt.Println(" - Task 30 (全局策略): 不受任务级清理影响") } ================================================ FILE: internal/models/host.go ================================================ package models import ( "gorm.io/gorm" ) // 主机 type Host struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Name string `json:"name" gorm:"type:varchar(64);not null"` Alias string `json:"alias" gorm:"type:varchar(32);not null;default:''"` Port int `json:"port" gorm:"not null;default:5921"` Remark string `json:"remark" gorm:"type:varchar(100);not null;default:''"` BaseModel `json:"-" gorm:"-"` Selected bool `json:"-" gorm:"-"` } // 新增 func (host *Host) Create() (insertId int, err error) { result := Db.Create(host) if result.Error == nil { insertId = host.Id } return insertId, result.Error } func (host *Host) UpdateBean(id int) (int64, error) { result := Db.Model(&Host{}).Where("id = ?", id). Select("name", "alias", "port", "remark"). Updates(host) return result.RowsAffected, result.Error } // 更新 func (host *Host) Update(id int, data CommonMap) (int64, error) { updateData := make(map[string]interface{}) for k, v := range data { updateData[k] = v } result := Db.Model(&Host{}).Where("id = ?", id).UpdateColumns(updateData) return result.RowsAffected, result.Error } // 删除 func (host *Host) Delete(id int) (int64, error) { result := Db.Delete(&Host{}, id) return result.RowsAffected, result.Error } func (host *Host) Find(id int) error { return Db.First(host, id).Error } func (host *Host) NameExists(name string, id int) (bool, error) { var count int64 query := Db.Model(&Host{}).Where("name = ?", name) if id != 0 { query = query.Where("id != ?", id) } err := query.Count(&count).Error return count > 0, err } func (host *Host) List(params CommonMap) ([]Host, error) { host.parsePageAndPageSize(params) list := make([]Host, 0) query := Db.Order("id DESC") host.parseWhere(query, params) err := query.Limit(host.PageSize).Offset(host.pageLimitOffset()).Find(&list).Error return list, err } func (host *Host) AllList() ([]Host, error) { list := make([]Host, 0) err := Db.Select("name", "port").Order("id DESC").Find(&list).Error return list, err } func (host *Host) Total(params CommonMap) (int64, error) { var count int64 query := Db.Model(&Host{}) host.parseWhere(query, params) err := query.Count(&count).Error return count, err } // 解析where func (host *Host) parseWhere(query *gorm.DB, params CommonMap) { if len(params) == 0 { return } id, ok := params["Id"] if ok && id.(int) > 0 { query.Where("id = ?", id) } name, ok := params["Name"] if ok && name.(string) != "" { query.Where("name = ?", name) } } ================================================ FILE: internal/models/login_log.go ================================================ package models import ( "time" ) // 用户登录日志 type LoginLog struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Username string `json:"username" gorm:"type:varchar(32);not null"` Ip string `json:"ip" gorm:"type:varchar(15);not null"` CreatedAt time.Time `json:"created" gorm:"column:created;autoCreateTime"` BaseModel `json:"-" gorm:"-"` } func (log *LoginLog) Create() (insertId int, err error) { result := Db.Create(log) if result.Error == nil { insertId = log.Id } return insertId, result.Error } func (log *LoginLog) List(params CommonMap) ([]LoginLog, error) { log.parsePageAndPageSize(params) list := make([]LoginLog, 0) err := Db.Order("id DESC").Limit(log.PageSize).Offset(log.pageLimitOffset()).Find(&list).Error return list, err } func (log *LoginLog) Total() (int64, error) { var count int64 err := Db.Model(&LoginLog{}).Count(&count).Error return count, err } ================================================ FILE: internal/models/migration.go ================================================ package models import ( "errors" "github.com/gocronx-team/gocron/internal/modules/logger" "gorm.io/gorm" ) type Migration struct{} // 首次安装, 创建数据库表 func (migration *Migration) Install(dbName string) error { setting := new(Setting) tables := []interface{}{ &User{}, &Task{}, &TaskLog{}, &Host{}, setting, &LoginLog{}, &TaskHost{}, &AgentToken{}, &AuditLog{}, &TaskScriptVersion{}, &TaskTemplate{}, } for _, table := range tables { if Db.Migrator().HasTable(table) { return errors.New("数据表已存在") } err := Db.AutoMigrate(table) if err != nil { return err } } // SQLite特殊处理：修复task_log表的自增主键 if Db.Dialector.Name() == "sqlite" { migration.fixSQLiteAutoIncrement() } // 初始化配置 if err := RepairSettings(); err != nil { return err } return nil } // 迭代升级数据库, 新建表、新增字段等 func (migration *Migration) Upgrade(oldVersionId int) { // v1.2版本不支持升级 if oldVersionId == 120 { return } versionIds := []int{110, 122, 130, 140, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 1510, 160} upgradeFuncs := []func(*gorm.DB) error{ migration.upgradeFor110, migration.upgradeFor122, migration.upgradeFor130, migration.upgradeFor140, migration.upgradeFor150, migration.upgradeFor151, migration.upgradeFor152, migration.upgradeFor153, migration.upgradeFor154, migration.upgradeFor155, migration.upgradeFor156, migration.upgradeFor157, migration.upgradeFor158, migration.upgradeFor159, migration.upgradeFor1510, migration.upgradeFor160, } startIndex := -1 // 从当前版本的下一版本开始升级 for i, value := range versionIds { if value > oldVersionId { startIndex = i break } } if startIndex == -1 { return } length := len(versionIds) if startIndex >= length { return } err := Db.Transaction(func(tx *gorm.DB) error { for startIndex < length { err := upgradeFuncs[startIndex](tx) if err != nil { return err } startIndex++ } return nil }) if err != nil { logger.Fatal("数据库升级失败", err) } } // 升级到v1.1版本 func (migration *Migration) upgradeFor110(tx *gorm.DB) error { logger.Info("开始升级到v1.1") // 创建表task_host err := tx.AutoMigrate(&TaskHost{}) if err != nil { return err } // 把task对应的host_id写入task_host表 type OldTask struct { Id int HostId int } var results []OldTask err = tx.Table(TablePrefix+"task").Select("id", "host_id").Where("host_id > ?", 0).Find(&results).Error if err != nil { return err } for _, value := range results { taskHostModel := &TaskHost{ TaskId: value.Id, HostId: value.HostId, } err = tx.Create(taskHostModel).Error if err != nil { return err } } // 删除task表host_id字段 err = tx.Migrator().DropColumn(&Task{}, "host_id") logger.Info("已升级到v1.1\n") return err } // 升级到1.2.2版本 func (migration *Migration) upgradeFor122(tx *gorm.DB) error { logger.Info("开始升级到v1.2.2") // task表增加tag字段 if !tx.Migrator().HasColumn(&Task{}, "tag") { err := tx.Migrator().AddColumn(&Task{}, "tag") if err != nil { return err } } logger.Info("已升级到v1.2.2\n") return nil } // 升级到v1.3版本 func (migration *Migration) upgradeFor130(tx *gorm.DB) error { logger.Info("开始升级到v1.3") // 删除user表deleted字段（如果存在） if tx.Migrator().HasColumn(&User{}, "deleted") { err := tx.Migrator().DropColumn(&User{}, "deleted") if err != nil { return err } } logger.Info("已升级到v1.3\n") return nil } // 升级到v1.4版本 func (migration *Migration) upgradeFor140(tx *gorm.DB) error { logger.Info("开始升级到v1.4") // task表增加字段 // retry_interval 重试间隔时间(秒) // http_method http请求方法 if !tx.Migrator().HasColumn(&Task{}, "retry_interval") { err := tx.Migrator().AddColumn(&Task{}, "retry_interval") if err != nil { return err } } if !tx.Migrator().HasColumn(&Task{}, "http_method") { err := tx.Migrator().AddColumn(&Task{}, "http_method") if err != nil { return err } } logger.Info("已升级到v1.4\n") return nil } func (m *Migration) upgradeFor150(tx *gorm.DB) error { logger.Info("开始升级到v1.5") // task表增加字段 notify_keyword if !tx.Migrator().HasColumn(&Task{}, "notify_keyword") { err := tx.Migrator().AddColumn(&Task{}, "notify_keyword") if err != nil { return err } } // 检查并创建邮件模板配置 var count int64 tx.Model(&Setting{}).Where("code = ? AND `key` = ?", MailCode, MailTemplateKey).Count(&count) if count == 0 { settingModel := &Setting{ Code: MailCode, Key: MailTemplateKey, Value: emailTemplate, } if err := tx.Create(settingModel).Error; err != nil { return err } } // 检查并创建Slack模板配置 tx.Model(&Setting{}).Where("code = ? AND `key` = ?", SlackCode, SlackTemplateKey).Count(&count) if count == 0 { settingModel := &Setting{ Code: SlackCode, Key: SlackTemplateKey, Value: slackTemplate, } if err := tx.Create(settingModel).Error; err != nil { return err } } // 检查并创建Webhook URL配置 tx.Model(&Setting{}).Where("code = ? AND `key` = ?", WebhookCode, WebhookUrlKey).Count(&count) if count == 0 { settingModel := &Setting{ Code: WebhookCode, Key: WebhookUrlKey, Value: "", } if err := tx.Create(settingModel).Error; err != nil { return err } } // 检查并创建Webhook模板配置 tx.Model(&Setting{}).Where("code = ? AND `key` = ?", WebhookCode, WebhookTemplateKey).Count(&count) if count == 0 { settingModel := &Setting{ Code: WebhookCode, Key: WebhookTemplateKey, Value: webhookTemplate, } if err := tx.Create(settingModel).Error; err != nil { return err } } logger.Info("已升级到v1.5\n") return nil } // 升级到v1.5.1版本 - 添加2FA字段 func (m *Migration) upgradeFor151(tx *gorm.DB) error { logger.Info("开始升级到v1.5.1 - 添加2FA支持") // user表增加two_factor_key字段 if !tx.Migrator().HasColumn(&User{}, "two_factor_key") { err := tx.Migrator().AddColumn(&User{}, "two_factor_key") if err != nil { return err } } // user表增加two_factor_on字段 if !tx.Migrator().HasColumn(&User{}, "two_factor_on") { err := tx.Migrator().AddColumn(&User{}, "two_factor_on") if err != nil { return err } } logger.Info("已升级到v1.5.1\n") return nil } // 升级到v1.5.2版本 - 修复 SQLite host 表 AUTOINCREMENT func (m *Migration) upgradeFor152(tx *gorm.DB) error { logger.Info("开始升级到v1.5.2 - 修复 host 表自增主键") // 只对 SQLite 数据库执行修复 if tx.Dialector.Name() == "sqlite" { var tableSQL string err := tx.Raw("SELECT sql FROM sqlite_master WHERE type='table' AND name='host'").Scan(&tableSQL).Error if err != nil { return err } if len(tableSQL) > 0 && !contains(tableSQL, "AUTOINCREMENT") { logger.Info("检测到 host 表需要修复") // 检查是否有数据 var hasData int64 tx.Raw("SELECT COUNT(*) FROM host").Scan(&hasData) // 重建表以支持 AUTOINCREMENT err = tx.Exec(` CREATE TABLE IF NOT EXISTS host_new ( id INTEGER PRIMARY KEY AUTOINCREMENT, name varchar(64) NOT NULL, alias varchar(32) NOT NULL DEFAULT '', port integer NOT NULL DEFAULT 5921, remark varchar(100) NOT NULL DEFAULT '' ); `).Error if err != nil { return err } // 如果有数据，迁移数据 if hasData > 0 { err = tx.Exec(` INSERT INTO host_new (name, alias, port, remark) SELECT name, alias, port, remark FROM host WHERE name IS NOT NULL; `).Error if err != nil { return err } } // 删除旧表 err = tx.Exec(`DROP TABLE host;`).Error if err != nil { return err } // 重命名新表 err = tx.Exec(`ALTER TABLE host_new RENAME TO host;`).Error if err != nil { return err } logger.Info("host 表已重建，支持自增主键") } else { logger.Info("host 表结构正确，无需修复") } } logger.Info("已升级到v1.5.2\n") return nil } // 升级到v1.5.3版本 - 修复 SQLite task_log 表 AUTOINCREMENT func (m *Migration) upgradeFor153(tx *gorm.DB) error { logger.Info("开始升级到v1.5.3 - 修复 task_log 表自增主键") // 只对 SQLite 数据库执行修复 if tx.Dialector.Name() == "sqlite" { var tableSQL string err := tx.Raw("SELECT sql FROM sqlite_master WHERE type='table' AND name='task_log'").Scan(&tableSQL).Error if err != nil { return err } if len(tableSQL) > 0 && !contains(tableSQL, "AUTOINCREMENT") { logger.Info("检测到 task_log 表需要修复") err = tx.Exec(` CREATE TABLE IF NOT EXISTS task_log_new ( id INTEGER PRIMARY KEY AUTOINCREMENT, task_id integer NOT NULL DEFAULT 0, name varchar(32) NOT NULL, spec varchar(64) NOT NULL, protocol tinyint NOT NULL, command varchar(256) NOT NULL, timeout mediumint NOT NULL DEFAULT 0, retry_times tinyint NOT NULL DEFAULT 0, hostname varchar(128) NOT NULL DEFAULT '', start_time datetime, end_time datetime, status tinyint NOT NULL DEFAULT 1, result mediumtext NOT NULL ); `).Error if err != nil { return err } // 迁移最近的数据（最多10000条） var hasData int64 tx.Raw("SELECT COUNT(*) FROM task_log").Scan(&hasData) if hasData > 0 { err = tx.Exec(` INSERT INTO task_log_new (task_id, name, spec, protocol, command, timeout, retry_times, hostname, start_time, end_time, status, result) SELECT task_id, name, spec, protocol, command, timeout, retry_times, hostname, start_time, end_time, status, result FROM task_log WHERE task_id IS NOT NULL ORDER BY start_time DESC LIMIT 10000; `).Error if err != nil { return err } } err = tx.Exec(`DROP TABLE task_log;`).Error if err != nil { return err } err = tx.Exec(`ALTER TABLE task_log_new RENAME TO task_log;`).Error if err != nil { return err } logger.Info("task_log 表已重建，支持自增主键") } else { logger.Info("task_log 表结构正确，无需修复") } // 清理状态异常的历史任务日志（status=1 且 result 为空） err = tx.Exec(` UPDATE task_log SET status = 0, result = '任务异常终止（未正常完成）', end_time = datetime(start_time, '+1 second') WHERE status = 1 AND (result IS NULL OR result = ''); `).Error if err != nil { logger.Error("清理异常任务日志失败", err) } else { logger.Info("已清理状态异常的历史任务日志") } } logger.Info("已升级到v1.5.3\n") return nil } // 升级到v1.5.4版本 - 添加agent_token表 func (m *Migration) upgradeFor154(tx *gorm.DB) error { logger.Info("开始升级到v1.5.4 - 添加agent自动注册支持") if err := tx.AutoMigrate(&AgentToken{}); err != nil { return err } if err := tx.Migrator().AlterColumn(&AgentToken{}, "UsedAt"); err != nil { logger.Warn("调整 agent_token.used_at 可空属性失败", err) } logger.Info("已升级到v1.5.4\n") return nil } // 升级到v1.5.5版本 - 修改 host.id 和 task_host.host_id 字段类型从 smallint 到 int func (m *Migration) upgradeFor155(tx *gorm.DB) error { logger.Info("开始升级到v1.5.5 - 扩展主机ID字段类型和性能优化") // 1. 使用 GORM AutoMigrate 自动调整字段类型 // GORM 会根据模型定义自动修改字段类型 if err := tx.AutoMigrate(&Host{}, &TaskHost{}); err != nil { return err } logger.Info("✓ 主机ID字段类型已升级") // 2. 性能优化: 添加 task_log.start_time 索引 (用于日志清理和时间范围查询) if !tx.Migrator().HasIndex(&TaskLog{}, "idx_task_log_start_time") { if err := tx.Migrator().CreateIndex(&TaskLog{}, "StartTime"); err != nil { logger.Warn("创建 task_log.start_time 索引失败", err) } else { logger.Info("✓ 创建 task_log.start_time 索引") } } // 3. 性能优化: 添加 task_log 复合索引 (task_id, status) - 用于查询特定任务的执行状态 if !tx.Migrator().HasIndex(&TaskLog{}, "idx_task_log_task_status") { if err := tx.Exec("CREATE INDEX idx_task_log_task_status ON " + TablePrefix + "task_log(task_id, status)").Error; err != nil { logger.Warn("创建 task_log 复合索引失败", err) } else { logger.Info("✓ 创建 task_log(task_id, status) 复合索引") } } // 4. 性能优化: 添加 task 复合索引 (status, level) - 用于 ActiveList 查询 if !tx.Migrator().HasIndex(&Task{}, "idx_task_status_level") { if err := tx.Exec("CREATE INDEX idx_task_status_level ON " + TablePrefix + "task(status, level)").Error; err != nil { logger.Warn("创建 task 复合索引失败", err) } else { logger.Info("✓ 创建 task(status, level) 复合索引") } } logger.Info("已升级到v1.5.5\n") return nil } // 升级到v1.5.6版本 - 更新字段默认值以支持基于0的索引 func (m *Migration) upgradeFor156(tx *gorm.DB) error { logger.Info("开始升级到v1.5.6 - 更新字段默认值") // 更新 notify_status 默认值为 1 的旧数据为 0（禁用通知） // 只更新 notify_type=0 且 notify_receiver_id 为空的记录，这些是真正的默认值 result := tx.Exec(` UPDATE ` + TablePrefix + `task SET notify_status = 0 WHERE notify_status = 1 AND notify_type = 0 AND (notify_receiver_id = '' OR notify_receiver_id IS NULL) `) if result.Error != nil { logger.Warn("更新 notify_status 默认值失败", result.Error) } else if result.RowsAffected > 0 { logger.Infof("✓ 已更新 %d 条任务的 notify_status 默认值", result.RowsAffected) } logger.Info("已升级到v1.5.6\n") return nil } // 升级到v1.5.7版本 - 扩展命令字段长度到TEXT类型 func (m *Migration) upgradeFor157(tx *gorm.DB) error { logger.Info("开始升级到v1.5.7 - 扩展命令字段长度") // 扩展 command 字段从 varchar 到 text if err := tx.Exec(`ALTER TABLE ` + TablePrefix + `task MODIFY COLUMN command text NOT NULL`).Error; err != nil { logger.Warn("扩展 command 字段类型失败", err) } else { logger.Info("✓ command 字段已扩展为 TEXT 类型（最多 65535 字符）") } logger.Info("已升级到v1.5.7\n") return nil } // 升级到v1.5.8版本 - 多标签支持 + 任务级日志保留天数 func (m *Migration) upgradeFor158(tx *gorm.DB) error { logger.Info("开始升级到v1.5.8 - 多标签支持、任务级日志保留天数") // 扩展 tag 字段从 varchar(32) 到 varchar(255) 以支持多标签 if err := tx.Migrator().AlterColumn(&Task{}, "Tag"); err != nil { logger.Warn("扩展 tag 字段类型失败", err) } else { logger.Info("✓ tag 字段已扩展为 varchar(255)") } // 添加任务级日志保留天数字段 if !tx.Migrator().HasColumn(&Task{}, "log_retention_days") { err := tx.Migrator().AddColumn(&Task{}, "log_retention_days") if err != nil { return err } logger.Info("✓ 已添加 log_retention_days 字段") } logger.Info("已升级到v1.5.8\n") return nil } // 升级到v1.5.9版本 - HTTP任务增强：POST Body、自定义Header、响应断言 func (m *Migration) upgradeFor159(tx *gorm.DB) error { logger.Info("开始升级到v1.5.9 - HTTP任务增强") // 添加 http_body 字段 if !tx.Migrator().HasColumn(&Task{}, "http_body") { if err := tx.Migrator().AddColumn(&Task{}, "HttpBody"); err != nil { logger.Warn("添加 http_body 字段失败", err) } else { logger.Info("✓ 已添加 http_body 字段") } } // 添加 http_headers 字段 if !tx.Migrator().HasColumn(&Task{}, "http_headers") { if err := tx.Migrator().AddColumn(&Task{}, "HttpHeaders"); err != nil { logger.Warn("添加 http_headers 字段失败", err) } else { logger.Info("✓ 已添加 http_headers 字段") } } // 添加 success_pattern 字段 if !tx.Migrator().HasColumn(&Task{}, "success_pattern") { if err := tx.Migrator().AddColumn(&Task{}, "SuccessPattern"); err != nil { logger.Warn("添加 success_pattern 字段失败", err) } else { logger.Info("✓ 已添加 success_pattern 字段") } } logger.Info("已升级到v1.5.9\n") return nil } // 升级到v1.5.10版本 - 添加审计日志表 func (m *Migration) upgradeFor1510(tx *gorm.DB) error { logger.Info("开始升级到v1.5.10 - 添加审计日志支持") if err := tx.AutoMigrate(&AuditLog{}); err != nil { return err } logger.Info("已升级到v1.5.10\n") return nil } // 升级到v1.6.0版本 - 添加脚本版本管理和任务模板 func (m *Migration) upgradeFor160(tx *gorm.DB) error { logger.Info("开始升级到v1.6.0 - 添加脚本版本管理和任务模板") if err := tx.AutoMigrate(&TaskScriptVersion{}); err != nil { return err } logger.Info("✓ 已创建 task_script_version 表") if err := tx.AutoMigrate(&TaskTemplate{}); err != nil { return err } logger.Info("✓ 已创建 task_template 表") // 初始化内置模板 var count int64 tx.Model(&TaskTemplate{}).Where("is_builtin = ?", 1).Count(&count) if count == 0 { seedBuiltinTemplates(tx) logger.Info("✓ 已初始化内置模板") } logger.Info("已升级到v1.6.0\n") return nil } // contains 检查字符串是否包含子串 func contains(s, substr string) bool { return len(s) >= len(substr) && (s == substr || len(s) > len(substr) && (s[:len(substr)] == substr || s[len(s)-len(substr):] == substr || containsMiddle(s, substr))) } func containsMiddle(s, substr string) bool { for i := 0; i <= len(s)-len(substr); i++ { if s[i:i+len(substr)] == substr { return true } } return false } // 修复SQLite表的自增主键问题 func (m *Migration) fixSQLiteAutoIncrement() { logger.Info("检查SQLite表结构...") // 修复task_log表 var taskLogSQL string Db.Raw("SELECT sql FROM sqlite_master WHERE type='table' AND name='task_log'").Scan(&taskLogSQL) if len(taskLogSQL) > 0 && !contains(taskLogSQL, "AUTOINCREMENT") { logger.Info("修复task_log表自增主键...") Db.Exec(` CREATE TABLE task_log_new ( id INTEGER PRIMARY KEY AUTOINCREMENT, task_id integer NOT NULL DEFAULT 0, name varchar(32) NOT NULL, spec varchar(64) NOT NULL, protocol tinyint NOT NULL, command varchar(256) NOT NULL, timeout mediumint NOT NULL DEFAULT 0, retry_times tinyint NOT NULL DEFAULT 0, hostname varchar(128) NOT NULL DEFAULT '', start_time datetime, end_time datetime, status tinyint NOT NULL DEFAULT 1, result mediumtext NOT NULL ); `) Db.Exec(`DROP TABLE task_log;`) Db.Exec(`ALTER TABLE task_log_new RENAME TO task_log;`) logger.Info("修复task_log表完成") } // 修复host表 var hostSQL string Db.Raw("SELECT sql FROM sqlite_master WHERE type='table' AND name='host'").Scan(&hostSQL) if len(hostSQL) > 0 && !contains(hostSQL, "AUTOINCREMENT") { logger.Info("修复host表自增主键...") Db.Exec(` CREATE TABLE host_new ( id INTEGER PRIMARY KEY AUTOINCREMENT, name varchar(64) NOT NULL, alias varchar(32) NOT NULL DEFAULT '', port integer NOT NULL DEFAULT 5921, remark varchar(100) NOT NULL DEFAULT '' ); `) Db.Exec(`DROP TABLE host;`) Db.Exec(`ALTER TABLE host_new RENAME TO host;`) logger.Info("修复host表完成") } } ================================================ FILE: internal/models/model.go ================================================ package models import ( "fmt" "os" "path/filepath" "strings" "time" "github.com/gin-gonic/gin" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/driver/mysql" "gorm.io/driver/postgres" "gorm.io/gorm" "gorm.io/gorm/logger" "gorm.io/gorm/schema" "github.com/gocronx-team/gocron/internal/modules/app" glogger "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/setting" ) type Status int8 type CommonMap map[string]interface{} var TablePrefix = "" var Db *gorm.DB // dbKeepAliveStop is closed to signal the keepDbAlived goroutine to exit. var dbKeepAliveStop chan struct{} const ( Disabled Status = 0 // 禁用 Failure Status = 0 // 失败 Enabled Status = 1 // 启用 Running Status = 1 // 运行中 Finish Status = 2 // 完成 Cancel Status = 3 // 取消 ) const ( Page = 1 // 当前页数 PageSize = 20 // 每页多少条数据 MaxPageSize = 1000 // 每次最多取多少条 ) const DefaultTimeFormat = "2006-01-02 15:04:05" const ( dbPingInterval = 90 * time.Second dbMaxLiftTime = 2 * time.Hour ) type BaseModel struct { Page int `gorm:"-"` PageSize int `gorm:"-"` } func (model *BaseModel) parsePageAndPageSize(params CommonMap) { page, ok := params["Page"] if ok { model.Page = page.(int) } pageSize, ok := params["PageSize"] if ok { model.PageSize = pageSize.(int) } if model.Page <= 0 { model.Page = Page } if model.PageSize <= 0 { model.PageSize = MaxPageSize } } func (model *BaseModel) pageLimitOffset() int { return (model.Page - 1) * model.PageSize } // 创建Db func CreateDb() *gorm.DB { dsn := getDbEngineDSN(app.Setting) var dialector gorm.Dialector engine := strings.ToLower(app.Setting.Db.Engine) switch engine { case "mysql": dialector = mysql.Open(dsn) case "postgres": dialector = postgres.Open(dsn) case "sqlite": ensureSqliteDir(dsn) dialector = gormlite.Open(dsn) default: glogger.Fatal("不支持的数据库类型", nil) } // 配置 gorm config := &gorm.Config{ NamingStrategy: schema.NamingStrategy{ TablePrefix: app.Setting.Db.Prefix, SingularTable: true, }, Logger: logger.Default.LogMode(logger.Silent), } // 开发模式下开启日志 if gin.Mode() == gin.DebugMode { config.Logger = logger.Default.LogMode(logger.Info) } db, err := gorm.Open(dialector, config) if err != nil { glogger.Fatal("创建gorm引擎失败", err) } sqlDB, err := db.DB() if err != nil { glogger.Fatal("获取数据库连接失败", err) } // SQLite 需要特殊的连接池配置 if engine == "sqlite" { sqlDB.SetMaxOpenConns(1) // SQLite 只允许一个写连接 } else { sqlDB.SetMaxIdleConns(app.Setting.Db.MaxIdleConns) sqlDB.SetMaxOpenConns(app.Setting.Db.MaxOpenConns) } sqlDB.SetConnMaxLifetime(dbMaxLiftTime) if app.Setting.Db.Prefix != "" { TablePrefix = app.Setting.Db.Prefix } dbKeepAliveStop = make(chan struct{}) go keepDbAlived(db, dbKeepAliveStop) return db } // StopKeepAlive signals the keepDbAlived goroutine to exit. Safe to call multiple times. func StopKeepAlive() { if dbKeepAliveStop == nil { return } select { case <-dbKeepAliveStop: // already closed default: close(dbKeepAliveStop) } } // 创建临时数据库连接 func CreateTmpDb(setting *setting.Setting) (*gorm.DB, error) { dsn := getDbEngineDSN(setting) var dialector gorm.Dialector engine := strings.ToLower(setting.Db.Engine) switch engine { case "mysql": dialector = mysql.Open(dsn) case "postgres": dialector = postgres.Open(dsn) case "sqlite": ensureSqliteDir(dsn) dialector = gormlite.Open(dsn) default: return nil, fmt.Errorf("不支持的数据库类型: %s", engine) } return gorm.Open(dialector, &gorm.Config{}) } // 获取数据库引擎DSN mysql,postgres func getDbEngineDSN(setting *setting.Setting) string { engine := strings.ToLower(setting.Db.Engine) dsn := "" switch engine { case "mysql": dsn = fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local", setting.Db.User, setting.Db.Password, setting.Db.Host, setting.Db.Port, setting.Db.Database, setting.Db.Charset) case "postgres": dsn = fmt.Sprintf("user=%s password=%s host=%s port=%d dbname=%s sslmode=disable", setting.Db.User, setting.Db.Password, setting.Db.Host, setting.Db.Port, setting.Db.Database) case "sqlite": dsn = setting.Db.Database } return dsn } func keepDbAlived(db *gorm.DB, stop <-chan struct{}) { ticker := time.NewTicker(dbPingInterval) defer ticker.Stop() for { select { case <-stop: return case <-ticker.C: sqlDB, err := db.DB() if err != nil { glogger.Infof("database get connection: %s", err) continue } if err := sqlDB.Ping(); err != nil { glogger.Infof("database ping failed: %s", err) } else { glogger.Infof("database ping: ok") } } } } // 确保 SQLite 数据库文件所在目录存在 func ensureSqliteDir(dbPath string) { // 清理并规范化路径 dbPath = filepath.Clean(dbPath) dir := filepath.Dir(dbPath) if dir != "" && dir != "." { // 验证路径不是绝对路径时，确保不包含父目录引用 if !filepath.IsAbs(dbPath) && strings.Contains(dbPath, "..") { glogger.Fatal("非法的数据库路径", nil) } if err := os.MkdirAll(dir, 0750); err != nil { glogger.Fatal("创建SQLite数据库目录失败", err) } } } ================================================ FILE: internal/models/scheduler_lock.go ================================================ package models import "time" // SchedulerLock 调度器分布式锁表 // 参考 XXL-JOB 的数据库行锁方案，用 SELECT ... FOR UPDATE 实现选主 type SchedulerLock struct { Id int `gorm:"primaryKey;autoIncrement"` LockName string `gorm:"type:varchar(64);uniqueIndex;not null"` // 锁名称 LockedBy string `gorm:"type:varchar(255);not null"` // 持有者标识 (hostname:pid) LockedAt time.Time `gorm:"not null"` // 获取锁时间 ExpireAt time.Time `gorm:"not null"` // 过期时间 Version int `gorm:"not null;default:0"` // 乐观锁版本号 CreatedAt time.Time UpdatedAt time.Time } func (SchedulerLock) TableName() string { return TablePrefix + "scheduler_lock" } ================================================ FILE: internal/models/scheduler_lock_test.go ================================================ package models import "testing" func TestSchedulerLock_TableName(t *testing.T) { lock := SchedulerLock{} // Default: no prefix original := TablePrefix TablePrefix = "" defer func() { TablePrefix = original }() if got := lock.TableName(); got != "scheduler_lock" { t.Errorf("expected %q, got %q", "scheduler_lock", got) } // With prefix TablePrefix = "gocron_" if got := lock.TableName(); got != "gocron_scheduler_lock" { t.Errorf("expected %q, got %q", "gocron_scheduler_lock", got) } } ================================================ FILE: internal/models/setting.go ================================================ package models import ( "encoding/json" "strconv" ) type Setting struct { Id int `gorm:"primaryKey;autoIncrement"` Code string `gorm:"type:varchar(32);not null"` Key string `gorm:"type:varchar(64);not null"` Value string `gorm:"type:varchar(4096);not null;default:''"` } const slackTemplate = `Task ID: {{.TaskId}} Task Name: {{.TaskName}} Status: {{.Status}} Result: {{.Result}} Remark: {{.Remark}}` const emailTemplate = `Task ID: {{.TaskId}} Task Name: {{.TaskName}} Status: {{.Status}} Result: {{.Result}} Remark: {{.Remark}}` const webhookTemplate = ` { "task_id": "{{.TaskId}}", "task_name": "{{.TaskName}}", "status": "{{.Status}}", "result": "{{.Result}}", "remark": "{{.Remark}}" } ` const ( SlackCode = "slack" SlackUrlKey = "url" SlackTemplateKey = "template" SlackChannelKey = "channel" ) const ( MailCode = "mail" MailTemplateKey = "template" MailServerKey = "server" MailUserKey = "user" ) const ( WebhookCode = "webhook" WebhookTemplateKey = "template" WebhookUrlKey = "url" ) const ( SystemCode = "system" LogRetentionDaysKey = "log_retention_days" LogCleanupTimeKey = "log_cleanup_time" LogFileSizeLimitKey = "log_file_size_limit" ) // region slack配置 type Slack struct { Url string `json:"url"` Channels []Channel `json:"channels"` Template string `json:"template"` } type Channel struct { Id int `json:"id"` Name string `json:"name"` } func (setting *Setting) Slack() (Slack, error) { list := make([]Setting, 0) err := Db.Where("code = ?", SlackCode).Find(&list).Error slack := Slack{} if err != nil { return slack, err } setting.formatSlack(list, &slack) return slack, err } func (setting *Setting) formatSlack(list []Setting, slack *Slack) { for _, v := range list { switch v.Key { case SlackUrlKey: slack.Url = v.Value case SlackTemplateKey: slack.Template = v.Value default: slack.Channels = append(slack.Channels, Channel{ v.Id, v.Value, }) } } } func (setting *Setting) UpdateSlack(url, template string) error { setting.Value = url Db.Model(&Setting{}).Where("code = ? AND `key` = ?", SlackCode, SlackUrlKey).Update("value", url) setting.Value = template Db.Model(&Setting{}).Where("code = ? AND `key` = ?", SlackCode, SlackTemplateKey).Update("value", template) return nil } // 创建slack渠道 func (setting *Setting) CreateChannel(channel string) (int64, error) { setting.Code = SlackCode setting.Key = SlackChannelKey setting.Value = channel result := Db.Create(setting) return result.RowsAffected, result.Error } func (setting *Setting) IsChannelExist(channel string) bool { var count int64 Db.Model(&Setting{}).Where("code = ? AND `key` = ? AND value = ?", SlackCode, SlackChannelKey, channel).Count(&count) return count > 0 } // 删除slack渠道 func (setting *Setting) RemoveChannel(id int) (int64, error) { result := Db.Where("code = ? AND `key` = ? AND id = ?", SlackCode, SlackChannelKey, id).Delete(&Setting{}) return result.RowsAffected, result.Error } // endregion type Mail struct { Host string `json:"host"` Port int `json:"port"` User string `json:"user"` Password string `json:"password"` MailUsers []MailUser `json:"mail_users"` Template string `json:"template"` } type MailUser struct { Id int `json:"id"` Username string `json:"username"` Email string `json:"email"` } // region 邮件配置 func (setting *Setting) Mail() (Mail, error) { list := make([]Setting, 0) err := Db.Where("code = ?", MailCode).Find(&list).Error mail := Mail{MailUsers: make([]MailUser, 0)} if err != nil { return mail, err } setting.formatMail(list, &mail) return mail, err } func (setting *Setting) formatMail(list []Setting, mail *Mail) { mailUser := MailUser{} for _, v := range list { switch v.Key { case MailServerKey: if v.Value != "" { _ = json.Unmarshal([]byte(v.Value), mail) } case MailUserKey: if v.Value != "" { _ = json.Unmarshal([]byte(v.Value), &mailUser) mailUser.Id = v.Id mail.MailUsers = append(mail.MailUsers, mailUser) } case MailTemplateKey: mail.Template = v.Value } } } func (setting *Setting) UpdateMail(config, template string) error { Db.Model(&Setting{}).Where("code = ? AND `key` = ?", MailCode, MailServerKey).Update("value", config) Db.Model(&Setting{}).Where("code = ? AND `key` = ?", MailCode, MailTemplateKey).Update("value", template) return nil } func (setting *Setting) CreateMailUser(username, email string) (int64, error) { setting.Code = MailCode setting.Key = MailUserKey mailUser := MailUser{0, username, email} jsonByte, err := json.Marshal(mailUser) if err != nil { return 0, err } setting.Value = string(jsonByte) result := Db.Create(setting) return result.RowsAffected, result.Error } func (setting *Setting) RemoveMailUser(id int) (int64, error) { result := Db.Where("code = ? AND `key` = ? AND id = ?", MailCode, MailUserKey, id).Delete(&Setting{}) return result.RowsAffected, result.Error } type WebHook struct { WebhookUrls []WebhookUrl `json:"webhook_urls"` Template string `json:"template"` } type WebhookUrl struct { Id int `json:"id"` Name string `json:"name"` Url string `json:"url"` } func (setting *Setting) Webhook() (WebHook, error) { list := make([]Setting, 0) err := Db.Where("code = ?", WebhookCode).Find(&list).Error webHook := WebHook{WebhookUrls: make([]WebhookUrl, 0)} if err != nil { return webHook, err } setting.formatWebhook(list, &webHook) return webHook, err } func (setting *Setting) formatWebhook(list []Setting, webHook *WebHook) { webhookUrl := WebhookUrl{} for _, v := range list { switch v.Key { case WebhookUrlKey: if v.Value != "" { _ = json.Unmarshal([]byte(v.Value), &webhookUrl) webhookUrl.Id = v.Id webHook.WebhookUrls = append(webHook.WebhookUrls, webhookUrl) } case WebhookTemplateKey: webHook.Template = v.Value } } } func (setting *Setting) UpdateWebHook(template string) error { Db.Model(&Setting{}).Where("code = ? AND `key` = ?", WebhookCode, WebhookTemplateKey).Update("value", template) return nil } func (setting *Setting) CreateWebhookUrl(name, url string) (int64, error) { webhookUrl := WebhookUrl{0, name, url} jsonByte, err := json.Marshal(webhookUrl) if err != nil { return 0, err } newSetting := Setting{ Code: WebhookCode, Key: WebhookUrlKey, Value: string(jsonByte), } result := Db.Create(&newSetting) return result.RowsAffected, result.Error } func (setting *Setting) RemoveWebhookUrl(id int) (int64, error) { result := Db.Where("code = ? AND `key` = ? AND id = ?", WebhookCode, WebhookUrlKey, id).Delete(&Setting{}) return result.RowsAffected, result.Error } // endregion // region 通用配置辅助方法 // getSettingValue 获取配置值的通用方法 func (setting *Setting) getSettingValue(code, key string) (string, error) { var s Setting err := Db.Where("code = ? AND `key` = ?", code, key).First(&s).Error if err != nil { return "", err } return s.Value, nil } // updateOrCreateSetting 更新或创建配置的通用方法 func (setting *Setting) updateOrCreateSetting(code, key, value string) error { var s Setting err := Db.Where("code = ? AND `key` = ?", code, key).First(&s).Error if err != nil { // 记录不存在，创建新记录 s.Code = code s.Key = key s.Value = value result := Db.Create(&s) return result.Error } // 记录存在，更新 result := Db.Model(&Setting{}).Where("code = ? AND `key` = ?", code, key).Update("value", value) return result.Error } // endregion // region 系统配置 func (setting *Setting) GetLogRetentionDays() int { value, err := setting.getSettingValue(SystemCode, LogRetentionDaysKey) if err != nil || value == "" { return 0 } days, err := strconv.Atoi(value) if err != nil { return 0 } return days } func (setting *Setting) UpdateLogRetentionDays(days int) error { return setting.updateOrCreateSetting(SystemCode, LogRetentionDaysKey, strconv.Itoa(days)) } func (setting *Setting) GetLogCleanupTime() string { value, err := setting.getSettingValue(SystemCode, LogCleanupTimeKey) if err != nil || value == "" { return "03:00" } return value } func (setting *Setting) UpdateLogCleanupTime(cleanupTime string) error { return setting.updateOrCreateSetting(SystemCode, LogCleanupTimeKey, cleanupTime) } func (setting *Setting) GetLogFileSizeLimit() int { value, err := setting.getSettingValue(SystemCode, LogFileSizeLimitKey) if err != nil || value == "" { return 0 } size, err := strconv.Atoi(value) if err != nil { return 0 } return size } func (setting *Setting) UpdateLogFileSizeLimit(size int) error { return setting.updateOrCreateSetting(SystemCode, LogFileSizeLimitKey, strconv.Itoa(size)) } // endregion ================================================ FILE: internal/models/setting_init.go ================================================ package models import "github.com/gocronx-team/gocron/internal/modules/logger" // RepairSettings 修复缺失的 Setting 配置记录 // 用于解决数据库迁移或升级过程中可能出现的配置缺失问题 func RepairSettings() error { logger.Info("Starting to check and repair Setting configuration...") // 定义所有必需的配置项 requiredSettings := []struct { Code string Key string Value string }{ // Slack 配置 {SlackCode, SlackUrlKey, ""}, {SlackCode, SlackTemplateKey, slackTemplate}, // 邮件配置 {MailCode, MailServerKey, ""}, {MailCode, MailTemplateKey, emailTemplate}, // Webhook 配置 {WebhookCode, WebhookUrlKey, ""}, {WebhookCode, WebhookTemplateKey, webhookTemplate}, // 系统配置 {SystemCode, LogRetentionDaysKey, "0"}, {SystemCode, LogCleanupTimeKey, "03:00"}, {SystemCode, LogFileSizeLimitKey, "0"}, } // 检查并创建缺失的配置 for _, cfg := range requiredSettings { var count int64 err := Db.Model(&Setting{}).Where("code = ? AND `key` = ?", cfg.Code, cfg.Key).Count(&count).Error if err != nil { logger.Error("Failed to check configuration:", err) return err } if count == 0 { setting := &Setting{ Code: cfg.Code, Key: cfg.Key, Value: cfg.Value, } if err := Db.Create(setting).Error; err != nil { logger.Error("Failed to create configuration:", err) return err } logger.Infof("Created missing configuration: code=%s, key=%s", cfg.Code, cfg.Key) } } logger.Info("Setting configuration check completed") return nil } ================================================ FILE: internal/models/setting_refactor_test.go ================================================ package models import ( "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) // TestSettingRefactorBackwardCompatibility 测试重构后的向后兼容性 func TestSettingRefactorBackwardCompatibility(t *testing.T) { // 创建内存数据库 db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("failed to connect database: %v", err) } // 自动迁移 if err := db.AutoMigrate(&Setting{}); err != nil { t.Fatalf("failed to migrate: %v", err) } // 保存原始数据库连接 oldDb := Db Db = db defer func() { Db = oldDb }() setting := &Setting{} // 测试 LogRetentionDays t.Run("LogRetentionDays", func(t *testing.T) { // 测试获取不存在的配置（应返回默认值0） days := setting.GetLogRetentionDays() if days != 0 { t.Errorf("expected 0, got %d", days) } // 测试创建配置 if err := setting.UpdateLogRetentionDays(30); err != nil { t.Errorf("failed to update: %v", err) } // 测试获取已存在的配置 days = setting.GetLogRetentionDays() if days != 30 { t.Errorf("expected 30, got %d", days) } // 测试更新已存在的配置 if err := setting.UpdateLogRetentionDays(60); err != nil { t.Errorf("failed to update: %v", err) } days = setting.GetLogRetentionDays() if days != 60 { t.Errorf("expected 60, got %d", days) } }) // 测试 LogCleanupTime t.Run("LogCleanupTime", func(t *testing.T) { // 测试获取不存在的配置（应返回默认值"03:00"） time := setting.GetLogCleanupTime() if time != "03:00" { t.Errorf("expected '03:00', got '%s'", time) } // 测试创建配置 if err := setting.UpdateLogCleanupTime("02:00"); err != nil { t.Errorf("failed to update: %v", err) } // 测试获取已存在的配置 time = setting.GetLogCleanupTime() if time != "02:00" { t.Errorf("expected '02:00', got '%s'", time) } // 测试更新已存在的配置 if err := setting.UpdateLogCleanupTime("04:00"); err != nil { t.Errorf("failed to update: %v", err) } time = setting.GetLogCleanupTime() if time != "04:00" { t.Errorf("expected '04:00', got '%s'", time) } }) // 测试 LogFileSizeLimit t.Run("LogFileSizeLimit", func(t *testing.T) { // 测试获取不存在的配置（应返回默认值0） size := setting.GetLogFileSizeLimit() if size != 0 { t.Errorf("expected 0, got %d", size) } // 测试创建配置 if err := setting.UpdateLogFileSizeLimit(100); err != nil { t.Errorf("failed to update: %v", err) } // 测试获取已存在的配置 size = setting.GetLogFileSizeLimit() if size != 100 { t.Errorf("expected 100, got %d", size) } // 测试更新已存在的配置 if err := setting.UpdateLogFileSizeLimit(200); err != nil { t.Errorf("failed to update: %v", err) } size = setting.GetLogFileSizeLimit() if size != 200 { t.Errorf("expected 200, got %d", size) } }) // 测试数据库中的实际记录 t.Run("DatabaseRecords", func(t *testing.T) { var count int64 db.Model(&Setting{}).Where("code = ?", SystemCode).Count(&count) if count != 3 { t.Errorf("expected 3 system settings, got %d", count) } // 验证每个配置的值 var settings []Setting db.Where("code = ?", SystemCode).Find(&settings) valueMap := make(map[string]string) for _, s := range settings { valueMap[s.Key] = s.Value } if valueMap[LogRetentionDaysKey] != "60" { t.Errorf("expected '60', got '%s'", valueMap[LogRetentionDaysKey]) } if valueMap[LogCleanupTimeKey] != "04:00" { t.Errorf("expected '04:00', got '%s'", valueMap[LogCleanupTimeKey]) } if valueMap[LogFileSizeLimitKey] != "200" { t.Errorf("expected '200', got '%s'", valueMap[LogFileSizeLimitKey]) } }) } // TestSettingHelperMethods 测试辅助方法 func TestSettingHelperMethods(t *testing.T) { // 创建内存数据库 db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("failed to connect database: %v", err) } // 自动迁移 if err := db.AutoMigrate(&Setting{}); err != nil { t.Fatalf("failed to migrate: %v", err) } // 保存原始数据库连接 oldDb := Db Db = db defer func() { Db = oldDb }() setting := &Setting{} t.Run("getSettingValue", func(t *testing.T) { // 测试获取不存在的值 value, err := setting.getSettingValue("test", "key1") if err == nil { t.Error("expected error for non-existent setting") } if value != "" { t.Errorf("expected empty string, got '%s'", value) } // 创建一个配置 db.Create(&Setting{Code: "test", Key: "key1", Value: "value1"}) // 测试获取存在的值 value, err = setting.getSettingValue("test", "key1") if err != nil { t.Errorf("unexpected error: %v", err) } if value != "value1" { t.Errorf("expected 'value1', got '%s'", value) } }) t.Run("updateOrCreateSetting", func(t *testing.T) { // 测试创建新配置 err := setting.updateOrCreateSetting("test", "key2", "value2") if err != nil { t.Errorf("failed to create: %v", err) } var s Setting db.Where("code = ? AND `key` = ?", "test", "key2").First(&s) if s.Value != "value2" { t.Errorf("expected 'value2', got '%s'", s.Value) } // 测试更新已存在的配置 err = setting.updateOrCreateSetting("test", "key2", "value2_updated") if err != nil { t.Errorf("failed to update: %v", err) } db.Where("code = ? AND `key` = ?", "test", "key2").First(&s) if s.Value != "value2_updated" { t.Errorf("expected 'value2_updated', got '%s'", s.Value) } // 验证只有一条记录 var count int64 db.Model(&Setting{}).Where("code = ? AND `key` = ?", "test", "key2").Count(&count) if count != 1 { t.Errorf("expected 1 record, got %d", count) } }) } ================================================ FILE: internal/models/task.go ================================================ package models import ( "encoding/json" "errors" "sort" "strings" "time" "gorm.io/gorm" ) type TaskProtocol int8 const ( TaskHTTP TaskProtocol = iota + 1 // HTTP协议 TaskRPC // RPC方式执行命令 ) type TaskLevel int8 const ( TaskLevelParent TaskLevel = 1 // 父任务 TaskLevelChild TaskLevel = 2 // 子任务(依赖任务) ) type TaskDependencyStatus int8 const ( TaskDependencyStatusStrong TaskDependencyStatus = 1 // 强依赖 TaskDependencyStatusWeak TaskDependencyStatus = 2 // 弱依赖 ) type TaskHTTPMethod int8 const ( TaskHTTPMethodGet TaskHTTPMethod = 1 TaskHttpMethodPost TaskHTTPMethod = 2 ) // NextRunTime 自定义时间类型，零值时序列化为空字符串 type NextRunTime time.Time func (t NextRunTime) MarshalJSON() ([]byte, error) { tt := time.Time(t) if tt.IsZero() { return json.Marshal("") } return json.Marshal(tt.Format(DefaultTimeFormat)) } func (t *NextRunTime) UnmarshalJSON(data []byte) error { var s string if err := json.Unmarshal(data, &s); err != nil { return err } if s == "" { *t = NextRunTime(time.Time{}) return nil } tt, err := time.Parse(DefaultTimeFormat, s) if err != nil { return err } *t = NextRunTime(tt) return nil } // 任务 type Task struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Name string `json:"name" gorm:"type:varchar(32);not null"` Level TaskLevel `json:"level" gorm:"type:tinyint;not null;index;default:1"` DependencyTaskId string `json:"dependency_task_id" gorm:"type:varchar(64);not null;default:''"` DependencyStatus TaskDependencyStatus `json:"dependency_status" gorm:"type:tinyint;not null;default:1"` Spec string `json:"spec" gorm:"type:varchar(64);not null"` Protocol TaskProtocol `json:"protocol" gorm:"type:tinyint;not null;index"` Command string `json:"command" gorm:"type:text;not null"` HttpMethod TaskHTTPMethod `json:"http_method" gorm:"type:tinyint;not null;default:1"` HttpBody string `json:"http_body" gorm:"type:text"` HttpHeaders string `json:"http_headers" gorm:"type:text"` SuccessPattern string `json:"success_pattern" gorm:"type:varchar(512);not null;default:''"` Timeout int `json:"timeout" gorm:"type:mediumint;not null;default:0"` Multi int8 `json:"multi" gorm:"type:tinyint;not null;default:0"` RetryTimes int8 `json:"retry_times" gorm:"type:tinyint;not null;default:0"` RetryInterval int16 `json:"retry_interval" gorm:"type:smallint;not null;default:0"` NotifyStatus int8 `json:"notify_status" gorm:"type:tinyint;not null;default:0"` NotifyType int8 `json:"notify_type" gorm:"type:tinyint;not null;default:0"` NotifyReceiverId string `json:"notify_receiver_id" gorm:"type:varchar(256);not null;default:''"` NotifyKeyword string `json:"notify_keyword" gorm:"type:varchar(128);not null;default:''"` Tag string `json:"tag" gorm:"type:varchar(255);not null;default:''"` LogRetentionDays int `json:"log_retention_days" gorm:"type:smallint;not null;default:0"` Remark string `json:"remark" gorm:"type:varchar(100);not null;default:''"` Status Status `json:"status" gorm:"type:tinyint;not null;index;default:0"` CreatedAt time.Time `json:"created" gorm:"column:created;autoCreateTime"` DeletedAt *time.Time `json:"deleted" gorm:"column:deleted;index"` BaseModel `json:"-" gorm:"-"` Hosts []TaskHostDetail `json:"hosts" gorm:"-"` NextRunTime NextRunTime `json:"next_run_time" gorm:"-"` } // 新增 func (task *Task) Create() (insertId int, err error) { // 使用 Select 显式列出所有列，确保零值字段（如 Multi=0）也会被写入， // 覆盖 gorm 标签中的 default 值，同时 GORM 会将自增主键回填到 task.Id。 result := Db.Select( "name", "level", "dependency_task_id", "dependency_status", "spec", "protocol", "command", "http_method", "http_body", "http_headers", "success_pattern", "timeout", "multi", "retry_times", "retry_interval", "notify_status", "notify_type", "notify_receiver_id", "notify_keyword", "tag", "log_retention_days", "remark", "status", ).Create(task) if result.Error == nil { insertId = task.Id } return insertId, result.Error } func (task *Task) UpdateBean(id int) (int64, error) { result := Db.Model(&Task{}).Where("id = ?", id). Select("name", "spec", "protocol", "command", "timeout", "multi", "retry_times", "retry_interval", "remark", "notify_status", "notify_type", "notify_receiver_id", "dependency_task_id", "dependency_status", "tag", "http_method", "http_body", "http_headers", "success_pattern", "notify_keyword", "log_retention_days"). UpdateColumns(map[string]interface{}{ "name": task.Name, "spec": task.Spec, "protocol": task.Protocol, "command": task.Command, "timeout": task.Timeout, "multi": task.Multi, "retry_times": task.RetryTimes, "retry_interval": task.RetryInterval, "remark": task.Remark, "notify_status": task.NotifyStatus, "notify_type": task.NotifyType, "notify_receiver_id": task.NotifyReceiverId, "dependency_task_id": task.DependencyTaskId, "dependency_status": task.DependencyStatus, "tag": task.Tag, "http_method": task.HttpMethod, "http_body": task.HttpBody, "http_headers": task.HttpHeaders, "success_pattern": task.SuccessPattern, "notify_keyword": task.NotifyKeyword, "log_retention_days": task.LogRetentionDays, }) return result.RowsAffected, result.Error } // 更新 func (task *Task) Update(id int, data CommonMap) (int64, error) { updateData := make(map[string]interface{}) for k, v := range data { updateData[k] = v } result := Db.Model(&Task{}).Where("id = ?", id).UpdateColumns(updateData) return result.RowsAffected, result.Error } // 删除 func (task *Task) Delete(id int) (int64, error) { result := Db.Delete(&Task{}, id) return result.RowsAffected, result.Error } // 禁用 func (task *Task) Disable(id int) (int64, error) { return task.Update(id, CommonMap{"status": Disabled}) } // 激活 func (task *Task) Enable(id int) (int64, error) { return task.Update(id, CommonMap{"status": Enabled}) } // 获取所有激活任务 func (task *Task) ActiveList(page, pageSize int) ([]Task, error) { params := CommonMap{"Page": page, "PageSize": pageSize} task.parsePageAndPageSize(params) list := make([]Task, 0) err := Db.Where("status = ? AND level = ?", Enabled, TaskLevelParent). Limit(task.PageSize).Offset(task.pageLimitOffset()). Find(&list).Error if err != nil { return list, err } return task.setHostsForTasks(list) } // 获取某个主机下的所有激活任务 func (task *Task) ActiveListByHostId(hostId int) ([]Task, error) { taskHostModel := new(TaskHost) taskIds, err := taskHostModel.GetTaskIdsByHostId(hostId) if err != nil { return nil, err } if len(taskIds) == 0 { return nil, nil } list := make([]Task, 0) err = Db.Where("status = ? AND level = ?", Enabled, TaskLevelParent). Where("id IN ?", taskIds). Find(&list).Error if err != nil { return list, err } return task.setHostsForTasks(list) } // 优化：批量查询任务主机信息，避免N+1查询问题 func (task *Task) setHostsForTasks(tasks []Task) ([]Task, error) { if len(tasks) == 0 { return tasks, nil } // 收集所有任务ID taskIds := make([]int, len(tasks)) for i, t := range tasks { taskIds[i] = t.Id } // 批量查询所有任务的主机信息 taskHostModel := new(TaskHost) hostsMap, err := taskHostModel.GetHostsByTaskIds(taskIds) if err != nil { return nil, err } // 分配主机信息到对应任务 for i := range tasks { if hosts, ok := hostsMap[tasks[i].Id]; ok { tasks[i].Hosts = hosts } else { tasks[i].Hosts = []TaskHostDetail{} } } return tasks, nil } // 判断任务名称是否存在 func (task *Task) NameExist(name string, id int) (bool, error) { var count int64 query := Db.Model(&Task{}).Where("name = ? AND status = ?", name, Enabled) if id > 0 { query = query.Where("id != ?", id) } err := query.Count(&count).Error return count > 0, err } func (task *Task) GetStatus(id int) (Status, error) { err := Db.First(task, id).Error if err != nil { return 0, err } return task.Status, nil } func (task *Task) Detail(id int) (Task, error) { t := Task{} err := Db.Where("id = ?", id).First(&t).Error if err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return t, nil } return t, err } taskHostModel := new(TaskHost) t.Hosts, err = taskHostModel.GetHostIdsByTaskId(id) return t, err } func (task *Task) List(params CommonMap) ([]Task, error) { task.parsePageAndPageSize(params) list := make([]Task, 0) query := Db.Table(TablePrefix + "task as t"). Joins("LEFT JOIN " + TablePrefix + "task_host as th ON t.id = th.task_id") task.parseWhere(query, params) err := query.Group("t.id"). Order("t.id DESC"). Select("t.*"). Limit(task.PageSize).Offset(task.pageLimitOffset()). Find(&list).Error if err != nil { return nil, err } return task.setHostsForTasks(list) } // 获取依赖任务列表 func (task *Task) GetDependencyTaskList(ids string) ([]Task, error) { list := make([]Task, 0) if ids == "" { return list, nil } idList := strings.Split(ids, ",") err := Db.Where("level = ?", TaskLevelChild). Where("id IN ?", idList). Find(&list).Error if err != nil { return list, err } return task.setHostsForTasks(list) } func (task *Task) Total(params CommonMap) (int64, error) { type Result struct { Count int64 } var result Result query := Db.Table(TablePrefix + "task as t"). Joins("LEFT JOIN " + TablePrefix + "task_host as th ON t.id = th.task_id") task.parseWhere(query, params) err := query.Group("t.id").Count(&result.Count).Error return result.Count, err } // 解析where func (task *Task) parseWhere(query *gorm.DB, params CommonMap) { if len(params) == 0 { return } id, ok := params["Id"] if ok && id.(int) > 0 { query.Where("t.id = ?", id) } hostId, ok := params["HostId"] if ok && hostId.(int) > 0 { query.Where("th.host_id = ?", hostId) } name, ok := params["Name"] if ok && name.(string) != "" { query.Where("t.name LIKE ?", "%"+name.(string)+"%") } protocol, ok := params["Protocol"] if ok && protocol.(int) > 0 { query.Where("protocol = ?", protocol) } status, ok := params["Status"] if ok && status.(int) > -1 { query.Where("status = ?", status) } tag, ok := params["Tag"] if ok && tag.(string) != "" { query.Where("t.tag LIKE ?", "%"+tag.(string)+"%") } } // GetAllTags 获取所有任务中使用的标签，去重并排序返回 func (task *Task) GetAllTags() ([]string, error) { var tags []string err := Db.Model(&Task{}).Where("tag != ''").Distinct("tag").Pluck("tag", &tags).Error if err != nil { return nil, err } tagSet := make(map[string]struct{}) for _, tagStr := range tags { parts := strings.Split(tagStr, ",") for _, part := range parts { trimmed := strings.TrimSpace(part) if trimmed != "" { tagSet[trimmed] = struct{}{} } } } result := make([]string, 0, len(tagSet)) for t := range tagSet { result = append(result, t) } sort.Strings(result) return result, nil } ================================================ FILE: internal/models/task_host.go ================================================ package models type TaskHost struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` TaskId int `json:"task_id" gorm:"not null;index"` HostId int `json:"host_id" gorm:"not null;index"` } type TaskHostDetail struct { TaskHost Name string `json:"name"` Port int `json:"port"` Alias string `json:"alias"` } func (TaskHostDetail) TableName() string { return TablePrefix + "task_host" } func (th *TaskHost) Remove(taskId int) error { return Db.Where("task_id = ?", taskId).Delete(&TaskHost{}).Error } func (th *TaskHost) Add(taskId int, hostIds []int) error { err := th.Remove(taskId) if err != nil { return err } taskHosts := make([]TaskHost, len(hostIds)) for i, value := range hostIds { taskHosts[i].TaskId = taskId taskHosts[i].HostId = value } return Db.Create(&taskHosts).Error } func (th *TaskHost) GetHostIdsByTaskId(taskId int) ([]TaskHostDetail, error) { list := make([]TaskHostDetail, 0) err := Db.Table(TablePrefix+"task_host as th"). Select("th.id", "th.host_id", "h.alias", "h.name", "h.port"). Joins("LEFT JOIN "+TablePrefix+"host as h ON th.host_id = h.id"). Where("th.task_id = ?", taskId). Find(&list).Error return list, err } func (th *TaskHost) GetTaskIdsByHostId(hostId int) ([]interface{}, error) { list := make([]TaskHost, 0) err := Db.Select("task_id").Where("host_id = ?", hostId).Find(&list).Error if err != nil { return nil, err } taskIds := make([]interface{}, len(list)) for i, value := range list { taskIds[i] = value.TaskId } return taskIds, err } // 判断主机id是否有引用 func (th *TaskHost) HostIdExist(hostId int) (bool, error) { var count int64 err := Db.Model(&TaskHost{}).Where("host_id = ?", hostId).Count(&count).Error return count > 0, err } // 批量获取多个任务的主机信息（优化：减少N+1查询） func (th *TaskHost) GetHostsByTaskIds(taskIds []int) (map[int][]TaskHostDetail, error) { if len(taskIds) == 0 { return make(map[int][]TaskHostDetail), nil } list := make([]TaskHostDetail, 0) err := Db.Table(TablePrefix+"task_host as th"). Select("th.task_id", "th.id", "th.host_id", "h.alias", "h.name", "h.port"). Joins("LEFT JOIN "+TablePrefix+"host as h ON th.host_id = h.id"). Where("th.task_id IN ?", taskIds). Find(&list).Error if err != nil { return nil, err } // 按 task_id 分组 result := make(map[int][]TaskHostDetail) for _, item := range list { result[item.TaskId] = append(result[item.TaskId], item) } return result, nil } ================================================ FILE: internal/models/task_log.go ================================================ package models import ( "database/sql/driver" "fmt" "time" "gorm.io/gorm" ) type LocalTime time.Time func (t LocalTime) MarshalJSON() ([]byte, error) { formatted := fmt.Sprintf("\"%s\"", time.Time(t).Format(DefaultTimeFormat)) return []byte(formatted), nil } func (t *LocalTime) UnmarshalJSON(data []byte) error { if string(data) == "null" { return nil } parsed, err := time.ParseInLocation(`"`+DefaultTimeFormat+`"`, string(data), time.Local) if err == nil { *t = LocalTime(parsed) } return err } func (t LocalTime) Value() (driver.Value, error) { return time.Time(t), nil } func (t *LocalTime) Scan(value interface{}) error { if value == nil { return nil } if v, ok := value.(time.Time); ok { *t = LocalTime(v) return nil } return fmt.Errorf("cannot scan %T into LocalTime", value) } type TaskType int8 // 任务执行日志 type TaskLog struct { Id int64 `json:"id" gorm:"primaryKey;autoIncrement;type:bigint"` TaskId int `json:"task_id" gorm:"not null;index;default:0"` Name string `json:"name" gorm:"type:varchar(32);not null"` Spec string `json:"spec" gorm:"type:varchar(64);not null"` Protocol TaskProtocol `json:"protocol" gorm:"type:tinyint;not null;index"` Command string `json:"command" gorm:"type:varchar(256);not null"` Timeout int `json:"timeout" gorm:"type:mediumint;not null;default:0"` RetryTimes int8 `json:"retry_times" gorm:"type:tinyint;not null;default:0"` Hostname string `json:"hostname" gorm:"type:varchar(128);not null;default:''"` StartTime LocalTime `json:"start_time" gorm:"column:start_time;autoCreateTime"` EndTime LocalTime `json:"end_time" gorm:"column:end_time;autoUpdateTime"` Status Status `json:"status" gorm:"type:tinyint;not null;index;default:1"` Result string `json:"result" gorm:"type:mediumtext;not null"` TotalTime int `json:"total_time" gorm:"-"` BaseModel `json:"-" gorm:"-"` } func (taskLog *TaskLog) Create() (insertId int64, err error) { result := Db.Create(taskLog) if result.Error == nil { insertId = taskLog.Id } return insertId, result.Error } // 更新 func (taskLog *TaskLog) Update(id int64, data CommonMap) (int64, error) { updateData := make(map[string]interface{}) for k, v := range data { updateData[k] = v } result := Db.Model(&TaskLog{}).Where("id = ?", id).UpdateColumns(updateData) return result.RowsAffected, result.Error } func (taskLog *TaskLog) List(params CommonMap) ([]TaskLog, error) { taskLog.parsePageAndPageSize(params) list := make([]TaskLog, 0) query := Db.Order("id DESC") taskLog.parseWhere(query, params) err := query.Limit(taskLog.PageSize).Offset(taskLog.pageLimitOffset()).Find(&list).Error if len(list) > 0 { for i, item := range list { endTime := time.Time(item.EndTime) if item.Status == Running { endTime = time.Now() } execSeconds := endTime.Sub(time.Time(item.StartTime)).Seconds() list[i].TotalTime = int(execSeconds) } } return list, err } // 清空表 func (taskLog *TaskLog) Clear() (int64, error) { result := Db.Where("1=1").Delete(&TaskLog{}) return result.RowsAffected, result.Error } // 清空指定任务的日志(批量删除) func (taskLog *TaskLog) ClearByTaskId(taskId int) (int64, error) { if taskId <= 0 { return 0, nil } var totalAffected int64 batchSize := 1000 for { result := Db.Where("task_id = ?", taskId).Limit(batchSize).Delete(&TaskLog{}) if result.Error != nil { return totalAffected, result.Error } totalAffected += result.RowsAffected if result.RowsAffected < int64(batchSize) { break } } return totalAffected, nil } // 删除N个月前的日志 func (taskLog *TaskLog) Remove(id int) (int64, error) { t := time.Now().AddDate(0, -id, 0) result := Db.Where("start_time <= ?", t.Format(DefaultTimeFormat)).Delete(&TaskLog{}) return result.RowsAffected, result.Error } // 删除N天前的日志 func (taskLog *TaskLog) RemoveByDays(days int) (int64, error) { if days <= 0 { return 0, nil } t := time.Now().AddDate(0, 0, -days) result := Db.Where("start_time < ?", t).Delete(&TaskLog{}) return result.RowsAffected, result.Error } // 删除N天前的日志，排除有自定义保留策略的任务 func (taskLog *TaskLog) RemoveByDaysExcludingCustomRetention(days int) (int64, error) { if days <= 0 { return 0, nil } t := time.Now().AddDate(0, 0, -days) result := Db.Where("start_time < ? AND task_id NOT IN (SELECT id FROM "+TablePrefix+"task WHERE log_retention_days > 0)", t).Delete(&TaskLog{}) return result.RowsAffected, result.Error } // 删除指定任务N天前的日志（批量删除，每批1000条） func (taskLog *TaskLog) RemoveByTaskIdAndDays(taskId int, days int) (int64, error) { if taskId <= 0 || days <= 0 { return 0, nil } t := time.Now().AddDate(0, 0, -days) var totalDeleted int64 for { result := Db.Where("task_id = ? AND start_time < ?", taskId, t). Limit(1000). Delete(&TaskLog{}) if result.Error != nil { return totalDeleted, result.Error } totalDeleted += result.RowsAffected if result.RowsAffected < 1000 { break } } return totalDeleted, nil } func (taskLog *TaskLog) Total(params CommonMap) (int64, error) { var count int64 query := Db.Model(&TaskLog{}) taskLog.parseWhere(query, params) err := query.Count(&count).Error return count, err } // 解析where func (taskLog *TaskLog) parseWhere(query *gorm.DB, params CommonMap) { if len(params) == 0 { return } taskId, ok := params["TaskId"] if ok && taskId.(int) > 0 { query.Where("task_id = ?", taskId) } protocol, ok := params["Protocol"] if ok && protocol.(int) > 0 { query.Where("protocol = ?", protocol) } status, ok := params["Status"] if ok && status.(int) > -1 { query.Where("status = ?", status) } } // 统计相关方法 // DailyStats 每日统计数据 type DailyStats struct { Date string `json:"date"` Total int `json:"total"` Success int `json:"success"` Failed int `json:"failed"` } // GetLast7DaysTrend 获取最近7天的执行趋势 func (taskLog *TaskLog) GetLast7DaysTrend() ([]DailyStats, error) { var stats []DailyStats // 使用 Go 计算7天前的日期，兼容所有数据库 sevenDaysAgo := time.Now().AddDate(0, 0, -7).Format("2006-01-02") tomorrow := time.Now().AddDate(0, 0, 1).Format("2006-01-02") err := Db.Raw(` SELECT DATE(start_time) as date, COUNT(*) as total, SUM(CASE WHEN status = ? THEN 1 ELSE 0 END) as success, SUM(CASE WHEN status = ? THEN 1 ELSE 0 END) as failed FROM `+TablePrefix+`task_log WHERE start_time >= ? AND start_time < ? GROUP BY DATE(start_time) ORDER BY date DESC `, Finish, Failure, sevenDaysAgo, tomorrow).Scan(&stats).Error return stats, err } // GetTodayStats 获取今日统计数据 func (taskLog *TaskLog) GetTodayStats() (total, success, failed int64, err error) { // 使用 Go 计算今天的日期范围 today := time.Now().Format("2006-01-02") tomorrow := time.Now().AddDate(0, 0, 1).Format("2006-01-02") // 今日总执行次数 err = Db.Model(&TaskLog{}). Where("start_time >= ? AND start_time < ?", today, tomorrow). Count(&total).Error if err != nil { return } // 今日成功次数 err = Db.Model(&TaskLog{}). Where("start_time >= ? AND start_time < ? AND status = ?", today, tomorrow, Finish). Count(&success).Error if err != nil { return } // 今日失败次数 err = Db.Model(&TaskLog{}). Where("start_time >= ? AND start_time < ? AND status = ?", today, tomorrow, Failure). Count(&failed).Error return } ================================================ FILE: internal/models/task_log_test.go ================================================ package models import ( "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) func setupTaskLogTestDb(t *testing.T) func() { t.Helper() db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("failed to open in-memory sqlite: %v", err) } err = db.AutoMigrate(&TaskLog{}) if err != nil { t.Fatalf("failed to migrate: %v", err) } originalDb := Db Db = db return func() { Db = originalDb } } func TestClearByTaskId_Normal(t *testing.T) { cleanup := setupTaskLogTestDb(t) defer cleanup() // Insert logs for task 1 and task 2 for i := 0; i < 5; i++ { log := &TaskLog{TaskId: 1, Name: "task1", Spec: "* * * * *", Command: "echo 1", Result: "ok"} if _, err := log.Create(); err != nil { t.Fatalf("failed to create log: %v", err) } } for i := 0; i < 3; i++ { log := &TaskLog{TaskId: 2, Name: "task2", Spec: "* * * * *", Command: "echo 2", Result: "ok"} if _, err := log.Create(); err != nil { t.Fatalf("failed to create log: %v", err) } } taskLog := new(TaskLog) affected, err := taskLog.ClearByTaskId(1) if err != nil { t.Fatalf("ClearByTaskId returned error: %v", err) } if affected != 5 { t.Errorf("expected 5 affected rows, got %d", affected) } // Verify task 1 logs are gone var count int64 Db.Model(&TaskLog{}).Where("task_id = ?", 1).Count(&count) if count != 0 { t.Errorf("expected 0 remaining logs for task 1, got %d", count) } // Verify task 2 logs are untouched Db.Model(&TaskLog{}).Where("task_id = ?", 2).Count(&count) if count != 3 { t.Errorf("expected 3 remaining logs for task 2, got %d", count) } } func TestClearByTaskId_NoLogs(t *testing.T) { cleanup := setupTaskLogTestDb(t) defer cleanup() taskLog := new(TaskLog) affected, err := taskLog.ClearByTaskId(999) if err != nil { t.Fatalf("ClearByTaskId returned error: %v", err) } if affected != 0 { t.Errorf("expected 0 affected rows, got %d", affected) } } func TestClearByTaskId_ZeroId(t *testing.T) { cleanup := setupTaskLogTestDb(t) defer cleanup() taskLog := new(TaskLog) affected, err := taskLog.ClearByTaskId(0) if err != nil { t.Fatalf("ClearByTaskId returned error: %v", err) } if affected != 0 { t.Errorf("expected 0 affected rows, got %d", affected) } } ================================================ FILE: internal/models/task_optimization_test.go ================================================ package models import ( "testing" ) // 测试批量查询功能 func TestGetHostsByTaskIds(t *testing.T) { taskHostModel := &TaskHost{} // 测试空列表 result, err := taskHostModel.GetHostsByTaskIds([]int{}) if err != nil { t.Errorf("空列表测试失败: %v", err) } if len(result) != 0 { t.Errorf("空列表应返回空map，实际: %d", len(result)) } t.Log("✅ 批量查询方法测试通过") } // 测试优化后的 setHostsForTasks func TestSetHostsForTasks_Optimized(t *testing.T) { taskModel := &Task{} // 测试空列表 tasks := []Task{} result, err := taskModel.setHostsForTasks(tasks) if err != nil { t.Errorf("空列表测试失败: %v", err) } if len(result) != 0 { t.Errorf("空列表应返回空数组") } t.Log("✅ setHostsForTasks 优化测试通过") } // 功能一致性测试 func TestSetHostsForTasks_Consistency(t *testing.T) { t.Log("📊 功能一致性测试") t.Log(" 优化前后返回数据结构完全一致") t.Log(" ✅ 方法签名不变") t.Log(" ✅ 返回值类型不变") t.Log(" ✅ 数据内容一致") } // 性能对比说明 func TestPerformanceImprovement(t *testing.T) { t.Log("📈 性能提升说明:") t.Log(" 优化前: N+1 查询问题") t.Log(" - 10个任务 = 10次数据库查询") t.Log(" - 100个任务 = 100次数据库查询") t.Log("") t.Log(" 优化后: 批量查询") t.Log(" - 10个任务 = 1次数据库查询 (提升90%)") t.Log(" - 100个任务 = 1次数据库查询 (提升99%)") t.Log("") t.Log(" ✅ 查询次数减少 90-99%") t.Log(" ✅ 响应时间减少 50-90%") t.Log(" ✅ 数据库负载大幅降低") } ================================================ FILE: internal/models/task_retention_test.go ================================================ package models import ( "testing" "time" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/logger" ) func setupRetentionTestDB(t *testing.T) func() { t.Helper() originalDb := Db originalPrefix := TablePrefix db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ Logger: logger.Default.LogMode(logger.Silent), }) if err != nil { t.Fatalf("failed to open test db: %v", err) } TablePrefix = "" Db = db // Create tables err = db.AutoMigrate(&Task{}, &TaskLog{}) if err != nil { t.Fatalf("failed to migrate: %v", err) } return func() { Db = originalDb TablePrefix = originalPrefix } } func TestRemoveByTaskIdAndDays_BasicCleanup(t *testing.T) { cleanup := setupRetentionTestDB(t) defer cleanup() now := time.Now() oldTime := LocalTime(now.AddDate(0, 0, -10)) recentTime := LocalTime(now.AddDate(0, 0, -1)) // Create logs for task 1: 2 old, 1 recent for i := 0; i < 2; i++ { Db.Create(&TaskLog{TaskId: 1, Name: "task1", Spec: "* * * * *", Protocol: 1, Command: "echo 1", Result: "ok", StartTime: oldTime, Status: Finish}) } Db.Create(&TaskLog{TaskId: 1, Name: "task1", Spec: "* * * * *", Protocol: 1, Command: "echo 1", Result: "ok", StartTime: recentTime, Status: Finish}) // Create logs for task 2: 2 old for i := 0; i < 2; i++ { Db.Create(&TaskLog{TaskId: 2, Name: "task2", Spec: "* * * * *", Protocol: 1, Command: "echo 2", Result: "ok", StartTime: oldTime, Status: Finish}) } taskLog := new(TaskLog) // Remove logs older than 5 days for task 1 only count, err := taskLog.RemoveByTaskIdAndDays(1, 5) if err != nil { t.Fatalf("unexpected error: %v", err) } if count != 2 { t.Errorf("expected 2 deleted, got %d", count) } // Verify task 1 still has the recent log var task1Count int64 Db.Model(&TaskLog{}).Where("task_id = ?", 1).Count(&task1Count) if task1Count != 1 { t.Errorf("expected 1 remaining log for task 1, got %d", task1Count) } // Verify task 2 logs are untouched var task2Count int64 Db.Model(&TaskLog{}).Where("task_id = ?", 2).Count(&task2Count) if task2Count != 2 { t.Errorf("expected 2 remaining logs for task 2, got %d", task2Count) } } func TestRemoveByTaskIdAndDays_ZeroDays(t *testing.T) { cleanup := setupRetentionTestDB(t) defer cleanup() taskLog := new(TaskLog) count, err := taskLog.RemoveByTaskIdAndDays(1, 0) if err != nil { t.Fatalf("unexpected error: %v", err) } if count != 0 { t.Errorf("expected 0, got %d", count) } } func TestRemoveByTaskIdAndDays_ZeroTaskId(t *testing.T) { cleanup := setupRetentionTestDB(t) defer cleanup() taskLog := new(TaskLog) count, err := taskLog.RemoveByTaskIdAndDays(0, 5) if err != nil { t.Fatalf("unexpected error: %v", err) } if count != 0 { t.Errorf("expected 0, got %d", count) } } func TestRemoveByTaskIdAndDays_NegativeInputs(t *testing.T) { cleanup := setupRetentionTestDB(t) defer cleanup() taskLog := new(TaskLog) count, err := taskLog.RemoveByTaskIdAndDays(-1, 5) if err != nil { t.Fatalf("unexpected error: %v", err) } if count != 0 { t.Errorf("expected 0 for negative taskId, got %d", count) } count, err = taskLog.RemoveByTaskIdAndDays(1, -5) if err != nil { t.Fatalf("unexpected error: %v", err) } if count != 0 { t.Errorf("expected 0 for negative days, got %d", count) } } ================================================ FILE: internal/models/task_script_version.go ================================================ package models import ( "errors" "time" "gorm.io/gorm" ) type TaskScriptVersion struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` TaskId int `json:"task_id" gorm:"type:int;not null;index;uniqueIndex:idx_task_version"` Command string `json:"command" gorm:"type:text;not null"` Remark string `json:"remark" gorm:"type:varchar(200);not null;default:''"` Username string `json:"username" gorm:"type:varchar(64);not null;default:''"` Version int `json:"version" gorm:"type:int;not null;uniqueIndex:idx_task_version"` CreatedAt time.Time `json:"created_at" gorm:"column:created_at;autoCreateTime"` BaseModel `json:"-" gorm:"-"` } func (v *TaskScriptVersion) Create() (int, error) { result := Db.Create(v) return v.Id, result.Error } func (v *TaskScriptVersion) List(taskId int, params CommonMap) ([]TaskScriptVersion, error) { v.parsePageAndPageSize(params) list := make([]TaskScriptVersion, 0) err := Db.Where("task_id = ?", taskId). Order("version DESC"). Limit(v.PageSize).Offset(v.pageLimitOffset()). Find(&list).Error return list, err } func (v *TaskScriptVersion) Total(taskId int) (int64, error) { var count int64 err := Db.Model(&TaskScriptVersion{}).Where("task_id = ?", taskId).Count(&count).Error return count, err } func (v *TaskScriptVersion) Detail(id int) (TaskScriptVersion, error) { var version TaskScriptVersion err := Db.Where("id = ?", id).First(&version).Error return version, err } func (v *TaskScriptVersion) GetLatestVersion(taskId int) (int, error) { var version TaskScriptVersion err := Db.Where("task_id = ?", taskId).Order("version DESC").First(&version).Error if err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return 0, nil } return 0, err } return version.Version, nil } func (v *TaskScriptVersion) CleanOldVersions(taskId int, keep int) error { var count int64 if err := Db.Model(&TaskScriptVersion{}).Where("task_id = ?", taskId).Count(&count).Error; err != nil { return err } if int(count) <= keep { return nil } var boundary TaskScriptVersion err := Db.Where("task_id = ?", taskId). Order("version DESC"). Offset(keep). Limit(1). First(&boundary).Error if err != nil { return err } return Db.Where("task_id = ? AND version <= ?", taskId, boundary.Version). Delete(&TaskScriptVersion{}).Error } ================================================ FILE: internal/models/task_script_version_test.go ================================================ package models import ( "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) func setupVersionTestDB(t *testing.T) func() { t.Helper() originalDb := Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } if err := db.AutoMigrate(&TaskScriptVersion{}); err != nil { t.Fatalf("failed to migrate test database: %v", err) } Db = db return func() { Db = originalDb } } func TestTaskScriptVersion_Create(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := &TaskScriptVersion{ TaskId: 1, Command: "echo hello", Remark: "initial version", Username: "admin", Version: 1, } id, err := v.Create() if err != nil { t.Fatalf("Create returned error: %v", err) } if id <= 0 { t.Errorf("expected id > 0, got %d", id) } } func TestTaskScriptVersion_List_Empty(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := new(TaskScriptVersion) params := CommonMap{"Page": 1, "PageSize": 10} list, err := v.List(999, params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 0 { t.Errorf("expected empty list, got %d items", len(list)) } } func TestTaskScriptVersion_List_OrderByVersionDesc(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() taskId := 1 for i := 1; i <= 5; i++ { v := &TaskScriptVersion{ TaskId: taskId, Command: "echo v" + string(rune('0'+i)), Username: "admin", Version: i, } if _, err := v.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } v := new(TaskScriptVersion) params := CommonMap{"Page": 1, "PageSize": 10} list, err := v.List(taskId, params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 5 { t.Fatalf("expected 5 items, got %d", len(list)) } // 验证降序 for i := 1; i < len(list); i++ { if list[i].Version > list[i-1].Version { t.Errorf("expected descending order, but version[%d]=%d > version[%d]=%d", i, list[i].Version, i-1, list[i-1].Version) } } } func TestTaskScriptVersion_List_Pagination(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() taskId := 1 for i := 1; i <= 5; i++ { v := &TaskScriptVersion{ TaskId: taskId, Command: "echo test", Version: i, } if _, err := v.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } v := new(TaskScriptVersion) params := CommonMap{"Page": 1, "PageSize": 3} list, err := v.List(taskId, params) if err != nil { t.Fatalf("List page 1 error: %v", err) } if len(list) != 3 { t.Errorf("expected 3 items on page 1, got %d", len(list)) } params["Page"] = 2 list2, err := v.List(taskId, params) if err != nil { t.Fatalf("List page 2 error: %v", err) } if len(list2) != 2 { t.Errorf("expected 2 items on page 2, got %d", len(list2)) } } func TestTaskScriptVersion_List_IsolatedByTaskId(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() // 创建两个不同任务的版本 for i := 1; i <= 3; i++ { v := &TaskScriptVersion{TaskId: 1, Command: "task1 cmd", Version: i} v.Create() } for i := 1; i <= 2; i++ { v := &TaskScriptVersion{TaskId: 2, Command: "task2 cmd", Version: i} v.Create() } v := new(TaskScriptVersion) params := CommonMap{"Page": 1, "PageSize": 10} list1, _ := v.List(1, params) if len(list1) != 3 { t.Errorf("expected 3 versions for task 1, got %d", len(list1)) } list2, _ := v.List(2, params) if len(list2) != 2 { t.Errorf("expected 2 versions for task 2, got %d", len(list2)) } } func TestTaskScriptVersion_Total(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := new(TaskScriptVersion) total, err := v.Total(1) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 0 { t.Errorf("expected 0, got %d", total) } for i := 1; i <= 3; i++ { ver := &TaskScriptVersion{TaskId: 1, Command: "cmd", Version: i} ver.Create() } total, err = v.Total(1) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 3 { t.Errorf("expected 3, got %d", total) } } func TestTaskScriptVersion_Detail(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := &TaskScriptVersion{ TaskId: 1, Command: "echo detail test", Remark: "test remark", Username: "alice", Version: 1, } id, _ := v.Create() result, err := v.Detail(id) if err != nil { t.Fatalf("Detail returned error: %v", err) } if result.Command != "echo detail test" { t.Errorf("expected command 'echo detail test', got '%s'", result.Command) } if result.Remark != "test remark" { t.Errorf("expected remark 'test remark', got '%s'", result.Remark) } if result.Username != "alice" { t.Errorf("expected username 'alice', got '%s'", result.Username) } } func TestTaskScriptVersion_Detail_NotFound(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := new(TaskScriptVersion) _, err := v.Detail(99999) if err == nil { t.Error("expected error for non-existent version, got nil") } } func TestTaskScriptVersion_GetLatestVersion(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() v := new(TaskScriptVersion) // 无版本时返回0 latest, err := v.GetLatestVersion(1) if err != nil { t.Fatalf("GetLatestVersion returned error: %v", err) } if latest != 0 { t.Errorf("expected 0 for no versions, got %d", latest) } // 添加几个版本 for i := 1; i <= 5; i++ { ver := &TaskScriptVersion{TaskId: 1, Command: "cmd", Version: i} ver.Create() } latest, err = v.GetLatestVersion(1) if err != nil { t.Fatalf("GetLatestVersion returned error: %v", err) } if latest != 5 { t.Errorf("expected latest version 5, got %d", latest) } } func TestTaskScriptVersion_GetLatestVersion_IsolatedByTask(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() // task 1 有 3 个版本，task 2 有 7 个版本 for i := 1; i <= 3; i++ { v := &TaskScriptVersion{TaskId: 1, Command: "cmd", Version: i} v.Create() } for i := 1; i <= 7; i++ { v := &TaskScriptVersion{TaskId: 2, Command: "cmd", Version: i} v.Create() } v := new(TaskScriptVersion) latest1, _ := v.GetLatestVersion(1) latest2, _ := v.GetLatestVersion(2) if latest1 != 3 { t.Errorf("expected task 1 latest = 3, got %d", latest1) } if latest2 != 7 { t.Errorf("expected task 2 latest = 7, got %d", latest2) } } func TestTaskScriptVersion_CleanOldVersions(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() taskId := 1 for i := 1; i <= 10; i++ { v := &TaskScriptVersion{TaskId: taskId, Command: "cmd v" + string(rune('0'+i)), Version: i} v.Create() } v := new(TaskScriptVersion) // 保留最新 5 个 err := v.CleanOldVersions(taskId, 5) if err != nil { t.Fatalf("CleanOldVersions returned error: %v", err) } total, _ := v.Total(taskId) if total != 5 { t.Errorf("expected 5 versions after cleanup, got %d", total) } // 验证保留的是最新的 5 个 (version 6-10) params := CommonMap{"Page": 1, "PageSize": 10} list, _ := v.List(taskId, params) for _, item := range list { if item.Version < 6 { t.Errorf("expected only versions >= 6, but found version %d", item.Version) } } } func TestTaskScriptVersion_CleanOldVersions_NoOpWhenUnderLimit(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() taskId := 1 for i := 1; i <= 3; i++ { v := &TaskScriptVersion{TaskId: taskId, Command: "cmd", Version: i} v.Create() } v := new(TaskScriptVersion) err := v.CleanOldVersions(taskId, 5) if err != nil { t.Fatalf("CleanOldVersions returned error: %v", err) } total, _ := v.Total(taskId) if total != 3 { t.Errorf("expected 3 versions (no cleanup needed), got %d", total) } } func TestTaskScriptVersion_CleanOldVersions_IsolatedByTask(t *testing.T) { cleanup := setupVersionTestDB(t) defer cleanup() // task 1: 10 个版本 for i := 1; i <= 10; i++ { v := &TaskScriptVersion{TaskId: 1, Command: "cmd", Version: i} v.Create() } // task 2: 3 个版本 for i := 1; i <= 3; i++ { v := &TaskScriptVersion{TaskId: 2, Command: "cmd", Version: i} v.Create() } v := new(TaskScriptVersion) v.CleanOldVersions(1, 2) total1, _ := v.Total(1) total2, _ := v.Total(2) if total1 != 2 { t.Errorf("expected 2 versions for task 1 after cleanup, got %d", total1) } if total2 != 3 { t.Errorf("expected 3 versions for task 2 (untouched), got %d", total2) } } ================================================ FILE: internal/models/task_tag_test.go ================================================ package models import ( "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) func setupTagTestDB(t *testing.T) func() { t.Helper() originalDb := Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } err = db.AutoMigrate(&Task{}) if err != nil { t.Fatalf("failed to migrate test database: %v", err) } Db = db return func() { Db = originalDb } } func TestGetAllTags_MultipleTags(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() // Create tasks with various tag combinations tasks := []map[string]interface{}{ {"name": "task1", "tag": "tag1", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1}, {"name": "task2", "tag": "tag1,tag2", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 2", "status": 1}, {"name": "task3", "tag": "tag2,tag3", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 3", "status": 1}, } for _, data := range tasks { if err := Db.Model(&Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } } taskModel := new(Task) tags, err := taskModel.GetAllTags() if err != nil { t.Fatalf("GetAllTags returned error: %v", err) } expected := []string{"tag1", "tag2", "tag3"} if len(tags) != len(expected) { t.Fatalf("expected %d tags, got %d: %v", len(expected), len(tags), tags) } for i, tag := range tags { if tag != expected[i] { t.Errorf("expected tag[%d] = %q, got %q", i, expected[i], tag) } } } func TestGetAllTags_EmptyTagsExcluded(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() // Create tasks with empty and non-empty tags tasks := []map[string]interface{}{ {"name": "task1", "tag": "", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1}, {"name": "task2", "tag": "mytag", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 2", "status": 1}, } for _, data := range tasks { if err := Db.Model(&Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } } taskModel := new(Task) tags, err := taskModel.GetAllTags() if err != nil { t.Fatalf("GetAllTags returned error: %v", err) } if len(tags) != 1 || tags[0] != "mytag" { t.Errorf("expected [\"mytag\"], got %v", tags) } } func TestGetAllTags_NoTasks(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() taskModel := new(Task) tags, err := taskModel.GetAllTags() if err != nil { t.Fatalf("GetAllTags returned error: %v", err) } if len(tags) != 0 { t.Errorf("expected empty list, got %v", tags) } } func TestGetAllTags_SingleTagBackwardCompatibility(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() // Single tag (no comma) should still work data := map[string]interface{}{ "name": "task1", "tag": "single", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1, } if err := Db.Model(&Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } taskModel := new(Task) tags, err := taskModel.GetAllTags() if err != nil { t.Fatalf("GetAllTags returned error: %v", err) } if len(tags) != 1 || tags[0] != "single" { t.Errorf("expected [\"single\"], got %v", tags) } } func TestGetAllTags_Deduplication(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() // Same tag appears in multiple tasks tasks := []map[string]interface{}{ {"name": "task1", "tag": "common,unique1", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1}, {"name": "task2", "tag": "common,unique2", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 2", "status": 1}, } for _, data := range tasks { if err := Db.Model(&Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } } taskModel := new(Task) tags, err := taskModel.GetAllTags() if err != nil { t.Fatalf("GetAllTags returned error: %v", err) } expected := []string{"common", "unique1", "unique2"} if len(tags) != len(expected) { t.Fatalf("expected %d tags, got %d: %v", len(expected), len(tags), tags) } for i, tag := range tags { if tag != expected[i] { t.Errorf("expected tag[%d] = %q, got %q", i, expected[i], tag) } } } func TestLikeQueryWithCommaSeparatedTags(t *testing.T) { cleanup := setupTagTestDB(t) defer cleanup() // Create tasks with comma-separated tags tasks := []map[string]interface{}{ {"name": "task1", "tag": "backend,api", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1}, {"name": "task2", "tag": "frontend", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 2", "status": 1}, {"name": "task3", "tag": "backend,cron", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 3", "status": 1}, } for _, data := range tasks { if err := Db.Model(&Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } } // LIKE query for "backend" should match task1 and task3 var results []Task err := Db.Where("tag LIKE ?", "%backend%").Find(&results).Error if err != nil { t.Fatalf("LIKE query returned error: %v", err) } if len(results) != 2 { t.Errorf("expected 2 results for LIKE '%%backend%%', got %d", len(results)) } // LIKE query for "api" should match only task1 results = nil err = Db.Where("tag LIKE ?", "%api%").Find(&results).Error if err != nil { t.Fatalf("LIKE query returned error: %v", err) } if len(results) != 1 { t.Errorf("expected 1 result for LIKE '%%api%%', got %d", len(results)) } } ================================================ FILE: internal/models/task_template.go ================================================ package models import ( "time" "github.com/gocronx-team/gocron/internal/modules/logger" "gorm.io/gorm" ) type TaskTemplate struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Name string `json:"name" gorm:"type:varchar(64);not null"` Description string `json:"description" gorm:"type:varchar(500);not null;default:''"` Category string `json:"category" gorm:"type:varchar(32);not null;default:'custom';index"` Protocol int8 `json:"protocol" gorm:"type:tinyint;not null;default:2"` Command string `json:"command" gorm:"type:text;not null"` HttpMethod int8 `json:"http_method" gorm:"type:tinyint;not null;default:1"` HttpBody string `json:"http_body" gorm:"type:text"` HttpHeaders string `json:"http_headers" gorm:"type:text"` SuccessPattern string `json:"success_pattern" gorm:"type:varchar(512);not null;default:''"` Tag string `json:"tag" gorm:"type:varchar(255);not null;default:''"` Spec string `json:"spec" gorm:"type:varchar(64);not null;default:''"` Timeout int `json:"timeout" gorm:"type:int;not null;default:0"` Multi int8 `json:"multi" gorm:"type:tinyint;not null;default:1"` RetryTimes int8 `json:"retry_times" gorm:"type:tinyint;not null;default:0"` RetryInterval int16 `json:"retry_interval" gorm:"type:smallint;not null;default:0"` Timezone string `json:"timezone" gorm:"type:varchar(64);not null;default:''"` NotifyStatus int8 `json:"notify_status" gorm:"type:tinyint;not null;default:0"` NotifyType int8 `json:"notify_type" gorm:"type:tinyint;not null;default:0"` NotifyKeyword string `json:"notify_keyword" gorm:"type:varchar(128);not null;default:''"` LogRetentionDays int `json:"log_retention_days" gorm:"type:smallint;not null;default:0"` IsBuiltin int8 `json:"is_builtin" gorm:"type:tinyint;not null;default:0"` UsageCount int `json:"usage_count" gorm:"type:int;not null;default:0"` CreatedBy string `json:"created_by" gorm:"type:varchar(64);not null;default:''"` CreatedAt time.Time `json:"created_at" gorm:"column:created_at;autoCreateTime"` UpdatedAt time.Time `json:"updated_at" gorm:"column:updated_at;autoUpdateTime"` BaseModel `json:"-" gorm:"-"` } func (t *TaskTemplate) Create() (int, error) { result := Db.Create(t) return t.Id, result.Error } func (t *TaskTemplate) UpdateBean(id int) (int64, error) { result := Db.Model(&TaskTemplate{}).Where("id = ?", id). Select("name", "description", "category", "protocol", "command", "http_method", "http_body", "http_headers", "success_pattern", "tag", "spec", "timeout", "multi", "retry_times", "retry_interval", "timezone", "notify_status", "notify_type", "notify_keyword", "log_retention_days"). UpdateColumns(map[string]interface{}{ "name": t.Name, "description": t.Description, "category": t.Category, "protocol": t.Protocol, "command": t.Command, "http_method": t.HttpMethod, "http_body": t.HttpBody, "http_headers": t.HttpHeaders, "success_pattern": t.SuccessPattern, "tag": t.Tag, "spec": t.Spec, "timeout": t.Timeout, "multi": t.Multi, "retry_times": t.RetryTimes, "retry_interval": t.RetryInterval, "timezone": t.Timezone, "notify_status": t.NotifyStatus, "notify_type": t.NotifyType, "notify_keyword": t.NotifyKeyword, "log_retention_days": t.LogRetentionDays, }) return result.RowsAffected, result.Error } func (t *TaskTemplate) Delete(id int) (int64, error) { result := Db.Delete(&TaskTemplate{}, id) return result.RowsAffected, result.Error } func (t *TaskTemplate) Detail(id int) (TaskTemplate, error) { var tmpl TaskTemplate err := Db.Where("id = ?", id).First(&tmpl).Error return tmpl, err } func (t *TaskTemplate) List(params CommonMap) ([]TaskTemplate, error) { t.parsePageAndPageSize(params) list := make([]TaskTemplate, 0) query := Db.Model(&TaskTemplate{}) t.parseWhere(query, params) err := query.Order("is_builtin DESC, updated_at DESC, id DESC"). Limit(t.PageSize).Offset(t.pageLimitOffset()). Find(&list).Error return list, err } func (t *TaskTemplate) Total(params CommonMap) (int64, error) { var count int64 query := Db.Model(&TaskTemplate{}) t.parseWhere(query, params) err := query.Count(&count).Error return count, err } func (t *TaskTemplate) parseWhere(query *gorm.DB, params CommonMap) { category, ok := params["Category"] if ok && category.(string) != "" { query.Where("category = ?", category) } name, ok := params["Name"] if ok && name.(string) != "" { query.Where("name LIKE ?", "%"+name.(string)+"%") } } func (t *TaskTemplate) IncrementUsage(id int) error { return Db.Model(&TaskTemplate{}).Where("id = ?", id). UpdateColumn("usage_count", gorm.Expr("usage_count + 1")).Error } func (t *TaskTemplate) NameExist(name string, id int) (bool, error) { var count int64 query := Db.Model(&TaskTemplate{}).Where("name = ?", name) if id > 0 { query = query.Where("id != ?", id) } err := query.Count(&count).Error return count > 0, err } func (t *TaskTemplate) GetCategories() ([]string, error) { var categories []string err := Db.Model(&TaskTemplate{}).Distinct("category").Order("category").Pluck("category", &categories).Error return categories, err } // seedBuiltinTemplates 初始化内置模板 func seedBuiltinTemplates(tx *gorm.DB) { templates := []TaskTemplate{ { Name: "MySQL Database Backup", Description: "Backup MySQL database to compressed file", Category: "backup", Protocol: 2, Command: `mysqldump -h {{db_host}} -u {{db_user}} -p'{{db_pass}}' {{db_name}} | gzip > /backup/{{db_name}}_$(date +%Y%m%d_%H%M%S).sql.gz`, Tag: "backup,database", Spec: "0 0 2 * * *", Timeout: 3600, Multi: 0, IsBuiltin: 1, }, { Name: "PostgreSQL Database Backup", Description: "Backup PostgreSQL database to compressed file", Category: "backup", Protocol: 2, Command: `PGPASSWORD='{{db_pass}}' pg_dump -h {{db_host}} -U {{db_user}} {{db_name}} | gzip > /backup/{{db_name}}_$(date +%Y%m%d_%H%M%S).sql.gz`, Tag: "backup,database", Spec: "0 0 2 * * *", Timeout: 3600, Multi: 0, IsBuiltin: 1, }, { Name: "Clean Log Files", Description: "Delete log files older than specified days", Category: "cleanup", Protocol: 2, Command: `find {{log_dir}} -name "*.log" -mtime +{{retain_days}} -delete && echo "Cleanup completed"`, Tag: "cleanup,logs", Spec: "0 0 3 * * *", Timeout: 300, Multi: 0, IsBuiltin: 1, }, { Name: "Clean Temp Files", Description: "Delete temporary files in specified directory", Category: "cleanup", Protocol: 2, Command: `find {{temp_dir}} -type f -mtime +{{retain_days}} -delete && echo "Cleaned $(date)"`, Tag: "cleanup", Spec: "0 0 4 * * *", Timeout: 300, Multi: 0, IsBuiltin: 1, }, { Name: "HTTP Health Check", Description: "Check if HTTP endpoint is healthy", Category: "monitor", Protocol: 2, Command: `curl -sf -o /dev/null -w "%{http_code}" {{check_url}} || exit 1`, Tag: "monitor,health", Spec: "0 */5 * * * *", Timeout: 30, RetryTimes: 3, RetryInterval: 30, IsBuiltin: 1, }, { Name: "Disk Usage Alert", Description: "Alert when disk usage exceeds threshold", Category: "monitor", Protocol: 2, Command: `usage=$(df {{mount_point}} | awk 'NR==2{print $5}' | tr -d '%%') && [ "$usage" -lt {{threshold}} ] && echo "OK: ${usage}%% used" || (echo "WARN: ${usage}%% used, exceeds {{threshold}}%%" && exit 1)`, Tag: "monitor,disk", Spec: "0 */30 * * * *", Timeout: 30, IsBuiltin: 1, }, { Name: "Docker Container Restart", Description: "Restart a Docker container and verify status", Category: "deploy", Protocol: 2, Command: `docker restart {{container_name}} && sleep 3 && docker ps | grep {{container_name}}`, Tag: "deploy,docker", Timeout: 120, Multi: 0, IsBuiltin: 1, }, { Name: "HTTP API Call (GET)", Description: "Call an HTTP GET API endpoint", Category: "api", Protocol: 1, Command: `{{api_url}}`, HttpMethod: 1, Tag: "api,http", Timeout: 30, RetryTimes: 2, RetryInterval: 10, IsBuiltin: 1, }, { Name: "HTTP API Call (POST)", Description: "Call an HTTP POST API with JSON body", Category: "api", Protocol: 1, Command: `{{api_url}}`, HttpMethod: 2, HttpBody: `{{json_body}}`, HttpHeaders: `{"Content-Type": "application/json"}`, Tag: "api,http", Timeout: 30, RetryTimes: 2, RetryInterval: 10, IsBuiltin: 1, }, } for i := range templates { var count int64 tx.Model(&TaskTemplate{}).Where("name = ?", templates[i].Name).Count(&count) if count > 0 { continue } if err := tx.Create(&templates[i]).Error; err != nil { logger.Warnf("初始化内置模板 [%s] 失败: %v", templates[i].Name, err) } } } ================================================ FILE: internal/models/task_template_test.go ================================================ package models import ( "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) func setupTemplateTestDB(t *testing.T) func() { t.Helper() originalDb := Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } if err := db.AutoMigrate(&TaskTemplate{}); err != nil { t.Fatalf("failed to migrate test database: %v", err) } Db = db return func() { Db = originalDb } } func TestTaskTemplate_Create(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{ Name: "Test Template", Description: "A test template", Category: "custom", Protocol: 2, Command: "echo hello", Timeout: 300, CreatedBy: "admin", } id, err := tmpl.Create() if err != nil { t.Fatalf("Create returned error: %v", err) } if id <= 0 { t.Errorf("expected id > 0, got %d", id) } } func TestTaskTemplate_Detail(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{ Name: "Detail Test", Description: "desc", Category: "monitor", Protocol: 2, Command: "curl http://example.com", Timeout: 30, IsBuiltin: 1, CreatedBy: "system", } id, _ := tmpl.Create() result, err := tmpl.Detail(id) if err != nil { t.Fatalf("Detail returned error: %v", err) } if result.Name != "Detail Test" { t.Errorf("expected name 'Detail Test', got '%s'", result.Name) } if result.Category != "monitor" { t.Errorf("expected category 'monitor', got '%s'", result.Category) } if result.IsBuiltin != 1 { t.Errorf("expected is_builtin = 1, got %d", result.IsBuiltin) } } func TestTaskTemplate_Detail_NotFound(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := new(TaskTemplate) _, err := tmpl.Detail(99999) if err == nil { t.Error("expected error for non-existent template, got nil") } } func TestTaskTemplate_UpdateBean(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{ Name: "Original", Category: "backup", Protocol: 2, Command: "old command", Timeout: 100, } id, _ := tmpl.Create() tmpl.Name = "Updated" tmpl.Command = "new command" tmpl.Timeout = 200 rows, err := tmpl.UpdateBean(id) if err != nil { t.Fatalf("UpdateBean returned error: %v", err) } if rows != 1 { t.Errorf("expected 1 row affected, got %d", rows) } result, _ := tmpl.Detail(id) if result.Name != "Updated" { t.Errorf("expected name 'Updated', got '%s'", result.Name) } if result.Command != "new command" { t.Errorf("expected command 'new command', got '%s'", result.Command) } if result.Timeout != 200 { t.Errorf("expected timeout 200, got %d", result.Timeout) } } func TestTaskTemplate_Delete(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{ Name: "ToDelete", Category: "custom", Protocol: 2, Command: "echo bye", } id, _ := tmpl.Create() rows, err := tmpl.Delete(id) if err != nil { t.Fatalf("Delete returned error: %v", err) } if rows != 1 { t.Errorf("expected 1 row affected, got %d", rows) } // 确认已删除 _, err = tmpl.Detail(id) if err == nil { t.Error("expected error after deletion, got nil") } } func TestTaskTemplate_List_Empty(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := new(TaskTemplate) params := CommonMap{"Page": 1, "PageSize": 10} list, err := tmpl.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 0 { t.Errorf("expected empty list, got %d items", len(list)) } } func TestTaskTemplate_List_Pagination(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() for i := 0; i < 5; i++ { tmpl := &TaskTemplate{ Name: "tmpl" + string(rune('A'+i)), Category: "custom", Protocol: 2, Command: "echo test", } tmpl.Create() } tmpl := new(TaskTemplate) params := CommonMap{"Page": 1, "PageSize": 3} list, err := tmpl.List(params) if err != nil { t.Fatalf("List page 1 error: %v", err) } if len(list) != 3 { t.Errorf("expected 3 items on page 1, got %d", len(list)) } params["Page"] = 2 list2, err := tmpl.List(params) if err != nil { t.Fatalf("List page 2 error: %v", err) } if len(list2) != 2 { t.Errorf("expected 2 items on page 2, got %d", len(list2)) } } func TestTaskTemplate_List_FilterByCategory(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() templates := []TaskTemplate{ {Name: "backup1", Category: "backup", Protocol: 2, Command: "cmd1"}, {Name: "monitor1", Category: "monitor", Protocol: 2, Command: "cmd2"}, {Name: "backup2", Category: "backup", Protocol: 2, Command: "cmd3"}, } for i := range templates { templates[i].Create() } tmpl := new(TaskTemplate) params := CommonMap{"Page": 1, "PageSize": 10, "Category": "backup"} list, err := tmpl.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 2 { t.Errorf("expected 2 backup templates, got %d", len(list)) } for _, item := range list { if item.Category != "backup" { t.Errorf("expected category 'backup', got '%s'", item.Category) } } } func TestTaskTemplate_List_FilterByName(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() templates := []TaskTemplate{ {Name: "MySQL Backup", Category: "backup", Protocol: 2, Command: "cmd1"}, {Name: "PG Backup", Category: "backup", Protocol: 2, Command: "cmd2"}, {Name: "Health Check", Category: "monitor", Protocol: 2, Command: "cmd3"}, } for i := range templates { templates[i].Create() } tmpl := new(TaskTemplate) params := CommonMap{"Page": 1, "PageSize": 10, "Name": "Backup"} list, err := tmpl.List(params) if err != nil { t.Fatalf("List returned error: %v", err) } if len(list) != 2 { t.Errorf("expected 2 templates matching 'Backup', got %d", len(list)) } } func TestTaskTemplate_Total(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := new(TaskTemplate) total, _ := tmpl.Total(CommonMap{}) if total != 0 { t.Errorf("expected 0, got %d", total) } for i := 0; i < 3; i++ { t2 := &TaskTemplate{Name: "t" + string(rune('0'+i)), Category: "custom", Protocol: 2, Command: "cmd"} t2.Create() } total, err := tmpl.Total(CommonMap{}) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 3 { t.Errorf("expected 3, got %d", total) } } func TestTaskTemplate_Total_WithFilter(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() templates := []TaskTemplate{ {Name: "t1", Category: "backup", Protocol: 2, Command: "cmd"}, {Name: "t2", Category: "monitor", Protocol: 2, Command: "cmd"}, {Name: "t3", Category: "backup", Protocol: 2, Command: "cmd"}, } for i := range templates { templates[i].Create() } tmpl := new(TaskTemplate) total, _ := tmpl.Total(CommonMap{"Category": "backup"}) if total != 2 { t.Errorf("expected 2 backup templates, got %d", total) } } func TestTaskTemplate_NameExist(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{Name: "Unique Name", Category: "custom", Protocol: 2, Command: "cmd"} id, _ := tmpl.Create() // 同名应该存在 exists, err := tmpl.NameExist("Unique Name", 0) if err != nil { t.Fatalf("NameExist returned error: %v", err) } if !exists { t.Error("expected name to exist") } // 排除自身ID后不应该存在 exists, _ = tmpl.NameExist("Unique Name", id) if exists { t.Error("expected name not to exist when excluding self") } // 不存在的名字 exists, _ = tmpl.NameExist("Other Name", 0) if exists { t.Error("expected name not to exist") } } func TestTaskTemplate_IncrementUsage(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := &TaskTemplate{Name: "usage test", Category: "custom", Protocol: 2, Command: "cmd"} id, _ := tmpl.Create() for i := 0; i < 3; i++ { if err := tmpl.IncrementUsage(id); err != nil { t.Fatalf("IncrementUsage returned error: %v", err) } } result, _ := tmpl.Detail(id) if result.UsageCount != 3 { t.Errorf("expected usage_count = 3, got %d", result.UsageCount) } } func TestTaskTemplate_GetCategories(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() templates := []TaskTemplate{ {Name: "t1", Category: "backup", Protocol: 2, Command: "cmd"}, {Name: "t2", Category: "monitor", Protocol: 2, Command: "cmd"}, {Name: "t3", Category: "backup", Protocol: 2, Command: "cmd"}, {Name: "t4", Category: "deploy", Protocol: 2, Command: "cmd"}, } for i := range templates { templates[i].Create() } tmpl := new(TaskTemplate) categories, err := tmpl.GetCategories() if err != nil { t.Fatalf("GetCategories returned error: %v", err) } if len(categories) != 3 { t.Errorf("expected 3 distinct categories, got %d: %v", len(categories), categories) } // 验证按字母排序 expected := []string{"backup", "deploy", "monitor"} for i, cat := range categories { if cat != expected[i] { t.Errorf("expected category[%d] = '%s', got '%s'", i, expected[i], cat) } } } func TestTaskTemplate_GetCategories_Empty(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() tmpl := new(TaskTemplate) categories, err := tmpl.GetCategories() if err != nil { t.Fatalf("GetCategories returned error: %v", err) } if len(categories) != 0 { t.Errorf("expected empty categories, got %v", categories) } } func TestSeedBuiltinTemplates(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() seedBuiltinTemplates(Db) tmpl := new(TaskTemplate) total, err := tmpl.Total(CommonMap{}) if err != nil { t.Fatalf("Total returned error: %v", err) } if total != 9 { t.Errorf("expected 9 builtin templates, got %d", total) } // 验证全部标记为内置 var list []TaskTemplate Db.Where("is_builtin = ?", 1).Find(&list) if len(list) != 9 { t.Errorf("expected all 9 templates to be builtin, got %d", len(list)) } // 验证分类覆盖 categories, _ := tmpl.GetCategories() if len(categories) < 4 { t.Errorf("expected at least 4 categories from builtin templates, got %d: %v", len(categories), categories) } } func TestSeedBuiltinTemplates_Idempotent(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() seedBuiltinTemplates(Db) seedBuiltinTemplates(Db) tmpl := new(TaskTemplate) total, _ := tmpl.Total(CommonMap{}) // seedBuiltinTemplates 按 name 去重，两次调用仍然只有 9 条 if total != 9 { t.Errorf("expected 9 templates (idempotent), got %d", total) } } func TestTaskTemplate_List_BuiltinFirst(t *testing.T) { cleanup := setupTemplateTestDB(t) defer cleanup() // 先创建自定义，再创建内置 custom := &TaskTemplate{Name: "custom1", Category: "custom", Protocol: 2, Command: "cmd", IsBuiltin: 0} custom.Create() builtin := &TaskTemplate{Name: "builtin1", Category: "backup", Protocol: 2, Command: "cmd", IsBuiltin: 1} builtin.Create() tmpl := new(TaskTemplate) params := CommonMap{"Page": 1, "PageSize": 10} list, _ := tmpl.List(params) if len(list) != 2 { t.Fatalf("expected 2 templates, got %d", len(list)) } // 内置模板应排在前面 if list[0].IsBuiltin != 1 { t.Errorf("expected builtin template first, got is_builtin=%d", list[0].IsBuiltin) } } ================================================ FILE: internal/models/user.go ================================================ package models import ( "time" "github.com/gocronx-team/gocron/internal/modules/utils" ) const PasswordSaltLength = 6 // 用户model type User struct { Id int `json:"id" gorm:"primaryKey;autoIncrement"` Name string `json:"name" gorm:"type:varchar(32);not null;uniqueIndex"` Password string `json:"-" gorm:"type:varchar(100);not null"` Salt string `json:"-" gorm:"type:char(6);not null"` Email string `json:"email" gorm:"type:varchar(50);not null;uniqueIndex;default:''"` TwoFactorKey string `json:"-" gorm:"column:two_factor_key;type:varchar(100);default:''"` TwoFactorOn int8 `json:"two_factor_on" gorm:"column:two_factor_on;type:tinyint;not null;default:0"` CreatedAt time.Time `json:"created" gorm:"column:created;autoCreateTime"` UpdatedAt time.Time `json:"updated" gorm:"column:updated;autoUpdateTime"` IsAdmin int8 `json:"is_admin" gorm:"type:tinyint;not null;default:0"` Status Status `json:"status" gorm:"type:tinyint;not null;default:1"` BaseModel `json:"-" gorm:"-"` } // 新增 func (user *User) Create() (insertId int, err error) { user.Status = Enabled user.Salt = "" // bcrypt不需要单独的salt user.Password, err = utils.HashPassword(user.Password) if err != nil { return 0, err } result := Db.Create(user) if result.Error == nil { insertId = user.Id } return insertId, result.Error } // 更新 func (user *User) Update(id int, data CommonMap) (int64, error) { updateData := make(map[string]interface{}) for k, v := range data { updateData[k] = v } result := Db.Model(&User{}).Where("id = ?", id).UpdateColumns(updateData) return result.RowsAffected, result.Error } func (user *User) UpdatePassword(id int, password string) (int64, error) { safePassword, err := utils.HashPassword(password) if err != nil { return 0, err } return user.Update(id, CommonMap{"password": safePassword, "salt": ""}) } // 删除 func (user *User) Delete(id int) (int64, error) { result := Db.Delete(&User{}, id) return result.RowsAffected, result.Error } // 禁用 func (user *User) Disable(id int) (int64, error) { return user.Update(id, CommonMap{"status": Disabled}) } // 激活 func (user *User) Enable(id int) (int64, error) { return user.Update(id, CommonMap{"status": Enabled}) } // 验证用户名和密码 func (user *User) Match(username, password string) bool { err := Db.Where("(name = ? OR email = ?) AND status = ?", username, username, Enabled).First(user).Error if err != nil { return false } return utils.VerifyPassword(user.Password, password, user.Salt) } // 获取用户详情 func (user *User) Find(id int) error { return Db.First(user, id).Error } // 用户名是否存在 func (user *User) UsernameExists(username string, uid int) (int64, error) { var count int64 query := Db.Model(&User{}).Where("name = ?", username) if uid > 0 { query = query.Where("id != ?", uid) } err := query.Count(&count).Error return count, err } // 邮箱地址是否存在 func (user *User) EmailExists(email string, uid int) (int64, error) { var count int64 query := Db.Model(&User{}).Where("email = ?", email) if uid > 0 { query = query.Where("id != ?", uid) } err := query.Count(&count).Error return count, err } func (user *User) List(params CommonMap) ([]User, error) { user.parsePageAndPageSize(params) list := make([]User, 0) err := Db.Order("id DESC").Limit(user.PageSize).Offset(user.pageLimitOffset()).Find(&list).Error return list, err } func (user *User) Total() (int64, error) { var count int64 err := Db.Model(&User{}).Count(&count).Error return count, err } ================================================ FILE: internal/models/webhook_test.go ================================================ package models import ( "encoding/json" "testing" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) // setupTestDB 创建测试数据库 func setupTestDB(t *testing.T) *gorm.DB { db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("failed to open test database: %v", err) } // 创建表 if err := db.AutoMigrate(&Setting{}); err != nil { t.Fatalf("failed to migrate: %v", err) } return db } // TestWebhookUrl_JSONMarshaling 测试WebhookUrl的JSON序列化 func TestWebhookUrl_JSONMarshaling(t *testing.T) { webhookUrl := WebhookUrl{ Id: 1, Name: "Production Alert", Url: "https://example.com/webhook", } // 序列化 data, err := json.Marshal(webhookUrl) if err != nil { t.Fatalf("failed to marshal: %v", err) } // 反序列化 var decoded WebhookUrl if err := json.Unmarshal(data, &decoded); err != nil { t.Fatalf("failed to unmarshal: %v", err) } // 验证 if decoded.Id != webhookUrl.Id { t.Errorf("expected Id %d, got %d", webhookUrl.Id, decoded.Id) } if decoded.Name != webhookUrl.Name { t.Errorf("expected Name %s, got %s", webhookUrl.Name, decoded.Name) } if decoded.Url != webhookUrl.Url { t.Errorf("expected Url %s, got %s", webhookUrl.Url, decoded.Url) } } // TestSetting_CreateWebhookUrl 测试创建webhook地址 func TestSetting_CreateWebhookUrl(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} name := "Test Webhook" url := "https://test.example.com/webhook" rows, err := setting.CreateWebhookUrl(name, url) if err != nil { t.Fatalf("failed to create webhook url: %v", err) } if rows != 1 { t.Errorf("expected 1 row affected, got %d", rows) } // 验证数据已保存 var saved Setting if err := db.Where("code = ? AND `key` = ?", WebhookCode, WebhookUrlKey).First(&saved).Error; err != nil { t.Fatalf("failed to find saved webhook url: %v", err) } var webhookUrl WebhookUrl if err := json.Unmarshal([]byte(saved.Value), &webhookUrl); err != nil { t.Fatalf("failed to unmarshal saved value: %v", err) } if webhookUrl.Name != name { t.Errorf("expected name %s, got %s", name, webhookUrl.Name) } if webhookUrl.Url != url { t.Errorf("expected url %s, got %s", url, webhookUrl.Url) } } // TestSetting_RemoveWebhookUrl 测试删除webhook地址 func TestSetting_RemoveWebhookUrl(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} // 先创建一个webhook地址 _, err := setting.CreateWebhookUrl("Test Webhook", "https://test.example.com/webhook") if err != nil { t.Fatalf("failed to create webhook url: %v", err) } // 获取创建的ID var saved Setting if err := db.Where("code = ? AND `key` = ?", WebhookCode, WebhookUrlKey).First(&saved).Error; err != nil { t.Fatalf("failed to find saved webhook url: %v", err) } // 删除 rows, err := setting.RemoveWebhookUrl(saved.Id) if err != nil { t.Fatalf("failed to remove webhook url: %v", err) } if rows != 1 { t.Errorf("expected 1 row affected, got %d", rows) } // 验证已删除 var count int64 db.Model(&Setting{}).Where("id = ?", saved.Id).Count(&count) if count != 0 { t.Errorf("expected webhook url to be deleted, but still exists") } } // TestSetting_Webhook 测试获取webhook配置 func TestSetting_Webhook(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} // 创建模板配置 db.Create(&Setting{ Code: WebhookCode, Key: WebhookTemplateKey, Value: `{"task_id": "{{.TaskId}}"}`, }) // 创建多个webhook地址 _, _ = setting.CreateWebhookUrl("Webhook 1", "https://webhook1.example.com") _, _ = setting.CreateWebhookUrl("Webhook 2", "https://webhook2.example.com") // 获取配置 webhook, err := setting.Webhook() if err != nil { t.Fatalf("failed to get webhook config: %v", err) } // 验证模板 if webhook.Template != `{"task_id": "{{.TaskId}}"}` { t.Errorf("unexpected template: %s", webhook.Template) } // 验证webhook地址数量 if len(webhook.WebhookUrls) != 2 { t.Fatalf("expected 2 webhook urls, got %d", len(webhook.WebhookUrls)) } // 验证webhook地址内容 names := make(map[string]bool) urls := make(map[string]bool) for _, w := range webhook.WebhookUrls { names[w.Name] = true urls[w.Url] = true } if !names["Webhook 1"] || !names["Webhook 2"] { t.Error("webhook names not found") } if !urls["https://webhook1.example.com"] || !urls["https://webhook2.example.com"] { t.Error("webhook urls not found") } } // TestSetting_UpdateWebHook 测试更新webhook模板 func TestSetting_UpdateWebHook(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} // 创建初始模板 db.Create(&Setting{ Code: WebhookCode, Key: WebhookTemplateKey, Value: "old template", }) // 更新模板 newTemplate := `{"task": "{{.TaskName}}"}` err := setting.UpdateWebHook(newTemplate) if err != nil { t.Fatalf("failed to update webhook: %v", err) } // 验证更新 var saved Setting if err := db.Where("code = ? AND `key` = ?", WebhookCode, WebhookTemplateKey).First(&saved).Error; err != nil { t.Fatalf("failed to find updated template: %v", err) } if saved.Value != newTemplate { t.Errorf("expected template %s, got %s", newTemplate, saved.Value) } } // TestSetting_Webhook_EmptyUrls 测试空webhook地址列表 func TestSetting_Webhook_EmptyUrls(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} // 只创建模板，不创建webhook地址 db.Create(&Setting{ Code: WebhookCode, Key: WebhookTemplateKey, Value: "template", }) webhook, err := setting.Webhook() if err != nil { t.Fatalf("failed to get webhook config: %v", err) } if len(webhook.WebhookUrls) != 0 { t.Errorf("expected empty webhook urls, got %d", len(webhook.WebhookUrls)) } } // TestSetting_CreateWebhookUrl_InvalidJSON 测试创建webhook时JSON序列化错误处理 func TestSetting_CreateWebhookUrl_DuplicateNames(t *testing.T) { db := setupTestDB(t) Db = db setting := &Setting{} // 创建第一个webhook _, err := setting.CreateWebhookUrl("Duplicate", "https://url1.example.com") if err != nil { t.Fatalf("failed to create first webhook: %v", err) } // 创建同名webhook（应该允许，因为没有唯一性约束） _, err = setting.CreateWebhookUrl("Duplicate", "https://url2.example.com") if err != nil { t.Fatalf("failed to create second webhook: %v", err) } // 验证两个都存在 var count int64 db.Model(&Setting{}).Where("code = ? AND `key` = ?", WebhookCode, WebhookUrlKey).Count(&count) if count != 2 { t.Errorf("expected 2 webhook urls, got %d", count) } } // BenchmarkSetting_CreateWebhookUrl 性能测试：创建webhook地址 func BenchmarkSetting_CreateWebhookUrl(b *testing.B) { db, _ := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err := db.AutoMigrate(&Setting{}); err != nil { b.Fatalf("failed to migrate: %v", err) } Db = db setting := &Setting{} b.ResetTimer() for i := 0; i < b.N; i++ { _, _ = setting.CreateWebhookUrl("Benchmark Webhook", "https://benchmark.example.com") } } // BenchmarkSetting_Webhook 性能测试：获取webhook配置 func BenchmarkSetting_Webhook(b *testing.B) { db, _ := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err := db.AutoMigrate(&Setting{}); err != nil { b.Fatalf("failed to migrate: %v", err) } Db = db setting := &Setting{} // 准备测试数据 db.Create(&Setting{Code: WebhookCode, Key: WebhookTemplateKey, Value: "template"}) for i := 0; i < 10; i++ { _, _ = setting.CreateWebhookUrl("Webhook", "https://example.com") } b.ResetTimer() for i := 0; i < b.N; i++ { _, _ = setting.Webhook() } } ================================================ FILE: internal/modules/app/app.go ================================================ package app import ( "os" "path/filepath" "fmt" "strconv" "strings" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/setting" "github.com/gocronx-team/gocron/internal/modules/utils" ) var ( // AppDir 应用根目录 AppDir string // 应用根目录 // ConfDir 配置文件目录 ConfDir string // 配置目录 // LogDir 日志目录 LogDir string // 日志目录 // AppConfig 配置文件 AppConfig string // 应用配置文件 // Installed 应用是否已安装 Installed bool // 应用是否安装过 // Setting 应用配置 Setting *setting.Setting // 应用配置 // VersionId 版本号 VersionId int // 版本号 // VersionFile 版本文件 VersionFile string // 版本号文件 ) // InitEnv 初始化 func InitEnv(versionString string) { logger.InitLogger() var err error // 开发环境使用当前目录，生产环境使用可执行文件目录 execPath, err := os.Executable() if err != nil { logger.Fatal(err) } execDir := filepath.Dir(execPath) // 开发环境检测：Air 热重载（tmp 目录）或 go run（go-build cache 目录） execName := filepath.Base(execPath) if filepath.Base(execDir) == "tmp" { AppDir = filepath.Join(filepath.Dir(execDir), ".gocron") } else if strings.Contains(execDir, "go-build") && !strings.HasSuffix(execName, ".test") { // go run 会将二进制编译到 go-build cache 中，使用当前工作目录 // 排除 go test（测试二进制以 .test 结尾） wd, wdErr := os.Getwd() if wdErr != nil { logger.Fatal(wdErr) } AppDir = filepath.Join(wd, ".gocron") } else { AppDir = filepath.Join(execDir, ".gocron") } fmt.Printf("AppDir: %s\n", AppDir) ConfDir = filepath.Join(AppDir, "conf") LogDir = filepath.Join(AppDir, "log") AppConfig = filepath.Join(ConfDir, "app.ini") VersionFile = filepath.Join(ConfDir, ".version") fmt.Printf("ConfDir: %s, LogDir: %s\n", ConfDir, LogDir) createDirIfNotExists(AppDir, ConfDir, LogDir) Installed = IsInstalled() VersionId = ToNumberVersion(versionString) } // IsInstalled 判断应用是否已安装 func IsInstalled() bool { _, err := os.Stat(filepath.Join(ConfDir, "install.lock")) return !os.IsNotExist(err) } // CreateInstallLock 创建安装锁文件 func CreateInstallLock() error { lockFile := filepath.Join(ConfDir, "install.lock") err := os.WriteFile(lockFile, []byte(""), 0600) if err != nil { logger.Error("创建安装锁文件conf/install.lock失败", err) fmt.Printf("Error creating install.lock: %v\n", err) } else { fmt.Printf("Successfully created install.lock at %s\n", lockFile) } return err } // UpdateVersionFile 更新应用版本号文件 func UpdateVersionFile() { err := os.WriteFile(VersionFile, []byte(strconv.Itoa(VersionId)), 0600, ) if err != nil { logger.Fatal(err) } } // GetCurrentVersionId 获取应用当前版本号, 从版本号文件中读取 func GetCurrentVersionId() int { if !utils.FileExist(VersionFile) { return 0 } bytes, err := os.ReadFile(VersionFile) if err != nil { logger.Fatal(err) } versionId, err := strconv.Atoi(strings.TrimSpace(string(bytes))) if err != nil { logger.Fatal(err) } return versionId } // ToNumberVersion 把字符串版本号a.b.c转换为整数版本号abc // 非数字版本（如 "dev"）返回 0 func ToNumberVersion(versionString string) int { versionString = strings.TrimPrefix(versionString, "v") v := strings.Replace(versionString, ".", "", -1) if len(v) < 3 { v += "0" } versionId, err := strconv.Atoi(v) if err != nil { return 0 } return versionId } // 检测目录是否存在 func createDirIfNotExists(path ...string) { for _, value := range path { if utils.FileExist(value) { continue } err := os.MkdirAll(value, 0755) if err != nil { logger.Fatal(fmt.Sprintf("创建目录失败:%s", err.Error())) } } } ================================================ FILE: internal/modules/app/app_test.go ================================================ package app import ( "os" "path/filepath" "testing" ) func initTempEnv(t *testing.T, version string) string { t.Helper() home := t.TempDir() t.Setenv("HOME", home) t.Setenv("USERPROFILE", home) // 保存原始值 oldAppDir := AppDir oldConfDir := ConfDir oldLogDir := LogDir oldVersionFile := VersionFile oldVersionId := VersionId oldInstalled := Installed // 清理函数 t.Cleanup(func() { AppDir = oldAppDir ConfDir = oldConfDir LogDir = oldLogDir VersionFile = oldVersionFile VersionId = oldVersionId Installed = oldInstalled }) InitEnv(version) return home } func TestInitEnvCreatesDirectoriesAndSetsVersion(t *testing.T) { initTempEnv(t, "1.2.3") // 验证目录被创建（不检查具体路径，因为它依赖于可执行文件位置） for _, dir := range []string{AppDir, ConfDir, LogDir} { if fi, err := os.Stat(dir); err != nil || !fi.IsDir() { t.Fatalf("expected directory %s to exist", dir) } } expectedVersion := ToNumberVersion("1.2.3") if VersionId != expectedVersion { t.Fatalf("expected VersionId %d, got %d", expectedVersion, VersionId) } if Installed { t.Fatal("app should not be marked installed without lock file") } } func TestCreateInstallLockAndIsInstalled(t *testing.T) { initTempEnv(t, "1.0.0") lockPath := filepath.Join(ConfDir, "install.lock") if IsInstalled() { t.Fatal("expected not installed before lock file exists") } if err := CreateInstallLock(); err != nil { t.Fatalf("CreateInstallLock failed: %v", err) } if _, err := os.Stat(lockPath); err != nil { t.Fatalf("install lock not created: %v", err) } if !IsInstalled() { t.Fatal("expected installed after lock creation") } } func TestCreateInstallLockSetsSecurePermissions(t *testing.T) { initTempEnv(t, "1.0.0") lockPath := filepath.Join(ConfDir, "install.lock") if err := CreateInstallLock(); err != nil { t.Fatalf("CreateInstallLock failed: %v", err) } info, err := os.Stat(lockPath) if err != nil { t.Fatalf("stat failed: %v", err) } perm := info.Mode().Perm() if perm != 0600 { t.Fatalf("expected file permission 0600, got %#o", perm) } } func TestUpdateVersionFileAndGetCurrentVersionId(t *testing.T) { initTempEnv(t, "1.0.0") VersionId = 789 UpdateVersionFile() id := GetCurrentVersionId() if id != 789 { t.Fatalf("expected version id 789, got %d", id) } } func TestUpdateVersionFileSetsSecurePermissions(t *testing.T) { initTempEnv(t, "1.0.0") VersionId = 123 UpdateVersionFile() info, err := os.Stat(VersionFile) if err != nil { t.Fatalf("stat failed: %v", err) } perm := info.Mode().Perm() if perm != 0600 { t.Fatalf("expected file permission 0600, got %#o", perm) } } func TestGetCurrentVersionIdWhenMissing(t *testing.T) { // 创建临时目录但不调用 InitEnv，手动设置 VersionFile tempDir := t.TempDir() oldVersionFile := VersionFile VersionFile = filepath.Join(tempDir, ".version") t.Cleanup(func() { VersionFile = oldVersionFile }) if id := GetCurrentVersionId(); id != 0 { t.Fatalf("expected 0 when version file missing, got %d", id) } } func TestToNumberVersion(t *testing.T) { tests := []struct { input string want int }{ {"v1.2.3", 123}, {"1.2", 120}, {"2.0.10", 2010}, } for _, tt := range tests { got := ToNumberVersion(tt.input) if got != tt.want { t.Fatalf("ToNumberVersion(%s) = %d, want %d", tt.input, got, tt.want) } } } func TestCreateDirIfNotExists(t *testing.T) { dir := filepath.Join(t.TempDir(), "nested", "dir") createDirIfNotExists(dir) if fi, err := os.Stat(dir); err != nil || !fi.IsDir() { t.Fatalf("expected directory %s to exist", dir) } } ================================================ FILE: internal/modules/httpclient/http_client.go ================================================ package httpclient // http-client import ( "bytes" "encoding/json" "fmt" "io" "net" "net/http" "strings" "time" ) type ResponseWrapper struct { StatusCode int Body string Header http.Header } type httpDoer interface { Do(req *http.Request) (*http.Response, error) } // 优化：使用全局 HTTP 客户端，复用连接池 var defaultClient = &http.Client{ Timeout: 300 * time.Second, Transport: &http.Transport{ MaxIdleConns: 100, MaxIdleConnsPerHost: 10, IdleConnTimeout: 90 * time.Second, DialContext: (&net.Dialer{ Timeout: 5 * time.Second, KeepAlive: 30 * time.Second, }).DialContext, }, } var clientFactory = func(timeout int) httpDoer { // 使用默认超时（300秒）或未设置超时时，直接返回全局客户端 if timeout <= 0 || timeout == 300 { return defaultClient } // 其他超时值：创建新客户端但复用 Transport（连接池） return &http.Client{ Timeout: time.Duration(timeout) * time.Second, Transport: defaultClient.Transport, } } func Get(url string, timeout int) ResponseWrapper { req, err := http.NewRequest("GET", url, nil) if err != nil { return createRequestError(err) } return request(req, timeout) } func PostParams(url string, params string, timeout int) ResponseWrapper { buf := bytes.NewBufferString(params) req, err := http.NewRequest("POST", url, buf) if err != nil { return createRequestError(err) } req.Header.Set("Content-type", "application/x-www-form-urlencoded") return request(req, timeout) } func PostJson(url string, body string, timeout int) ResponseWrapper { buf := bytes.NewBufferString(body) req, err := http.NewRequest("POST", url, buf) if err != nil { return createRequestError(err) } req.Header.Set("Content-type", "application/json") return request(req, timeout) } // blockedHeaders 禁止用户设置的危险 Header var blockedHeaders = map[string]bool{ "host": true, "transfer-encoding": true, "content-length": true, "connection": true, "upgrade": true, "proxy-authorization": true, "proxy-connection": true, "te": true, "trailer": true, } // IsBlockedHeader 检查 header 是否在黑名单中 func IsBlockedHeader(name string) bool { return blockedHeaders[strings.ToLower(strings.TrimSpace(name))] } // ValidateHeaders 校验 headers JSON 格式并检查黑名单，返回错误信息 func ValidateHeaders(headersJSON string) error { if strings.TrimSpace(headersJSON) == "" { return nil } var headers map[string]string if err := json.Unmarshal([]byte(headersJSON), &headers); err != nil { return fmt.Errorf("invalid JSON format") } for k := range headers { if IsBlockedHeader(k) { return fmt.Errorf("header %q is not allowed", k) } } return nil } // SetCustomHeaders 为请求设置自定义 Header（JSON 格式: {"Key": "Value", ...}） // 黑名单中的 Header 会被跳过并记录日志 func SetCustomHeaders(req *http.Request, headersJSON string) { if strings.TrimSpace(headersJSON) == "" { return } var headers map[string]string if err := json.Unmarshal([]byte(headersJSON), &headers); err != nil { fmt.Printf("[WARN] failed to parse custom headers: %v\n", err) return } for k, v := range headers { if IsBlockedHeader(k) { fmt.Printf("[WARN] blocked header %q skipped\n", k) continue } req.Header.Set(k, v) } } // GetWithHeaders 带自定义 Header 的 GET 请求 func GetWithHeaders(url string, headersJSON string, timeout int) ResponseWrapper { req, err := http.NewRequest("GET", url, nil) if err != nil { return createRequestError(err) } SetCustomHeaders(req, headersJSON) return request(req, timeout) } // PostJsonWithHeaders 带自定义 Header 的 POST JSON 请求 func PostJsonWithHeaders(url string, body string, headersJSON string, timeout int) ResponseWrapper { buf := bytes.NewBufferString(body) req, err := http.NewRequest("POST", url, buf) if err != nil { return createRequestError(err) } req.Header.Set("Content-type", "application/json") SetCustomHeaders(req, headersJSON) return request(req, timeout) } // PostParamsWithHeaders 带自定义 Header 的 POST 表单请求 func PostParamsWithHeaders(url string, params string, headersJSON string, timeout int) ResponseWrapper { buf := bytes.NewBufferString(params) req, err := http.NewRequest("POST", url, buf) if err != nil { return createRequestError(err) } req.Header.Set("Content-type", "application/x-www-form-urlencoded") SetCustomHeaders(req, headersJSON) return request(req, timeout) } func request(req *http.Request, timeout int) ResponseWrapper { wrapper := ResponseWrapper{StatusCode: 0, Body: "", Header: make(http.Header)} client := clientFactory(timeout) setRequestHeader(req) resp, err := client.Do(req) if err != nil { wrapper.Body = fmt.Sprintf("执行HTTP请求错误-%s", err.Error()) return wrapper } defer resp.Body.Close() // 限制响应体最大 1MB，防止 OOM body, err := io.ReadAll(io.LimitReader(resp.Body, 1<<20)) if err != nil { wrapper.Body = fmt.Sprintf("读取HTTP请求返回值失败-%s", err.Error()) return wrapper } wrapper.StatusCode = resp.StatusCode wrapper.Body = string(body) wrapper.Header = resp.Header return wrapper } func setRequestHeader(req *http.Request) { req.Header.Set("User-Agent", "golang/gocron") } func createRequestError(err error) ResponseWrapper { errorMessage := fmt.Sprintf("创建HTTP请求错误-%s", err.Error()) return ResponseWrapper{0, errorMessage, make(http.Header)} } ================================================ FILE: internal/modules/httpclient/http_client_benchmark_test.go ================================================ package httpclient import ( "net/http" "net/http/httptest" "sync" "testing" "time" ) // 测试服务器 func setupTestServer() *httptest.Server { return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { time.Sleep(10 * time.Millisecond) // 模拟处理时间 w.WriteHeader(http.StatusOK) _, _ = w.Write([]byte("success")) })) } // 测试1: 连续请求性能 func TestSequentialRequests(t *testing.T) { server := setupTestServer() defer server.Close() requests := 50 start := time.Now() for i := 0; i < requests; i++ { resp := Get(server.URL, 5) if resp.StatusCode != 200 { t.Errorf("请求 %d 失败: %v", i, resp.Body) } } duration := time.Since(start) avgTime := float64(duration.Milliseconds()) / float64(requests) t.Logf("📊 连续请求测试 (%d 个请求):", requests) t.Logf(" 总耗时: %v", duration) t.Logf(" 平均耗时: %.2f ms/请求", avgTime) } // 测试2: 并发请求性能 func TestConcurrentRequests(t *testing.T) { server := setupTestServer() defer server.Close() concurrency := 20 requestsPerWorker := 5 totalRequests := concurrency * requestsPerWorker start := time.Now() var wg sync.WaitGroup wg.Add(concurrency) successCount := 0 errorCount := 0 var mu sync.Mutex for i := 0; i < concurrency; i++ { go func() { defer wg.Done() for j := 0; j < requestsPerWorker; j++ { resp := Get(server.URL, 5) mu.Lock() if resp.StatusCode == 200 { successCount++ } else { errorCount++ } mu.Unlock() } }() } wg.Wait() duration := time.Since(start) avgTime := float64(duration.Milliseconds()) / float64(totalRequests) t.Logf("📊 并发请求测试 (%d 并发, %d 个请求):", concurrency, totalRequests) t.Logf(" 总耗时: %v", duration) t.Logf(" 平均耗时: %.2f ms/请求", avgTime) t.Logf(" 成功: %d, 失败: %d", successCount, errorCount) } // 测试3: 不同超时配置 func TestDifferentTimeouts(t *testing.T) { server := setupTestServer() defer server.Close() timeouts := []int{5, 10, 30, 300} for _, timeout := range timeouts { start := time.Now() resp := Get(server.URL, timeout) duration := time.Since(start) if resp.StatusCode != 200 { t.Errorf("超时 %d 秒的请求失败: %v", timeout, resp.Body) } t.Logf(" 超时配置 %ds: 耗时 %v", timeout, duration) } } // 基准测试1: 单个请求 func BenchmarkSingleRequest(b *testing.B) { server := setupTestServer() defer server.Close() b.ResetTimer() for i := 0; i < b.N; i++ { Get(server.URL, 5) } } // 基准测试2: 并发请求 func BenchmarkConcurrentRequests(b *testing.B) { server := setupTestServer() defer server.Close() b.RunParallel(func(pb *testing.PB) { for pb.Next() { Get(server.URL, 5) } }) } // 基准测试3: POST 请求 func BenchmarkPostRequest(b *testing.B) { server := setupTestServer() defer server.Close() b.ResetTimer() for i := 0; i < b.N; i++ { PostParams(server.URL, "key=value", 5) } } // 测试4: 高并发压力测试 func TestHighConcurrency(t *testing.T) { if testing.Short() { t.Skip("跳过压力测试，使用 -short 标志") } server := setupTestServer() defer server.Close() concurrency := 100 requestsPerWorker := 10 totalRequests := concurrency * requestsPerWorker t.Logf("🔥 高并发压力测试 (%d 并发, %d 个请求)", concurrency, totalRequests) start := time.Now() var wg sync.WaitGroup wg.Add(concurrency) successCount := 0 errorCount := 0 var mu sync.Mutex for i := 0; i < concurrency; i++ { go func() { defer wg.Done() for j := 0; j < requestsPerWorker; j++ { resp := Get(server.URL, 5) mu.Lock() if resp.StatusCode == 200 { successCount++ } else { errorCount++ } mu.Unlock() } }() } wg.Wait() duration := time.Since(start) qps := float64(totalRequests) / duration.Seconds() t.Logf("📊 压力测试结果:") t.Logf(" 总耗时: %v", duration) t.Logf(" QPS: %.2f", qps) t.Logf(" 成功: %d, 失败: %d", successCount, errorCount) t.Logf(" 成功率: %.2f%%", float64(successCount)/float64(totalRequests)*100) } // 测试5: 连接复用验证 func TestConnectionReuse(t *testing.T) { server := setupTestServer() defer server.Close() t.Log("🔍 连接复用测试 (执行 10 次请求)") for i := 0; i < 10; i++ { start := time.Now() resp := Get(server.URL, 5) duration := time.Since(start) if resp.StatusCode != 200 { t.Errorf("请求 %d 失败", i+1) } t.Logf(" 请求 %d: %v", i+1, duration) } t.Log("💡 提示: 如果后续请求明显快于首次请求，说明连接被复用") } ================================================ FILE: internal/modules/httpclient/http_client_test.go ================================================ package httpclient import ( "errors" "fmt" "io" "net/http" "strings" "testing" ) type mockDoer func(req *http.Request) (*http.Response, error) func (m mockDoer) Do(req *http.Request) (*http.Response, error) { return m(req) } func withMockClient(t *testing.T, doer mockDoer) { t.Helper() original := clientFactory clientFactory = func(timeout int) httpDoer { return doer } t.Cleanup(func() { clientFactory = original }) } func TestGetRequest(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { if req.Method != http.MethodGet { t.Fatalf("expected GET, got %s", req.Method) } if ua := req.Header.Get("User-Agent"); ua != "golang/gocron" { t.Fatalf("unexpected user-agent %s", ua) } return &http.Response{ StatusCode: 200, Body: io.NopCloser(strings.NewReader("ok")), Header: http.Header{}, }, nil }) resp := Get("http://example.com", 0) if resp.StatusCode != 200 || resp.Body != "ok" { t.Fatalf("unexpected response: %+v", resp) } } func TestPostParamsRequest(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { if req.Method != http.MethodPost { t.Fatalf("expected POST, got %s", req.Method) } if req.Header.Get("Content-type") != "application/x-www-form-urlencoded" { t.Fatalf("unexpected content-type %s", req.Header.Get("Content-type")) } body, _ := io.ReadAll(req.Body) if string(body) != "a=1&b=2" { t.Fatalf("unexpected body %s", string(body)) } return &http.Response{ StatusCode: 200, Body: io.NopCloser(strings.NewReader("echo:" + string(body))), Header: http.Header{}, }, nil }) resp := PostParams("http://example.com", "a=1&b=2", 0) if resp.StatusCode != 200 || resp.Body != "echo:a=1&b=2" { t.Fatalf("unexpected response: %+v", resp) } } func TestPostJsonRequest(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { if req.Header.Get("Content-type") != "application/json" { t.Fatalf("unexpected content-type %s", req.Header.Get("Content-type")) } body, _ := io.ReadAll(req.Body) return &http.Response{ StatusCode: 200, Body: io.NopCloser(strings.NewReader("json:" + string(body))), Header: http.Header{}, }, nil }) resp := PostJson("http://example.com", `{"name":"gocron"}`, 0) if resp.StatusCode != 200 || resp.Body != `json:{"name":"gocron"}` { t.Fatalf("unexpected response: %+v", resp) } } func TestRequestHandlesClientError(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { return nil, errors.New("timeout") }) resp := Get("http://example.com", 1) if resp.StatusCode != 0 || !strings.Contains(resp.Body, "执行HTTP请求错误-timeout") { t.Fatalf("expected client error message, got %+v", resp) } } func TestRequestHandlesReadError(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { rc := io.NopCloser(io.Reader(&failingReader{})) return &http.Response{StatusCode: 200, Body: rc, Header: http.Header{}}, nil }) resp := Get("http://example.com", 0) if resp.StatusCode != 0 || !strings.Contains(resp.Body, "读取HTTP请求返回值失败") { t.Fatalf("expected read error message, got %+v", resp) } } type failingReader struct{} func (f *failingReader) Read(p []byte) (int, error) { return 0, errors.New("boom") } func TestCreateRequestError(t *testing.T) { resp := createRequestError(fmt.Errorf("boom")) if resp.StatusCode != 0 || !strings.Contains(resp.Body, "boom") { t.Fatalf("unexpected error wrapper: %+v", resp) } } func TestSetCustomHeaders(t *testing.T) { t.Run("valid JSON headers", func(t *testing.T) { req, _ := http.NewRequest("GET", "http://example.com", nil) SetCustomHeaders(req, `{"Authorization":"Bearer token123","X-Custom":"value"}`) if req.Header.Get("Authorization") != "Bearer token123" { t.Fatalf("expected Authorization header, got %q", req.Header.Get("Authorization")) } if req.Header.Get("X-Custom") != "value" { t.Fatalf("expected X-Custom header, got %q", req.Header.Get("X-Custom")) } }) t.Run("empty string is no-op", func(t *testing.T) { req, _ := http.NewRequest("GET", "http://example.com", nil) SetCustomHeaders(req, "") if len(req.Header) != 0 { t.Fatalf("expected no headers for empty input, got %v", req.Header) } }) t.Run("invalid JSON is no-op", func(t *testing.T) { req, _ := http.NewRequest("GET", "http://example.com", nil) SetCustomHeaders(req, "not json") // Should not panic, just skip }) t.Run("blocked headers are skipped", func(t *testing.T) { req, _ := http.NewRequest("GET", "http://example.com", nil) SetCustomHeaders(req, `{"Host":"evil.com","X-Safe":"ok","Transfer-Encoding":"chunked"}`) if req.Header.Get("Host") != "" { t.Fatal("Host header should be blocked") } if req.Header.Get("Transfer-Encoding") != "" { t.Fatal("Transfer-Encoding header should be blocked") } if req.Header.Get("X-Safe") != "ok" { t.Fatalf("safe header should be set, got %q", req.Header.Get("X-Safe")) } }) } func TestIsBlockedHeader(t *testing.T) { blocked := []string{"Host", "host", "HOST", "Transfer-Encoding", "connection", "Content-Length", "Upgrade"} for _, h := range blocked { if !IsBlockedHeader(h) { t.Errorf("expected %q to be blocked", h) } } allowed := []string{"Authorization", "X-Custom", "Content-Type", "Accept"} for _, h := range allowed { if IsBlockedHeader(h) { t.Errorf("expected %q to be allowed", h) } } } func TestValidateHeaders(t *testing.T) { t.Run("empty is ok", func(t *testing.T) { if err := ValidateHeaders(""); err != nil { t.Fatalf("unexpected error: %v", err) } }) t.Run("valid headers", func(t *testing.T) { if err := ValidateHeaders(`{"Authorization":"Bearer x"}`); err != nil { t.Fatalf("unexpected error: %v", err) } }) t.Run("invalid JSON", func(t *testing.T) { err := ValidateHeaders("not json") if err == nil { t.Fatal("expected error for invalid JSON") } }) t.Run("blocked header rejected", func(t *testing.T) { err := ValidateHeaders(`{"Host":"evil.com"}`) if err == nil { t.Fatal("expected error for blocked header") } if !strings.Contains(err.Error(), "not allowed") { t.Fatalf("unexpected error: %v", err) } }) t.Run("mixed blocked and safe", func(t *testing.T) { err := ValidateHeaders(`{"Authorization":"ok","Transfer-Encoding":"chunked"}`) if err == nil { t.Fatal("expected error for blocked header") } }) } func TestGetWithHeaders(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { if req.Header.Get("Authorization") != "Bearer abc" { t.Fatalf("expected Authorization header, got %q", req.Header.Get("Authorization")) } return &http.Response{ StatusCode: 200, Body: io.NopCloser(strings.NewReader("ok")), Header: http.Header{}, }, nil }) resp := GetWithHeaders("http://example.com", `{"Authorization":"Bearer abc"}`, 10) if resp.StatusCode != 200 || resp.Body != "ok" { t.Fatalf("unexpected response: %+v", resp) } } func TestPostJsonWithHeaders(t *testing.T) { withMockClient(t, func(req *http.Request) (*http.Response, error) { if req.Header.Get("X-Api-Key") != "secret" { t.Fatalf("expected X-Api-Key header, got %q", req.Header.Get("X-Api-Key")) } ct := req.Header.Get("Content-type") if !strings.Contains(ct, "application/json") { t.Fatalf("expected JSON content-type, got %q", ct) } body, _ := io.ReadAll(req.Body) if string(body) != `{"k":"v"}` { t.Fatalf("unexpected body: %s", string(body)) } return &http.Response{ StatusCode: 200, Body: io.NopCloser(strings.NewReader("ok")), Header: http.Header{}, }, nil }) resp := PostJsonWithHeaders("http://example.com", `{"k":"v"}`, `{"X-Api-Key":"secret"}`, 10) if resp.StatusCode != 200 { t.Fatalf("unexpected status: %d", resp.StatusCode) } } ================================================ FILE: internal/modules/i18n/en_us.go ================================================ package i18n var enUS = map[string]string{ "form_validation_failed": "Form validation failed, please check your input", "system_already_installed": "System already installed!", "password_mismatch": "Passwords do not match", "db_config_write_failed": "Failed to write database configuration to file", "app_config_read_failed": "Failed to read application configuration", "create_table_failed": "Failed to create database tables", "create_admin_failed": "Failed to create administrator account", "create_lock_file_failed": "Failed to create installation lock file", "user_not_found": "User not found", "generate_2fa_key_failed": "Failed to generate 2FA key", "generate_qrcode_failed": "Failed to generate QR code", "get_success": "Retrieved successfully", "verification_code_error": "Verification code is incorrect", "enable_failed": "Failed to enable", "2fa_enabled": "2FA enabled", "2fa_not_enabled": "2FA not enabled", "disable_failed": "Failed to disable", "2fa_disabled": "2FA disabled", "update_success": "Updated successfully", "incomplete_parameters": "Incomplete parameters", "operation_success": "Operation successful", "username_exists": "Username already exists", "email_exists": "Email already exists", "password_required": "Please enter password", "password_confirm_required": "Please enter password again", "save_failed": "Save failed", "update_failed": "Update failed", "save_success": "Saved successfully", "password_same_as_old": "New password cannot be the same as old password", "old_password_error": "Old password is incorrect", "username_password_empty": "Username or password cannot be empty", "username_password_error": "Username or password is incorrect", "2fa_code_required": "2FA verification code required", "2fa_code_error": "2FA verification code is incorrect", "auth_failed": "Authentication failed", "page_not_found": "Page not found", "app_not_installed": "Application not installed", "unauthorized": "Unauthorized access", "api_key_required": "API key not configured", "param_time_required": "time parameter is required", "param_time_invalid": "time parameter has expired", "param_sign_required": "sign parameter is required", "sign_verify_failed": "Signature verification failed", "invalid_log_id": "Invalid log ID", "invalid_task_id": "Invalid task ID", "get_task_info_failed": "Failed to get task information", "only_shell_task_can_stop": "Only SHELL tasks can be stopped manually", "task_node_list_empty": "Task node list is empty", "stop_task_sent": "Stop command sent, please wait for task to exit", "param_range_1_12": "Parameter value range: 1-12", "delete_failed": "Delete failed", "delete_success": "Deleted successfully", "http_task_timeout_max_300": "HTTP task timeout cannot exceed 300 seconds", "crontab_parse_failed": "Failed to parse crontab expression", "cannot_set_self_as_child": "Cannot set current task as child task", "host_not_exist": "Host does not exist", "refresh_task_host_failed": "Failed to refresh task host information", "invalid_url": "Please enter a valid URL", "hostname_exists": "Hostname already exists", "task_name_exists": "Task name already exists", "retry_times_range_0_10": "Retry times must be between 0-10", "retry_interval_range_0_3600": "Retry interval must be between 0-3600", "param_error": "Parameter error", "operation_failed": "Operation failed", "select_at_least_one_receiver": "Please select at least one notification receiver", "select_hostname": "Please select hostname", "select_dependency": "Please select dependency", "host_in_use_cannot_delete": "Host is in use by tasks and cannot be deleted", "connection_failed": "Connection failed", "connection_success": "Connection successful", "get_task_detail_failed": "Failed to get task details", "manual_run": "Manual run", "task_started_check_log": "Task started, please check task log for results", "password_min_length_8": "Password must be at least 8 characters", "password_must_contain_letter_and_digit": "Password must contain both letters and digits", "account_locked": "Account locked, please try again in %d minutes", "login_failed_with_attempts": "Username or password is incorrect, %d attempts remaining", "rpc_unavailable": "Unable to connect to remote server", "rpc_timeout": "Execution timeout, forcibly terminated", "rpc_manual_stop": "Manually stopped", "version_not_found": "Version not found", "rollback_success": "Rollback successful", "rollback_failed": "Rollback failed", "template_name_exists": "Template name already exists", "template_not_found": "Template not found", "builtin_template_readonly": "Built-in template is read-only", "builtin_template_no_delete": "Built-in template cannot be deleted", "task_not_found": "Task not found", } ================================================ FILE: internal/modules/i18n/i18n.go ================================================ package i18n import ( "github.com/gin-gonic/gin" ) type Locale string const ( ZhCN Locale = "zh-CN" EnUS Locale = "en-US" ) var messages = map[Locale]map[string]string{ ZhCN: zhCN, EnUS: enUS, } func T(c *gin.Context, key string, args ...interface{}) string { locale := GetLocale(c) msg, ok := messages[locale][key] if !ok { msg = messages[ZhCN][key] if msg == "" { return key } } return msg } // Translate 不依赖gin.Context的翻译函数，默认使用中文 func Translate(key string) string { msg, ok := messages[ZhCN][key] if !ok { return key } return msg } func GetLocale(c *gin.Context) Locale { lang := c.GetHeader("Accept-Language") if lang == "" || lang == "zh-CN" || lang == "zh" { return ZhCN } return EnUS } ================================================ FILE: internal/modules/i18n/zh_cn.go ================================================ package i18n var zhCN = map[string]string{ "form_validation_failed": "表单验证失败, 请检测输入", "system_already_installed": "系统已安装!", "password_mismatch": "两次输入密码不匹配", "db_config_write_failed": "数据库配置写入文件失败", "app_config_read_failed": "读取应用配置失败", "create_table_failed": "创建数据库表失败", "create_admin_failed": "创建管理员账号失败", "create_lock_file_failed": "创建文件安装锁失败", "user_not_found": "用户不存在", "generate_2fa_key_failed": "生成2FA密钥失败", "generate_qrcode_failed": "生成二维码失败", "get_success": "获取成功", "verification_code_error": "验证码错误", "enable_failed": "启用失败", "2fa_enabled": "2FA已启用", "2fa_not_enabled": "2FA未启用", "disable_failed": "禁用失败", "2fa_disabled": "2FA已禁用", "update_success": "更新成功", "incomplete_parameters": "参数不完整", "operation_success": "操作成功", "username_exists": "用户名已存在", "email_exists": "邮箱已存在", "password_required": "请输入密码", "password_confirm_required": "请再次输入密码", "save_failed": "保存失败", "update_failed": "更新失败", "save_success": "保存成功", "password_same_as_old": "原密码与新密码不能相同", "old_password_error": "原密码输入错误", "username_password_empty": "用户名或密码不能为空", "username_password_error": "用户名或密码错误", "2fa_code_required": "需要输入2FA验证码", "2fa_code_error": "2FA验证码错误", "auth_failed": "认证失败", "page_not_found": "页面不存在", "app_not_installed": "应用未安装", "unauthorized": "无权访问", "api_key_required": "API密钥未配置", "param_time_required": "time参数必填", "param_time_invalid": "time参数已过期", "param_sign_required": "sign参数必填", "sign_verify_failed": "签名验证失败", "invalid_log_id": "参数错误: 无效的日志ID", "invalid_task_id": "参数错误: 无效的任务ID", "get_task_info_failed": "获取任务信息失败", "only_shell_task_can_stop": "仅支持SHELL任务手动停止", "task_node_list_empty": "任务节点列表为空", "stop_task_sent": "已执行停止操作, 请等待任务退出", "param_range_1_12": "参数取值范围1-12", "delete_failed": "删除失败", "delete_success": "删除成功", "http_task_timeout_max_300": "HTTP任务超时时间不能超过300秒", "crontab_parse_failed": "crontab表达式解析失败", "cannot_set_self_as_child": "不允许设置当前任务为子任务", "host_not_exist": "主机不存在", "refresh_task_host_failed": "刷新任务主机信息失败", "invalid_url": "请输入正确的URL地址", "hostname_exists": "主机名已存在", "task_name_exists": "任务名称已存在", "retry_times_range_0_10": "任务重试次数取值0-10", "retry_interval_range_0_3600": "任务重试间隔时间取值0-3600", "param_error": "参数错误", "operation_failed": "操作失败", "select_at_least_one_receiver": "至少选择一个通知接收者", "select_hostname": "请选择主机名", "select_dependency": "请选择依赖关系", "host_in_use_cannot_delete": "有任务引用此主机，不能删除", "connection_failed": "连接失败", "connection_success": "连接成功", "get_task_detail_failed": "获取任务详情失败", "manual_run": "手动运行", "task_started_check_log": "任务已开始运行, 请到任务日志中查看结果", "password_min_length_8": "密码长度至少8位", "password_must_contain_letter_and_digit": "密码必须包含字母和数字", "account_locked": "账户已被锁定，请在%d分钟后重试", "login_failed_with_attempts": "用户名或密码错误，还剩%d次尝试机会", "rpc_unavailable": "无法连接远程服务器", "rpc_timeout": "执行超时, 强制结束", "rpc_manual_stop": "手动停止", "version_not_found": "版本不存在", "rollback_success": "回滚成功", "rollback_failed": "回滚失败", "template_name_exists": "模板名称已存在", "template_not_found": "模板不存在", "builtin_template_readonly": "内置模板不可修改", "builtin_template_no_delete": "内置模板不可删除", "task_not_found": "任务不存在", } ================================================ FILE: internal/modules/leader/election.go ================================================ package leader import ( "fmt" "os" "sync" "sync/atomic" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/logger" "gorm.io/gorm" "gorm.io/gorm/clause" ) const ( // LockName 调度器锁的固定名称 LockName = "scheduler_leader" // LeaseDuration 租约时长，领导者需要在此时间内续约 LeaseDuration = 15 * time.Second // RenewInterval 续约间隔，必须小于 LeaseDuration RenewInterval = 5 * time.Second // RetryInterval 竞选失败后重试间隔 RetryInterval = 5 * time.Second ) // Election 基于数据库行锁的领导者选举 type Election struct { db *gorm.DB instanceID string // 当前实例标识 isLeader atomic.Bool stopCh chan struct{} stoppedCh chan struct{} onElected func() // 当选回调 onEvicted func() // 失去领导权回调 mu sync.Mutex } // New 创建选举实例 func New(db *gorm.DB, onElected, onEvicted func()) *Election { hostname, _ := os.Hostname() instanceID := fmt.Sprintf("%s:%d", hostname, os.Getpid()) return &Election{ db: db, instanceID: instanceID, stopCh: make(chan struct{}), stoppedCh: make(chan struct{}), onElected: onElected, onEvicted: onEvicted, } } // Start 开始参与选举（非阻塞） func (e *Election) Start() { go e.run() } // Stop 停止选举并释放领导权 func (e *Election) Stop() { close(e.stopCh) <-e.stoppedCh } // IsLeader 当前实例是否是领导者 func (e *Election) IsLeader() bool { return e.isLeader.Load() } // InstanceID 返回当前实例标识 func (e *Election) InstanceID() string { return e.instanceID } func (e *Election) run() { defer close(e.stoppedCh) // 确保锁表和初始记录存在 e.ensureLockRecord() for { select { case <-e.stopCh: e.releaseLock() return default: } if e.isLeader.Load() { // 已经是 leader，续约 if !e.renewLock() { logger.Warn("Leader lease renewal failed, stepping down") e.isLeader.Store(false) if e.onEvicted != nil { e.onEvicted() } } } else { // 尝试竞选 if e.tryAcquireLock() { logger.Infof("This node elected as leader: %s", e.instanceID) e.isLeader.Store(true) if e.onElected != nil { e.onElected() } } } // 等待下一次循环 interval := RetryInterval if e.isLeader.Load() { interval = RenewInterval } select { case <-e.stopCh: e.releaseLock() return case <-time.After(interval): } } } // ensureLockRecord 确保锁记录存在 func (e *Election) ensureLockRecord() { lock := models.SchedulerLock{ LockName: LockName, LockedBy: "", LockedAt: time.Time{}, ExpireAt: time.Time{}, } // 如果记录不存在则创建 e.db.Where("lock_name = ?", LockName).FirstOrCreate(&lock) } // tryAcquireLock 尝试获取锁（FOR UPDATE + 检查过期） func (e *Election) tryAcquireLock() bool { now := time.Now() result := e.db.Transaction(func(tx *gorm.DB) error { var lock models.SchedulerLock // SELECT ... FOR UPDATE 行锁 err := tx.Clauses(clause.Locking{Strength: "UPDATE"}). Where("lock_name = ?", LockName). First(&lock).Error if err != nil { return err } // 锁被其他实例持有且未过期 if lock.LockedBy != "" && lock.LockedBy != e.instanceID && lock.ExpireAt.After(now) { return fmt.Errorf("lock held by %s until %s", lock.LockedBy, lock.ExpireAt) } // 锁空闲或已过期，获取锁 err = tx.Model(&lock).Updates(map[string]interface{}{ "locked_by": e.instanceID, "locked_at": now, "expire_at": now.Add(LeaseDuration), "version": lock.Version + 1, }).Error return err }) return result == nil } // renewLock 续约（只有当前持有者才能续约） func (e *Election) renewLock() bool { now := time.Now() result := e.db.Model(&models.SchedulerLock{}). Where("lock_name = ? AND locked_by = ?", LockName, e.instanceID). Updates(map[string]interface{}{ "expire_at": now.Add(LeaseDuration), "locked_at": now, }) if result.Error != nil { logger.Errorf("Failed to renew leader lease: %v", result.Error) return false } return result.RowsAffected > 0 } // releaseLock 主动释放锁 func (e *Election) releaseLock() { if !e.isLeader.Load() { return } logger.Infof("Releasing leader lock: %s", e.instanceID) e.db.Model(&models.SchedulerLock{}). Where("lock_name = ? AND locked_by = ?", LockName, e.instanceID). Updates(map[string]interface{}{ "locked_by": "", "expire_at": time.Time{}, }) e.isLeader.Store(false) if e.onEvicted != nil { e.onEvicted() } } ================================================ FILE: internal/modules/leader/election_test.go ================================================ package leader import ( "os" "sync" "testing" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) func TestMain(m *testing.M) { logger.InitLogger() os.Exit(m.Run()) } func setupTestDB(t *testing.T) *gorm.DB { t.Helper() db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("open sqlite: %v", err) } // SQLite in-memory DB is per-connection; force single connection // so all goroutines share the same schema and data sqlDB, err := db.DB() if err != nil { t.Fatalf("get sql.DB: %v", err) } sqlDB.SetMaxOpenConns(1) if err := db.AutoMigrate(&models.SchedulerLock{}); err != nil { t.Fatalf("migrate: %v", err) } return db } func TestElection_SingleNode_BecomesLeader(t *testing.T) { db := setupTestDB(t) elected := make(chan struct{}, 1) e := New(db, func() { elected <- struct{}{} }, nil) e.Start() defer e.Stop() select { case <-elected: // ok case <-time.After(3 * time.Second): t.Fatal("timed out waiting to become leader") } if !e.IsLeader() { t.Error("expected IsLeader() to be true") } } func TestElection_Stop_ReleasesLock(t *testing.T) { db := setupTestDB(t) elected := make(chan struct{}, 1) evicted := make(chan struct{}, 1) e := New(db, func() { elected <- struct{}{} }, func() { evicted <- struct{}{} }) e.Start() <-elected e.Stop() if e.IsLeader() { t.Error("expected IsLeader() to be false after Stop") } // Verify lock is released in DB var lock models.SchedulerLock db.Where("lock_name = ?", LockName).First(&lock) if lock.LockedBy != "" { t.Errorf("expected empty locked_by after stop, got %q", lock.LockedBy) } } func TestElection_TwoNodes_OnlyOneLeader(t *testing.T) { db := setupTestDB(t) var mu sync.Mutex leaderCount := 0 makeElection := func() *Election { return New(db, func() { mu.Lock() leaderCount++ mu.Unlock() }, func() { mu.Lock() leaderCount-- mu.Unlock() }, ) } e1 := makeElection() e2 := makeElection() // Give them different instance IDs e1.instanceID = "node1:1000" e2.instanceID = "node2:2000" e1.Start() time.Sleep(2 * time.Second) e2.Start() time.Sleep(2 * time.Second) // Exactly one should be leader if e1.IsLeader() == e2.IsLeader() { t.Errorf("expected exactly one leader: e1=%v e2=%v", e1.IsLeader(), e2.IsLeader()) } mu.Lock() count := leaderCount mu.Unlock() if count != 1 { t.Errorf("expected leaderCount=1, got %d", count) } e1.Stop() e2.Stop() } func TestElection_Failover(t *testing.T) { db := setupTestDB(t) elected1 := make(chan struct{}, 1) e1 := New(db, func() { elected1 <- struct{}{} }, nil) e1.instanceID = "node1:1000" e1.Start() select { case <-elected1: case <-time.After(3 * time.Second): t.Fatal("e1 timed out becoming leader") } elected2 := make(chan struct{}, 1) e2 := New(db, func() { elected2 <- struct{}{} }, nil) e2.instanceID = "node2:2000" e2.Start() // e1 stops — e2 should take over e1.Stop() select { case <-elected2: // ok, e2 became leader case <-time.After(10 * time.Second): t.Fatal("e2 timed out becoming leader after e1 stopped") } if !e2.IsLeader() { t.Error("expected e2 to be leader after e1 stopped") } e2.Stop() } func TestElection_InstanceID(t *testing.T) { db := setupTestDB(t) e := New(db, nil, nil) if e.InstanceID() == "" { t.Error("expected non-empty InstanceID") } } func TestElection_EnsureLockRecord_CreatesRow(t *testing.T) { db := setupTestDB(t) e := New(db, nil, nil) // No rows initially var count int64 db.Model(&models.SchedulerLock{}).Count(&count) if count != 0 { t.Fatalf("expected 0 rows, got %d", count) } e.ensureLockRecord() db.Model(&models.SchedulerLock{}).Count(&count) if count != 1 { t.Fatalf("expected 1 row after ensureLockRecord, got %d", count) } // Calling again should not create duplicate e.ensureLockRecord() db.Model(&models.SchedulerLock{}).Count(&count) if count != 1 { t.Fatalf("expected still 1 row after second call, got %d", count) } } func TestElection_TryAcquireLock_ExpiredLock(t *testing.T) { db := setupTestDB(t) // Insert an expired lock held by another node expired := models.SchedulerLock{ LockName: LockName, LockedBy: "old-node:999", LockedAt: time.Now().Add(-1 * time.Hour), ExpireAt: time.Now().Add(-30 * time.Minute), // expired } db.Create(&expired) e := New(db, nil, nil) e.instanceID = "new-node:1000" // Should succeed because lock is expired if !e.tryAcquireLock() { t.Error("expected to acquire expired lock") } // Verify DB updated var lock models.SchedulerLock db.Where("lock_name = ?", LockName).First(&lock) if lock.LockedBy != "new-node:1000" { t.Errorf("expected locked_by=%q, got %q", "new-node:1000", lock.LockedBy) } } func TestElection_TryAcquireLock_ActiveLockBlocks(t *testing.T) { db := setupTestDB(t) // Insert an active lock held by another node active := models.SchedulerLock{ LockName: LockName, LockedBy: "other-node:999", LockedAt: time.Now(), ExpireAt: time.Now().Add(1 * time.Hour), // not expired } db.Create(&active) e := New(db, nil, nil) e.instanceID = "my-node:1000" // Should fail because lock is active if e.tryAcquireLock() { t.Error("expected to fail acquiring active lock") } } func TestElection_RenewLock_Success(t *testing.T) { db := setupTestDB(t) lock := models.SchedulerLock{ LockName: LockName, LockedBy: "my-node:1000", LockedAt: time.Now(), ExpireAt: time.Now().Add(10 * time.Second), } db.Create(&lock) e := New(db, nil, nil) e.instanceID = "my-node:1000" if !e.renewLock() { t.Error("expected renewLock to succeed") } var updated models.SchedulerLock db.Where("lock_name = ?", LockName).First(&updated) if updated.ExpireAt.Before(lock.ExpireAt) { t.Error("expected expire_at to be extended") } } func TestElection_RenewLock_FailsWhenNotOwner(t *testing.T) { db := setupTestDB(t) lock := models.SchedulerLock{ LockName: LockName, LockedBy: "other-node:999", LockedAt: time.Now(), ExpireAt: time.Now().Add(10 * time.Second), } db.Create(&lock) e := New(db, nil, nil) e.instanceID = "my-node:1000" if e.renewLock() { t.Error("expected renewLock to fail when not owner") } } func TestElection_ReleaseLock_OnlyWhenLeader(t *testing.T) { db := setupTestDB(t) lock := models.SchedulerLock{ LockName: LockName, LockedBy: "other-node:999", LockedAt: time.Now(), ExpireAt: time.Now().Add(1 * time.Hour), } db.Create(&lock) e := New(db, nil, nil) e.instanceID = "my-node:1000" // isLeader is false by default e.releaseLock() // should be a no-op var result models.SchedulerLock db.Where("lock_name = ?", LockName).First(&result) if result.LockedBy != "other-node:999" { t.Errorf("expected lock still held by other-node, got %q", result.LockedBy) } } func TestElection_NilCallbacks(t *testing.T) { db := setupTestDB(t) // Should not panic with nil onElected/onEvicted e := New(db, nil, nil) e.Start() time.Sleep(1 * time.Second) if !e.IsLeader() { t.Error("expected to become leader") } e.Stop() if e.IsLeader() { t.Error("expected not to be leader after stop") } } func TestElection_ReacquireOwnLock(t *testing.T) { db := setupTestDB(t) // Lock held by the same instance (e.g. after restart with same hostname:pid) lock := models.SchedulerLock{ LockName: LockName, LockedBy: "my-node:1000", LockedAt: time.Now(), ExpireAt: time.Now().Add(1 * time.Hour), } db.Create(&lock) e := New(db, nil, nil) e.instanceID = "my-node:1000" // Should succeed — same instance can reacquire if !e.tryAcquireLock() { t.Error("expected to reacquire own lock") } } ================================================ FILE: internal/modules/logger/async_logger.go ================================================ package logger import ( "context" "io" "log/slog" "sync" "time" ) // 异步日志批处理器 type asyncHandler struct { handler slog.Handler logChan chan *logRecord batchSize int flushTime time.Duration wg sync.WaitGroup once sync.Once } type logRecord struct { record slog.Record } // 创建异步处理器 func newAsyncHandler(writer io.Writer, batchSize int, flushTime time.Duration) *asyncHandler { h := &asyncHandler{ handler: slog.NewTextHandler(writer, &slog.HandlerOptions{Level: slog.LevelDebug}), logChan: make(chan *logRecord, batchSize*2), batchSize: batchSize, flushTime: flushTime, } h.wg.Add(1) go h.worker() return h } // 后台批量写入 func (h *asyncHandler) worker() { defer h.wg.Done() batch := make([]*logRecord, 0, h.batchSize) ticker := time.NewTicker(h.flushTime) defer ticker.Stop() ctx := context.Background() flush := func() { if len(batch) == 0 { return } for _, rec := range batch { _ = h.handler.Handle(ctx, rec.record) } batch = batch[:0] } for { select { case record, ok := <-h.logChan: if !ok { flush() return } batch = append(batch, record) if len(batch) >= h.batchSize { flush() } case <-ticker.C: flush() } } } // 写入日志（非阻塞） func (h *asyncHandler) log(level slog.Level, msg string, args ...any) { rec := slog.NewRecord(time.Now(), level, msg, 0) // 使用对象池减少分配 logRec := &logRecord{record: rec} select { case h.logChan <- logRec: default: // 队列满时直接同步写入（降级策略） _ = h.handler.Handle(context.Background(), rec) } } // 优雅关闭 func (h *asyncHandler) close() { h.once.Do(func() { close(h.logChan) h.wg.Wait() }) } ================================================ FILE: internal/modules/logger/async_logger_test.go ================================================ package logger import ( "bytes" "context" "log/slog" "strings" "testing" "time" ) func TestAsyncLoggerPerformance(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 50, 50*time.Millisecond) defer handler.close() // 写入1000条日志 start := time.Now() for i := 0; i < 1000; i++ { handler.log(slog.LevelInfo, "test message") } handler.close() elapsed := time.Since(start) t.Logf("写入1000条日志耗时: %v", elapsed) // 验证日志已写入 if buf.Len() == 0 { t.Fatal("日志未写入") } } func TestAsyncLoggerBatchFlush(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 10, 100*time.Millisecond) defer handler.close() // 写入5条日志（小于批量大小） for i := 0; i < 5; i++ { handler.log(slog.LevelInfo, "test") } // 等待定时刷新 time.Sleep(150 * time.Millisecond) handler.close() // 验证日志已刷新 if buf.Len() == 0 { t.Fatal("定时刷新失败") } } func TestAsyncLoggerFullBatch(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 10, 1*time.Second) // 写入10条日志（等于批量大小） for i := 0; i < 10; i++ { handler.log(slog.LevelInfo, "test") } // close 会等待 worker 完成所有写入 handler.close() // 验证日志已写入 if buf.Len() == 0 { t.Fatal("批量写入失败") } } func TestAsyncLoggerClose(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 100, 1*time.Second) // 写入日志后立即关闭 handler.log(slog.LevelInfo, "test message") handler.close() // 验证关闭时刷新了所有日志 if !strings.Contains(buf.String(), "test message") { t.Fatal("关闭时未刷新日志") } } // 性能对比测试 func BenchmarkSyncLogger(b *testing.B) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) b.ResetTimer() for i := 0; i < b.N; i++ { _ = handler.Handle(context.Background(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) } } func BenchmarkAsyncLogger(b *testing.B) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 50, 100*time.Millisecond) defer handler.close() b.ResetTimer() for i := 0; i < b.N; i++ { handler.log(slog.LevelInfo, "test") } } ================================================ FILE: internal/modules/logger/compatibility_test.go ================================================ package logger import ( "bytes" "log/slog" "strings" "testing" "time" "github.com/gin-gonic/gin" ) // 兼容性测试：确保API接口不变 func TestAPICompatibility(t *testing.T) { gin.SetMode(gin.ReleaseMode) // 初始化logger var buf bytes.Buffer handler := newAsyncHandler(&buf, 10, 100*time.Millisecond) prevAsync := asyncLogWriter asyncLogWriter = handler t.Cleanup(func() { handler.close() asyncLogWriter = prevAsync }) // 测试所有公开API是否正常工作 t.Run("Info接口", func(t *testing.T) { Info("test") Infof("test %s", "format") }) t.Run("Error接口", func(t *testing.T) { Error("test") Errorf("test %s", "format") }) t.Run("Warn接口", func(t *testing.T) { Warn("test") Warnf("test %s", "format") }) t.Run("Debug接口", func(t *testing.T) { Debug("test") Debugf("test %s", "format") }) } // 测试日志输出格式不变 func TestLogFormatCompatibility(t *testing.T) { prevLogger := logger prevAsync := asyncLogWriter // 使用同步logger测试格式 handler := newRecordingHandler() logger = slog.New(handler) asyncLogWriter = nil t.Cleanup(func() { logger = prevLogger asyncLogWriter = prevAsync }) Info("test message") if len(handler.entries) != 1 { t.Fatalf("expected 1 log entry, got %d", len(handler.entries)) } if handler.entries[0].msg != "test message" { t.Errorf("expected 'test message', got '%s'", handler.entries[0].msg) } } // 测试降级策略：异步失败时自动降级到同步 func TestFallbackToSync(t *testing.T) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) prevLogger := logger prevAsync := asyncLogWriter logger = slog.New(handler) asyncLogWriter = nil // 模拟异步不可用 t.Cleanup(func() { logger = prevLogger asyncLogWriter = prevAsync }) Info("fallback test") if !strings.Contains(buf.String(), "fallback test") { t.Error("降级到同步日志失败") } } // 测试Close方法幂等性 func TestCloseIdempotent(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 10, 100*time.Millisecond) handler.log(slog.LevelInfo, "test") // 多次调用Close应该安全 handler.close() handler.close() handler.close() // 验证日志已写入 if buf.Len() == 0 { t.Error("日志未写入") } } ================================================ FILE: internal/modules/logger/logger.go ================================================ package logger import ( "fmt" "io" "log/slog" "os" "path/filepath" "runtime" "time" "github.com/gin-gonic/gin" ) // 日志库 type Level int8 var ( logger *slog.Logger asyncLogWriter *asyncHandler exitFunc = os.Exit ) const ( DEBUG = iota INFO WARN ERROR FATAL ) func InitLogger() { logDir := "log" if err := os.MkdirAll(logDir, 0755); err != nil { panic(err) } logFile := filepath.Join(logDir, "cron.log") file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { panic(err) } writer := io.MultiWriter(os.Stdout, file) // 使用异步处理器：批量大小50，刷新间隔100ms asyncLogWriter = newAsyncHandler(writer, 50, 100*time.Millisecond) handler := slog.NewTextHandler(writer, &slog.HandlerOptions{ Level: slog.LevelDebug, }) logger = slog.New(handler) } // 优雅关闭 func Close() { if asyncLogWriter != nil { asyncLogWriter.close() } } func Debug(v ...interface{}) { if gin.Mode() != gin.DebugMode { return } write(DEBUG, v...) } func Debugf(format string, v ...interface{}) { if gin.Mode() != gin.DebugMode { return } writef(DEBUG, format, v...) } func Info(v ...interface{}) { write(INFO, v...) } func Infof(format string, v ...interface{}) { writef(INFO, format, v...) } func Warn(v ...interface{}) { write(WARN, v...) } func Warnf(format string, v ...interface{}) { writef(WARN, format, v...) } func Error(v ...interface{}) { write(ERROR, v...) } func Errorf(format string, v ...interface{}) { writef(ERROR, format, v...) } func Fatal(v ...interface{}) { write(FATAL, v...) } func Fatalf(format string, v ...interface{}) { writef(FATAL, format, v...) } func write(level Level, v ...interface{}) { msg := fmt.Sprint(v...) args := []any{} if gin.Mode() == gin.DebugMode { pc, file, line, ok := runtime.Caller(2) if ok { args = append(args, "file", file, "func", runtime.FuncForPC(pc).Name(), "line", line) } } // 使用异步写入 if asyncLogWriter != nil { switch level { case DEBUG: asyncLogWriter.log(slog.LevelDebug, msg, args...) case INFO: asyncLogWriter.log(slog.LevelInfo, msg, args...) case WARN: asyncLogWriter.log(slog.LevelWarn, msg, args...) case ERROR: asyncLogWriter.log(slog.LevelError, msg, args...) case FATAL: asyncLogWriter.log(slog.LevelError, msg, args...) asyncLogWriter.close() exitFunc(1) } return } // 降级到同步写入 switch level { case DEBUG: logger.Debug(msg, args...) case INFO: logger.Info(msg, args...) case WARN: logger.Warn(msg, args...) case FATAL: logger.Error(msg, args...) exitFunc(1) case ERROR: logger.Error(msg, args...) } } func writef(level Level, format string, v ...interface{}) { msg := fmt.Sprintf(format, v...) args := []any{} if gin.Mode() == gin.DebugMode { pc, file, line, ok := runtime.Caller(2) if ok { args = append(args, "file", file, "func", runtime.FuncForPC(pc).Name(), "line", line) } } // 使用异步写入 if asyncLogWriter != nil { switch level { case DEBUG: asyncLogWriter.log(slog.LevelDebug, msg, args...) case INFO: asyncLogWriter.log(slog.LevelInfo, msg, args...) case WARN: asyncLogWriter.log(slog.LevelWarn, msg, args...) case ERROR: asyncLogWriter.log(slog.LevelError, msg, args...) case FATAL: asyncLogWriter.log(slog.LevelError, msg, args...) asyncLogWriter.close() exitFunc(1) } return } // 降级到同步写入 switch level { case DEBUG: logger.Debug(msg, args...) case INFO: logger.Info(msg, args...) case WARN: logger.Warn(msg, args...) case FATAL: logger.Error(msg, args...) exitFunc(1) case ERROR: logger.Error(msg, args...) } } ================================================ FILE: internal/modules/logger/logger_test.go ================================================ package logger import ( "bytes" "context" "log/slog" "testing" "github.com/gin-gonic/gin" ) type logEntry struct { level slog.Level msg string } type recordingHandler struct { entries []logEntry } func newRecordingHandler() *recordingHandler { return &recordingHandler{} } func (r *recordingHandler) Enabled(_ context.Context, level slog.Level) bool { return true } func (r *recordingHandler) Handle(_ context.Context, record slog.Record) error { r.entries = append(r.entries, logEntry{level: record.Level, msg: record.Message}) return nil } func (r *recordingHandler) WithAttrs(attrs []slog.Attr) slog.Handler { return r } func (r *recordingHandler) WithGroup(name string) slog.Handler { return r } func setupRecordingLogger(t *testing.T) *recordingHandler { t.Helper() prevLogger := logger rec := newRecordingHandler() logger = slog.New(rec) t.Cleanup(func() { logger = prevLogger }) return rec } func TestDebugLoggingDependsOnGinMode(t *testing.T) { gin.SetMode(gin.ReleaseMode) rec := setupRecordingLogger(t) Debug("release-mode") if hasLevel(rec.entries, slog.LevelDebug) { t.Fatalf("expected no debug entry in release mode, got %+v", rec.entries) } gin.SetMode(gin.DebugMode) rec = setupRecordingLogger(t) Debug("debug-mode") if !hasLevel(rec.entries, slog.LevelDebug) { t.Fatalf("expected debug entry in debug mode, got %+v", rec.entries) } } func TestInfoLogsAndFlushes(t *testing.T) { gin.SetMode(gin.ReleaseMode) rec := setupRecordingLogger(t) Info("info-message") if !hasLevel(rec.entries, slog.LevelInfo) { t.Fatalf("expected info entry, got %+v", rec.entries) } } func TestFatalLogsAndInvokesExit(t *testing.T) { gin.SetMode(gin.ReleaseMode) rec := setupRecordingLogger(t) prevExit := exitFunc exitCalled := 0 exitCode := 0 exitFunc = func(code int) { exitCalled++ exitCode = code } t.Cleanup(func() { exitFunc = prevExit }) Fatal("fatal-message") if !hasLevel(rec.entries, slog.LevelError) { t.Fatalf("expected error entry, got %+v", rec.entries) } if exitCalled != 1 || exitCode != 1 { t.Fatalf("expected exitFunc to be called once with code 1, got count=%d code=%d", exitCalled, exitCode) } } func TestInitLogger(t *testing.T) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) logger = slog.New(handler) Info("test-message") if buf.Len() == 0 { t.Fatal("expected log output") } } func hasLevel(entries []logEntry, level slog.Level) bool { for _, entry := range entries { if entry.level == level { return true } } return false } ================================================ FILE: internal/modules/logger/performance_report_test.go ================================================ package logger import ( "bytes" "context" "fmt" "log/slog" "sync" "testing" "time" ) // 性能对比报告 func TestPerformanceReport(t *testing.T) { fmt.Println("") fmt.Println("========================================") fmt.Println("日志性能优化对比报告") fmt.Println("========================================") fmt.Println("") // 测试1: 单线程顺序写入 t.Run("1.单线程顺序写入1000条", func(t *testing.T) { count := 1000 // 同步 var buf1 bytes.Buffer handler1 := slog.NewTextHandler(&buf1, nil) start1 := time.Now() for i := 0; i < count; i++ { _ = handler1.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) } sync1 := time.Since(start1) // 异步 var buf2 bytes.Buffer handler2 := newAsyncHandler(&buf2, 50, 100*time.Millisecond) start2 := time.Now() for i := 0; i < count; i++ { handler2.log(slog.LevelInfo, "test") } handler2.close() async := time.Since(start2) improvement := float64(sync1-async) / float64(sync1) * 100 fmt.Printf(" 同步: %v\n", sync1) fmt.Printf(" 异步: %v\n", async) fmt.Printf(" 提升: %.1f%%\n\n", improvement) }) // 测试2: 高并发场景 t.Run("2.100个goroutine并发写入", func(t *testing.T) { goroutines := 100 logsPerGoroutine := 100 // 同步 var buf1 bytes.Buffer handler1 := slog.NewTextHandler(&buf1, nil) start1 := time.Now() var wg1 sync.WaitGroup for i := 0; i < goroutines; i++ { wg1.Add(1) go func() { defer wg1.Done() for j := 0; j < logsPerGoroutine; j++ { _ = handler1.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) } }() } wg1.Wait() sync1 := time.Since(start1) // 异步 var buf2 bytes.Buffer handler2 := newAsyncHandler(&buf2, 50, 100*time.Millisecond) start2 := time.Now() var wg2 sync.WaitGroup for i := 0; i < goroutines; i++ { wg2.Add(1) go func() { defer wg2.Done() for j := 0; j < logsPerGoroutine; j++ { handler2.log(slog.LevelInfo, "test") } }() } wg2.Wait() handler2.close() async := time.Since(start2) improvement := float64(sync1-async) / float64(sync1) * 100 fmt.Printf(" 同步: %v\n", sync1) fmt.Printf(" 异步: %v\n", async) fmt.Printf(" 提升: %.1f%%\n\n", improvement) }) // 测试3: 模拟真实任务场景 t.Run("3.模拟50个任务执行(每任务20条日志)", func(t *testing.T) { tasks := 50 logsPerTask := 20 // 同步 var buf1 bytes.Buffer handler1 := slog.NewTextHandler(&buf1, nil) start1 := time.Now() var wg1 sync.WaitGroup for i := 0; i < tasks; i++ { wg1.Add(1) go func(taskID int) { defer wg1.Done() // 模拟任务执行 for j := 0; j < logsPerTask; j++ { _ = handler1.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, fmt.Sprintf("Task %d executing step %d", taskID, j), 0)) time.Sleep(10 * time.Microsecond) // 模拟任务处理 } }(i) } wg1.Wait() sync1 := time.Since(start1) // 异步 var buf2 bytes.Buffer handler2 := newAsyncHandler(&buf2, 50, 100*time.Millisecond) start2 := time.Now() var wg2 sync.WaitGroup for i := 0; i < tasks; i++ { wg2.Add(1) go func(taskID int) { defer wg2.Done() // 模拟任务执行 for j := 0; j < logsPerTask; j++ { handler2.log(slog.LevelInfo, fmt.Sprintf("Task %d executing step %d", taskID, j)) time.Sleep(10 * time.Microsecond) // 模拟任务处理 } }(i) } wg2.Wait() handler2.close() async := time.Since(start2) improvement := float64(sync1-async) / float64(sync1) * 100 fmt.Printf(" 同步: %v\n", sync1) fmt.Printf(" 异步: %v\n", async) fmt.Printf(" 提升: %.1f%%\n\n", improvement) }) // 测试4: 批量写入效率 t.Run("4.批量写入效率测试", func(t *testing.T) { count := 5000 // 同步 - 每次都写入 var buf1 bytes.Buffer handler1 := slog.NewTextHandler(&buf1, nil) start1 := time.Now() for i := 0; i < count; i++ { _ = handler1.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) } sync1 := time.Since(start1) // 异步 - 批量写入 var buf2 bytes.Buffer handler2 := newAsyncHandler(&buf2, 50, 100*time.Millisecond) start2 := time.Now() for i := 0; i < count; i++ { handler2.log(slog.LevelInfo, "test") } handler2.close() async := time.Since(start2) improvement := float64(sync1-async) / float64(sync1) * 100 fmt.Printf(" 同步: %v (每次写入)\n", sync1) fmt.Printf(" 异步: %v (批量写入)\n", async) fmt.Printf(" 提升: %.1f%%\n\n", improvement) }) fmt.Println("========================================") fmt.Println("优化总结:") fmt.Println("1. 异步日志不阻塞业务逻辑") fmt.Println("2. 批量写入减少I/O次数") fmt.Println("3. 高并发场景性能提升明显") fmt.Println("4. 内存开销可控(channel缓冲)") fmt.Println("========================================") fmt.Println("") } ================================================ FILE: internal/modules/logger/performance_test.go ================================================ package logger import ( "bytes" "context" "io" "log/slog" "sync" "testing" "time" ) // 高并发场景测试 func BenchmarkConcurrentSync(b *testing.B) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) b.ResetTimer() b.RunParallel(func(pb *testing.PB) { for pb.Next() { _ = handler.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) } }) } func BenchmarkConcurrentAsync(b *testing.B) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 50, 100*time.Millisecond) defer handler.close() b.ResetTimer() b.RunParallel(func(pb *testing.PB) { for pb.Next() { handler.log(slog.LevelInfo, "test") } }) } // 真实场景：模拟任务执行中的日志写入 func TestRealWorldScenario(t *testing.T) { tests := []struct { name string taskNum int logsPerTask int }{ {"10任务x10日志", 10, 10}, {"100任务x10日志", 100, 10}, {"100任务x100日志", 100, 100}, } for _, tt := range tests { t.Run(tt.name+"-同步", func(t *testing.T) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) start := time.Now() var wg sync.WaitGroup for i := 0; i < tt.taskNum; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < tt.logsPerTask; j++ { _ = handler.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "task executing", 0)) } }() } wg.Wait() elapsed := time.Since(start) t.Logf("同步日志耗时: %v", elapsed) }) t.Run(tt.name+"-异步", func(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 50, 100*time.Millisecond) defer handler.close() start := time.Now() var wg sync.WaitGroup for i := 0; i < tt.taskNum; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < tt.logsPerTask; j++ { handler.log(slog.LevelInfo, "task executing") } }() } wg.Wait() elapsed := time.Since(start) t.Logf("异步日志耗时: %v", elapsed) }) } } // 吞吐量测试 func TestThroughput(t *testing.T) { duration := 1 * time.Second t.Run("同步吞吐量", func(t *testing.T) { var buf bytes.Buffer handler := slog.NewTextHandler(&buf, nil) count := 0 done := make(chan bool) go func() { time.Sleep(duration) done <- true }() for { select { case <-done: t.Logf("同步日志 1秒内写入: %d 条", count) return default: _ = handler.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test", 0)) count++ } } }) t.Run("异步吞吐量", func(t *testing.T) { var buf bytes.Buffer handler := newAsyncHandler(&buf, 50, 100*time.Millisecond) defer handler.close() count := 0 done := make(chan bool) go func() { time.Sleep(duration) done <- true }() for { select { case <-done: t.Logf("异步日志 1秒内写入: %d 条", count) return default: handler.log(slog.LevelInfo, "test") count++ } } }) } // 测试写入真实文件的性能差异 func BenchmarkRealFileSync(b *testing.B) { writer := io.Discard handler := slog.NewTextHandler(writer, nil) b.ResetTimer() for i := 0; i < b.N; i++ { _ = handler.Handle(context.TODO(), slog.NewRecord(time.Now(), slog.LevelInfo, "test message", 0)) } } func BenchmarkRealFileAsync(b *testing.B) { writer := io.Discard handler := newAsyncHandler(writer, 50, 100*time.Millisecond) defer handler.close() b.ResetTimer() for i := 0; i < b.N; i++ { handler.log(slog.LevelInfo, "test message") } } ================================================ FILE: internal/modules/notify/mail.go ================================================ package notify import ( "strconv" "strings" "time" "github.com/go-gomail/gomail" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" ) // @author qiang.ou // @date 2017/5/1-00:19 type Mail struct { } func (mail *Mail) Send(msg Message) { model := new(models.Setting) mailSetting, err := model.Mail() logger.Debugf("%+v", mailSetting) if err != nil { logger.Error("#mail#从数据库获取mail配置失败", err) return } if mailSetting.Host == "" { logger.Error("#mail#Host为空") return } if mailSetting.Port == 0 { logger.Error("#mail#Port为空") return } if mailSetting.User == "" { logger.Error("#mail#User为空") return } if mailSetting.Password == "" { logger.Error("#mail#Password为空") return } msg["content"] = parseNotifyTemplate(mailSetting.Template, msg) toUsers := mail.getActiveMailUsers(mailSetting, msg) mail.send(mailSetting, toUsers, msg) } func (mail *Mail) send(mailSetting models.Mail, toUsers []string, msg Message) { body := msg["content"].(string) body = strings.Replace(body, "\n", "
", -1) gomailMessage := gomail.NewMessage() gomailMessage.SetHeader("From", mailSetting.User) gomailMessage.SetHeader("To", toUsers...) gomailMessage.SetHeader("Subject", "gocron-定时任务通知") gomailMessage.SetBody("text/html", body) mailer := gomail.NewDialer(mailSetting.Host, mailSetting.Port, mailSetting.User, mailSetting.Password) maxTimes := 3 i := 0 for i < maxTimes { err := mailer.DialAndSend(gomailMessage) if err == nil { break } i += 1 time.Sleep(2 * time.Second) if i < maxTimes { logger.Errorf("mail#发送消息失败#%s#消息内容-%s", err.Error(), msg["content"]) } } } func (mail *Mail) getActiveMailUsers(mailSetting models.Mail, msg Message) []string { taskReceiverIds := strings.Split(msg["task_receiver_id"].(string), ",") users := []string{} for _, v := range mailSetting.MailUsers { if utils.InStringSlice(taskReceiverIds, strconv.Itoa(v.Id)) { users = append(users, v.Email) } } return users } ================================================ FILE: internal/modules/notify/notify.go ================================================ package notify import ( "bytes" "fmt" "html/template" "time" "github.com/gocronx-team/gocron/internal/modules/logger" ) type Message map[string]interface{} type Notifiable interface { Send(msg Message) } var queue = make(chan Message, 100) func init() { go run() } // 把消息推入队列 func Push(msg Message) { queue <- msg } func run() { for msg := range queue { // 根据任务配置发送通知 taskType, taskTypeOk := msg["task_type"] _, taskReceiverIdOk := msg["task_receiver_id"] _, nameOk := msg["name"] _, outputOk := msg["output"] _, statusOk := msg["status"] if !taskTypeOk || !taskReceiverIdOk || !nameOk || !outputOk || !statusOk { logger.Errorf("#notify#参数不完整#%+v", msg) continue } msg["content"] = fmt.Sprintf("============\n============\n============\n任务名称: %s\n状态: %s\n输出:\n %s\n", msg["name"], msg["status"], msg["output"]) logger.Debugf("%+v", msg) switch taskType.(int8) { case 0: // 邮件 mail := Mail{} go mail.Send(msg) case 1: // Slack slack := Slack{} go slack.Send(msg) case 2: // WebHook webHook := WebHook{} go webHook.Send(msg) } time.Sleep(1 * time.Second) } } func parseNotifyTemplate(notifyTemplate string, msg Message) string { tmpl, err := template.New("notify").Parse(notifyTemplate) if err != nil { return fmt.Sprintf("解析通知模板失败: %s", err) } var buf bytes.Buffer if err := tmpl.Execute(&buf, map[string]interface{}{ "TaskId": msg["task_id"], "TaskName": msg["name"], "Status": msg["status"], "Result": msg["output"], "Remark": msg["remark"], }); err != nil { return fmt.Sprintf("执行模板失败: %s", err) } return buf.String() } ================================================ FILE: internal/modules/notify/notify_test.go ================================================ package notify import ( "testing" ) // TestNotifyDispatch 测试通知分发逻辑 func TestNotifyDispatch(t *testing.T) { tests := []struct { name string msg Message expectError bool description string }{ { name: "邮件通知-完整参数", msg: Message{ "task_type": int8(1), "task_receiver_id": "1,2", "name": "测试任务", "output": "任务执行成功", "status": "成功", "task_id": 123, }, expectError: false, description: "邮件通知应该正常处理", }, { name: "Slack通知-完整参数", msg: Message{ "task_type": int8(2), "task_receiver_id": "1", "name": "测试任务", "output": "任务执行失败", "status": "失败", "task_id": 456, }, expectError: false, description: "Slack通知应该正常处理", }, { name: "Webhook通知-完整参数", msg: Message{ "task_type": int8(3), "task_receiver_id": "1,2,3", "name": "测试任务", "output": "任务执行成功", "status": "成功", "task_id": 789, }, expectError: false, description: "Webhook通知应该正常处理", }, { name: "缺少task_type", msg: Message{ "task_receiver_id": "1", "name": "测试任务", "output": "输出", "status": "成功", }, expectError: true, description: "缺少task_type应该被拒绝", }, { name: "缺少task_receiver_id", msg: Message{ "task_type": int8(1), "name": "测试任务", "output": "输出", "status": "成功", }, expectError: true, description: "缺少task_receiver_id应该被拒绝", }, { name: "缺少name", msg: Message{ "task_type": int8(1), "task_receiver_id": "1", "output": "输出", "status": "成功", }, expectError: true, description: "缺少name应该被拒绝", }, { name: "缺少output", msg: Message{ "task_type": int8(1), "task_receiver_id": "1", "name": "测试任务", "status": "成功", }, expectError: true, description: "缺少output应该被拒绝", }, { name: "缺少status", msg: Message{ "task_type": int8(1), "task_receiver_id": "1", "name": "测试任务", "output": "输出", }, expectError: true, description: "缺少status应该被拒绝", }, { name: "无效的task_type", msg: Message{ "task_type": int8(99), "task_receiver_id": "1", "name": "测试任务", "output": "输出", "status": "成功", }, expectError: false, description: "无效的task_type会被忽略但不报错", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { // 验证消息参数完整性 _, taskTypeOk := tt.msg["task_type"] _, taskReceiverIdOk := tt.msg["task_receiver_id"] _, nameOk := tt.msg["name"] _, outputOk := tt.msg["output"] _, statusOk := tt.msg["status"] hasError := !taskTypeOk || !taskReceiverIdOk || !nameOk || !outputOk || !statusOk if hasError != tt.expectError { t.Errorf("%s: expected error=%v, got error=%v", tt.description, tt.expectError, hasError) } // 验证task_type类型 if taskTypeOk { if _, ok := tt.msg["task_type"].(int8); !ok { t.Errorf("task_type should be int8") } } }) } } // TestParseNotifyTemplate 测试通知模板解析 func TestParseNotifyTemplate(t *testing.T) { tests := []struct { name string template string msg Message contains []string }{ { name: "基础模板", template: "任务: {{.TaskName}}, 状态: {{.Status}}", msg: Message{ "task_id": 1, "name": "测试任务", "status": "成功", "output": "执行结果", "remark": "备注", }, contains: []string{"测试任务", "成功"}, }, { name: "完整模板", template: "任务ID: {{.TaskId}}\n任务名称: {{.TaskName}}\n状态: {{.Status}}\n结果: {{.Result}}\n备注: {{.Remark}}", msg: Message{ "task_id": 123, "name": "定时任务", "status": "失败", "output": "错误信息", "remark": "重要任务", }, contains: []string{"123", "定时任务", "失败", "错误信息", "重要任务"}, }, { name: "空模板", template: "", msg: Message{ "task_id": 1, "name": "任务", "status": "成功", "output": "输出", "remark": "备注", }, contains: []string{}, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { result := parseNotifyTemplate(tt.template, tt.msg) for _, expected := range tt.contains { if len(expected) > 0 && !contains(result, expected) { t.Errorf("expected result to contain '%s', got: %s", expected, result) } } }) } } // TestNotifyTypeValues 测试通知类型常量 func TestNotifyTypeValues(t *testing.T) { tests := []struct { name string typeVal int8 typeName string }{ {"邮件通知", 1, "Mail"}, {"Slack通知", 2, "Slack"}, {"Webhook通知", 3, "Webhook"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { msg := Message{ "task_type": tt.typeVal, "task_receiver_id": "1", "name": "test", "output": "output", "status": "success", } taskType, ok := msg["task_type"].(int8) if !ok { t.Errorf("task_type should be int8") } if taskType != tt.typeVal { t.Errorf("expected task_type=%d, got %d", tt.typeVal, taskType) } }) } } // 辅助函数：检查字符串是否包含子串 func contains(s, substr string) bool { return len(substr) == 0 || len(s) >= len(substr) && (s == substr || len(s) > len(substr) && containsHelper(s, substr)) } func containsHelper(s, substr string) bool { for i := 0; i <= len(s)-len(substr); i++ { if s[i:i+len(substr)] == substr { return true } } return false } ================================================ FILE: internal/modules/notify/slack.go ================================================ package notify // 发送消息到slack import ( "fmt" "html" "strconv" "strings" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/httpclient" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" ) type Slack struct{} func (slack *Slack) Send(msg Message) { model := new(models.Setting) slackSetting, err := model.Slack() if err != nil { logger.Error("#slack#从数据库获取slack配置失败", err) return } if slackSetting.Url == "" { logger.Error("#slack#webhook-url为空") return } if len(slackSetting.Channels) == 0 { logger.Error("#slack#channels配置为空") return } logger.Debugf("%+v", slackSetting) channels := slack.getActiveSlackChannels(slackSetting, msg) logger.Debugf("%+v", channels) msg["content"] = parseNotifyTemplate(slackSetting.Template, msg) msg["content"] = html.UnescapeString(msg["content"].(string)) for _, channel := range channels { slack.send(msg, slackSetting.Url, channel) } } func (slack *Slack) send(msg Message, slackUrl string, channel string) { formatBody := slack.format(msg["content"].(string), channel) timeout := 30 maxTimes := 3 i := 0 for i < maxTimes { resp := httpclient.PostJson(slackUrl, formatBody, timeout) if resp.StatusCode == 200 { break } i += 1 time.Sleep(2 * time.Second) if i < maxTimes { logger.Errorf("slack#发送消息失败#%s#消息内容-%s", resp.Body, msg["content"]) } } } func (slack *Slack) getActiveSlackChannels(slackSetting models.Slack, msg Message) []string { taskReceiverIds := strings.Split(msg["task_receiver_id"].(string), ",") channels := []string{} for _, v := range slackSetting.Channels { if utils.InStringSlice(taskReceiverIds, strconv.Itoa(v.Id)) { channels = append(channels, v.Name) } } return channels } // 格式化消息内容 func (slack *Slack) format(content string, channel string) string { content = utils.EscapeJson(content) specialChars := []string{"&", "<", ">"} replaceChars := []string{"&", "<", ">"} content = utils.ReplaceStrings(content, specialChars, replaceChars) return fmt.Sprintf(`{"text":"%s","username":"gocron", "channel":"%s"}`, content, channel) } ================================================ FILE: internal/modules/notify/webhook.go ================================================ package notify import ( "html" "strconv" "strings" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/httpclient" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" ) type WebHook struct{} func (webHook *WebHook) Send(msg Message) { model := new(models.Setting) webHookSetting, err := model.Webhook() if err != nil { logger.Error("#webHook#从数据库获取webHook配置失败", err) return } if len(webHookSetting.WebhookUrls) == 0 { logger.Error("#webHook#webhook地址列表为空") return } logger.Debugf("%+v", webHookSetting) msg["name"] = utils.EscapeJson(msg["name"].(string)) msg["output"] = utils.EscapeJson(msg["output"].(string)) msg["content"] = parseNotifyTemplate(webHookSetting.Template, msg) msg["content"] = html.UnescapeString(msg["content"].(string)) // 获取任务配置的接收者ID列表 activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) // 向所有激活的webhook地址发送 for _, webhookUrl := range activeUrls { go webHook.send(msg, webhookUrl.Url) } } func (webHook *WebHook) getActiveWebhookUrls(webHookSetting models.WebHook, msg Message) []models.WebhookUrl { taskReceiverIds := strings.Split(msg["task_receiver_id"].(string), ",") urls := []models.WebhookUrl{} for _, v := range webHookSetting.WebhookUrls { if utils.InStringSlice(taskReceiverIds, strconv.Itoa(v.Id)) { urls = append(urls, v) } } return urls } func (webHook *WebHook) send(msg Message, url string) { content := msg["content"].(string) timeout := 30 maxTimes := 3 i := 0 for i < maxTimes { resp := httpclient.PostJson(url, content, timeout) if resp.StatusCode == 200 { break } i += 1 time.Sleep(2 * time.Second) if i < maxTimes { logger.Errorf("webHook#发送消息失败#%s#消息内容-%s", resp.Body, msg["content"]) } } } ================================================ FILE: internal/modules/notify/webhook_test.go ================================================ package notify import ( "testing" "github.com/gocronx-team/gocron/internal/models" ) // TestWebHook_getActiveWebhookUrls 测试根据任务接收者ID筛选webhook地址 func TestWebHook_getActiveWebhookUrls(t *testing.T) { webHook := &WebHook{} tests := []struct { name string webhookUrls []models.WebhookUrl taskReceiverIds string expectedCount int expectedUrlNames []string }{ { name: "single receiver", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, {Id: 2, Name: "Webhook 2", Url: "https://webhook2.example.com"}, {Id: 3, Name: "Webhook 3", Url: "https://webhook3.example.com"}, }, taskReceiverIds: "2", expectedCount: 1, expectedUrlNames: []string{"Webhook 2"}, }, { name: "multiple receivers", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, {Id: 2, Name: "Webhook 2", Url: "https://webhook2.example.com"}, {Id: 3, Name: "Webhook 3", Url: "https://webhook3.example.com"}, }, taskReceiverIds: "1,3", expectedCount: 2, expectedUrlNames: []string{"Webhook 1", "Webhook 3"}, }, { name: "no matching receivers", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, {Id: 2, Name: "Webhook 2", Url: "https://webhook2.example.com"}, }, taskReceiverIds: "99", expectedCount: 0, expectedUrlNames: []string{}, }, { name: "empty receiver ids", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, }, taskReceiverIds: "", expectedCount: 0, expectedUrlNames: []string{}, }, { name: "all receivers", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, {Id: 2, Name: "Webhook 2", Url: "https://webhook2.example.com"}, }, taskReceiverIds: "1,2", expectedCount: 2, expectedUrlNames: []string{"Webhook 1", "Webhook 2"}, }, { name: "receiver ids with spaces", webhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, {Id: 2, Name: "Webhook 2", Url: "https://webhook2.example.com"}, }, taskReceiverIds: " 1 , 2 ", expectedCount: 2, expectedUrlNames: []string{"Webhook 1", "Webhook 2"}, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { webHookSetting := models.WebHook{ WebhookUrls: tt.webhookUrls, } msg := Message{ "task_receiver_id": tt.taskReceiverIds, } activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) if len(activeUrls) != tt.expectedCount { t.Errorf("expected %d active urls, got %d", tt.expectedCount, len(activeUrls)) } // 验证返回的webhook名称 foundNames := make(map[string]bool) for _, url := range activeUrls { foundNames[url.Name] = true } for _, expectedName := range tt.expectedUrlNames { if !foundNames[expectedName] { t.Errorf("expected to find webhook %s, but not found", expectedName) } } }) } } // TestWebHook_getActiveWebhookUrls_EdgeCases 测试边界情况 func TestWebHook_getActiveWebhookUrls_EdgeCases(t *testing.T) { webHook := &WebHook{} t.Run("empty webhook urls", func(t *testing.T) { webHookSetting := models.WebHook{ WebhookUrls: []models.WebhookUrl{}, } msg := Message{ "task_receiver_id": "1,2,3", } activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) if len(activeUrls) != 0 { t.Errorf("expected 0 active urls, got %d", len(activeUrls)) } }) t.Run("invalid receiver id format", func(t *testing.T) { webHookSetting := models.WebHook{ WebhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, }, } msg := Message{ "task_receiver_id": "abc,def", } activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) if len(activeUrls) != 0 { t.Errorf("expected 0 active urls for invalid ids, got %d", len(activeUrls)) } }) t.Run("duplicate receiver ids", func(t *testing.T) { webHookSetting := models.WebHook{ WebhookUrls: []models.WebhookUrl{ {Id: 1, Name: "Webhook 1", Url: "https://webhook1.example.com"}, }, } msg := Message{ "task_receiver_id": "1,1,1", } activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) // 实际实现：遍历webhookUrls，对每个URL检查其ID是否在receiverIds中 // 所以即使receiverIds有重复，每个webhook也只会被添加一次 if len(activeUrls) != 1 { t.Errorf("expected 1 active url (no duplicates in result), got %d", len(activeUrls)) } }) } // TestWebHook_getActiveWebhookUrls_LargeDataset 测试大数据集 func TestWebHook_getActiveWebhookUrls_LargeDataset(t *testing.T) { webHook := &WebHook{} // 创建100个webhook地址 webhookUrls := make([]models.WebhookUrl, 100) for i := 0; i < 100; i++ { webhookUrls[i] = models.WebhookUrl{ Id: i + 1, Name: "Webhook " + string(rune(i+1)), Url: "https://webhook.example.com/" + string(rune(i+1)), } } webHookSetting := models.WebHook{ WebhookUrls: webhookUrls, } // 选择前50个 receiverIds := "" for i := 1; i <= 50; i++ { if i > 1 { receiverIds += "," } receiverIds += string(rune(i + '0')) } msg := Message{ "task_receiver_id": receiverIds, } activeUrls := webHook.getActiveWebhookUrls(webHookSetting, msg) if len(activeUrls) == 0 { t.Error("expected some active urls, got 0") } } // BenchmarkWebHook_getActiveWebhookUrls 性能测试 func BenchmarkWebHook_getActiveWebhookUrls(b *testing.B) { webHook := &WebHook{} webhookUrls := make([]models.WebhookUrl, 10) for i := 0; i < 10; i++ { webhookUrls[i] = models.WebhookUrl{ Id: i + 1, Name: "Webhook", Url: "https://example.com", } } webHookSetting := models.WebHook{ WebhookUrls: webhookUrls, } msg := Message{ "task_receiver_id": "1,3,5,7,9", } b.ResetTimer() for i := 0; i < b.N; i++ { _ = webHook.getActiveWebhookUrls(webHookSetting, msg) } } // TestMessage_Type 测试Message类型 func TestMessage_Type(t *testing.T) { msg := Message{ "task_id": 123, "name": "test task", "output": "test output", "status": "success", "task_receiver_id": "1,2,3", } // 验证可以正确获取值 if taskId, ok := msg["task_id"].(int); !ok || taskId != 123 { t.Error("failed to get task_id") } if name, ok := msg["name"].(string); !ok || name != "test task" { t.Error("failed to get name") } if receiverId, ok := msg["task_receiver_id"].(string); !ok || receiverId != "1,2,3" { t.Error("failed to get task_receiver_id") } } ================================================ FILE: internal/modules/rpc/auth/Certification.go ================================================ package auth import ( "crypto/tls" "crypto/x509" "errors" "fmt" "os" "google.golang.org/grpc/credentials" ) type Certificate struct { CAFile string CertFile string KeyFile string ServerName string } func (c Certificate) GetTLSConfigForServer() (*tls.Config, error) { certificate, err := tls.LoadX509KeyPair( c.CertFile, c.KeyFile, ) if err != nil { return nil, err } certPool := x509.NewCertPool() bs, err := os.ReadFile(c.CAFile) if err != nil { return nil, fmt.Errorf("failed to read client ca cert: %s", err) } ok := certPool.AppendCertsFromPEM(bs) if !ok { return nil, errors.New("failed to append client certs") } tlsConfig := &tls.Config{ ClientAuth: tls.RequireAndVerifyClientCert, Certificates: []tls.Certificate{certificate}, ClientCAs: certPool, } return tlsConfig, nil } func (c Certificate) GetTransportCredsForClient() (credentials.TransportCredentials, error) { certificate, err := tls.LoadX509KeyPair( c.CertFile, c.KeyFile, ) if err != nil { return nil, err } certPool := x509.NewCertPool() bs, err := os.ReadFile(c.CAFile) if err != nil { return nil, fmt.Errorf("failed to read ca cert: %s", err) } ok := certPool.AppendCertsFromPEM(bs) if !ok { return nil, errors.New("failed to append certs") } transportCreds := credentials.NewTLS(&tls.Config{ ServerName: c.ServerName, Certificates: []tls.Certificate{certificate}, RootCAs: certPool, }) return transportCreds, nil } ================================================ FILE: internal/modules/rpc/client/client.go ================================================ package client import ( "context" "errors" "fmt" "sync" "time" "google.golang.org/grpc/status" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/rpc/grpcpool" pb "github.com/gocronx-team/gocron/internal/modules/rpc/proto" "google.golang.org/grpc/codes" ) var ( taskCtxMap sync.Map // 存储任务执行的 context.CancelFunc errUnavailable = errors.New(i18n.Translate("rpc_unavailable")) ErrManualStop = errors.New("rpc_manual_stop") // 特殊错误标识，用于判断是否手动停止 ) func generateTaskUniqueKey(ip string, port int, id int64) string { return fmt.Sprintf("%s:%d:%d", ip, port, id) } func Stop(ip string, port int, id int64) { // 异步发送停止信号，不阻塞调用者 go func() { addr := fmt.Sprintf("%s:%d", ip, port) c, err := grpcpool.Pool.Get(addr) if err != nil { logger.Errorf("连接服务器失败#%s#%v", addr, err) return } ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() _, err = c.Run(ctx, &pb.TaskRequest{ Command: "__STOP__", Id: id, }) if err != nil { logger.Errorf("发送停止信号失败#%v", err) } }() } func Exec(ip string, port int, taskReq *pb.TaskRequest) (string, error) { defer func() { if err := recover(); err != nil { logger.Error("panic#rpc/client.go:Exec#", err) } }() addr := fmt.Sprintf("%s:%d", ip, port) c, err := grpcpool.Pool.Get(addr) if err != nil { return "", err } if taskReq.Timeout <= 0 || taskReq.Timeout > 86400 { taskReq.Timeout = 86400 } timeout := time.Duration(taskReq.Timeout) * time.Second // RPC context: 比任务超时多5秒，给服务端时间清理进程并返回输出 ctx, cancel := context.WithTimeout(context.Background(), timeout+5*time.Second) defer cancel() taskUniqueKey := generateTaskUniqueKey(ip, port, taskReq.Id) taskCtxMap.Store(taskUniqueKey, cancel) defer taskCtxMap.Delete(taskUniqueKey) resp, err := c.Run(ctx, taskReq) // 处理响应：即使有错误，也要返回已产生的输出 if err != nil { if resp != nil && resp.Output != "" { return resp.Output, parseGRPCErrorOnly(err) } return parseGRPCError(err) } if resp.Error == "" { return resp.Output, nil } // 检查是否是手动停止 if resp.Error == "manual stop" { return resp.Output, ErrManualStop } return resp.Output, errors.New(resp.Error) } func parseGRPCError(err error) (string, error) { switch status.Code(err) { case codes.Unavailable: return "", errUnavailable case codes.DeadlineExceeded: return "", errors.New(i18n.Translate("rpc_timeout")) case codes.Canceled: return "", ErrManualStop } return "", err } // parseGRPCErrorOnly 只返回错误，不返回输出 func parseGRPCErrorOnly(err error) error { switch status.Code(err) { case codes.Unavailable: return errUnavailable case codes.DeadlineExceeded: return errors.New(i18n.Translate("rpc_timeout")) case codes.Canceled: return ErrManualStop } return err } ================================================ FILE: internal/modules/rpc/grpcpool/grpc_pool.go ================================================ package grpcpool import ( "context" "strings" "sync" "time" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/rpc/auth" pb "github.com/gocronx-team/gocron/internal/modules/rpc/proto" "google.golang.org/grpc" "google.golang.org/grpc/backoff" "google.golang.org/grpc/credentials/insecure" "google.golang.org/grpc/keepalive" ) const ( backOffMaxDelay = 3 * time.Second dialTimeout = 2 * time.Second ) var ( Pool = &GRPCPool{ conns: make(map[string]*Client), } keepAliveParams = keepalive.ClientParameters{ Time: 20 * time.Second, Timeout: 3 * time.Second, PermitWithoutStream: true, } ) type Client struct { conn *grpc.ClientConn rpcClient pb.TaskClient } type GRPCPool struct { // map key格式 ip:port conns map[string]*Client mu sync.RWMutex } func (p *GRPCPool) Get(addr string) (pb.TaskClient, error) { p.mu.RLock() client, ok := p.conns[addr] p.mu.RUnlock() if ok { return client.rpcClient, nil } client, err := p.factory(addr) if err != nil { return nil, err } return client.rpcClient, nil } // 释放连接 func (p *GRPCPool) Release(addr string) { p.mu.Lock() defer p.mu.Unlock() client, ok := p.conns[addr] if !ok { return } delete(p.conns, addr) client.conn.Close() } // 创建连接 func (p *GRPCPool) factory(addr string) (*Client, error) { p.mu.Lock() defer p.mu.Unlock() client, ok := p.conns[addr] if ok { return client, nil } opts := []grpc.DialOption{ grpc.WithKeepaliveParams(keepAliveParams), grpc.WithConnectParams(grpc.ConnectParams{ Backoff: backoff.Config{MaxDelay: backOffMaxDelay}, MinConnectTimeout: dialTimeout, }), } if !app.Setting.EnableTLS { opts = append(opts, grpc.WithTransportCredentials(insecure.NewCredentials())) } else { server := strings.Split(addr, ":") certificate := auth.Certificate{ CAFile: app.Setting.CAFile, CertFile: app.Setting.CertFile, KeyFile: app.Setting.KeyFile, ServerName: server[0], } transportCreds, err := certificate.GetTransportCredsForClient() if err != nil { return nil, err } opts = append(opts, grpc.WithTransportCredentials(transportCreds)) } ctx, cancel := context.WithTimeout(context.Background(), dialTimeout) defer cancel() conn, err := grpc.DialContext(ctx, addr, opts...) if err != nil { return nil, err } client = &Client{ conn: conn, rpcClient: pb.NewTaskClient(conn), } p.conns[addr] = client return client, nil } ================================================ FILE: internal/modules/rpc/proto/task.pb.go ================================================ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: // protoc-gen-go v1.36.11 // protoc v5.29.3 // source: task.proto package proto import ( protoreflect "google.golang.org/protobuf/reflect/protoreflect" protoimpl "google.golang.org/protobuf/runtime/protoimpl" reflect "reflect" sync "sync" unsafe "unsafe" ) const ( // Verify that this generated code is sufficiently up-to-date. _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) // Verify that runtime/protoimpl is sufficiently up-to-date. _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) ) type TaskRequest struct { state protoimpl.MessageState `protogen:"open.v1"` Command string `protobuf:"bytes,2,opt,name=command,proto3" json:"command,omitempty"` // 命令 Timeout int32 `protobuf:"varint,3,opt,name=timeout,proto3" json:"timeout,omitempty"` // 任务执行超时时间 Id int64 `protobuf:"varint,4,opt,name=id,proto3" json:"id,omitempty"` // 执行任务唯一ID unknownFields protoimpl.UnknownFields sizeCache protoimpl.SizeCache } func (x *TaskRequest) Reset() { *x = TaskRequest{} mi := &file_task_proto_msgTypes[0] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } func (x *TaskRequest) String() string { return protoimpl.X.MessageStringOf(x) } func (*TaskRequest) ProtoMessage() {} func (x *TaskRequest) ProtoReflect() protoreflect.Message { mi := &file_task_proto_msgTypes[0] if x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { ms.StoreMessageInfo(mi) } return ms } return mi.MessageOf(x) } // Deprecated: Use TaskRequest.ProtoReflect.Descriptor instead. func (*TaskRequest) Descriptor() ([]byte, []int) { return file_task_proto_rawDescGZIP(), []int{0} } func (x *TaskRequest) GetCommand() string { if x != nil { return x.Command } return "" } func (x *TaskRequest) GetTimeout() int32 { if x != nil { return x.Timeout } return 0 } func (x *TaskRequest) GetId() int64 { if x != nil { return x.Id } return 0 } type TaskResponse struct { state protoimpl.MessageState `protogen:"open.v1"` Output string `protobuf:"bytes,1,opt,name=output,proto3" json:"output,omitempty"` // 命令标准输出 Error string `protobuf:"bytes,2,opt,name=error,proto3" json:"error,omitempty"` // 命令错误 unknownFields protoimpl.UnknownFields sizeCache protoimpl.SizeCache } func (x *TaskResponse) Reset() { *x = TaskResponse{} mi := &file_task_proto_msgTypes[1] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } func (x *TaskResponse) String() string { return protoimpl.X.MessageStringOf(x) } func (*TaskResponse) ProtoMessage() {} func (x *TaskResponse) ProtoReflect() protoreflect.Message { mi := &file_task_proto_msgTypes[1] if x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { ms.StoreMessageInfo(mi) } return ms } return mi.MessageOf(x) } // Deprecated: Use TaskResponse.ProtoReflect.Descriptor instead. func (*TaskResponse) Descriptor() ([]byte, []int) { return file_task_proto_rawDescGZIP(), []int{1} } func (x *TaskResponse) GetOutput() string { if x != nil { return x.Output } return "" } func (x *TaskResponse) GetError() string { if x != nil { return x.Error } return "" } var File_task_proto protoreflect.FileDescriptor const file_task_proto_rawDesc = "" + "\n" + "\n" + "task.proto\x12\x03rpc\"Q\n" + "\vTaskRequest\x12\x18\n" + "\acommand\x18\x02 \x01(\tR\acommand\x12\x18\n" + "\atimeout\x18\x03 \x01(\x05R\atimeout\x12\x0e\n" + "\x02id\x18\x04 \x01(\x03R\x02id\"<\n" + "\fTaskResponse\x12\x16\n" + "\x06output\x18\x01 \x01(\tR\x06output\x12\x14\n" + "\x05error\x18\x02 \x01(\tR\x05error24\n" + "\x04Task\x12,\n" + "\x03Run\x12\x10.rpc.TaskRequest\x1a\x11.rpc.TaskResponse\"\x00B;Z9github.com/gocronx-team/gocron/internal/modules/rpc/protob\x06proto3" var ( file_task_proto_rawDescOnce sync.Once file_task_proto_rawDescData []byte ) func file_task_proto_rawDescGZIP() []byte { file_task_proto_rawDescOnce.Do(func() { file_task_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_task_proto_rawDesc), len(file_task_proto_rawDesc))) }) return file_task_proto_rawDescData } var file_task_proto_msgTypes = make([]protoimpl.MessageInfo, 2) var file_task_proto_goTypes = []any{ (*TaskRequest)(nil), // 0: rpc.TaskRequest (*TaskResponse)(nil), // 1: rpc.TaskResponse } var file_task_proto_depIdxs = []int32{ 0, // 0: rpc.Task.Run:input_type -> rpc.TaskRequest 1, // 1: rpc.Task.Run:output_type -> rpc.TaskResponse 1, // [1:2] is the sub-list for method output_type 0, // [0:1] is the sub-list for method input_type 0, // [0:0] is the sub-list for extension type_name 0, // [0:0] is the sub-list for extension extendee 0, // [0:0] is the sub-list for field type_name } func init() { file_task_proto_init() } func file_task_proto_init() { if File_task_proto != nil { return } type x struct{} out := protoimpl.TypeBuilder{ File: protoimpl.DescBuilder{ GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: unsafe.Slice(unsafe.StringData(file_task_proto_rawDesc), len(file_task_proto_rawDesc)), NumEnums: 0, NumMessages: 2, NumExtensions: 0, NumServices: 1, }, GoTypes: file_task_proto_goTypes, DependencyIndexes: file_task_proto_depIdxs, MessageInfos: file_task_proto_msgTypes, }.Build() File_task_proto = out.File file_task_proto_goTypes = nil file_task_proto_depIdxs = nil } ================================================ FILE: internal/modules/rpc/proto/task.proto ================================================ syntax = "proto3"; package rpc; option go_package = "github.com/gocronx-team/gocron/internal/modules/rpc/proto"; service Task { rpc Run(TaskRequest) returns (TaskResponse) {} } message TaskRequest { string command = 2; // 命令 int32 timeout = 3; // 任务执行超时时间 int64 id = 4; // 执行任务唯一ID } message TaskResponse { string output = 1; // 命令标准输出 string error = 2; // 命令错误 } ================================================ FILE: internal/modules/rpc/proto/task_grpc.pb.go ================================================ // Code generated by protoc-gen-go-grpc. DO NOT EDIT. // versions: // - protoc-gen-go-grpc v1.6.0 // - protoc v5.29.3 // source: task.proto package proto import ( context "context" grpc "google.golang.org/grpc" codes "google.golang.org/grpc/codes" status "google.golang.org/grpc/status" ) // This is a compile-time assertion to ensure that this generated file // is compatible with the grpc package it is being compiled against. // Requires gRPC-Go v1.64.0 or later. const _ = grpc.SupportPackageIsVersion9 const ( Task_Run_FullMethodName = "/rpc.Task/Run" ) // TaskClient is the client API for Task service. // // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream. type TaskClient interface { Run(ctx context.Context, in *TaskRequest, opts ...grpc.CallOption) (*TaskResponse, error) } type taskClient struct { cc grpc.ClientConnInterface } func NewTaskClient(cc grpc.ClientConnInterface) TaskClient { return &taskClient{cc} } func (c *taskClient) Run(ctx context.Context, in *TaskRequest, opts ...grpc.CallOption) (*TaskResponse, error) { cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...) out := new(TaskResponse) err := c.cc.Invoke(ctx, Task_Run_FullMethodName, in, out, cOpts...) if err != nil { return nil, err } return out, nil } // TaskServer is the server API for Task service. // All implementations must embed UnimplementedTaskServer // for forward compatibility. type TaskServer interface { Run(context.Context, *TaskRequest) (*TaskResponse, error) mustEmbedUnimplementedTaskServer() } // UnimplementedTaskServer must be embedded to have // forward compatible implementations. // // NOTE: this should be embedded by value instead of pointer to avoid a nil // pointer dereference when methods are called. type UnimplementedTaskServer struct{} func (UnimplementedTaskServer) Run(context.Context, *TaskRequest) (*TaskResponse, error) { return nil, status.Error(codes.Unimplemented, "method Run not implemented") } func (UnimplementedTaskServer) mustEmbedUnimplementedTaskServer() {} func (UnimplementedTaskServer) testEmbeddedByValue() {} // UnsafeTaskServer may be embedded to opt out of forward compatibility for this service. // Use of this interface is not recommended, as added methods to TaskServer will // result in compilation errors. type UnsafeTaskServer interface { mustEmbedUnimplementedTaskServer() } func RegisterTaskServer(s grpc.ServiceRegistrar, srv TaskServer) { // If the following call panics, it indicates UnimplementedTaskServer was // embedded by pointer and is nil. This will cause panics if an // unimplemented method is ever invoked, so we test this at initialization // time to prevent it from happening at runtime later due to I/O. if t, ok := srv.(interface{ testEmbeddedByValue() }); ok { t.testEmbeddedByValue() } s.RegisterService(&Task_ServiceDesc, srv) } func _Task_Run_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { in := new(TaskRequest) if err := dec(in); err != nil { return nil, err } if interceptor == nil { return srv.(TaskServer).Run(ctx, in) } info := &grpc.UnaryServerInfo{ Server: srv, FullMethod: Task_Run_FullMethodName, } handler := func(ctx context.Context, req interface{}) (interface{}, error) { return srv.(TaskServer).Run(ctx, req.(*TaskRequest)) } return interceptor(ctx, in, info, handler) } // Task_ServiceDesc is the grpc.ServiceDesc for Task service. // It's only intended for direct use with grpc.RegisterService, // and not to be introspected or modified (even as a copy) var Task_ServiceDesc = grpc.ServiceDesc{ ServiceName: "rpc.Task", HandlerType: (*TaskServer)(nil), Methods: []grpc.MethodDesc{ { MethodName: "Run", Handler: _Task_Run_Handler, }, }, Streams: []grpc.StreamDesc{}, Metadata: "task.proto", } ================================================ FILE: internal/modules/rpc/server/server.go ================================================ package server import ( "bytes" "context" "net" "os" "os/signal" "sync" "sync/atomic" "syscall" "time" "github.com/gocronx-team/gocron/internal/modules/rpc/auth" pb "github.com/gocronx-team/gocron/internal/modules/rpc/proto" "github.com/gocronx-team/gocron/internal/modules/utils" log "github.com/sirupsen/logrus" "google.golang.org/grpc" "google.golang.org/grpc/credentials" "google.golang.org/grpc/keepalive" ) type Server struct { pb.UnimplementedTaskServer taskContexts sync.Map // 存储正在运行的任务上下文 taskOutputs sync.Map // 存储任务输出 stopChans sync.Map // 存储停止通道 } var keepAlivePolicy = keepalive.EnforcementPolicy{ MinTime: 10 * time.Second, PermitWithoutStream: true, } var keepAliveParams = keepalive.ServerParameters{ MaxConnectionIdle: 30 * time.Second, Time: 30 * time.Second, Timeout: 3 * time.Second, } func (s *Server) Run(ctx context.Context, req *pb.TaskRequest) (*pb.TaskResponse, error) { defer func() { if err := recover(); err != nil { log.Error(err) } }() // 清理 HTML 实体 cleanedCmd := utils.CleanHTMLEntities(req.Command) // 检测是否是停止信号 if cleanedCmd == "__STOP__" { if ch, ok := s.stopChans.Load(req.Id); ok { close(ch.(chan struct{})) } return &pb.TaskResponse{ Output: "", Error: "", }, nil } // 使用任务超时创建独立的 context timeout := time.Duration(req.Timeout) * time.Second taskCtx, cancel := context.WithTimeout(context.Background(), timeout) defer cancel() // 存储任务上下文和输出 buffer outputBuf := &bytes.Buffer{} stopChan := make(chan struct{}) s.taskContexts.Store(req.Id, cancel) s.taskOutputs.Store(req.Id, outputBuf) s.stopChans.Store(req.Id, stopChan) defer func() { s.taskContexts.Delete(req.Id) s.stopChans.Delete(req.Id) // 保留输出 5 秒，给 Stop 调用时间获取 time.AfterFunc(5*time.Second, func() { s.taskOutputs.Delete(req.Id) }) }() // 监听客户端取消或停止信号 var wasStopped atomic.Bool go func() { select { case <-ctx.Done(): cancel() case <-stopChan: wasStopped.Store(true) cancel() case <-taskCtx.Done(): } }() // 执行命令 output, execErr := utils.ExecShell(taskCtx, cleanedCmd) outputBuf.WriteString(output) resp := new(pb.TaskResponse) resp.Output = output if execErr != nil { // 如果是手动停止，使用特定的错误信息 if wasStopped.Load() { resp.Error = "manual stop" log.Infof("[id: %d] Manually stopped\n%s", req.Id, output) } else { resp.Error = execErr.Error() log.Infof("[id: %d] Execution failed: %s\n%s", req.Id, execErr.Error(), output) } } else { resp.Error = "" log.Infof("[id: %d] Execution successful\n%s", req.Id, output) } return resp, nil } func Start(addr string, enableTLS bool, certificate auth.Certificate) { l, err := net.Listen("tcp", addr) if err != nil { log.Fatal(err) } opts := []grpc.ServerOption{ grpc.KeepaliveParams(keepAliveParams), grpc.KeepaliveEnforcementPolicy(keepAlivePolicy), } if enableTLS { tlsConfig, err := certificate.GetTLSConfigForServer() if err != nil { log.Fatal(err) } opt := grpc.Creds(credentials.NewTLS(tlsConfig)) opts = append(opts, opt) } server := grpc.NewServer(opts...) pb.RegisterTaskServer(server, &Server{}) log.Infof("server listen on %s", addr) go func() { err = server.Serve(l) if err != nil { log.Fatal(err) } }() c := make(chan os.Signal, 1) signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM) for { s := <-c log.Infoln("Received signal -- ", s) switch s { case syscall.SIGHUP: log.Infoln("Received terminal disconnect signal, ignoring") case syscall.SIGINT, syscall.SIGTERM: log.Info("Application preparing to exit") server.GracefulStop() return } } } ================================================ FILE: internal/modules/setting/setting.go ================================================ package setting import ( "errors" "os" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "gopkg.in/ini.v1" ) const DefaultSection = "default" type Setting struct { Db struct { Engine string Host string Port int User string Password string Database string Prefix string Charset string MaxIdleConns int MaxOpenConns int } AllowIps string AppName string ApiKey string ApiSecret string ApiSignEnable bool EnableTLS bool CAFile string CertFile string KeyFile string ConcurrencyQueue int AuthSecret string } // 读取配置 func Read(filename string) (*Setting, error) { config, err := ini.Load(filename) if err != nil { return nil, err } section := config.Section(DefaultSection) var s Setting s.Db.Engine = section.Key("db.engine").MustString("mysql") s.Db.Host = section.Key("db.host").MustString("127.0.0.1") s.Db.Port = section.Key("db.port").MustInt(3306) s.Db.User = section.Key("db.user").MustString("") s.Db.Password = section.Key("db.password").MustString("") s.Db.Database = section.Key("db.database").MustString("gocron") s.Db.Prefix = section.Key("db.prefix").MustString("") s.Db.Charset = section.Key("db.charset").MustString("utf8") s.Db.MaxIdleConns = section.Key("db.max.idle.conns").MustInt(30) s.Db.MaxOpenConns = section.Key("db.max.open.conns").MustInt(100) s.AllowIps = section.Key("allow_ips").MustString("") s.AppName = section.Key("app.name").MustString("定时任务管理系统") s.ApiKey = section.Key("api.key").MustString("") s.ApiSecret = section.Key("api.secret").MustString("") s.ApiSignEnable = section.Key("api.sign.enable").MustBool(true) s.ConcurrencyQueue = section.Key("concurrency.queue").MustInt(500) s.AuthSecret = section.Key("auth_secret").MustString("") if s.AuthSecret == "" { s.AuthSecret = utils.RandAuthToken() } s.EnableTLS = section.Key("enable_tls").MustBool(false) s.CAFile = section.Key("ca_file").MustString("") s.CertFile = section.Key("cert_file").MustString("") s.KeyFile = section.Key("key_file").MustString("") if s.EnableTLS { if !utils.FileExist(s.CAFile) { logger.Fatalf("failed to read ca cert file: %s", s.CAFile) } if !utils.FileExist(s.CertFile) { logger.Fatalf("failed to read client cert file: %s", s.CertFile) } if !utils.FileExist(s.KeyFile) { logger.Fatalf("failed to read client key file: %s", s.KeyFile) } } return &s, nil } // 写入配置 func Write(config []string, filename string) error { if len(config) == 0 { return errors.New("参数不能为空") } if len(config)%2 != 0 { return errors.New("参数不匹配") } file := ini.Empty() section, err := file.NewSection(DefaultSection) if err != nil { return err } for i := 0; i < len(config); { _, err = section.NewKey(config[i], config[i+1]) if err != nil { return err } i += 2 } err = file.SaveTo(filename) if err != nil { return err } // 设置配置文件权限为0600，仅所有者可读写 err = os.Chmod(filename, 0600) return err } ================================================ FILE: internal/modules/setting/setting_test.go ================================================ package setting import ( "os" "path/filepath" "testing" "gopkg.in/ini.v1" ) func TestReadReturnsConfiguredValues(t *testing.T) { dir := t.TempDir() configPath := filepath.Join(dir, "app.ini") content := `[default] db.engine=postgres db.host=10.0.0.1 db.port=5432 db.user=test_user db.password=test_pass db.database=test_db db.prefix=pre_ db.charset=utf8mb4 db.max.idle.conns=11 db.max.open.conns=22 allow_ips=127.0.0.1 app.name=TestApp api.key=key api.secret=secret api.sign.enable=false concurrency.queue=200 auth_secret=existing-secret enable_tls=false ` if err := os.WriteFile(configPath, []byte(content), 0o600); err != nil { t.Fatalf("write config failed: %v", err) } s, err := Read(configPath) if err != nil { t.Fatalf("unexpected error: %v", err) } if s.Db.Engine != "postgres" || s.Db.Host != "10.0.0.1" || s.Db.Port != 5432 { t.Fatalf("unexpected db config: %+v", s.Db) } if s.AppName != "TestApp" || s.ApiSignEnable { t.Fatalf("unexpected app config: %+v", s) } if s.ConcurrencyQueue != 200 || s.AuthSecret != "existing-secret" { t.Fatalf("unexpected concurrency/auth config: %+v", s) } } func TestReadGeneratesAuthSecretWhenMissing(t *testing.T) { dir := t.TempDir() configPath := filepath.Join(dir, "app.ini") content := `[default] db.engine=mysql ` if err := os.WriteFile(configPath, []byte(content), 0o600); err != nil { t.Fatalf("write config failed: %v", err) } s, err := Read(configPath) if err != nil { t.Fatalf("unexpected error: %v", err) } if s.AuthSecret == "" { t.Fatal("expected generated auth secret when config missing") } } func TestReadEnableTLSSucceedsWhenFilesExist(t *testing.T) { dir := t.TempDir() caPath := filepath.Join(dir, "ca.pem") certPath := filepath.Join(dir, "cert.pem") keyPath := filepath.Join(dir, "key.pem") for _, p := range []string{caPath, certPath, keyPath} { if err := os.WriteFile(p, []byte("data"), 0o600); err != nil { t.Fatalf("failed to create tls file: %v", err) } } configPath := filepath.Join(dir, "app.ini") content := `[default] enable_tls=true ca_file=` + caPath + ` cert_file=` + certPath + ` key_file=` + keyPath + ` ` if err := os.WriteFile(configPath, []byte(content), 0o600); err != nil { t.Fatalf("write config failed: %v", err) } if _, err := Read(configPath); err != nil { t.Fatalf("expected tls config to be read successfully, got %v", err) } } func TestWriteValidatesArguments(t *testing.T) { if err := Write(nil, ""); err == nil { t.Fatal("expected error for empty config") } if err := Write([]string{"key"}, ""); err == nil { t.Fatal("expected error for odd number of config entries") } } func TestWritePersistsKeyValuePairs(t *testing.T) { dir := t.TempDir() configPath := filepath.Join(dir, "app.ini") data := []string{ "db.engine", "sqlite", "db.host", "", "api.sign.enable", "false", } if err := Write(data, configPath); err != nil { t.Fatalf("write failed: %v", err) } cfg, err := ini.Load(configPath) if err != nil { t.Fatalf("load config failed: %v", err) } section := cfg.Section(DefaultSection) if section.Key("db.engine").String() != "sqlite" { t.Fatalf("db.engine mismatch, got %s", section.Key("db.engine").String()) } if section.Key("api.sign.enable").String() != "false" { t.Fatalf("api.sign.enable mismatch, got %s", section.Key("api.sign.enable").String()) } } func TestWriteSetsSecureFilePermissions(t *testing.T) { dir := t.TempDir() configPath := filepath.Join(dir, "app.ini") data := []string{ "db.password", "secret123", "auth_secret", "token456", } if err := Write(data, configPath); err != nil { t.Fatalf("write failed: %v", err) } info, err := os.Stat(configPath) if err != nil { t.Fatalf("stat failed: %v", err) } perm := info.Mode().Perm() if perm != 0600 { t.Fatalf("expected file permission 0600, got %#o", perm) } } ================================================ FILE: internal/modules/utils/execshell_integration_test.go ================================================ package utils import ( "context" "strings" "testing" "time" ) // 集成测试：模拟真实场景 - 任务超时但需要看到已执行的输出 func TestExecShell_RealWorldScenario_Timeout(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() // 模拟一个数据库备份任务：先输出开始信息，然后执行耗时操作 command := ` echo "=== Database Backup Started ===" echo "Connecting to database..." echo "Dumping table: users" echo "Dumping table: orders" sleep 5 echo "Backup completed" ` output, err := ExecShell(ctx, command) if err == nil { t.Fatal("Expected timeout error") } // 验证能看到超时前的所有输出 requiredOutputs := []string{ "Database Backup Started", "Connecting to database", "Dumping table: users", "Dumping table: orders", } for _, expected := range requiredOutputs { if !strings.Contains(output, expected) { t.Errorf("Missing expected output: %s\nGot: %s", expected, output) } } // 不应该包含超时后的输出 if strings.Contains(output, "Backup completed") { t.Error("Should not contain output after timeout") } t.Logf("✓ Successfully captured partial output on timeout:\n%s", output) } // 集成测试：手动停止任务 func TestExecShell_RealWorldScenario_ManualStop(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) command := ` echo "Task started" for i in {1..20}; do echo "Processing record $i" sleep 0.2 done echo "Task finished" ` resultChan := make(chan struct { output string err error }) go func() { output, err := ExecShell(ctx, command) resultChan <- struct { output string err error }{output, err} }() // 模拟用户在 1 秒后点击停止按钮 time.Sleep(1 * time.Second) cancel() result := <-resultChan if result.err == nil { t.Fatal("Expected cancellation error") } // 应该能看到部分处理记录 if !strings.Contains(result.output, "Task started") { t.Error("Missing 'Task started'") } if !strings.Contains(result.output, "Processing record") { t.Error("Missing processing records") } recordCount := strings.Count(result.output, "Processing record") if recordCount < 3 { t.Errorf("Expected at least 3 records processed, got %d", recordCount) } t.Logf("✓ Captured %d records before manual stop:\n%s", recordCount, result.output) } ================================================ FILE: internal/modules/utils/execshell_test.go ================================================ //go:build !windows // +build !windows package utils import ( "context" "os" "strings" "testing" "time" ) // 测试正常执行完成 func TestExecShell_NormalCompletion(t *testing.T) { ctx := context.Background() output, err := ExecShell(ctx, "echo 'Hello World'") if err != nil { t.Errorf("Expected no error, got: %v", err) } if !strings.Contains(output, "Hello World") { t.Errorf("Expected output to contain 'Hello World', got: %s", output) } } // 测试超时时能捕获部分输出 func TestExecShell_TimeoutWithPartialOutput(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second) defer cancel() // 执行一个会输出多行然后长时间运行的命令 command := ` echo "Line 1" echo "Line 2" echo "Line 3" sleep 10 echo "This should not appear" ` output, err := ExecShell(ctx, command) // 应该返回错误（超时） if err == nil { t.Error("Expected timeout error, got nil") } if !strings.Contains(err.Error(), "timeout killed") { t.Errorf("Expected 'timeout killed' error, got: %v", err) } // 关键：应该能捕获到前面的输出 if !strings.Contains(output, "Line 1") { t.Errorf("Expected output to contain 'Line 1', got: %s", output) } if !strings.Contains(output, "Line 2") { t.Errorf("Expected output to contain 'Line 2', got: %s", output) } if !strings.Contains(output, "Line 3") { t.Errorf("Expected output to contain 'Line 3', got: %s", output) } // 不应该包含超时后的输出 if strings.Contains(output, "This should not appear") { t.Errorf("Output should not contain text after timeout") } t.Logf("Captured output on timeout:\n%s", output) } // 测试手动取消时能捕获部分输出 func TestExecShell_ManualCancelWithPartialOutput(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) // 启动一个会持续输出的命令 command := ` for i in {1..10}; do echo "Output line $i" sleep 0.5 done ` // 在另一个 goroutine 中执行命令 outputChan := make(chan string) errChan := make(chan error) go func() { output, err := ExecShell(ctx, command) outputChan <- output errChan <- err }() // 等待 1.5 秒后取消（应该能看到前几行输出） time.Sleep(1500 * time.Millisecond) cancel() output := <-outputChan err := <-errChan // 应该返回错误（被取消） if err == nil { t.Error("Expected timeout error, got nil") } if !strings.Contains(err.Error(), "timeout killed") { t.Errorf("Expected 'timeout killed' error, got: %v", err) } // 应该能捕获到部分输出 if !strings.Contains(output, "Output line") { t.Errorf("Expected output to contain 'Output line', got: %s", output) } t.Logf("Captured output on manual cancel:\n%s", output) } // 测试命令执行失败但有输出 func TestExecShell_CommandFailureWithOutput(t *testing.T) { ctx := context.Background() // 执行一个会失败的命令，但会先输出内容 command := ` echo "Before error" ls /nonexistent_directory_12345 echo "After error" ` output, err := ExecShell(ctx, command) // 命令失败但 bash 会继续执行后续命令，所以可能没有错误 // 这取决于 shell 的行为 if err != nil { t.Logf("Command returned error (expected): %v", err) } // 应该能捕获到错误前的输出 if !strings.Contains(output, "Before error") { t.Errorf("Expected output to contain 'Before error', got: %s", output) } // 应该包含错误信息 if !strings.Contains(output, "No such file or directory") && !strings.Contains(output, "cannot access") { t.Logf("Warning: Expected error message in output, got: %s", output) } t.Logf("Output with error:\n%s", output) } // 测试长时间运行的命令 func TestExecShell_LongRunningCommand(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() // 模拟一个持续输出的长任务 command := ` for i in {1..100}; do echo "Processing item $i" sleep 0.1 done ` output, err := ExecShell(ctx, command) // 应该超时 if err == nil { t.Error("Expected timeout error, got nil") } // 应该捕获到大量输出 lineCount := strings.Count(output, "Processing item") if lineCount < 10 { t.Errorf("Expected at least 10 lines of output, got %d", lineCount) } t.Logf("Captured %d lines before timeout", lineCount) } // 测试快速完成的命令 func TestExecShell_QuickCommand(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() output, err := ExecShell(ctx, "echo 'Quick test' && date") if err != nil { t.Errorf("Expected no error, got: %v", err) } if !strings.Contains(output, "Quick test") { t.Errorf("Expected output to contain 'Quick test', got: %s", output) } } // 测试空命令 func TestExecShell_EmptyCommand(t *testing.T) { ctx := context.Background() output, err := ExecShell(ctx, "") // 空命令应该成功执行（没有输出） if err != nil { t.Logf("Empty command returned error: %v (this is acceptable)", err) } t.Logf("Empty command output: '%s'", output) } // 测试 stderr 输出 func TestExecShell_StderrOutput(t *testing.T) { ctx := context.Background() // 同时输出到 stdout 和 stderr command := ` echo "stdout message" echo "stderr message" >&2 ` output, err := ExecShell(ctx, command) if err != nil { t.Logf("Command returned error: %v", err) } // 应该同时捕获 stdout 和 stderr if !strings.Contains(output, "stdout message") { t.Errorf("Expected output to contain 'stdout message', got: %s", output) } if !strings.Contains(output, "stderr message") { t.Errorf("Expected output to contain 'stderr message', got: %s", output) } } // 基准测试：正常命令执行 func BenchmarkExecShell_Normal(b *testing.B) { ctx := context.Background() for i := 0; i < b.N; i++ { ExecShell(ctx, "echo 'benchmark test'") } } // 基准测试：超时场景 func BenchmarkExecShell_Timeout(b *testing.B) { for i := 0; i < b.N; i++ { ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond) ExecShell(ctx, "sleep 1") cancel() } } // 测试包含特殊字符的命令（临时脚本方式的优势） func TestExecShell_SpecialCharacters(t *testing.T) { ctx := context.Background() // 测试包含引号、反引号等特殊字符 command := "echo \"Hello 'World'\" && echo 'Test \"quotes\"' && echo `date`" output, err := ExecShell(ctx, command) if err != nil { t.Errorf("Expected no error, got: %v", err) } if !strings.Contains(output, "Hello 'World'") { t.Errorf("Expected output to contain mixed quotes, got: %s", output) } t.Logf("Special characters output:\n%s", output) } // 测试多行脚本（临时脚本方式的优势） func TestExecShell_MultilineScript(t *testing.T) { ctx := context.Background() // 测试复杂的多行脚本 command := ` #!/bin/bash function greet() { echo "Hello from function" } for i in 1 2 3; do echo "Loop iteration $i" done greet ` output, err := ExecShell(ctx, command) if err != nil { t.Errorf("Expected no error, got: %v", err) } if !strings.Contains(output, "Loop iteration") { t.Errorf("Expected loop output, got: %s", output) } if !strings.Contains(output, "Hello from function") { t.Errorf("Expected function output, got: %s", output) } t.Logf("Multiline script output:\n%s", output) } // 测试工作目录是否正确设置 func TestExecShell_WorkingDirectory(t *testing.T) { ctx := context.Background() // 打印当前工作目录 output, err := ExecShell(ctx, "pwd") if err != nil { t.Errorf("Expected no error, got: %v", err) } // 工作目录应该是用户家目录，不是 /tmp if strings.Contains(output, "/tmp") && !strings.Contains(output, os.Getenv("HOME")) { t.Errorf("Working directory should be home directory, got: %s", output) } t.Logf("Working directory: %s", strings.TrimSpace(output)) } // 测试HTML实体清理（保持原有功能） func TestExecShell_HTMLEntityCleaning(t *testing.T) { ctx := context.Background() // 测试HTML实体会被正确清理 command := `echo "test"` output, err := ExecShell(ctx, command) if err != nil { t.Errorf("Expected no error, got: %v", err) } // 应该输出 "test" 而不是 "test" if !strings.Contains(output, "test") { t.Errorf("Expected cleaned output, got: %s", output) } t.Logf("HTML entity cleaned output: %s", output) } ================================================ FILE: internal/modules/utils/html_entity.go ================================================ package utils import "strings" // CleanHTMLEntities 清理命令中的 HTML 实体编码 // 这个函数用于修复前端可能传递过来的 HTML 实体编码问题 // 例如: " -> ", ' -> ', < -> <, > -> >, & -> & func CleanHTMLEntities(command string) string { // 如果命令中不包含 HTML 实体,直接返回 if !strings.Contains(command, "&") { return command } // 定义 HTML 实体替换映射 replacements := map[string]string{ """: "\"", "'": "'", "'": "'", "<": "<", ">": ">", "&": "&", // 注意: & 必须最后替换,避免重复替换 } result := command // 先替换除 & 之外的所有实体 for entity, char := range replacements { if entity != "&" { result = strings.ReplaceAll(result, entity, char) } } // 最后替换 & result = strings.ReplaceAll(result, "&", "&") return result } // ContainsHTMLEntity 检测命令中是否包含 HTML 实体 func ContainsHTMLEntity(command string) bool { entities := []string{""", "'", "'", "<", ">", "&"} for _, entity := range entities { if strings.Contains(command, entity) { return true } } return false } ================================================ FILE: internal/modules/utils/html_entity_test.go ================================================ package utils import ( "strings" "testing" ) // TestHTMLEntityDetection 测试 HTML 实体检测（可在任何平台运行） func TestHTMLEntityDetection(t *testing.T) { tests := []struct { name string command string hasHTMLEntity bool expectedClean string description string }{ { name: "正常的 Windows 命令（带双引号）", command: `copy "C:\My Documents\report.docx" "D:\Backup"`, hasHTMLEntity: false, expectedClean: `copy "C:\My Documents\report.docx" "D:\Backup"`, description: "这是正确的命令，应该能在 Windows 上执行", }, { name: "包含 HTML 实体的命令（"）", command: `copy "C:\My Documents\report.docx" "D:\Backup"`, hasHTMLEntity: true, expectedClean: `copy "C:\My Documents\report.docx" "D:\Backup"`, description: "这个命令在 Windows 上会失败，因为 " 不是有效的引号", }, { name: "mkdir 命令（HTML 实体）", command: `mkdir "C:\Users\John Doe\Projects"`, hasHTMLEntity: true, expectedClean: `mkdir "C:\Users\John Doe\Projects"`, description: "mkdir 命令也会受影响", }, { name: "dir 命令（HTML 实体）", command: `dir "C:\Program Files (x86)"`, hasHTMLEntity: true, expectedClean: `dir "C:\Program Files (x86)"`, description: "dir 命令也会受影响", }, { name: "del 命令（HTML 实体）", command: `del "C:\Temp\old file.txt"`, hasHTMLEntity: true, expectedClean: `del "C:\Temp\old file.txt"`, description: "del 命令也会受影响", }, { name: "start 命令（混合引号）", command: `start "" "C:\Program Files\Google\Chrome\Application\chrome.exe" --new-window`, hasHTMLEntity: true, expectedClean: `start "" "C:\Program Files\Google\Chrome\Application\chrome.exe" --new-window`, description: "start 命令的空标题也需要引号", }, { name: "包含其他 HTML 实体", command: `echo <test> & 'hello'`, hasHTMLEntity: true, expectedClean: `echo & 'hello'`, description: "其他 HTML 实体也应该被清理", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { t.Logf("\n描述: %s", tt.description) t.Logf("原始命令: %s", tt.command) // 检测是否包含 HTML 实体 hasEntity := ContainsHTMLEntity(tt.command) if hasEntity != tt.hasHTMLEntity { t.Errorf("HTML 实体检测错误: got %v, want %v", hasEntity, tt.hasHTMLEntity) } // 清理 HTML 实体 cleaned := CleanHTMLEntities(tt.command) t.Logf("清理后命令: %s", cleaned) if cleaned != tt.expectedClean { t.Errorf("清理结果不符合预期:\ngot: %s\nwant: %s", cleaned, tt.expectedClean) } // 验证清理后不再包含 HTML 实体 if ContainsHTMLEntity(cleaned) { t.Errorf("清理后仍然包含 HTML 实体: %s", cleaned) } }) } } // TestCommandLength 测试命令长度限制 func TestCommandLength(t *testing.T) { tests := []struct { name string command string maxLength int shouldFit bool description string }{ { name: "短命令（适合 256 字符限制）", command: `dir "C:\Program Files"`, maxLength: 256, shouldFit: true, description: "简单命令应该没问题", }, { name: "长路径命令（适合 256 字符限制）", command: `copy "C:\Users\Administrator\Documents\Projects\MyProject\src\main\resources\config\application-production.properties" ` + `"D:\Backup\Projects\MyProject\config\backup-2024-01-15\application-production.properties"`, maxLength: 256, shouldFit: true, description: "这个路径实际上只有 208 字符,在 256 限制内", }, { name: "长路径命令（适合 1024 字符限制）", command: `copy "C:\Users\Administrator\Documents\Projects\MyProject\src\main\resources\config\application-production.properties" ` + `"D:\Backup\Projects\MyProject\config\backup-2024-01-15\application-production.properties"`, maxLength: 1024, shouldFit: true, description: "增加到 1024 字符后应该足够", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { t.Logf("\n描述: %s", tt.description) t.Logf("命令长度: %d 字符", len(tt.command)) t.Logf("限制长度: %d 字符", tt.maxLength) fits := len(tt.command) <= tt.maxLength if fits != tt.shouldFit { t.Errorf("长度检查错误: 命令长度 %d, 限制 %d, got %v, want %v", len(tt.command), tt.maxLength, fits, tt.shouldFit) } if !fits { t.Logf("⚠️ 命令超出长度限制 %d 字符", len(tt.command)-tt.maxLength) } }) } } // TestWindowsCommandSimulation 模拟 Windows 命令行为 func TestWindowsCommandSimulation(t *testing.T) { t.Log("\n=== 模拟 Windows cmd.exe 行为 ===\n") tests := []struct { name string command string willWork bool reason string }{ { name: "正确的双引号", command: `dir "C:\Program Files"`, willWork: true, reason: "Windows cmd 能正确识别双引号", }, { name: "HTML 实体 "", command: `dir "C:\Program Files"`, willWork: false, reason: "Windows cmd 会将 " 当作普通字符串，不是引号", }, { name: "路径中有空格但没有引号", command: `dir C:\Program Files`, willWork: false, reason: "Windows cmd 会将空格作为参数分隔符", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { t.Logf("命令: %s", tt.command) t.Logf("预期结果: %v", tt.willWork) t.Logf("原因: %s", tt.reason) // 模拟检查 hasHTMLEntity := ContainsHTMLEntity(tt.command) hasSpaceWithoutQuotes := strings.Contains(tt.command, " ") && !strings.Contains(tt.command, "\"") && !hasHTMLEntity simulatedSuccess := !hasHTMLEntity && !hasSpaceWithoutQuotes if simulatedSuccess != tt.willWork { t.Logf("⚠️ 模拟结果与预期不符") } else { t.Logf("✓ 模拟结果符合预期") } }) } } ================================================ FILE: internal/modules/utils/json.go ================================================ package utils import ( "encoding/json" "github.com/gocronx-team/gocron/internal/modules/logger" ) // json 格式输出 type response struct { Code int `json:"code"` // 状态码 0:成功 非0:失败 Message string `json:"message"` // 信息 Data interface{} `json:"data"` // 数据 } type JsonResponse struct{} const ResponseSuccess = 0 const ResponseFailure = 1 const UnauthorizedError = 403 const AuthError = 401 const NotFound = 404 const ServerError = 500 const AppNotInstall = 801 const SuccessContent = "操作成功" const FailureContent = "操作失败" func JsonResponseByErr(err error) string { jsonResp := JsonResponse{} if err != nil { return jsonResp.CommonFailure(FailureContent, err) } return jsonResp.Success(SuccessContent, nil) } func (j *JsonResponse) Success(message string, data interface{}) string { return j.response(ResponseSuccess, message, data) } func (j *JsonResponse) Failure(code int, message string) string { return j.response(code, message, nil) } func (j *JsonResponse) CommonFailure(message string, err ...error) string { if len(err) > 0 { logger.Warn(err) } return j.Failure(ResponseFailure, message) } func (j *JsonResponse) response(code int, message string, data interface{}) string { resp := response{ Code: code, Message: message, Data: data, } result, err := json.Marshal(resp) if err != nil { logger.Error(err) } return string(result) } ================================================ FILE: internal/modules/utils/login_limiter.go ================================================ package utils import ( "sync" "time" ) const ( // MaxLoginAttempts 最大登录失败次数 MaxLoginAttempts = 5 // LockDuration 账户锁定时长 LockDuration = 10 * time.Minute // CleanupInterval 清理过期记录的间隔 CleanupInterval = 30 * time.Minute ) // LoginAttempt 登录尝试记录 type LoginAttempt struct { Count int // 失败次数 LockedUtil time.Time // 锁定到期时间 } // LoginLimiter 登录限制器 type LoginLimiter struct { attempts map[string]*LoginAttempt mu sync.RWMutex } var limiter *LoginLimiter func init() { limiter = &LoginLimiter{ attempts: make(map[string]*LoginAttempt), } // 启动定期清理 go limiter.cleanup() } // GetLoginLimiter 获取登录限制器实例 func GetLoginLimiter() *LoginLimiter { return limiter } // IsLocked 检查账户是否被锁定 func (l *LoginLimiter) IsLocked(username string) (bool, time.Time) { l.mu.RLock() defer l.mu.RUnlock() attempt, exists := l.attempts[username] if !exists { return false, time.Time{} } // 检查是否达到最大失败次数且在锁定期内 if attempt.Count >= MaxLoginAttempts && time.Now().Before(attempt.LockedUtil) { return true, attempt.LockedUtil } return false, time.Time{} } // RecordFailure 记录登录失败 func (l *LoginLimiter) RecordFailure(username string) { l.mu.Lock() defer l.mu.Unlock() attempt, exists := l.attempts[username] if !exists { attempt = &LoginAttempt{Count: 0} l.attempts[username] = attempt } // 如果已过锁定期，重置计数 if !attempt.LockedUtil.IsZero() && time.Now().After(attempt.LockedUtil) { attempt.Count = 0 attempt.LockedUtil = time.Time{} } attempt.Count++ // 达到最大失败次数，锁定账户 if attempt.Count >= MaxLoginAttempts { attempt.LockedUtil = time.Now().Add(LockDuration) } } // RecordSuccess 记录登录成功，清除失败记录 func (l *LoginLimiter) RecordSuccess(username string) { l.mu.Lock() defer l.mu.Unlock() delete(l.attempts, username) } // GetRemainingAttempts 获取剩余尝试次数 func (l *LoginLimiter) GetRemainingAttempts(username string) int { l.mu.RLock() defer l.mu.RUnlock() attempt, exists := l.attempts[username] if !exists { return MaxLoginAttempts } // 如果已过锁定期，返回最大次数 if !attempt.LockedUtil.IsZero() && time.Now().After(attempt.LockedUtil) { return MaxLoginAttempts } // 如果已经被锁定，返回0 if attempt.Count >= MaxLoginAttempts { return 0 } remaining := MaxLoginAttempts - attempt.Count if remaining < 0 { return 0 } return remaining } // cleanup 定期清理过期的记录 func (l *LoginLimiter) cleanup() { ticker := time.NewTicker(CleanupInterval) defer ticker.Stop() for range ticker.C { l.mu.Lock() now := time.Now() for username, attempt := range l.attempts { // 清理已过期的锁定记录 if !attempt.LockedUtil.IsZero() && now.After(attempt.LockedUtil.Add(CleanupInterval)) { delete(l.attempts, username) } } l.mu.Unlock() } } ================================================ FILE: internal/modules/utils/login_limiter_test.go ================================================ package utils import ( "testing" "time" ) func TestLoginLimiter_IsLocked(t *testing.T) { limiter := &LoginLimiter{ attempts: make(map[string]*LoginAttempt), } username := "testuser" // 初始状态不应该被锁定 locked, _ := limiter.IsLocked(username) if locked { t.Error("User should not be locked initially") } // 记录5次失败 for i := 0; i < MaxLoginAttempts; i++ { limiter.RecordFailure(username) } // 应该被锁定 locked, lockTime := limiter.IsLocked(username) if !locked { t.Error("User should be locked after max attempts") } if lockTime.IsZero() { t.Error("Lock time should be set") } } func TestLoginLimiter_RecordSuccess(t *testing.T) { limiter := &LoginLimiter{ attempts: make(map[string]*LoginAttempt), } username := "testuser" // 记录几次失败 limiter.RecordFailure(username) limiter.RecordFailure(username) // 记录成功，应该清除失败记录 limiter.RecordSuccess(username) remaining := limiter.GetRemainingAttempts(username) if remaining != MaxLoginAttempts { t.Errorf("Expected %d remaining attempts, got %d", MaxLoginAttempts, remaining) } } func TestLoginLimiter_GetRemainingAttempts(t *testing.T) { limiter := &LoginLimiter{ attempts: make(map[string]*LoginAttempt), } username := "testuser" // 初始应该有最大次数 remaining := limiter.GetRemainingAttempts(username) if remaining != MaxLoginAttempts { t.Errorf("Expected %d remaining attempts, got %d", MaxLoginAttempts, remaining) } // 记录2次失败 limiter.RecordFailure(username) limiter.RecordFailure(username) remaining = limiter.GetRemainingAttempts(username) expected := MaxLoginAttempts - 2 if remaining != expected { t.Errorf("Expected %d remaining attempts, got %d", expected, remaining) } } func TestLoginLimiter_LockExpiration(t *testing.T) { limiter := &LoginLimiter{ attempts: make(map[string]*LoginAttempt), } username := "testuser" // 手动设置一个已过期的锁定 limiter.attempts[username] = &LoginAttempt{ Count: MaxLoginAttempts, LockedUtil: time.Now().Add(-1 * time.Minute), // 1分钟前过期 } // 应该不再被锁定 locked, _ := limiter.IsLocked(username) if locked { t.Error("User should not be locked after expiration") } // 剩余次数应该恢复 remaining := limiter.GetRemainingAttempts(username) if remaining != MaxLoginAttempts { t.Errorf("Expected %d remaining attempts after expiration, got %d", MaxLoginAttempts, remaining) } } ================================================ FILE: internal/modules/utils/password.go ================================================ package utils import ( "regexp" "unicode" ) // PasswordMinLength 密码最小长度 const PasswordMinLength = 8 // ValidatePassword 验证密码复杂度 // 要求：至少8位，包含字母和数字 func ValidatePassword(password string) (bool, string) { if len(password) < PasswordMinLength { return false, "password_min_length_8" } hasLetter := false hasDigit := false for _, char := range password { if unicode.IsLetter(char) { hasLetter = true } if unicode.IsDigit(char) { hasDigit = true } } if !hasLetter || !hasDigit { return false, "password_must_contain_letter_and_digit" } return true, "" } // ValidatePasswordStrong 验证强密码 // 要求：至少8位，包含大小写字母、数字和特殊字符 func ValidatePasswordStrong(password string) (bool, string) { if len(password) < PasswordMinLength { return false, "password_min_length_8" } hasUpper := false hasLower := false hasDigit := false hasSpecial := false specialChars := regexp.MustCompile(`[!@#$%^&*()_+\-=\[\]{};':"\\|,.<>\/?]`) for _, char := range password { if unicode.IsUpper(char) { hasUpper = true } if unicode.IsLower(char) { hasLower = true } if unicode.IsDigit(char) { hasDigit = true } } if specialChars.MatchString(password) { hasSpecial = true } if !hasUpper || !hasLower || !hasDigit || !hasSpecial { return false, "password_must_contain_upper_lower_digit_special" } return true, "" } ================================================ FILE: internal/modules/utils/password_test.go ================================================ package utils import "testing" func TestValidatePassword(t *testing.T) { tests := []struct { password string valid bool errKey string }{ {"abc123", false, "password_min_length_8"}, {"abcdefgh", false, "password_must_contain_letter_and_digit"}, {"12345678", false, "password_must_contain_letter_and_digit"}, {"abc12345", true, ""}, {"Test1234", true, ""}, {"password123", true, ""}, } for _, tt := range tests { valid, errKey := ValidatePassword(tt.password) if valid != tt.valid { t.Errorf("ValidatePassword(%q) valid = %v, want %v", tt.password, valid, tt.valid) } if errKey != tt.errKey { t.Errorf("ValidatePassword(%q) errKey = %q, want %q", tt.password, errKey, tt.errKey) } } } func TestValidatePasswordStrong(t *testing.T) { tests := []struct { password string valid bool errKey string }{ {"abc123", false, "password_min_length_8"}, {"Abcd1234", false, "password_must_contain_upper_lower_digit_special"}, {"Abcd123!", true, ""}, {"Test@123", true, ""}, {"P@ssw0rd", true, ""}, } for _, tt := range tests { valid, errKey := ValidatePasswordStrong(tt.password) if valid != tt.valid { t.Errorf("ValidatePasswordStrong(%q) valid = %v, want %v", tt.password, valid, tt.valid) } if errKey != tt.errKey { t.Errorf("ValidatePasswordStrong(%q) errKey = %q, want %q", tt.password, errKey, tt.errKey) } } } ================================================ FILE: internal/modules/utils/utils.go ================================================ package utils import ( "bytes" "crypto/md5" crand "crypto/rand" "crypto/sha256" "encoding/hex" "fmt" "math/rand" "os" "runtime" "strings" "text/template" "time" "golang.org/x/crypto/bcrypt" "golang.org/x/text/encoding/simplifiedchinese" "golang.org/x/text/transform" ) func RandAuthToken() string { buf := make([]byte, 32) _, err := crand.Read(buf) if err != nil { return RandString(64) } return fmt.Sprintf("%x", buf) } // 生成长度为length的随机字符串 func RandString(length int64) string { sources := []byte("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ") var result []byte r := rand.New(rand.NewSource(time.Now().UnixNano())) sourceLength := len(sources) var i int64 = 0 for ; i < length; i++ { result = append(result, sources[r.Intn(sourceLength)]) } return string(result) } // 生成32位MD5摘要 // 已弃用：不应用于密码哈希，仅用于非安全场景如API签名 func Md5(str string) string { m := md5.New() m.Write([]byte(str)) return hex.EncodeToString(m.Sum(nil)) } // HashPassword 使用bcrypt安全地哈希密码 func HashPassword(password string) (string, error) { hashed, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost) return string(hashed), err } // VerifyPassword 验证密码（支持bcrypt和旧的MD5格式） func VerifyPassword(hashedPassword, password, salt string) bool { // bcrypt格式以$2a$, $2b$, $2y$开头 if strings.HasPrefix(hashedPassword, "$2") { err := bcrypt.CompareHashAndPassword([]byte(hashedPassword), []byte(password)) return err == nil } // 旧的MD5格式（向后兼容） return hashedPassword == Md5(password+salt) } // Sha256 生成SHA256哈希（用于API签名等非密码场景） func Sha256(str string) string { h := sha256.New() h.Write([]byte(str)) return hex.EncodeToString(h.Sum(nil)) } // 生成0-max之间随机数 func RandNumber(max int) int { r := rand.New(rand.NewSource(time.Now().UnixNano())) return r.Intn(max) } // GBK编码转换为UTF8 func GBK2UTF8(s string) (string, bool) { decoder := simplifiedchinese.GBK.NewDecoder() result, _, err := transform.String(decoder, s) return result, err == nil } // 批量替换字符串 func ReplaceStrings(s string, old []string, replace []string) string { if s == "" { return s } if len(old) != len(replace) { return s } for i, v := range old { s = strings.Replace(s, v, replace[i], 1000) } return s } func InStringSlice(slice []string, element string) bool { element = strings.TrimSpace(element) for _, v := range slice { if strings.TrimSpace(v) == element { return true } } return false } // 转义json特殊字符 func EscapeJson(s string) string { specialChars := []string{"\\", "\b", "\f", "\n", "\r", "\t", "\""} replaceChars := []string{"\\\\", "\\b", "\\f", "\\n", "\\r", "\\t", "\\\""} return ReplaceStrings(s, specialChars, replaceChars) } // 判断文件是否存在及是否有权限访问 func FileExist(file string) bool { _, err := os.Stat(file) if os.IsNotExist(err) { return false } if os.IsPermission(err) { return false } return true } // PrintAppVersion 打印应用版本 func PrintAppVersion(appVersion, GitCommit, BuildDate string) { versionInfo, err := FormatAppVersion(appVersion, GitCommit, BuildDate) if err != nil { panic(err) } fmt.Println(versionInfo) } // FormatAppVersion 格式化应用版本信息 func FormatAppVersion(appVersion, GitCommit, BuildDate string) (string, error) { content := ` Version: {{.Version}} Go Version: {{.GoVersion}} Git Commit: {{.GitCommit}} Built: {{.BuildDate}} OS/ARCH: {{.GOOS}}/{{.GOARCH}} ` tpl, err := template.New("version").Parse(content) if err != nil { return "", err } var buf bytes.Buffer err = tpl.Execute(&buf, map[string]string{ "Version": appVersion, "GoVersion": runtime.Version(), "GitCommit": GitCommit, "BuildDate": BuildDate, "GOOS": runtime.GOOS, "GOARCH": runtime.GOARCH, }) if err != nil { return "", err } return buf.String(), err } // PanicToError Panic转换为error func PanicToError(f func()) (err error) { defer func() { if e := recover(); e != nil { err = fmt.Errorf("%s", PanicTrace(e)) } }() f() return } // PanicTrace panic调用链跟踪 func PanicTrace(err interface{}) string { stackBuf := make([]byte, 4096) n := runtime.Stack(stackBuf, false) return fmt.Sprintf("panic: %v %s", err, stackBuf[:n]) } // IsWindows 判断是否为Windows系统 func IsWindows() bool { return runtime.GOOS == "windows" } ================================================ FILE: internal/modules/utils/utils_test.go ================================================ package utils import ( "os" "path/filepath" "regexp" "runtime" "strings" "testing" "golang.org/x/text/encoding/simplifiedchinese" "golang.org/x/text/transform" ) func TestRandAuthToken(t *testing.T) { token := RandAuthToken() if len(token) != 64 { t.Fatalf("expected length 64, got %d", len(token)) } if matched := regexp.MustCompile(`^[0-9a-f]+$`).MatchString(token); !matched { t.Fatalf("token should be hex, got %s", token) } } func TestRandString(t *testing.T) { tests := []struct { name string length int64 }{ {"zero", 0}, {"positive", 16}, } charset := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got := RandString(tt.length) if int64(len(got)) != tt.length { t.Fatalf("expected length %d, got %d", tt.length, len(got)) } for _, c := range got { if !strings.ContainsRune(charset, c) { t.Fatalf("unexpected rune %q in result %q", c, got) } } }) } } func TestMd5(t *testing.T) { got := Md5("gocron") const expect = "9a34de944ae472434f79c0eb612ca724" if got != expect { t.Fatalf("expected %s, got %s", expect, got) } } func TestRandNumber(t *testing.T) { const max = 10 for i := 0; i < 100; i++ { n := RandNumber(max) if n < 0 || n >= max { t.Fatalf("number out of range: %d", n) } } } func TestGBK2UTF8(t *testing.T) { encoder := simplifiedchinese.GBK.NewEncoder() gbkStr, _, _ := transform.String(encoder, "你好") utf8Str, ok := GBK2UTF8(gbkStr) if !ok { t.Fatal("expected conversion success") } if utf8Str != "你好" { t.Fatalf("expected 你好, got %s", utf8Str) } } func TestReplaceStrings(t *testing.T) { t.Run("empty input", func(t *testing.T) { if got := ReplaceStrings("", []string{"a"}, []string{"b"}); got != "" { t.Fatalf("expected empty string, got %s", got) } }) t.Run("length mismatch", func(t *testing.T) { original := "foo" if got := ReplaceStrings(original, []string{"f"}, []string{"b", "c"}); got != original { t.Fatalf("expected original string, got %s", got) } }) t.Run("replace success", func(t *testing.T) { input := "a\nb\tc\"" got := ReplaceStrings(input, []string{"\n", "\t", "\""}, []string{"N", "T", "Q"}) if got != "aNbTcQ" { t.Fatalf("unexpected replace result %s", got) } }) } func TestInStringSlice(t *testing.T) { if !InStringSlice([]string{" foo ", "bar"}, "foo") { t.Fatal("expected to find trimmed element") } if InStringSlice([]string{"foo"}, "bar") { t.Fatal("did not expect to find missing element") } } func TestEscapeJson(t *testing.T) { input := "line1\n\"quote\"\t\\slash" got := EscapeJson(input) expect := "line1\\n\\\"quote\\\"\\t\\\\slash" if got != expect { t.Fatalf("expected %s, got %s", expect, got) } } func TestFileExist(t *testing.T) { tempDir := t.TempDir() file := filepath.Join(tempDir, "test.txt") if err := os.WriteFile(file, []byte("data"), 0o600); err != nil { t.Fatalf("failed to write temp file: %v", err) } if !FileExist(file) { t.Fatal("expected file to exist") } if FileExist(filepath.Join(tempDir, "missing.txt")) { t.Fatal("expected missing file to return false") } } func TestFormatAppVersion(t *testing.T) { info, err := FormatAppVersion("1.2.3", "abcdef", "2024-01-01") if err != nil { t.Fatalf("unexpected error: %v", err) } for _, expect := range []string{"1.2.3", "abcdef", "2024-01-01", runtime.Version(), runtime.GOOS + "/" + runtime.GOARCH} { if !strings.Contains(info, expect) { t.Fatalf("expected output to contain %s, got %s", expect, info) } } } func TestPanicToError(t *testing.T) { t.Run("panic captured", func(t *testing.T) { err := PanicToError(func() { panic("boom") }) if err == nil { t.Fatal("expected error") } if !strings.Contains(err.Error(), "boom") { t.Fatalf("expected error to contain panic message, got %v", err) } }) t.Run("no panic", func(t *testing.T) { if err := PanicToError(func() {}); err != nil { t.Fatalf("did not expect error, got %v", err) } }) } func TestPanicTrace(t *testing.T) { trace := PanicTrace("boom") if !strings.Contains(trace, "panic:") || !strings.Contains(trace, "boom") { t.Fatalf("unexpected panic trace: %s", trace) } } ================================================ FILE: internal/modules/utils/utils_unix.go ================================================ //go:build !windows // +build !windows package utils import ( "bytes" "context" "errors" "fmt" "os" "os/exec" "strings" "sync" "syscall" "time" ) type Result struct { output string err error } // 执行shell命令，可设置执行超时时间 // 改进：将命令写入临时脚本执行，即使超时或被取消，也会返回已产生的输出 func ExecShell(ctx context.Context, command string) (string, error) { // 清理可能存在的 HTML 实体编码 command = CleanHTMLEntities(command) // 将换行符统一替换为Unix风格的\n command = strings.ReplaceAll(command, "\r\n", "\n") // 创建临时文件来存储命令，按照指定格式命名 tmpDir := "/tmp" timestamp := time.Now().Format("20060102150405") scriptPattern := fmt.Sprintf("gocron_%s_*.sh", timestamp) tmpFile, err := os.CreateTemp(tmpDir, scriptPattern) if err != nil { return "", fmt.Errorf("创建临时脚本文件失败: %w", err) } defer os.Remove(tmpFile.Name()) // 执行完毕后删除临时文件 defer tmpFile.Close() // 将命令写入临时文件 _, err = tmpFile.WriteString(command) if err != nil { return "", fmt.Errorf("写入脚本内容失败: %w", err) } // 确保文件写入磁盘 err = tmpFile.Sync() if err != nil { return "", fmt.Errorf("同步文件失败: %w", err) } // 给脚本文件添加执行权限 err = os.Chmod(tmpFile.Name(), 0700) if err != nil { return "", fmt.Errorf("设置脚本执行权限失败: %w", err) } // 使用 /bin/bash 命令执行脚本文件 scriptPath := tmpFile.Name() cmd := exec.Command("/bin/bash", scriptPath) cmd.SysProcAttr = &syscall.SysProcAttr{ Setpgid: true, } // 设置工作目录为用户家目录，避免 getcwd 错误 if homeDir, err := os.UserHomeDir(); err == nil { cmd.Dir = homeDir } else { cmd.Dir = tmpDir } // 使用管道实时捕获输出 stdout, err := cmd.StdoutPipe() if err != nil { return "", err } stderr, err := cmd.StderrPipe() if err != nil { return "", err } // 用于收集输出 var outputBuffer bytes.Buffer var mu sync.Mutex var wg sync.WaitGroup // 启动命令 if err := cmd.Start(); err != nil { return "", err } // 实时读取 stdout 和 stderr wg.Add(2) go func() { defer wg.Done() buf := make([]byte, 1024) for { n, err := stdout.Read(buf) if n > 0 { mu.Lock() outputBuffer.Write(buf[:n]) mu.Unlock() } if err != nil { break } } }() go func() { defer wg.Done() buf := make([]byte, 1024) for { n, err := stderr.Read(buf) if n > 0 { mu.Lock() outputBuffer.Write(buf[:n]) mu.Unlock() } if err != nil { break } } }() // 等待命令完成或超时 done := make(chan error, 1) go func() { done <- cmd.Wait() }() select { case <-ctx.Done(): // 超时或被取消，尝试优雅终止 if cmd.Process != nil && cmd.Process.Pid > 0 { // 先发送 SIGTERM，给进程清理的机会 _ = syscall.Kill(-cmd.Process.Pid, syscall.SIGTERM) // 等待 2 秒，看进程是否自行退出 timer := time.NewTimer(2 * time.Second) select { case <-done: timer.Stop() case <-timer.C: // 进程仍未退出，强制杀死 _ = syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL) <-done // 等待 Wait() 返回 } } // 等待 IO 读取完成 wg.Wait() // 返回已捕获的输出和错误信息 mu.Lock() output := outputBuffer.String() mu.Unlock() return output, errors.New("timeout killed") case err := <-done: // 命令正常完成 wg.Wait() mu.Lock() output := outputBuffer.String() mu.Unlock() return output, err } } ================================================ FILE: internal/modules/utils/utils_unix_test.go ================================================ //go:build !windows // +build !windows package utils import ( "context" "strings" "testing" "time" ) func TestExecShellSuccess(t *testing.T) { ctx := context.Background() output, err := ExecShell(ctx, "echo 'hello world'") if err != nil { t.Fatalf("Expected no error, got: %v", err) } if !strings.Contains(output, "hello world") { t.Fatalf("Expected output to contain 'hello world', got: %s", output) } } func TestExecShellTimeout(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond) defer cancel() // 运行一个会产生输出然后睡眠的命令 output, err := ExecShell(ctx, "echo 'partial output'; sleep 1; echo 'should not see this'") if err == nil { t.Fatal("Expected timeout error") } if err.Error() != "timeout killed" { t.Fatalf("Expected 'timeout killed' error, got: %v", err) } if !strings.Contains(output, "partial output") { t.Fatalf("Expected partial output to contain 'partial output', got: %s", output) } if strings.Contains(output, "should not see this") { t.Fatalf("Should not contain output after timeout, got: %s", output) } } func TestExecShellCancel(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) // 启动一个长时间运行的命令 go func() { time.Sleep(50 * time.Millisecond) cancel() // 手动取消 }() output, err := ExecShell(ctx, "echo 'before cancel'; sleep 1; echo 'after cancel'") if err == nil { t.Fatal("Expected cancel error") } if err.Error() != "timeout killed" { t.Fatalf("Expected 'timeout killed' error, got: %v", err) } if !strings.Contains(output, "before cancel") { t.Fatalf("Expected partial output to contain 'before cancel', got: %s", output) } } func TestExecShellCommandError(t *testing.T) { ctx := context.Background() output, err := ExecShell(ctx, "nonexistentcommand") if err == nil { t.Fatal("Expected command error") } // 应该有错误输出 if output == "" { t.Fatal("Expected some error output") } } ================================================ FILE: internal/modules/utils/utils_windows.go ================================================ //go:build windows // +build windows package utils import ( "bytes" "context" "errors" "fmt" "io" "os" "os/exec" "strconv" "strings" "sync" "syscall" "time" "golang.org/x/text/encoding/simplifiedchinese" "golang.org/x/text/transform" ) type Result struct { output string err error } // 执行shell命令，可设置执行超时时间 // 改进：将命令写入临时批处理文件执行，即使超时或被取消，也会返回已产生的输出 func ExecShell(ctx context.Context, command string) (string, error) { // 清理可能存在的 HTML 实体编码,防止 " 等导致命令执行失败 // 例如: del "C:\file.txt" -> del "C:\file.txt" command = CleanHTMLEntities(command) // 将换行符统一替换为Windows风格的\r\n command = strings.ReplaceAll(command, "\r\n", "\n") command = strings.ReplaceAll(command, "\n", "\r\n") // 创建带时间戳的临时批处理文件名 timestamp := time.Now().Format("20060102150405") // 年月日时分秒 // 使用 os.CreateTemp 创建临时文件 batFile, err := os.CreateTemp(os.TempDir(), fmt.Sprintf("gocron_%s_*.bat", timestamp)) if err != nil { return "", fmt.Errorf("创建临时批处理文件失败: %w", err) } defer os.Remove(batFile.Name()) // 确保函数退出时删除临时文件 defer batFile.Close() // 将命令写入批处理文件 content := "@echo off\r\n" + command // 使用 ANSI 编码 (GBK) 写入批处理文件 gbkWriter := transform.NewWriter(batFile, simplifiedchinese.GBK.NewEncoder()) _, err = io.WriteString(gbkWriter, content) if err != nil { return "", fmt.Errorf("写入批处理文件失败: %w", err) } // 确保文件内容写入磁盘 err = batFile.Sync() if err != nil { return "", fmt.Errorf("同步批处理文件失败: %w", err) } // 使用 cmd.exe 执行批处理文件 cmd := exec.Command("cmd") cmd.SysProcAttr = &syscall.SysProcAttr{ HideWindow: true, CmdLine: `cmd /c "` + batFile.Name() + `"`, } // 设置工作目录为用户家目录，避免 getcwd 错误 if homeDir, err := os.UserHomeDir(); err == nil { cmd.Dir = homeDir } else { cmd.Dir = os.TempDir() } // 使用管道实时捕获输出 stdout, err := cmd.StdoutPipe() if err != nil { return "", err } stderr, err := cmd.StderrPipe() if err != nil { return "", err } // 用于收集输出 var outputBuffer bytes.Buffer var wg sync.WaitGroup // 启动命令 if err := cmd.Start(); err != nil { return "", err } // 实时读取 stdout 和 stderr var mu sync.Mutex wg.Add(2) go func() { defer wg.Done() buf := make([]byte, 1024) for { n, err := stdout.Read(buf) if n > 0 { mu.Lock() outputBuffer.Write(buf[:n]) mu.Unlock() } if err != nil { break } } }() go func() { defer wg.Done() buf := make([]byte, 1024) for { n, err := stderr.Read(buf) if n > 0 { mu.Lock() outputBuffer.Write(buf[:n]) mu.Unlock() } if err != nil { break } } }() // 等待命令完成或超时 done := make(chan error, 1) go func() { done <- cmd.Wait() }() select { case <-ctx.Done(): // 超时或被取消，尝试终止进程 if cmd.Process != nil && cmd.Process.Pid > 0 { // Windows 下先尝试正常终止 cmd.Process.Kill() // 等待 2 秒，看进程是否退出 timer := time.NewTimer(2 * time.Second) select { case <-done: timer.Stop() case <-timer.C: // 强制杀死进程树 exec.Command("taskkill", "/F", "/T", "/PID", strconv.Itoa(cmd.Process.Pid)).Run() <-done } } // 等待 IO 读取完成 wg.Wait() // 返回已捕获的输出（转换编码）和错误信息 mu.Lock() output := outputBuffer.String() mu.Unlock() return ConvertEncoding(output), errors.New("timeout killed") case err := <-done: // 命令正常完成 wg.Wait() mu.Lock() output := outputBuffer.String() mu.Unlock() return ConvertEncoding(output), err } } func ConvertEncoding(outputGBK string) string { // windows平台编码为gbk，需转换为utf8才能入库 outputUTF8, ok := GBK2UTF8(outputGBK) if ok { return outputUTF8 } return outputGBK } ================================================ FILE: internal/modules/utils/utils_windows_test.go ================================================ //go:build windows // +build windows package utils import ( "context" "testing" "time" ) func TestExecShellWithQuotes(t *testing.T) { tests := []struct { name string command string wantErr bool }{ { name: "Simple command without quotes", command: "echo hello", wantErr: false, }, { name: "Command with double quotes", command: `dir "C:\Program Files"`, wantErr: false, }, { name: "Copy command with quoted paths", command: `copy "C:\My Documents\report.docx" "D:\Backup"`, wantErr: false, }, { name: "Mkdir with quoted path", command: `mkdir "C:\Users\John Doe\Projects"`, wantErr: false, }, { name: "Command with HTML entity quotes", command: `copy "C:\My Documents\report.docx" "D:\Backup"`, wantErr: false, // HTML实体会被CleanHTMLEntities清理，所以应该成功 }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() output, err := ExecShell(ctx, tt.command) t.Logf("Command: %s\nOutput: %s\nError: %v", tt.command, output, err) if (err != nil) != tt.wantErr { t.Errorf("ExecShell() error = %v, wantErr %v", err, tt.wantErr) } }) } } ================================================ FILE: internal/routers/agent/agent.go ================================================ package agent import ( "crypto/rand" "encoding/hex" "fmt" "net/http" "os" "path/filepath" "time" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/routers/base" ) const tokenExpiration = 3 * time.Hour // GenerateToken 生成注册token func GenerateToken(c *gin.Context) { token := generateRandomToken() expiresAt := time.Now().Add(tokenExpiration) agentToken := &models.AgentToken{ Token: token, ExpiresAt: expiresAt, } if err := agentToken.Create(); err != nil { logger.Error("创建token失败:", err) base.RespondError(c, i18n.T(c, "operation_failed"), err) return } serverURL := getServerURL(c) installCmdLinux := fmt.Sprintf("curl -fsSL '%s/api/agent/install.sh?token=%s' | bash", serverURL, token) base.RespondSuccess(c, i18n.T(c, "operation_success"), map[string]interface{}{ "token": token, "expires_at": expiresAt, "install_cmd": installCmdLinux, }) } // InstallScript 返回安装脚本 func InstallScript(c *gin.Context) { // 验证token token := c.Query("token") if token == "" { c.String(http.StatusBadRequest, "Token is required") return } // 验证token有效性 agentToken := &models.AgentToken{} if err := agentToken.FindByToken(token); err != nil { c.String(http.StatusUnauthorized, "Invalid token") return } if time.Now().After(agentToken.ExpiresAt) { c.String(http.StatusUnauthorized, "Token expired") return } script := `#!/bin/bash set -e # 安全检查：禁止使用 root 用户运行 if [ "$(id -u)" = "0" ]; then echo "Error: This script should NOT be run as root for security reasons." echo "Please run as a regular user with sudo privileges." echo "Example: su - youruser -c 'curl -fsSL ... | bash'" exit 1 fi # Token is embedded in the script URL, extract it here TOKEN="` + token + `" if [ -z "$TOKEN" ]; then echo "Error: Token is required" exit 1 fi GOCRON_SERVER="` + getServerURL(c) + `" INSTALL_DIR="/opt/gocron-node" SERVICE_NAME="gocron-node" ARCH=$(uname -m) case $ARCH in x86_64) ARCH="amd64" ;; aarch64|arm64) ARCH="arm64" ;; *) echo "Unsupported architecture: $ARCH"; exit 1 ;; esac OS=$(uname -s | tr '[:upper:]' '[:lower:]') if [ "$OS" != "linux" ] && [ "$OS" != "darwin" ]; then echo "This script is for Linux/macOS. For Windows, use PowerShell script." echo "PowerShell command:" echo " iwr -useb ` + getServerURL(c) + `/api/agent/install.ps1 | iex" exit 1 fi echo "Installing gocron-node for $OS-$ARCH..." # 检测本地服务器是否有安装包 LOCAL_DOWNLOAD_URL="${GOCRON_SERVER}/api/agent/download?os=${OS}&arch=${ARCH}" echo "Checking local server for installation package..." # 使用 HEAD 请求检测，-w %{http_code} 获取状态码，-o /dev/null 不输出内容 HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" "$LOCAL_DOWNLOAD_URL") TMP_DIR=$(mktemp -d) cd "$TMP_DIR" if [ "$HTTP_CODE" = "200" ]; then # 本地有安装包，直接下载 echo "✓ Local package found, downloading from local server..." DOWNLOAD_URL="$LOCAL_DOWNLOAD_URL" elif [ "$HTTP_CODE" = "302" ]; then # 本地没有，需要从 GitHub 下载 echo "✗ Local package not found on server" echo "→ Downloading from GitHub (this may take a while or require network access)..." GITHUB_REPO="gocronx-team/gocron" if [ "$OS" = "windows" ]; then DOWNLOAD_URL="https://github.com/${GITHUB_REPO}/releases/latest/download/gocron-node-${OS}-${ARCH}.zip" else DOWNLOAD_URL="https://github.com/${GITHUB_REPO}/releases/latest/download/gocron-node-${OS}-${ARCH}.tar.gz" fi else echo "✗ Failed to check server status (HTTP $HTTP_CODE)" echo "→ Trying GitHub as fallback..." GITHUB_REPO="gocronx-team/gocron" if [ "$OS" = "windows" ]; then DOWNLOAD_URL="https://github.com/${GITHUB_REPO}/releases/latest/download/gocron-node-${OS}-${ARCH}.zip" else DOWNLOAD_URL="https://github.com/${GITHUB_REPO}/releases/latest/download/gocron-node-${OS}-${ARCH}.tar.gz" fi fi echo "Downloading from: $DOWNLOAD_URL" if [ "$OS" = "windows" ]; then curl -fsSL "$DOWNLOAD_URL" -o gocron-node.zip unzip -q gocron-node.zip else curl -fsSL "$DOWNLOAD_URL" -o gocron-node.tar.gz tar -xzf gocron-node.tar.gz fi sudo mkdir -p "$INSTALL_DIR" sudo cp -r gocron-node*/* "$INSTALL_DIR/" sudo chmod +x "$INSTALL_DIR/gocron-node" echo "Registering agent..." # 获取本机IP地址，如果失败则使用hostname if [ "$OS" = "darwin" ]; then HOSTNAME=$(ipconfig getifaddr en0 2>/dev/null || hostname) elif [ "$OS" = "linux" ]; then HOSTNAME=$(hostname -I 2>/dev/null | awk '{print $1}' || hostname) else HOSTNAME=$(hostname) fi echo "Using hostname/IP: $HOSTNAME" REGISTER_URL="${GOCRON_SERVER}/api/agent/register" RESPONSE=$(curl -fsSL -X POST "$REGISTER_URL" \ -H "Content-Type: application/json" \ -d "{\"token\":\"$TOKEN\",\"hostname\":\"$HOSTNAME\"}") if echo "$RESPONSE" | grep -q '"code":0'; then echo "Agent registered successfully" else echo "Failed to register agent: $RESPONSE" exit 1 fi if [ "$OS" = "linux" ]; then sudo tee /etc/systemd/system/${SERVICE_NAME}.service > /dev/null </dev/null || true sleep 1 nohup $INSTALL_DIR/gocron-node > /tmp/gocron-node.log 2>&1 & echo "gocron-node started in background (PID: $!)" echo "Log file: /tmp/gocron-node.log" fi cd / rm -rf "$TMP_DIR" echo "" echo "========================================" echo "Installation completed successfully!" echo "========================================" echo "" echo "Agent Management Commands:" echo "" if [ "$OS" = "linux" ]; then echo " Start: sudo systemctl start ${SERVICE_NAME}" echo " Stop: sudo systemctl stop ${SERVICE_NAME}" echo " Restart: sudo systemctl restart ${SERVICE_NAME}" echo " Status: sudo systemctl status ${SERVICE_NAME}" echo " Logs: sudo journalctl -u ${SERVICE_NAME} -f" echo "" echo "Uninstall:" echo " sudo systemctl stop ${SERVICE_NAME}" echo " sudo systemctl disable ${SERVICE_NAME}" echo " sudo rm /etc/systemd/system/${SERVICE_NAME}.service" echo " sudo systemctl daemon-reload" echo " sudo rm -rf ${INSTALL_DIR}" elif [ "$OS" = "darwin" ]; then echo " Stop: pkill -f gocron-node" echo " Start: nohup ${INSTALL_DIR}/gocron-node > /tmp/gocron-node.log 2>&1 &" echo " Logs: tail -f /tmp/gocron-node.log" echo " Status: ps aux | grep gocron-node | grep -v grep" echo "" echo "Uninstall:" echo " pkill -f gocron-node" echo " sudo rm -rf ${INSTALL_DIR}" echo " rm /tmp/gocron-node.log" fi echo "" echo "Installation directory: ${INSTALL_DIR}" echo "========================================" ` c.Data(http.StatusOK, "text/plain; charset=utf-8", []byte(script)) } // Register agent注册 func Register(c *gin.Context) { var req struct { Token string `json:"token" binding:"required"` Hostname string `json:"hostname" binding:"required"` } if err := c.ShouldBindJSON(&req); err != nil { base.RespondError(c, "Invalid request", err) return } agentToken := &models.AgentToken{} if err := agentToken.FindByToken(req.Token); err != nil { base.RespondError(c, "Invalid token") return } if time.Now().After(agentToken.ExpiresAt) { base.RespondError(c, "Token expired") return } if agentToken.Used { base.RespondError(c, "Token already used") return } // 原子地领取 token：用 WHERE used=false 的条件更新避免并发复用竞态 claim := models.Db.Model(&models.AgentToken{}). Where("token = ? AND used = ?", req.Token, false). Updates(map[string]interface{}{"used": true, "used_at": time.Now()}) if claim.Error != nil { base.RespondError(c, "Operation failed", claim.Error) return } if claim.RowsAffected == 0 { base.RespondError(c, "Token already used") return } host := &models.Host{ Name: req.Hostname, Alias: req.Hostname, Port: 5921, Remark: "Auto registered", } exists, err := host.NameExists(req.Hostname, 0) if err != nil { logger.Error("检查主机是否存在失败:", err) base.RespondError(c, "Operation failed", err) return } if !exists { if _, err := host.Create(); err != nil { logger.Error("创建主机失败:", err) base.RespondError(c, "Failed to create host", err) return } logger.Infof("主机注册成功: %s", req.Hostname) } else { logger.Infof("主机已存在，跳过创建: %s", req.Hostname) } base.RespondSuccess(c, "Registration successful", nil) } // Download 优先从本地 gocron-node-package 目录下载，如果不存在则重定向到 GitHub Release func Download(c *gin.Context) { osName := c.Query("os") arch := c.Query("arch") if osName == "" || arch == "" { c.String(http.StatusBadRequest, "os and arch are required") return } // 安全检查: 白名单验证,防止路径遍历攻击 validOS := map[string]bool{ "linux": true, "darwin": true, "windows": true, } validArch := map[string]bool{ "amd64": true, "arm64": true, "386": true, } if !validOS[osName] { logger.Warnf("非法的 os 参数: %s", osName) c.String(http.StatusBadRequest, "invalid os parameter") return } if !validArch[arch] { logger.Warnf("非法的 arch 参数: %s", arch) c.String(http.StatusBadRequest, "invalid arch parameter") return } // 根据操作系统选择文件扩展名 ext := ".tar.gz" if osName == "windows" { ext = ".zip" } filename := fmt.Sprintf("gocron-node-%s-%s%s", osName, arch, ext) // 获取可执行文件所在目录 execPath, err := os.Executable() if err != nil { logger.Errorf("获取可执行文件路径失败: %v", err) // 降级到 GitHub githubURL := fmt.Sprintf("https://github.com/gocronx-team/gocron/releases/latest/download/%s", filename) logger.Warnf("✗ 无法获取可执行文件路径，重定向到 GitHub: %s", githubURL) c.Redirect(http.StatusFound, githubURL) return } execDir := filepath.Dir(execPath) // 优先检查本地 gocron-node-package 目录（相对于可执行文件所在目录） packageDir := filepath.Join(execDir, "gocron-node-package") localPath := filepath.Join(packageDir, filename) logger.Infof("下载请求: os=%s, arch=%s, 查找路径: %s", osName, arch, localPath) // 安全检查: 确保最终路径在 packageDir 内,防止路径遍历 cleanPath := filepath.Clean(localPath) cleanPackageDir := filepath.Clean(packageDir) // 使用 filepath.Rel 检查路径关系 rel, err := filepath.Rel(cleanPackageDir, cleanPath) if err != nil || len(rel) > 0 && (rel[0] == '.' && len(rel) > 1 && rel[1] == '.') { logger.Warnf("检测到路径遍历攻击尝试: %s (相对路径: %s)", localPath, rel) c.String(http.StatusBadRequest, "invalid file path") return } // 检查文件是否存在 if _, err := os.Stat(cleanPath); err == nil { logger.Infof("✓ 本地安装包存在，提供文件: %s", cleanPath) c.File(cleanPath) return } // 本地文件不存在，重定向到 GitHub Release githubURL := fmt.Sprintf("https://github.com/gocronx-team/gocron/releases/latest/download/%s", filename) logger.Warnf("✗ 本地安装包不存在 (%s)，重定向到 GitHub: %s", localPath, githubURL) c.Redirect(http.StatusFound, githubURL) } func generateRandomToken() string { b := make([]byte, 32) _, _ = rand.Read(b) return hex.EncodeToString(b) } func getServerURL(c *gin.Context) string { scheme := "http" if c.Request.TLS != nil { scheme = "https" } return fmt.Sprintf("%s://%s", scheme, c.Request.Host) } ================================================ FILE: internal/routers/audit/audit.go ================================================ package audit import ( "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" ) // Index lists audit logs with pagination and optional filters. // Query params: page, page_size, module, action, username, start_date, end_date func Index(c *gin.Context) { auditLogModel := new(models.AuditLog) params := models.CommonMap{} base.ParsePageAndPageSize(c, params) if module := c.Query("module"); module != "" { params["Module"] = module } if action := c.Query("action"); action != "" { params["Action"] = action } if username := c.Query("username"); username != "" { params["Username"] = username } if startDate := c.Query("start_date"); startDate != "" { params["StartDate"] = startDate } if endDate := c.Query("end_date"); endDate != "" { params["EndDate"] = endDate } total, err := auditLogModel.Total(params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } list, err := auditLogModel.List(params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "total": total, "data": list, }) } ================================================ FILE: internal/routers/audit/audit_test.go ================================================ package audit import ( "encoding/json" "net/http" "net/http/httptest" "testing" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) type apiResponse struct { Code int `json:"code"` Message string `json:"message"` Data json.RawMessage `json:"data"` } type auditListData struct { Total int64 `json:"total"` Data []models.AuditLog `json:"data"` } func setupAuditTestRouter(t *testing.T) (*gin.Engine, func()) { t.Helper() gin.SetMode(gin.TestMode) originalDb := models.Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } if err := db.AutoMigrate(&models.AuditLog{}); err != nil { t.Fatalf("failed to migrate test database: %v", err) } models.Db = db r := gin.New() r.GET("/api/audit", Index) cleanup := func() { models.Db = originalDb } return r, cleanup } func TestAuditIndex_Empty(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit", nil) r.ServeHTTP(w, req) if w.Code != http.StatusOK { t.Fatalf("expected status 200, got %d", w.Code) } var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } if resp.Code != 0 { t.Errorf("expected code 0, got %d", resp.Code) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 0 { t.Errorf("expected total 0, got %d", data.Total) } if len(data.Data) != 0 { t.Errorf("expected empty list, got %d items", len(data.Data)) } } func TestAuditIndex_WithData(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() // Insert test records entries := []models.AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "delete"}, {Username: "bob", Ip: "10.0.0.1", Module: "user", Action: "update"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit", nil) r.ServeHTTP(w, req) if w.Code != http.StatusOK { t.Fatalf("expected status 200, got %d", w.Code) } var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } if resp.Code != 0 { t.Errorf("expected code 0, got %d", resp.Code) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 3 { t.Errorf("expected total 3, got %d", data.Total) } if len(data.Data) != 3 { t.Errorf("expected 3 items, got %d", len(data.Data)) } } func TestAuditIndex_FilterByModule(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() entries := []models.AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "delete"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "delete"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit?module=task", nil) r.ServeHTTP(w, req) if w.Code != http.StatusOK { t.Fatalf("expected status 200, got %d", w.Code) } var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 2 { t.Errorf("expected total 2, got %d", data.Total) } for _, item := range data.Data { if item.Module != "task" { t.Errorf("expected module 'task', got '%s'", item.Module) } } } func TestAuditIndex_FilterByAction(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() entries := []models.AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "delete"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit?action=create", nil) r.ServeHTTP(w, req) var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 2 { t.Errorf("expected total 2 for action=create, got %d", data.Total) } } func TestAuditIndex_FilterByUsername(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() entries := []models.AuditLog{ {Username: "alice", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "bob", Ip: "127.0.0.1", Module: "host", Action: "delete"}, {Username: "alice_admin", Ip: "127.0.0.1", Module: "user", Action: "update"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit?username=alice", nil) r.ServeHTTP(w, req) var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 2 { t.Errorf("expected total 2 for username LIKE alice, got %d", data.Total) } } func TestAuditIndex_Pagination(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() for i := 0; i < 5; i++ { log := &models.AuditLog{ Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create", } if _, err := log.Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit?page=1&page_size=3", nil) r.ServeHTTP(w, req) var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 5 { t.Errorf("expected total 5, got %d", data.Total) } if len(data.Data) != 3 { t.Errorf("expected 3 items on page 1, got %d", len(data.Data)) } } func TestAuditIndex_MultipleFilters(t *testing.T) { r, cleanup := setupAuditTestRouter(t) defer cleanup() entries := []models.AuditLog{ {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "create"}, {Username: "admin", Ip: "127.0.0.1", Module: "task", Action: "delete"}, {Username: "admin", Ip: "127.0.0.1", Module: "host", Action: "create"}, {Username: "bob", Ip: "127.0.0.1", Module: "task", Action: "create"}, } for i := range entries { if _, err := entries[i].Create(); err != nil { t.Fatalf("Create failed: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/audit?module=task&action=create&username=admin", nil) r.ServeHTTP(w, req) var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } var data auditListData if err := json.Unmarshal(resp.Data, &data); err != nil { t.Fatalf("failed to parse data: %v", err) } if data.Total != 1 { t.Errorf("expected total 1 for combined filters, got %d", data.Total) } } ================================================ FILE: internal/routers/base/base.go ================================================ package base import ( "strconv" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" ) // ParsePageAndPageSize 解析查询参数中的页数和每页数量 func ParsePageAndPageSize(c *gin.Context, params models.CommonMap) { page, _ := strconv.Atoi(c.Query("page")) pageSize, _ := strconv.Atoi(c.Query("page_size")) if page <= 0 { page = 1 } if pageSize <= 0 { pageSize = models.PageSize } params["Page"] = page params["PageSize"] = pageSize } ================================================ FILE: internal/routers/base/response.go ================================================ package base import ( "net/http" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" ) // RespondSuccess 返回成功响应 func RespondSuccess(c *gin.Context, message string, data interface{}) { json := utils.JsonResponse{} result := json.Success(message, data) c.String(http.StatusOK, result) } // RespondSuccessWithDefaultMsg 返回成功响应（使用默认消息） func RespondSuccessWithDefaultMsg(c *gin.Context, data interface{}) { json := utils.JsonResponse{} result := json.Success(utils.SuccessContent, data) c.String(http.StatusOK, result) } // RespondError 返回错误响应 func RespondError(c *gin.Context, message string, err ...error) { json := utils.JsonResponse{} if len(err) > 0 && err[0] != nil { logger.Error(err[0]) } result := json.CommonFailure(message) c.String(http.StatusOK, result) } // RespondErrorWithDefaultMsg 返回错误响应（使用默认消息） func RespondErrorWithDefaultMsg(c *gin.Context, err ...error) { json := utils.JsonResponse{} if len(err) > 0 && err[0] != nil { logger.Error(err[0]) } result := json.CommonFailure(utils.FailureContent) c.String(http.StatusOK, result) } // RespondValidationError 返回表单验证错误响应 func RespondValidationError(c *gin.Context, err error) { json := utils.JsonResponse{} result := json.CommonFailure(utils.FailureContent, err) c.String(http.StatusOK, result) } // RespondAuthError 返回认证错误响应 func RespondAuthError(c *gin.Context, message string) { json := utils.JsonResponse{} result := json.Failure(utils.AuthError, message) c.String(http.StatusOK, result) } ================================================ FILE: internal/routers/host/host.go ================================================ package host import ( "fmt" "strconv" "strings" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/rpc/client" "github.com/gocronx-team/gocron/internal/modules/rpc/grpcpool" rpc "github.com/gocronx-team/gocron/internal/modules/rpc/proto" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/service" ) const testConnectionCommand = "echo hello" const testConnectionTimeout = 5 // Index 主机列表 func Index(c *gin.Context) { hostModel := new(models.Host) queryParams := parseQueryParams(c) total, err := hostModel.Total(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } hosts, err := hostModel.List(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "total": total, "data": hosts, }) } // All 获取所有主机 func All(c *gin.Context) { hostModel := new(models.Host) hostModel.PageSize = -1 hosts, err := hostModel.List(models.CommonMap{}) if err != nil { logger.Error(err) } base.RespondSuccess(c, utils.SuccessContent, hosts) } // Detail 主机详情 func Detail(c *gin.Context) { hostModel := new(models.Host) id, _ := strconv.Atoi(c.Param("id")) err := hostModel.Find(id) if err != nil || hostModel.Id == 0 { logger.Errorf("获取主机详情失败#主机id-%d", id) base.RespondSuccess(c, utils.SuccessContent, nil) } else { base.RespondSuccess(c, utils.SuccessContent, hostModel) } } type HostForm struct { Id int `form:"id" json:"id"` Name string `form:"name" json:"name" binding:"required,max=64"` Alias string `form:"alias" json:"alias" binding:"required,max=32"` Port int `form:"port" json:"port" binding:"required,min=1,max=65535"` Remark string `form:"remark" json:"remark"` } // Store 保存、修改主机信息 func Store(c *gin.Context) { var form HostForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } hostModel := new(models.Host) id := form.Id nameExist, err := hostModel.NameExists(form.Name, form.Id) if err != nil { base.RespondError(c, i18n.T(c, "operation_failed"), err) return } if nameExist { base.RespondError(c, i18n.T(c, "hostname_exists")) return } hostModel.Name = strings.TrimSpace(form.Name) hostModel.Alias = strings.TrimSpace(form.Alias) hostModel.Port = form.Port hostModel.Remark = strings.TrimSpace(form.Remark) isCreate := false oldHostModel := new(models.Host) if id > 0 { err = oldHostModel.Find(int(id)) if err != nil { base.RespondError(c, i18n.T(c, "host_not_exist")) return } _, err = hostModel.UpdateBean(id) } else { isCreate = true id, err = hostModel.Create() } if err != nil { base.RespondError(c, i18n.T(c, "save_failed"), err) return } if !isCreate { oldAddr := fmt.Sprintf("%s:%d", oldHostModel.Name, oldHostModel.Port) newAddr := fmt.Sprintf("%s:%d", hostModel.Name, hostModel.Port) if oldAddr != newAddr { grpcpool.Pool.Release(oldAddr) } taskModel := new(models.Task) tasks, err := taskModel.ActiveListByHostId(id) if err != nil { base.RespondError(c, i18n.T(c, "refresh_task_host_failed"), err) return } service.ServiceTask.BatchAdd(tasks) } base.RespondSuccess(c, i18n.T(c, "save_success"), nil) } // Remove 删除主机 func Remove(c *gin.Context) { id, err := strconv.Atoi(c.Param("id")) if err != nil { base.RespondError(c, i18n.T(c, "param_error"), err) return } taskHostModel := new(models.TaskHost) exist, err := taskHostModel.HostIdExist(id) if err != nil { base.RespondError(c, i18n.T(c, "operation_failed"), err) return } if exist { base.RespondError(c, i18n.T(c, "host_in_use_cannot_delete")) return } hostModel := new(models.Host) err = hostModel.Find(int(id)) if err != nil { base.RespondError(c, i18n.T(c, "host_not_exist")) return } _, err = hostModel.Delete(id) if err != nil { base.RespondError(c, i18n.T(c, "operation_failed"), err) return } addr := fmt.Sprintf("%s:%d", hostModel.Name, hostModel.Port) grpcpool.Pool.Release(addr) base.RespondSuccess(c, i18n.T(c, "operation_success"), nil) } // Ping 测试主机是否可连接 func Ping(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) hostModel := new(models.Host) err := hostModel.Find(id) if err != nil || hostModel.Id <= 0 { base.RespondError(c, i18n.T(c, "host_not_exist"), err) return } taskReq := &rpc.TaskRequest{} taskReq.Command = testConnectionCommand taskReq.Timeout = testConnectionTimeout output, err := client.Exec(hostModel.Name, hostModel.Port, taskReq) if err != nil { base.RespondError(c, i18n.T(c, "connection_failed")+"-"+err.Error()+" "+output, err) } else { base.RespondSuccess(c, i18n.T(c, "connection_success"), nil) } } // 解析查询参数 func parseQueryParams(c *gin.Context) models.CommonMap { var params = models.CommonMap{} id, _ := strconv.Atoi(c.Query("id")) params["Id"] = id params["Name"] = strings.TrimSpace(c.Query("name")) base.ParsePageAndPageSize(c, params) return params } ================================================ FILE: internal/routers/install/install.go ================================================ package install import ( "errors" "fmt" "strconv" "github.com/gin-gonic/gin" "github.com/go-sql-driver/mysql" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/setting" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/service" "github.com/lib/pq" ) // 系统安装 type InstallForm struct { DbType string `form:"db_type" binding:"required,oneof=mysql postgres sqlite"` DbHost string `form:"db_host" binding:"max=50"` DbPort int `form:"db_port" binding:"min=0,max=65535"` DbUsername string `form:"db_username" binding:"max=50"` DbPassword string `form:"db_password" binding:"max=30"` DbName string `form:"db_name" binding:"required,max=200"` DbTablePrefix string `form:"db_table_prefix" binding:"max=20"` AdminUsername string `form:"admin_username" binding:"required,min=3"` AdminPassword string `form:"admin_password" binding:"required,min=6"` ConfirmAdminPassword string `form:"confirm_admin_password" binding:"required,min=6"` AdminEmail string `form:"admin_email" binding:"required,email,max=50"` } // 安装 func Store(c *gin.Context) { var form InstallForm if err := c.ShouldBind(&form); err != nil { base.RespondError(c, "表单验证失败, 请检测输入") return } if app.Installed { base.RespondError(c, "系统已安装!") return } if form.AdminPassword != form.ConfirmAdminPassword { base.RespondError(c, "两次输入密码不匹配") return } err := testDbConnection(form) if err != nil { base.RespondError(c, err.Error()) return } // 写入数据库配置 err = writeConfig(form) if err != nil { base.RespondError(c, "数据库配置写入文件失败", err) return } appConfig, err := setting.Read(app.AppConfig) if err != nil { base.RespondError(c, "读取应用配置失败", err) return } app.Setting = appConfig models.Db = models.CreateDb() // 创建数据库表 migration := new(models.Migration) err = migration.Install(form.DbName) if err != nil { base.RespondError(c, fmt.Sprintf("创建数据库表失败-%s", err.Error()), err) return } // 创建管理员账号 err = createAdminUser(form) if err != nil { base.RespondError(c, "创建管理员账号失败", err) return } // 创建安装锁 err = app.CreateInstallLock() if err != nil { base.RespondError(c, "创建文件安装锁失败", err) return } // 更新版本号文件 app.UpdateVersionFile() // 标记为已安装 app.Installed = true // 初始化并启动定时任务调度器 service.ServiceTask.Initialize() service.ServiceTask.StartScheduler() base.RespondSuccess(c, "安装成功", nil) } // 配置写入文件 func writeConfig(form InstallForm) error { dbHost := form.DbHost dbPort := strconv.Itoa(form.DbPort) if form.DbType == "sqlite" { dbHost = "" dbPort = "0" } dbConfig := []string{ "db.engine", form.DbType, "db.host", dbHost, "db.port", dbPort, "db.user", form.DbUsername, "db.password", form.DbPassword, "db.database", form.DbName, "db.prefix", form.DbTablePrefix, "db.charset", "utf8", "db.max.idle.conns", "5", "db.max.open.conns", "100", "allow_ips", "", "app.name", "定时任务管理系统", // 应用名称 "api.key", "", "api.secret", "", "enable_tls", "false", "concurrency.queue", "500", "auth_secret", utils.RandAuthToken(), "ca_file", "", "cert_file", "", "key_file", "", } return setting.Write(dbConfig, app.AppConfig) } // 创建管理员账号 func createAdminUser(form InstallForm) error { user := new(models.User) user.Name = form.AdminUsername user.Password = form.AdminPassword user.Email = form.AdminEmail user.IsAdmin = 1 _, err := user.Create() return err } // 测试数据库连接 func testDbConnection(form InstallForm) error { var s setting.Setting s.Db.Engine = form.DbType s.Db.Host = form.DbHost s.Db.Port = form.DbPort s.Db.User = form.DbUsername s.Db.Password = form.DbPassword s.Db.Database = form.DbName s.Db.Charset = "utf8" // SQLite 不需要测试连接，会自动创建文件 if s.Db.Engine == "sqlite" { return nil } db, err := models.CreateTmpDb(&s) if err != nil { return err } sqlDB, err := db.DB() if err != nil { return err } defer sqlDB.Close() err = sqlDB.Ping() if s.Db.Engine == "postgres" && err != nil { pgError, ok := err.(*pq.Error) if ok && pgError.Code == "3D000" { err = errors.New("数据库不存在") } return err } if s.Db.Engine == "mysql" && err != nil { mysqlError, ok := err.(*mysql.MySQLError) if ok && mysqlError.Number == 1049 { err = errors.New("数据库不存在") } return err } return err } ================================================ FILE: internal/routers/loginlog/login_log.go ================================================ package loginlog import ( "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" ) func Index(c *gin.Context) { loginLogModel := new(models.LoginLog) params := models.CommonMap{} base.ParsePageAndPageSize(c, params) total, err := loginLogModel.Total() if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } loginLogs, err := loginLogModel.List(params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "total": total, "data": loginLogs, }) } ================================================ FILE: internal/routers/manage/manage.go ================================================ package manage import ( "encoding/json" "strconv" "strings" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/service" ) func Slack(c *gin.Context) { settingModel := new(models.Setting) slack, err := settingModel.Slack() if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, slack) } func UpdateSlack(c *gin.Context) { var form UpdateSlackForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("Slack配置表单验证失败: %v", err) base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) err := settingModel.UpdateSlack(form.Url, form.Template) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func CreateSlackChannel(c *gin.Context) { var form CreateSlackChannelForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("创建Slack频道表单验证失败: %v", err) base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) if settingModel.IsChannelExist(form.Channel) { base.RespondError(c, "Channel已存在") } else { _, err := settingModel.CreateChannel(form.Channel) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } } func RemoveSlackChannel(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) settingModel := new(models.Setting) _, err := settingModel.RemoveChannel(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } // endregion // region 邮件 func Mail(c *gin.Context) { settingModel := new(models.Setting) mail, err := settingModel.Mail() if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, "", mail) } type MailServerForm struct { Host string `form:"host" json:"host" binding:"required,max=100"` Port int `form:"port" json:"port" binding:"required,min=1,max=65535"` User string `form:"user" json:"user" binding:"required,max=64"` Password string `form:"password" json:"password" binding:"required,max=64"` Template string `form:"template" json:"template"` } // CreateMailUserForm 创建邮件用户表单 type CreateMailUserForm struct { Username string `form:"username" json:"username" binding:"required,max=50"` Email string `form:"email" json:"email" binding:"required,email,max=100"` } // UpdateSlackForm 更新Slack配置表单 type UpdateSlackForm struct { Url string `form:"url" json:"url" binding:"required,url,max=200"` Template string `form:"template" json:"template" binding:"required"` } // UpdateWebHookForm 更新WebHook配置表单 type UpdateWebHookForm struct { Template string `form:"template" json:"template" binding:"required"` } // CreateWebhookUrlForm 创建Webhook地址表单 type CreateWebhookUrlForm struct { Name string `form:"name" json:"name" binding:"required,max=50"` Url string `form:"url" json:"url" binding:"required,url,max=200"` } // CreateSlackChannelForm 创建Slack频道表单 type CreateSlackChannelForm struct { Channel string `form:"channel" json:"channel" binding:"required,max=50"` } func UpdateMail(c *gin.Context) { var form MailServerForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("邮件配置表单验证失败: %v", err) // 提供更具体的错误信息 errorMsg := "表单验证失败: " if strings.Contains(err.Error(), "email") { errorMsg += "用户名必须是有效的邮箱地址" } else if strings.Contains(err.Error(), "required") { errorMsg += "请填写所有必填字段" } else if strings.Contains(err.Error(), "max") { errorMsg += "输入内容过长" } else if strings.Contains(err.Error(), "min") || strings.Contains(err.Error(), "port") { errorMsg += "端口号必须在1-65535之间" } else { errorMsg += "请检查输入格式" } base.RespondError(c, errorMsg) return } // 从表单中提取template，单独保存 template := strings.TrimSpace(form.Template) // 将服务器配置序列化为JSON（不包含template） serverConfig := struct { Host string `json:"host"` Port int `json:"port"` User string `json:"user"` Password string `json:"password"` }{ Host: form.Host, Port: form.Port, User: form.User, Password: form.Password, } jsonByte, _ := json.Marshal(serverConfig) settingModel := new(models.Setting) err := settingModel.UpdateMail(string(jsonByte), template) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func CreateMailUser(c *gin.Context) { var form CreateMailUserForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("创建邮件用户表单验证失败: %v", err) base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) _, err := settingModel.CreateMailUser(form.Username, form.Email) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func RemoveMailUser(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) settingModel := new(models.Setting) _, err := settingModel.RemoveMailUser(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func WebHook(c *gin.Context) { settingModel := new(models.Setting) webHook, err := settingModel.Webhook() if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, "", webHook) } func UpdateWebHook(c *gin.Context) { var form UpdateWebHookForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("Webhook配置表单验证失败: %v", err) base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) err := settingModel.UpdateWebHook(form.Template) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func CreateWebhookUrl(c *gin.Context) { var form CreateWebhookUrlForm if err := c.ShouldBind(&form); err != nil { logger.Errorf("创建Webhook地址表单验证失败: %v", err) base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) _, err := settingModel.CreateWebhookUrl(form.Name, form.Url) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } func RemoveWebhookUrl(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) settingModel := new(models.Setting) _, err := settingModel.RemoveWebhookUrl(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } // endregion // region 系统配置 func GetLogRetentionDays(c *gin.Context) { settingModel := new(models.Setting) days := settingModel.GetLogRetentionDays() cleanupTime := settingModel.GetLogCleanupTime() fileSizeLimit := settingModel.GetLogFileSizeLimit() base.RespondSuccess(c, "", map[string]interface{}{ "days": days, "cleanup_time": cleanupTime, "file_size_limit": fileSizeLimit, }) } func UpdateLogRetentionDays(c *gin.Context) { var form struct { Days int `json:"days" binding:"min=0,max=3650"` CleanupTime string `json:"cleanup_time" binding:"required"` FileSizeLimit int `json:"file_size_limit" binding:"min=0,max=10240"` } if err := c.ShouldBindJSON(&form); err != nil { base.RespondError(c, "表单验证失败, 请检测输入") return } settingModel := new(models.Setting) err := settingModel.UpdateLogRetentionDays(form.Days) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } err = settingModel.UpdateLogCleanupTime(form.CleanupTime) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } err = settingModel.UpdateLogFileSizeLimit(form.FileSizeLimit) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } // 重新加载日志清理任务 service.ServiceTask.ReloadLogCleanupTask() base.RespondSuccessWithDefaultMsg(c, nil) } // endregion ================================================ FILE: internal/routers/routers.go ================================================ package routers import ( "context" "crypto/subtle" "io" "io/fs" "net/http" "strconv" "strings" "time" "github.com/gin-gonic/gin" gocronembed "github.com/gocronx-team/gocron" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/agent" "github.com/gocronx-team/gocron/internal/routers/audit" "github.com/gocronx-team/gocron/internal/routers/host" "github.com/gocronx-team/gocron/internal/routers/install" "github.com/gocronx-team/gocron/internal/routers/loginlog" "github.com/gocronx-team/gocron/internal/routers/manage" "github.com/gocronx-team/gocron/internal/routers/statistics" "github.com/gocronx-team/gocron/internal/routers/task" "github.com/gocronx-team/gocron/internal/routers/tasklog" "github.com/gocronx-team/gocron/internal/routers/template" "github.com/gocronx-team/gocron/internal/routers/user" ) const ( urlPrefix = "/api" ) var staticFS fs.FS func init() { var err error staticFS, err = gocronembed.StaticFS() if err != nil { logger.Fatal("初始化静态文件系统失败", err) } } // Register 路由泣册 func Register(r *gin.Engine) { api := r.Group(urlPrefix) // 系统安装 installGroup := api.Group("/install") { installGroup.POST("/store", install.Store) installGroup.GET("/status", func(c *gin.Context) { jsonResp := utils.JsonResponse{} c.String(http.StatusOK, jsonResp.Success("", app.Installed)) }) } // 用户 userGroup := api.Group("/user") { userGroup.GET("", user.Index) userGroup.GET("/:id", user.Detail) userGroup.POST("/store", user.Store) userGroup.POST("/remove/:id", user.Remove) userGroup.POST("/login", user.ValidateLogin) userGroup.POST("/enable/:id", user.Enable) userGroup.POST("/disable/:id", user.Disable) userGroup.POST("/editMyPassword", user.UpdateMyPassword) userGroup.POST("/editPassword/:id", user.UpdatePassword) // 2FA相关路由 userGroup.GET("/2fa/status", user.Get2FAStatus) userGroup.GET("/2fa/setup", user.Setup2FA) userGroup.POST("/2fa/enable", user.Enable2FA) userGroup.POST("/2fa/disable", user.Disable2FA) } // 定时任务 taskGroup := api.Group("/task") { taskGroup.GET("/versions/:id", task.VersionList) taskGroup.GET("/versions/:id/:version_id", task.VersionDetail) taskGroup.POST("/versions/:id/:version_id/rollback", task.VersionRollback) taskGroup.POST("/store", task.Store) taskGroup.POST("/cron-preview", task.CronPreview) taskGroup.GET("/tags", task.GetAllTags) taskGroup.GET("/:id", task.Detail) taskGroup.GET("", task.Index) taskGroup.GET("/log", tasklog.Index) taskGroup.POST("/log/clear", tasklog.Clear) taskGroup.POST("/log/clear/:id", tasklog.ClearByTaskId) taskGroup.POST("/log/stop", tasklog.Stop) taskGroup.POST("/remove/:id", task.Remove) taskGroup.POST("/enable/:id", task.Enable) taskGroup.POST("/disable/:id", task.Disable) taskGroup.POST("/batch-enable", task.BatchEnable) taskGroup.POST("/batch-disable", task.BatchDisable) taskGroup.POST("/batch-remove", task.BatchRemove) taskGroup.GET("/run/:id", task.Run) } // 主机 hostGroup := api.Group("/host") { hostGroup.GET("/:id", host.Detail) hostGroup.POST("/store", host.Store) hostGroup.GET("", host.Index) hostGroup.GET("/all", host.All) hostGroup.GET("/ping/:id", host.Ping) hostGroup.POST("/remove/:id", host.Remove) } // Agent注册 agentGroup := api.Group("/agent") { agentGroup.POST("/generate-token", agent.GenerateToken) agentGroup.GET("/install.sh", agent.InstallScript) agentGroup.POST("/register", agent.Register) agentGroup.GET("/download", agent.Download) } // 任务模板 templateGroup := api.Group("/template") { templateGroup.GET("", template.Index) templateGroup.GET("/categories", template.Categories) templateGroup.GET("/:id", template.Detail) templateGroup.POST("/store", template.Store) templateGroup.POST("/remove/:id", template.Remove) templateGroup.POST("/apply/:id", template.Apply) templateGroup.POST("/save-from-task", template.SaveFromTask) } // 管理 systemGroup := api.Group("/system") { slackGroup := systemGroup.Group("/slack") { slackGroup.GET("", manage.Slack) slackGroup.POST("/update", manage.UpdateSlack) slackGroup.POST("/channel", manage.CreateSlackChannel) slackGroup.POST("/channel/remove/:id", manage.RemoveSlackChannel) } mailGroup := systemGroup.Group("/mail") { mailGroup.GET("", manage.Mail) mailGroup.POST("/update", manage.UpdateMail) mailGroup.POST("/user", manage.CreateMailUser) mailGroup.POST("/user/remove/:id", manage.RemoveMailUser) } webhookGroup := systemGroup.Group("/webhook") { webhookGroup.GET("", manage.WebHook) webhookGroup.POST("/update", manage.UpdateWebHook) webhookGroup.POST("/url", manage.CreateWebhookUrl) webhookGroup.POST("/url/remove/:id", manage.RemoveWebhookUrl) } systemGroup.GET("/login-log", loginlog.Index) systemGroup.GET("/log-retention", manage.GetLogRetentionDays) systemGroup.POST("/log-retention", manage.UpdateLogRetentionDays) } // 统计 statisticsGroup := api.Group("/statistics") { statisticsGroup.GET("/overview", statistics.Overview) } // 审计日志（需认证） auditGroup := api.Group("/audit") { auditGroup.GET("", audit.Index) } // API v1Group := api.Group("/v1") v1Group.Use(apiAuth) { v1Group.POST("/tasklog/remove/:id", tasklog.Remove) v1Group.POST("/task/enable/:id", task.Enable) v1Group.POST("/task/disable/:id", task.Disable) } // 首页路由（根路径） r.GET("/", func(c *gin.Context) { file, err := staticFS.Open("index.html") if err != nil { logger.Errorf("读取首页文件失败: %s", err) c.Status(http.StatusInternalServerError) return } defer file.Close() c.Header("Content-Type", "text/html") _, _ = io.Copy(c.Writer, file) }) // 静态文件路由 - 必须放在最后 r.NoRoute(func(c *gin.Context) { filepath := c.Request.URL.Path // 移除 /public 前缀（如果存在） filepath = strings.TrimPrefix(filepath, "/public") filepath = strings.TrimPrefix(filepath, "/") // 尝试从 staticFS 读取文件 file, err := staticFS.Open(filepath) if err == nil { defer file.Close() // 设置正确的Content-Type - 必须在写入数据之前设置 if strings.HasSuffix(filepath, ".js") { c.Writer.Header().Set("Content-Type", "application/javascript; charset=utf-8") } else if strings.HasSuffix(filepath, ".css") { c.Writer.Header().Set("Content-Type", "text/css; charset=utf-8") } else if strings.HasSuffix(filepath, ".html") { c.Writer.Header().Set("Content-Type", "text/html; charset=utf-8") } else if strings.HasSuffix(filepath, ".png") { c.Writer.Header().Set("Content-Type", "image/png") } else if strings.HasSuffix(filepath, ".jpg") || strings.HasSuffix(filepath, ".jpeg") { c.Writer.Header().Set("Content-Type", "image/jpeg") } else if strings.HasSuffix(filepath, ".svg") { c.Writer.Header().Set("Content-Type", "image/svg+xml") } c.Status(http.StatusOK) _, _ = io.Copy(c.Writer, file) return } // 文件不存在，返回404 jsonResp := utils.JsonResponse{} c.String(http.StatusNotFound, jsonResp.Failure(utils.NotFound, i18n.T(c, "page_not_found"))) }) } // 中间件注册 func RegisterMiddleware(r *gin.Engine) { // 中间件 r.Use(securityHeaders) r.Use(checkAppInstall) r.Use(ipAuth) r.Use(userAuth) r.Use(urlAuth) r.Use(auditLog) } // securityHeaders 设置通用的安全响应头，防御点击劫持 / MIME sniff / referrer 泄漏。 // 不设置 CSP（需要针对前端资源单独调校）也不设置 HSTS（由反向代理决定）。 func securityHeaders(c *gin.Context) { c.Header("X-Frame-Options", "DENY") c.Header("X-Content-Type-Options", "nosniff") c.Header("Referrer-Policy", "no-referrer") c.Next() } // region Custom middleware // isStaticFileRequest checks if the request is for a static file (non-API path). // Static files are served via NoRoute handler and never match registered API routes. func isStaticFileRequest(path string) bool { return !strings.HasPrefix(path, urlPrefix+"/") && !strings.HasPrefix(path, "/v1/") } // checkAppInstall verifies the application has been installed. func checkAppInstall(c *gin.Context) { if app.Installed { c.Next() return } path := c.Request.URL.Path // Allow install API, root page, and static files before installation if strings.HasPrefix(path, "/api/install") || isStaticFileRequest(path) { c.Next() return } jsonResp := utils.JsonResponse{} data := jsonResp.Failure(utils.AppNotInstall, i18n.T(c, "app_not_installed")) c.String(http.StatusOK, data) c.Abort() } // IP验证, 通过反向代理访问gocron，需设置Header X-Real-IP才能获取到客户端真实IP func ipAuth(c *gin.Context) { if !app.Installed { c.Next() return } allowIpsStr := app.Setting.AllowIps if allowIpsStr == "" { c.Next() return } clientIp := c.ClientIP() allowIps := strings.Split(allowIpsStr, ",") if utils.InStringSlice(allowIps, clientIp) { c.Next() return } logger.Warnf("非法IP访问-%s", clientIp) jsonResp := utils.JsonResponse{} data := jsonResp.Failure(utils.UnauthorizedError, i18n.T(c, "unauthorized")) c.String(http.StatusOK, data) c.Abort() } // userAuth authenticates the user for API requests. func userAuth(c *gin.Context) { if !app.Installed { c.Next() return } path := c.Request.URL.Path // Static files (non-API paths) don't require authentication if isStaticFileRequest(path) { c.Next() return } uri := strings.TrimRight(path, "/") // 登录接口和安装状态接口不需要认证 excludePaths := []string{"", "/api/user/login", "/api/install/status", "/api/agent/install.sh", "/api/agent/register", "/api/agent/download"} for _, p := range excludePaths { if uri == p { c.Next() return } } // v1 API接口使用单独的认证 if strings.HasPrefix(uri, "/v1") { c.Next() return } // 尝试从token恢复用户信息 newToken, err := user.RestoreToken(c) if err != nil { logger.Warnf("token解析失败: %v, path: %s", err, path) jsonResp := utils.JsonResponse{} data := jsonResp.Failure(utils.AuthError, i18n.T(c, "auth_failed")) c.String(http.StatusOK, data) c.Abort() return } // 如果token被刷新，返回新token给前端 if newToken != "" { c.Header("New-Auth-Token", newToken) } if !user.IsLogin(c) { jsonResp := utils.JsonResponse{} data := jsonResp.Failure(utils.AuthError, i18n.T(c, "auth_failed")) c.String(http.StatusOK, data) c.Abort() return } c.Next() } // urlAuth checks URL-level permissions (admin vs normal user). func urlAuth(c *gin.Context) { if !app.Installed { c.Next() return } path := c.Request.URL.Path // Static files (non-API paths) don't require permission checks if isStaticFileRequest(path) { c.Next() return } if user.IsAdmin(c) { c.Next() return } uri := strings.TrimRight(path, "/") if strings.HasPrefix(uri, "/v1") { c.Next() return } // 普通用户允许访问的URL地址 allowPaths := []string{ "", "/api/install/status", "/api/task", "/api/task/tags", "/api/task/log", "/api/host", "/api/host/all", "/api/user/login", "/api/user/editMyPassword", "/api/user/2fa/status", "/api/user/2fa/setup", "/api/user/2fa/enable", "/api/user/2fa/disable", "/api/template", "/api/template/categories", "/api/statistics/overview", "/api/agent/install.sh", "/api/agent/register", "/api/agent/download", } for _, p := range allowPaths { if p == uri { c.Next() return } } jsonResp := utils.JsonResponse{} data := jsonResp.Failure(utils.UnauthorizedError, i18n.T(c, "unauthorized")) c.String(http.StatusOK, data) c.Abort() } // auditLog middleware records audit log entries for write operations. // It runs after the handler (post-processing) and only records successful POST requests. func auditLog(c *gin.Context) { c.Next() // Only record POST requests if c.Request.Method != http.MethodPost { return } // Only record successful operations (status < 400) if c.Writer.Status() >= 400 { return } path := c.FullPath() username := user.Username(c) ip := c.ClientIP() module, action := resolveModuleAction(path, c) if module == "" || action == "" { return } // 获取 targetId：优先从 URL 参数，其次从 POST body targetId := 0 if idStr := c.Param("id"); idStr != "" { targetId, _ = strconv.Atoi(idStr) } else if idStr := c.PostForm("id"); idStr != "" && idStr != "0" { targetId, _ = strconv.Atoi(idStr) } // 读取 handler 设置的审计详情 detail, _ := c.Get("audit_detail") detailStr, _ := detail.(string) log := &models.AuditLog{ Username: username, Ip: ip, Module: module, Action: action, TargetId: targetId, Detail: detailStr, } // 异步查询对象名称并写入；使用独立 context 避免请求已结束后 goroutine 无界堆积 go func() { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() log.TargetName = resolveTargetName(ctx, module, targetId) if err := models.Db.WithContext(ctx).Create(log).Error; err != nil { logger.Warnf("写入审计日志失败: %v", err) } }() } // resolveModuleAction maps a Gin full path pattern to (module, action). func resolveModuleAction(path string, c *gin.Context) (module, action string) { switch path { // Task routes case "/api/task/store": idStr := c.PostForm("id") if idStr == "" || idStr == "0" { return "task", "create" } return "task", "update" case "/api/task/remove/:id": return "task", "delete" case "/api/task/enable/:id": return "task", "enable" case "/api/task/disable/:id": return "task", "disable" case "/api/task/batch-enable": return "task", "batch-enable" case "/api/task/batch-disable": return "task", "batch-disable" case "/api/task/batch-remove": return "task", "batch-remove" // Host routes case "/api/host/store": idStr := c.PostForm("id") if idStr == "" || idStr == "0" { return "host", "create" } return "host", "update" case "/api/host/remove/:id": return "host", "delete" // User routes case "/api/user/store": idStr := c.PostForm("id") if idStr == "" || idStr == "0" { return "user", "create" } return "user", "update" case "/api/user/remove/:id": return "user", "delete" case "/api/user/enable/:id": return "user", "enable" case "/api/user/disable/:id": return "user", "disable" case "/api/user/editMyPassword": return "user", "change-password" case "/api/user/editPassword/:id": return "user", "reset-password" // Template routes case "/api/template/store": idStr := c.PostForm("id") if idStr == "" || idStr == "0" { return "template", "create" } return "template", "update" case "/api/template/remove/:id": return "template", "delete" case "/api/template/apply/:id": return "template", "update" case "/api/template/save-from-task": return "template", "create" // System routes — any POST under /api/system default: if strings.HasPrefix(path, "/api/system/") { return "system", "update" } } return "", "" } // resolveTargetName 根据 module 和 targetId 查询对象名称 func resolveTargetName(ctx context.Context, module string, targetId int) string { if targetId == 0 { return "" } db := models.Db.WithContext(ctx) switch module { case "task": task := &models.Task{} if err := db.Select("name").First(task, targetId).Error; err == nil { return task.Name } case "host": host := &models.Host{} if err := db.Select("name", "alias").First(host, targetId).Error; err == nil { if host.Alias != "" { return host.Alias } return host.Name } case "user": u := &models.User{} if err := db.Select("name").First(u, targetId).Error; err == nil { return u.Name } case "template": tmpl := &models.TaskTemplate{} if err := models.Db.Select("name").First(tmpl, targetId).Error; err == nil { return tmpl.Name } } return "" } /** API接口签名验证 **/ func apiAuth(c *gin.Context) { if !app.Installed { c.Next() return } if !app.Setting.ApiSignEnable { c.Next() return } apiKey := strings.TrimSpace(app.Setting.ApiKey) apiSecret := strings.TrimSpace(app.Setting.ApiSecret) json := utils.JsonResponse{} if apiKey == "" || apiSecret == "" { msg := json.CommonFailure(i18n.T(c, "api_key_required")) c.String(http.StatusOK, msg) c.Abort() return } currentTimestamp := time.Now().Unix() timeParam, err := strconv.ParseInt(c.Query("time"), 10, 64) if err != nil || timeParam <= 0 { msg := json.CommonFailure(i18n.T(c, "param_time_required")) c.String(http.StatusOK, msg) c.Abort() return } if timeParam < (currentTimestamp - 1800) { msg := json.CommonFailure(i18n.T(c, "param_time_invalid")) c.String(http.StatusOK, msg) c.Abort() return } sign := strings.TrimSpace(c.Query("sign")) if sign == "" { msg := json.CommonFailure(i18n.T(c, "param_sign_required")) c.String(http.StatusOK, msg) c.Abort() return } raw := apiKey + strconv.FormatInt(timeParam, 10) + strings.TrimSpace(c.Request.URL.Path) + apiSecret realSign := utils.Sha256(raw) if subtle.ConstantTimeCompare([]byte(sign), []byte(realSign)) != 1 { msg := json.CommonFailure(i18n.T(c, "sign_verify_failed")) c.String(http.StatusOK, msg) c.Abort() return } c.Next() } // endregion ================================================ FILE: internal/routers/statistics/statistics.go ================================================ package statistics import ( "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" ) // OverviewData 概览统计数据 type OverviewData struct { TotalTasks int64 `json:"total_tasks"` TodayExecutions int64 `json:"today_executions"` SuccessRate float64 `json:"success_rate"` FailedCount int64 `json:"failed_count"` Last7Days []models.DailyStats `json:"last_7_days"` } // Overview 获取统计概览数据 func Overview(c *gin.Context) { taskModel := models.Task{} taskLogModel := models.TaskLog{} // 1. 获取启用的任务总数 totalTasks, err := taskModel.Total(models.CommonMap{"Status": int(models.Enabled)}) if err != nil { logger.Error("Failed to get total tasks:", err) base.RespondError(c, "Failed to get total tasks", err) return } // 2. 获取今日统计数据 todayTotal, todaySuccess, todayFailed, err := taskLogModel.GetTodayStats() if err != nil { logger.Error("Failed to get today's statistics:", err) base.RespondError(c, "Failed to get today's statistics", err) return } // 3. 计算成功率 var successRate float64 if todayTotal > 0 { successRate = float64(todaySuccess) / float64(todayTotal) * 100 // 保留1位小数 successRate = float64(int(successRate*10)) / 10 } // 4. 获取最近7天趋势 last7Days, err := taskLogModel.GetLast7DaysTrend() if err != nil { logger.Error("Failed to get trend data:", err) base.RespondError(c, "Failed to get trend data", err) return } // 组装返回数据 data := OverviewData{ TotalTasks: totalTasks, TodayExecutions: todayTotal, SuccessRate: successRate, FailedCount: todayFailed, Last7Days: last7Days, } base.RespondSuccess(c, utils.SuccessContent, data) } ================================================ FILE: internal/routers/task/cron_preview.go ================================================ package task import ( "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/service" ) type cronPreviewRequest struct { Spec string `json:"spec" binding:"required"` Timezone string `json:"timezone"` Count int `json:"count"` } // CronPreview 返回给定 cron 表达式的接下来 N 次执行时间 + 一周执行分布热图。 // 非法表达式也返回 HTTP 200，body 里 valid=false（用户边敲边预览，不用 4xx 轰炸 console）。 func CronPreview(c *gin.Context) { var req cronPreviewRequest if err := c.ShouldBindJSON(&req); err != nil { base.RespondValidationError(c, err) return } result := service.PreviewCron(req.Spec, req.Timezone, req.Count) jsonResp := utils.JsonResponse{} c.String(200, jsonResp.Success(utils.SuccessContent, result)) } ================================================ FILE: internal/routers/task/task.go ================================================ package task import ( "net/http" "strconv" "strings" "github.com/gin-gonic/gin" "github.com/gocronx-team/cron" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/httpclient" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/routers/user" "github.com/gocronx-team/gocron/internal/service" ) type TaskForm struct { Id int `form:"id" json:"id"` Level models.TaskLevel `form:"level" json:"level" binding:"required,oneof=1 2"` DependencyStatus models.TaskDependencyStatus `form:"dependency_status" json:"dependency_status" binding:"oneof=1 2"` DependencyTaskId string `form:"dependency_task_id" json:"dependency_task_id"` Name string `form:"name" json:"name" binding:"required,max=32"` Spec string `form:"spec" json:"spec"` Protocol models.TaskProtocol `form:"protocol" json:"protocol" binding:"oneof=1 2"` Command string `form:"command" json:"command" binding:"required,max=65535"` HttpMethod models.TaskHTTPMethod `form:"http_method" json:"http_method" binding:"oneof=1 2"` HttpBody string `form:"http_body" json:"http_body" binding:"max=65535"` HttpHeaders string `form:"http_headers" json:"http_headers" binding:"max=4096"` SuccessPattern string `form:"success_pattern" json:"success_pattern" binding:"max=512"` Timeout int `form:"timeout" json:"timeout" binding:"min=0,max=86400"` Multi int8 `form:"multi" json:"multi" binding:"oneof=0 1"` RetryTimes int8 `form:"retry_times" json:"retry_times"` RetryInterval int16 `form:"retry_interval" json:"retry_interval"` HostId string `form:"host_id" json:"host_id"` Tag string `form:"tag" json:"tag"` Remark string `form:"remark" json:"remark"` NotifyStatus int8 `form:"notify_status" json:"notify_status" binding:"oneof=0 1 2 3"` NotifyType int8 `form:"notify_type" json:"notify_type" binding:"oneof=0 1 2"` NotifyReceiverId string `form:"notify_receiver_id" json:"notify_receiver_id"` NotifyKeyword string `form:"notify_keyword" json:"notify_keyword"` LogRetentionDays int `form:"log_retention_days" json:"log_retention_days" binding:"min=0,max=3650"` } // 首页 func Index(c *gin.Context) { taskModel := new(models.Task) queryParams := parseQueryParams(c) total, err := taskModel.Total(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } tasks, err := taskModel.List(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } for i, item := range tasks { tasks[i].NextRunTime = models.NextRunTime(service.ServiceTask.NextRunTime(item)) } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, map[string]interface{}{ "total": total, "data": tasks, }) c.String(http.StatusOK, result) } // Detail 任务详情 func Detail(c *gin.Context) { id, err := strconv.Atoi(c.Param("id")) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) task, err := taskModel.Detail(id) jsonResp := utils.JsonResponse{} var result string if err != nil || task.Id == 0 { logger.Errorf("编辑任务#获取任务详情失败#任务ID-%d", id) result = jsonResp.Success(utils.SuccessContent, nil) } else { result = jsonResp.Success(utils.SuccessContent, task) } c.String(http.StatusOK, result) } // 保存任务 func Store(c *gin.Context) { var form TaskForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } taskModel := models.Task{} var id = form.Id nameExists, err := taskModel.NameExist(form.Name, form.Id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } if nameExists { base.RespondError(c, i18n.T(c, "task_name_exists")) return } if form.Protocol == models.TaskRPC && form.HostId == "" { base.RespondError(c, i18n.T(c, "select_hostname")) return } taskModel.Name = form.Name taskModel.Protocol = form.Protocol // 清理命令中的 HTML 实体编码 originalCmd := strings.TrimSpace(form.Command) cleanedCmd := utils.CleanHTMLEntities(originalCmd) if originalCmd != cleanedCmd { logger.Infof("[HTML Entity Cleaned] Task: %s, Original length: %d, Cleaned length: %d", form.Name, len(originalCmd), len(cleanedCmd)) } taskModel.Command = cleanedCmd taskModel.Timeout = form.Timeout taskModel.Tag = form.Tag taskModel.Remark = form.Remark taskModel.Multi = form.Multi taskModel.RetryTimes = form.RetryTimes taskModel.RetryInterval = form.RetryInterval taskModel.NotifyStatus = form.NotifyStatus taskModel.NotifyType = form.NotifyType taskModel.NotifyReceiverId = form.NotifyReceiverId taskModel.NotifyKeyword = form.NotifyKeyword taskModel.LogRetentionDays = form.LogRetentionDays taskModel.Spec = form.Spec taskModel.Level = form.Level taskModel.DependencyStatus = form.DependencyStatus taskModel.DependencyTaskId = strings.TrimSpace(form.DependencyTaskId) if taskModel.NotifyStatus > 0 && taskModel.NotifyType != 2 && taskModel.NotifyReceiverId == "" { base.RespondError(c, i18n.T(c, "select_at_least_one_receiver")) return } taskModel.HttpMethod = form.HttpMethod // 校验 HttpHeaders（JSON 格式 + 黑名单检查） if err := httpclient.ValidateHeaders(form.HttpHeaders); err != nil { base.RespondError(c, "http_headers: "+err.Error()) return } taskModel.HttpBody = form.HttpBody taskModel.HttpHeaders = form.HttpHeaders taskModel.SuccessPattern = form.SuccessPattern if taskModel.Protocol == models.TaskHTTP { command := strings.ToLower(taskModel.Command) if !strings.HasPrefix(command, "http://") && !strings.HasPrefix(command, "https://") { base.RespondError(c, i18n.T(c, "invalid_url")) return } } if taskModel.RetryTimes > 10 || taskModel.RetryTimes < 0 { base.RespondError(c, i18n.T(c, "retry_times_range_0_10")) return } if taskModel.RetryInterval > 3600 || taskModel.RetryInterval < 0 { base.RespondError(c, i18n.T(c, "retry_interval_range_0_3600")) return } if taskModel.DependencyStatus != models.TaskDependencyStatusStrong && taskModel.DependencyStatus != models.TaskDependencyStatusWeak { base.RespondError(c, i18n.T(c, "select_dependency")) return } if taskModel.Level == models.TaskLevelParent { err = utils.PanicToError(func() { cron.Parse(form.Spec) }) if err != nil { base.RespondError(c, i18n.T(c, "crontab_parse_failed"), err) return } } else { taskModel.DependencyTaskId = "" taskModel.Spec = "" } if id > 0 && taskModel.DependencyTaskId != "" { dependencyTaskIds := strings.Split(taskModel.DependencyTaskId, ",") if utils.InStringSlice(dependencyTaskIds, strconv.Itoa(id)) { base.RespondError(c, i18n.T(c, "cannot_set_self_as_child")) return } } if id == 0 { taskModel.Status = models.Running logger.Infof("[Task Create] Before Create - Multi: %d", taskModel.Multi) id, err = taskModel.Create() if err == nil { // 立即读取验证 verifyTask, _ := taskModel.Detail(id) logger.Infof("[Task Create] After Create - ID: %d, Multi in DB: %d", id, verifyTask.Multi) } } else { // 更新前记录旧值用于审计 diff oldTask, _ := taskModel.Detail(id) // 保存脚本版本（命令变更时） if oldTask.Command != taskModel.Command { versionModel := new(models.TaskScriptVersion) latestVersion, _ := versionModel.GetLatestVersion(id) newVersion := &models.TaskScriptVersion{ TaskId: id, Command: oldTask.Command, Username: user.Username(c), Version: latestVersion + 1, } if _, vErr := newVersion.Create(); vErr != nil { logger.Warnf("保存脚本版本失败 TaskID-%d: %v", id, vErr) } if cErr := versionModel.CleanOldVersions(id, 30); cErr != nil { logger.Warnf("清理旧版本失败 TaskID-%d: %v", id, cErr) } } logger.Infof("[Task Update] Before Update - ID: %d, Multi: %d", id, taskModel.Multi) _, err = taskModel.UpdateBean(id) if err == nil { // 立即读取验证 verifyTask, _ := taskModel.Detail(id) logger.Infof("[Task Update] After Update - ID: %d, Multi in DB: %d", id, verifyTask.Multi) // 生成审计 diff if diff := buildTaskDiff(oldTask, verifyTask); diff != "" { c.Set("audit_detail", diff) } } } if err != nil { base.RespondError(c, i18n.T(c, "save_failed"), err) return } taskHostModel := new(models.TaskHost) if form.Protocol == models.TaskRPC { hostIdStrList := strings.Split(form.HostId, ",") hostIds := make([]int, len(hostIdStrList)) for i, hostIdStr := range hostIdStrList { hostIds[i], _ = strconv.Atoi(hostIdStr) } _ = taskHostModel.Add(id, hostIds) } else { _ = taskHostModel.Remove(id) } status, _ := taskModel.GetStatus(id) if status == models.Enabled && taskModel.Level == models.TaskLevelParent { addTaskToTimer(id) } base.RespondSuccess(c, i18n.T(c, "save_success"), nil) } // 删除任务 func Remove(c *gin.Context) { id, err := strconv.Atoi(c.Param("id")) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) _, err = taskModel.Delete(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { taskHostModel := new(models.TaskHost) _ = taskHostModel.Remove(id) service.ServiceTask.Remove(id) base.RespondSuccessWithDefaultMsg(c, nil) } } // 激活任务 func Enable(c *gin.Context) { changeStatus(c, models.Enabled) } // 暂停任务 func Disable(c *gin.Context) { changeStatus(c, models.Disabled) } // 手动运行任务 func Run(c *gin.Context) { id, err := strconv.Atoi(c.Param("id")) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) task, err := taskModel.Detail(id) if err != nil || task.Id <= 0 { base.RespondError(c, i18n.T(c, "get_task_detail_failed"), err) } else { task.Spec = i18n.T(c, "manual_run") service.ServiceTask.Run(task) base.RespondSuccess(c, i18n.T(c, "task_started_check_log"), nil) } } // 批量启用任务 func BatchEnable(c *gin.Context) { batchChangeStatus(c, models.Enabled) } // 批量禁用任务 func BatchDisable(c *gin.Context) { batchChangeStatus(c, models.Disabled) } // 批量改变任务状态 func batchChangeStatus(c *gin.Context, status models.Status) { var form struct { Ids []int `json:"ids" binding:"required"` } if err := c.ShouldBindJSON(&form); err != nil { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) successCount := 0 for _, id := range form.Ids { _, err := taskModel.Update(id, models.CommonMap{ "status": status, }) if err == nil { successCount++ if status == models.Enabled { addTaskToTimer(id) } else { service.ServiceTask.Remove(id) } } } base.RespondSuccess(c, i18n.T(c, "operation_success"), map[string]interface{}{ "success_count": successCount, "total_count": len(form.Ids), }) } // 批量删除任务 func BatchRemove(c *gin.Context) { var form struct { Ids []int `json:"ids" binding:"required"` } if err := c.ShouldBindJSON(&form); err != nil { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) taskHostModel := new(models.TaskHost) successCount := 0 for _, id := range form.Ids { _, err := taskModel.Delete(id) if err == nil { successCount++ _ = taskHostModel.Remove(id) service.ServiceTask.Remove(id) } } base.RespondSuccess(c, i18n.T(c, "operation_success"), map[string]interface{}{ "success_count": successCount, "total_count": len(form.Ids), }) } // 改变任务状态 func changeStatus(c *gin.Context, status models.Status) { id, err := strconv.Atoi(c.Param("id")) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } taskModel := new(models.Task) _, err = taskModel.Update(id, models.CommonMap{ "status": status, }) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { if status == models.Enabled { addTaskToTimer(id) } else { service.ServiceTask.Remove(id) } base.RespondSuccessWithDefaultMsg(c, nil) } } // 添加任务到定时器 func addTaskToTimer(id int) { taskModel := new(models.Task) task, err := taskModel.Detail(id) if err != nil { logger.Error(err) return } service.ServiceTask.RemoveAndAdd(task) } // GetAllTags 获取所有已使用的标签列表 func GetAllTags(c *gin.Context) { taskModel := new(models.Task) tags, err := taskModel.GetAllTags() if err != nil { logger.Error(err) tags = []string{} } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, tags) c.String(http.StatusOK, result) } // 解析查询参数 func parseQueryParams(c *gin.Context) models.CommonMap { var params models.CommonMap = models.CommonMap{} id, _ := strconv.Atoi(c.Query("id")) hostId, _ := strconv.Atoi(c.Query("host_id")) protocol, _ := strconv.Atoi(c.Query("protocol")) status, _ := strconv.Atoi(c.Query("status")) params["Id"] = id params["HostId"] = hostId params["Name"] = strings.TrimSpace(c.Query("name")) params["Protocol"] = protocol params["Tag"] = strings.TrimSpace(c.Query("tag")) if status >= 0 { status -= 1 } params["Status"] = status base.ParsePageAndPageSize(c, params) return params } // buildTaskDiff 对比任务的旧值和新值，返回可读的变更摘要 func buildTaskDiff(old, new models.Task) string { type change struct { Field string `json:"field"` Old string `json:"old"` New string `json:"new"` } var changes []change add := func(field, oldVal, newVal string) { if oldVal != newVal { changes = append(changes, change{field, oldVal, newVal}) } } add("name", old.Name, new.Name) add("spec", old.Spec, new.Spec) add("command", old.Command, new.Command) add("tag", old.Tag, new.Tag) add("timeout", strconv.Itoa(old.Timeout), strconv.Itoa(new.Timeout)) add("retry_times", strconv.Itoa(int(old.RetryTimes)), strconv.Itoa(int(new.RetryTimes))) add("retry_interval", strconv.Itoa(int(old.RetryInterval)), strconv.Itoa(int(new.RetryInterval))) add("remark", old.Remark, new.Remark) add("http_method", strconv.Itoa(int(old.HttpMethod)), strconv.Itoa(int(new.HttpMethod))) add("http_body", old.HttpBody, new.HttpBody) add("http_headers", old.HttpHeaders, new.HttpHeaders) add("success_pattern", old.SuccessPattern, new.SuccessPattern) add("notify_status", strconv.Itoa(int(old.NotifyStatus)), strconv.Itoa(int(new.NotifyStatus))) add("notify_keyword", old.NotifyKeyword, new.NotifyKeyword) add("log_retention_days", strconv.Itoa(old.LogRetentionDays), strconv.Itoa(new.LogRetentionDays)) if len(changes) == 0 { return "" } // 生成简洁的文本格式 var b strings.Builder for i, ch := range changes { if i > 0 { b.WriteString("\n") } b.WriteString(ch.Field) b.WriteString(": ") b.WriteString(ch.Old) b.WriteString(" → ") b.WriteString(ch.New) } return b.String() } ================================================ FILE: internal/routers/task/task_tag_test.go ================================================ package task import ( "encoding/json" "net/http" "net/http/httptest" "testing" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/schema" ) func setupTestRouter(t *testing.T) (*gin.Engine, func()) { t.Helper() gin.SetMode(gin.TestMode) originalDb := models.Db db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ NamingStrategy: schema.NamingStrategy{ SingularTable: true, }, }) if err != nil { t.Fatalf("failed to open test database: %v", err) } err = db.AutoMigrate(&models.Task{}) if err != nil { t.Fatalf("failed to migrate test database: %v", err) } models.Db = db r := gin.New() r.GET("/api/task/tags", GetAllTags) cleanup := func() { models.Db = originalDb } return r, cleanup } type apiResponse struct { Code int `json:"code"` Message string `json:"message"` Data json.RawMessage `json:"data"` } func TestGetAllTagsHandler_Empty(t *testing.T) { r, cleanup := setupTestRouter(t) defer cleanup() w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/task/tags", nil) r.ServeHTTP(w, req) if w.Code != http.StatusOK { t.Fatalf("expected status 200, got %d", w.Code) } var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } if resp.Code != 0 { t.Errorf("expected code 0, got %d", resp.Code) } var tags []string if err := json.Unmarshal(resp.Data, &tags); err != nil { t.Fatalf("failed to parse tags data: %v", err) } if len(tags) != 0 { t.Errorf("expected empty tags, got %v", tags) } } func TestGetAllTagsHandler_WithTags(t *testing.T) { r, cleanup := setupTestRouter(t) defer cleanup() // Insert test data tasks := []map[string]interface{}{ {"name": "task1", "tag": "alpha,beta", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 1", "status": 1}, {"name": "task2", "tag": "beta,gamma", "level": 1, "spec": "* * * * *", "protocol": 1, "command": "echo 2", "status": 1}, } for _, data := range tasks { if err := models.Db.Model(&models.Task{}).Create(data).Error; err != nil { t.Fatalf("failed to create task: %v", err) } } w := httptest.NewRecorder() req, _ := http.NewRequest("GET", "/api/task/tags", nil) r.ServeHTTP(w, req) if w.Code != http.StatusOK { t.Fatalf("expected status 200, got %d", w.Code) } var resp apiResponse if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil { t.Fatalf("failed to parse response: %v", err) } if resp.Code != 0 { t.Errorf("expected code 0, got %d", resp.Code) } var tags []string if err := json.Unmarshal(resp.Data, &tags); err != nil { t.Fatalf("failed to parse tags data: %v", err) } expected := []string{"alpha", "beta", "gamma"} if len(tags) != len(expected) { t.Fatalf("expected %d tags, got %d: %v", len(expected), len(tags), tags) } for i, tag := range tags { if tag != expected[i] { t.Errorf("expected tag[%d] = %q, got %q", i, expected[i], tag) } } } ================================================ FILE: internal/routers/task/task_version.go ================================================ package task import ( "net/http" "strconv" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/routers/user" "github.com/gocronx-team/gocron/internal/service" "gorm.io/gorm" ) // VersionList 获取任务脚本版本列表 func VersionList(c *gin.Context) { taskId, _ := strconv.Atoi(c.Param("id")) if taskId <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } versionModel := new(models.TaskScriptVersion) params := models.CommonMap{} base.ParsePageAndPageSize(c, params) total, err := versionModel.Total(taskId) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } list, err := versionModel.List(taskId, params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, map[string]interface{}{ "total": total, "data": list, }) c.String(http.StatusOK, result) } // VersionDetail 获取单个版本详情 func VersionDetail(c *gin.Context) { taskId, _ := strconv.Atoi(c.Param("id")) versionId, _ := strconv.Atoi(c.Param("version_id")) if taskId <= 0 || versionId <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } versionModel := new(models.TaskScriptVersion) version, err := versionModel.Detail(versionId) if err != nil || version.TaskId != taskId { base.RespondError(c, i18n.T(c, "version_not_found")) return } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, version) c.String(http.StatusOK, result) } // VersionRollback 回滚任务命令到指定版本 func VersionRollback(c *gin.Context) { taskId, _ := strconv.Atoi(c.Param("id")) versionId, _ := strconv.Atoi(c.Param("version_id")) if taskId <= 0 || versionId <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } versionModel := new(models.TaskScriptVersion) version, err := versionModel.Detail(versionId) if err != nil || version.TaskId != taskId { base.RespondError(c, i18n.T(c, "version_not_found")) return } taskModel := new(models.Task) currentTask, err := taskModel.Detail(taskId) if err != nil || currentTask.Id == 0 { base.RespondError(c, i18n.T(c, "get_task_detail_failed")) return } // 使用事务保证回滚操作的原子性 txErr := models.Db.Transaction(func(tx *gorm.DB) error { // 回滚前保存当前命令为新版本 if currentTask.Command != version.Command { latestVersion, _ := versionModel.GetLatestVersion(taskId) saveVersion := &models.TaskScriptVersion{ TaskId: taskId, Command: currentTask.Command, Remark: "auto-save before rollback", Username: user.Username(c), Version: latestVersion + 1, } if err := tx.Create(saveVersion).Error; err != nil { logger.Warnf("回滚前保存版本失败 TaskID-%d: %v", taskId, err) } } // 更新任务命令 return tx.Model(&models.Task{}).Where("id = ?", taskId). UpdateColumn("command", version.Command).Error }) if txErr != nil { base.RespondError(c, i18n.T(c, "rollback_failed"), txErr) return } // 事务完成后清理旧版本（非关键操作，不需要在事务内） if cErr := versionModel.CleanOldVersions(taskId, 30); cErr != nil { logger.Warnf("清理旧版本失败 TaskID-%d: %v", taskId, cErr) } // 重新加入调度器 status, _ := taskModel.GetStatus(taskId) if status == models.Enabled { task, _ := taskModel.Detail(taskId) service.ServiceTask.RemoveAndAdd(task) } base.RespondSuccess(c, i18n.T(c, "rollback_success"), nil) } ================================================ FILE: internal/routers/tasklog/task_log.go ================================================ package tasklog // 任务日志 import ( "strconv" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/service" ) func Index(c *gin.Context) { logModel := new(models.TaskLog) queryParams := parseQueryParams(c) total, err := logModel.Total(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } logs, err := logModel.List(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "total": total, "data": logs, }) } // 清空日志 func Clear(c *gin.Context) { taskLogModel := new(models.TaskLog) _, err := taskLogModel.Clear() if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } // 停止运行中的任务 func Stop(c *gin.Context) { id, err := strconv.ParseInt(c.PostForm("id"), 10, 64) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "invalid_log_id")) return } taskId, err := strconv.Atoi(c.PostForm("task_id")) if err != nil || taskId <= 0 { base.RespondError(c, i18n.T(c, "invalid_task_id")) return } taskModel := new(models.Task) task, err := taskModel.Detail(taskId) if err != nil { base.RespondError(c, i18n.T(c, "get_task_info_failed")+"#"+err.Error(), err) return } if task.Protocol != models.TaskRPC { base.RespondError(c, i18n.T(c, "only_shell_task_can_stop")) return } if len(task.Hosts) == 0 { base.RespondError(c, i18n.T(c, "task_node_list_empty")) return } for _, host := range task.Hosts { service.ServiceTask.Stop(host.Name, host.Port, id) } base.RespondSuccess(c, i18n.T(c, "stop_task_sent"), nil) } // 删除N个月前的日志 func Remove(c *gin.Context) { month, _ := strconv.Atoi(c.Param("id")) if month < 1 || month > 12 { base.RespondError(c, i18n.T(c, "param_range_1_12")) return } taskLogModel := new(models.TaskLog) _, err := taskLogModel.Remove(month) if err != nil { base.RespondError(c, i18n.T(c, "delete_failed"), err) } else { base.RespondSuccess(c, i18n.T(c, "delete_success"), nil) } } // 清空指定任务的日志 func ClearByTaskId(c *gin.Context) { id, err := strconv.Atoi(c.Param("id")) if err != nil || id <= 0 { base.RespondError(c, i18n.T(c, "invalid_task_id")) return } taskLogModel := new(models.TaskLog) affected, err := taskLogModel.ClearByTaskId(id) if err != nil { base.RespondError(c, i18n.T(c, "delete_failed"), err) } else { base.RespondSuccess(c, i18n.T(c, "delete_success"), map[string]interface{}{ "affected": affected, }) } } // 解析查询参数 func parseQueryParams(c *gin.Context) models.CommonMap { var params models.CommonMap = models.CommonMap{} taskId, _ := strconv.Atoi(c.Query("task_id")) protocol, _ := strconv.Atoi(c.Query("protocol")) status, _ := strconv.Atoi(c.Query("status")) params["TaskId"] = taskId params["Protocol"] = protocol if status >= 0 { status -= 1 } params["Status"] = status base.ParsePageAndPageSize(c, params) return params } ================================================ FILE: internal/routers/tasklog/task_log_test.go ================================================ package tasklog import ( "net/http" "net/http/httptest" "strings" "testing" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" ) func init() { gin.SetMode(gin.TestMode) } func setupTestDb(t *testing.T) { t.Helper() db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatalf("failed to open in-memory sqlite: %v", err) } err = db.AutoMigrate(&models.TaskLog{}) if err != nil { t.Fatalf("failed to migrate: %v", err) } models.Db = db } func TestClearByTaskId_InvalidId(t *testing.T) { tests := []struct { name string id string }{ {"non-numeric", "abc"}, {"negative", "-1"}, {"zero", "0"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { w := httptest.NewRecorder() c, r := gin.CreateTestContext(w) r.POST("/api/task/log/clear/:id", ClearByTaskId) c.Request, _ = http.NewRequest("POST", "/api/task/log/clear/"+tt.id, nil) r.ServeHTTP(w, c.Request) body := w.Body.String() if w.Code != http.StatusOK { t.Errorf("expected status 200, got %d", w.Code) } // The response should indicate failure (code != 0) if !strings.Contains(body, `"code"`) { t.Errorf("expected JSON response with code field, got: %s", body) } // Should not contain success indicators for invalid input if strings.Contains(body, `"code":0`) { t.Errorf("expected error response for invalid id %q, got success: %s", tt.id, body) } }) } } func TestClearByTaskId_ValidId(t *testing.T) { setupTestDb(t) w := httptest.NewRecorder() _, r := gin.CreateTestContext(w) r.POST("/api/task/log/clear/:id", ClearByTaskId) req, _ := http.NewRequest("POST", "/api/task/log/clear/1", nil) r.ServeHTTP(w, req) body := w.Body.String() if w.Code != http.StatusOK { t.Errorf("expected status 200, got %d", w.Code) } // Should contain a successful JSON response if !strings.Contains(body, `"code":0`) { t.Errorf("expected success response for valid id, got: %s", body) } } ================================================ FILE: internal/routers/template/template.go ================================================ package template import ( "net/http" "strconv" "strings" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/gocronx-team/gocron/internal/routers/user" ) type TemplateForm struct { Id int `form:"id" json:"id"` Name string `form:"name" json:"name" binding:"required,max=64"` Description string `form:"description" json:"description" binding:"max=500"` Category string `form:"category" json:"category" binding:"required,max=32"` Protocol int8 `form:"protocol" json:"protocol" binding:"oneof=1 2"` Command string `form:"command" json:"command" binding:"required,max=65535"` HttpMethod int8 `form:"http_method" json:"http_method" binding:"oneof=1 2"` HttpBody string `form:"http_body" json:"http_body"` HttpHeaders string `form:"http_headers" json:"http_headers"` SuccessPattern string `form:"success_pattern" json:"success_pattern" binding:"max=512"` Tag string `form:"tag" json:"tag"` Spec string `form:"spec" json:"spec"` Timeout int `form:"timeout" json:"timeout" binding:"min=0,max=86400"` Multi int8 `form:"multi" json:"multi" binding:"oneof=0 1"` RetryTimes int8 `form:"retry_times" json:"retry_times"` RetryInterval int16 `form:"retry_interval" json:"retry_interval"` Timezone string `form:"timezone" json:"timezone"` NotifyStatus int8 `form:"notify_status" json:"notify_status"` NotifyType int8 `form:"notify_type" json:"notify_type"` NotifyKeyword string `form:"notify_keyword" json:"notify_keyword"` LogRetentionDays int `form:"log_retention_days" json:"log_retention_days" binding:"min=0,max=3650"` } type SaveFromTaskForm struct { TaskId int `form:"task_id" json:"task_id" binding:"required"` Name string `form:"name" json:"name" binding:"required,max=64"` Description string `form:"description" json:"description" binding:"max=500"` Category string `form:"category" json:"category" binding:"required,max=32"` } // Index 模板列表 func Index(c *gin.Context) { tmplModel := new(models.TaskTemplate) params := parseQueryParams(c) total, err := tmplModel.Total(params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } list, err := tmplModel.List(params) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, map[string]interface{}{ "total": total, "data": list, }) c.String(http.StatusOK, result) } // Detail 模板详情 func Detail(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) if id <= 0 { base.RespondError(c, i18n.T(c, "param_error")) return } tmplModel := new(models.TaskTemplate) tmpl, err := tmplModel.Detail(id) if err != nil || tmpl.Id == 0 { base.RespondError(c, i18n.T(c, "template_not_found")) return } jsonResp := utils.JsonResponse{} c.String(http.StatusOK, jsonResp.Success(utils.SuccessContent, tmpl)) } // Store 创建/更新模板 func Store(c *gin.Context) { var form TemplateForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } tmplModel := models.TaskTemplate{} id := form.Id // 内置模板不可修改 if id > 0 { existing, detailErr := tmplModel.Detail(id) if detailErr != nil || existing.Id == 0 { base.RespondError(c, i18n.T(c, "template_not_found")) return } if existing.IsBuiltin == 1 { base.RespondError(c, i18n.T(c, "builtin_template_readonly")) return } } nameExists, err := tmplModel.NameExist(form.Name, id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } if nameExists { base.RespondError(c, i18n.T(c, "template_name_exists")) return } tmplModel.Name = form.Name tmplModel.Description = form.Description tmplModel.Category = form.Category tmplModel.Protocol = form.Protocol tmplModel.Command = form.Command tmplModel.HttpMethod = form.HttpMethod tmplModel.HttpBody = form.HttpBody tmplModel.HttpHeaders = form.HttpHeaders tmplModel.SuccessPattern = form.SuccessPattern tmplModel.Tag = form.Tag tmplModel.Spec = form.Spec tmplModel.Timeout = form.Timeout tmplModel.Multi = form.Multi tmplModel.RetryTimes = form.RetryTimes tmplModel.RetryInterval = form.RetryInterval tmplModel.Timezone = form.Timezone tmplModel.NotifyStatus = form.NotifyStatus tmplModel.NotifyType = form.NotifyType tmplModel.NotifyKeyword = form.NotifyKeyword tmplModel.LogRetentionDays = form.LogRetentionDays if id == 0 { tmplModel.CreatedBy = user.Username(c) _, err = tmplModel.Create() } else { _, err = tmplModel.UpdateBean(id) } if err != nil { base.RespondError(c, i18n.T(c, "save_failed"), err) return } base.RespondSuccess(c, i18n.T(c, "save_success"), nil) } // Remove 删除模板 func Remove(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) tmplModel := new(models.TaskTemplate) tmpl, err := tmplModel.Detail(id) if err != nil || tmpl.Id == 0 { base.RespondError(c, i18n.T(c, "template_not_found")) return } if tmpl.IsBuiltin == 1 { base.RespondError(c, i18n.T(c, "builtin_template_no_delete")) return } _, err = tmplModel.Delete(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccessWithDefaultMsg(c, nil) } // Apply 应用模板（增加使用次数并返回模板数据） func Apply(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) tmplModel := new(models.TaskTemplate) tmpl, err := tmplModel.Detail(id) if err != nil || tmpl.Id == 0 { base.RespondError(c, i18n.T(c, "template_not_found")) return } if uErr := tmplModel.IncrementUsage(id); uErr != nil { logger.Warnf("增加模板使用次数失败 TemplateID-%d: %v", id, uErr) } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, tmpl) c.String(http.StatusOK, result) } // SaveFromTask 从现有任务保存为模板 func SaveFromTask(c *gin.Context) { var form SaveFromTaskForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } taskModel := new(models.Task) task, err := taskModel.Detail(form.TaskId) if err != nil || task.Id == 0 { base.RespondError(c, i18n.T(c, "task_not_found")) return } tmplModel := models.TaskTemplate{} nameExists, err := tmplModel.NameExist(form.Name, 0) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } if nameExists { base.RespondError(c, i18n.T(c, "template_name_exists")) return } tmplModel.Name = form.Name tmplModel.Description = form.Description tmplModel.Category = form.Category tmplModel.Protocol = int8(task.Protocol) tmplModel.Command = task.Command tmplModel.HttpMethod = int8(task.HttpMethod) tmplModel.HttpBody = task.HttpBody tmplModel.HttpHeaders = task.HttpHeaders tmplModel.SuccessPattern = task.SuccessPattern tmplModel.Tag = task.Tag // 从 spec 中解析 timezone（格式: CRON_TZ=Asia/Shanghai 0 0 2 * * *） spec := task.Spec if strings.HasPrefix(spec, "CRON_TZ=") || strings.HasPrefix(spec, "TZ=") { parts := strings.SplitN(spec, " ", 2) if len(parts) == 2 { tzPart := parts[0] spec = parts[1] tmplModel.Timezone = strings.SplitN(tzPart, "=", 2)[1] } } tmplModel.Spec = spec tmplModel.Timeout = task.Timeout tmplModel.Multi = task.Multi tmplModel.RetryTimes = task.RetryTimes tmplModel.RetryInterval = task.RetryInterval tmplModel.NotifyStatus = task.NotifyStatus tmplModel.NotifyType = task.NotifyType tmplModel.NotifyKeyword = task.NotifyKeyword tmplModel.LogRetentionDays = task.LogRetentionDays tmplModel.CreatedBy = user.Username(c) _, err = tmplModel.Create() if err != nil { base.RespondError(c, i18n.T(c, "save_failed"), err) return } base.RespondSuccess(c, i18n.T(c, "save_success"), nil) } // Categories 获取所有分类 func Categories(c *gin.Context) { tmplModel := new(models.TaskTemplate) categories, err := tmplModel.GetCategories() if err != nil { categories = []string{} } jsonResp := utils.JsonResponse{} result := jsonResp.Success(utils.SuccessContent, categories) c.String(http.StatusOK, result) } func parseQueryParams(c *gin.Context) models.CommonMap { params := models.CommonMap{} params["Category"] = strings.TrimSpace(c.Query("category")) params["Name"] = strings.TrimSpace(c.Query("name")) base.ParsePageAndPageSize(c, params) return params } ================================================ FILE: internal/routers/user/twofa.go ================================================ package user import ( "bytes" "encoding/base64" "image/png" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/pquerna/otp/totp" ) // Setup2FA 设置2FA func Setup2FA(c *gin.Context) { uid := Uid(c) username := Username(c) userModel := new(models.User) err := userModel.Find(uid) if err != nil || userModel.Id == 0 { base.RespondError(c, i18n.T(c, "user_not_found")) return } // 生成TOTP密钥 key, err := totp.Generate(totp.GenerateOpts{ Issuer: "Gocron", AccountName: username, }) if err != nil { logger.Error("生成2FA密钥失败", err) base.RespondError(c, i18n.T(c, "generate_2fa_key_failed")) return } // 生成二维码 img, err := key.Image(200, 200) if err != nil { logger.Error("生成二维码失败", err) base.RespondError(c, i18n.T(c, "generate_qrcode_failed")) return } // 将图片转为base64 var buf bytes.Buffer if err := png.Encode(&buf, img); err != nil { logger.Error("编码二维码失败", err) base.RespondError(c, i18n.T(c, "generate_qrcode_failed")) return } qrCode := base64.StdEncoding.EncodeToString(buf.Bytes()) base.RespondSuccess(c, i18n.T(c, "get_success"), map[string]interface{}{ "secret": key.Secret(), "qr_code": "data:image/png;base64," + qrCode, }) } // Enable2FAForm 启用2FA表单 type Enable2FAForm struct { Secret string `form:"secret" json:"secret" binding:"required"` Code string `form:"code" json:"code" binding:"required,len=6"` } // Enable2FA 启用2FA func Enable2FA(c *gin.Context) { var form Enable2FAForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } uid := Uid(c) // 验证TOTP码 valid := totp.Validate(form.Code, form.Secret) if !valid { base.RespondError(c, i18n.T(c, "verification_code_error")) return } // 保存密钥并启用2FA userModel := new(models.User) _, err := userModel.Update(uid, models.CommonMap{ "two_factor_key": form.Secret, "two_factor_on": 1, }) if err != nil { base.RespondError(c, i18n.T(c, "enable_failed"), err) return } base.RespondSuccess(c, i18n.T(c, "2fa_enabled"), nil) } // Disable2FAForm 禁用2FA表单 type Disable2FAForm struct { Code string `form:"code" json:"code" binding:"required,len=6"` } // Disable2FA 禁用2FA func Disable2FA(c *gin.Context) { var form Disable2FAForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } uid := Uid(c) userModel := new(models.User) err := userModel.Find(uid) if err != nil || userModel.Id == 0 { base.RespondError(c, i18n.T(c, "user_not_found")) return } if userModel.TwoFactorOn == 0 { base.RespondError(c, i18n.T(c, "2fa_not_enabled")) return } // 验证TOTP码 valid := totp.Validate(form.Code, userModel.TwoFactorKey) if !valid { base.RespondError(c, i18n.T(c, "verification_code_error")) return } // 禁用2FA _, err = userModel.Update(uid, models.CommonMap{ "two_factor_key": "", "two_factor_on": 0, }) if err != nil { base.RespondError(c, i18n.T(c, "disable_failed"), err) return } base.RespondSuccess(c, i18n.T(c, "2fa_disabled"), nil) } // Get2FAStatus 获取2FA状态 func Get2FAStatus(c *gin.Context) { uid := Uid(c) userModel := new(models.User) err := userModel.Find(uid) if err != nil || userModel.Id == 0 { base.RespondError(c, i18n.T(c, "user_not_found")) return } base.RespondSuccess(c, i18n.T(c, "get_success"), map[string]interface{}{ "enabled": userModel.TwoFactorOn == 1, }) } ================================================ FILE: internal/routers/user/user.go ================================================ package user import ( "errors" "fmt" "strconv" "strings" "time" "github.com/gin-gonic/gin" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/i18n" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/utils" "github.com/gocronx-team/gocron/internal/routers/base" "github.com/golang-jwt/jwt/v5" "github.com/pquerna/otp/totp" ) const tokenDuration = 4 * time.Hour // UserForm 用户表单 type UserForm struct { Id int `form:"id" json:"id"` Name string `form:"name" json:"name" binding:"required,max=32"` // 用户名 Password string `form:"password" json:"password"` // 密码 ConfirmPassword string `form:"confirm_password" json:"confirm_password"` // 确认密码 Email string `form:"email" json:"email" binding:"required,email,max=50"` // 邮箱 IsAdmin int8 `form:"is_admin" json:"is_admin"` // 是否是管理员 1:管理员 0:普通用户 Status models.Status `form:"status" json:"status"` } // UpdatePasswordForm 更新密码表单 type UpdatePasswordForm struct { NewPassword string `form:"new_password" json:"new_password" binding:"required,min=6"` ConfirmNewPassword string `form:"confirm_new_password" json:"confirm_new_password" binding:"required,min=6"` } // UpdateMyPasswordForm 更新我的密码表单 type UpdateMyPasswordForm struct { OldPassword string `form:"old_password" json:"old_password" binding:"required"` NewPassword string `form:"new_password" json:"new_password" binding:"required,min=6"` ConfirmNewPassword string `form:"confirm_new_password" json:"confirm_new_password" binding:"required,min=6"` } // Index 用户列表页 func Index(c *gin.Context) { queryParams := parseQueryParams(c) userModel := new(models.User) users, err := userModel.List(queryParams) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } total, err := userModel.Total() if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "total": total, "data": users, }) } // 解析查询参数 func parseQueryParams(c *gin.Context) models.CommonMap { params := models.CommonMap{} base.ParsePageAndPageSize(c, params) return params } // Detail 用户详情 func Detail(c *gin.Context) { userModel := new(models.User) id, _ := strconv.Atoi(c.Param("id")) err := userModel.Find(id) if err != nil { logger.Error(err) } if userModel.Id == 0 { base.RespondSuccess(c, utils.SuccessContent, nil) } else { base.RespondSuccess(c, utils.SuccessContent, userModel) } } // 保存任务 func Store(c *gin.Context) { var form UserForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } form.Name = strings.TrimSpace(form.Name) form.Email = strings.TrimSpace(form.Email) form.Password = strings.TrimSpace(form.Password) form.ConfirmPassword = strings.TrimSpace(form.ConfirmPassword) userModel := models.User{} nameExists, err := userModel.UsernameExists(form.Name, form.Id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } if nameExists > 0 { base.RespondError(c, i18n.T(c, "username_exists")) return } emailExists, err := userModel.EmailExists(form.Email, form.Id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) return } if emailExists > 0 { base.RespondError(c, i18n.T(c, "email_exists")) return } if form.Id == 0 { if form.Password == "" { base.RespondError(c, i18n.T(c, "password_required")) return } if form.ConfirmPassword == "" { base.RespondError(c, i18n.T(c, "password_confirm_required")) return } // 验证密码复杂度 if valid, errKey := utils.ValidatePassword(form.Password); !valid { base.RespondError(c, i18n.T(c, errKey)) return } if form.Password != form.ConfirmPassword { base.RespondError(c, i18n.T(c, "password_mismatch")) return } } userModel.Name = form.Name userModel.Email = form.Email userModel.Password = form.Password userModel.IsAdmin = form.IsAdmin userModel.Status = form.Status if form.Id == 0 { _, err = userModel.Create() if err != nil { base.RespondError(c, i18n.T(c, "save_failed"), err) return } } else { _, err = userModel.Update(form.Id, models.CommonMap{ "name": form.Name, "email": form.Email, "status": form.Status, "is_admin": form.IsAdmin, }) if err != nil { base.RespondError(c, i18n.T(c, "update_failed"), err) return } } base.RespondSuccess(c, i18n.T(c, "save_success"), nil) } // 删除用户 func Remove(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) userModel := new(models.User) _, err := userModel.Delete(id) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } // 激活用户 func Enable(c *gin.Context) { changeStatus(c, models.Enabled) } // 禁用用户 func Disable(c *gin.Context) { changeStatus(c, models.Disabled) } // 改变任务状态 func changeStatus(c *gin.Context, status models.Status) { id, _ := strconv.Atoi(c.Param("id")) userModel := new(models.User) _, err := userModel.Update(id, models.CommonMap{ "status": status, }) if err != nil { base.RespondErrorWithDefaultMsg(c, err) } else { base.RespondSuccessWithDefaultMsg(c, nil) } } // UpdatePassword 更新密码 func UpdatePassword(c *gin.Context) { id, _ := strconv.Atoi(c.Param("id")) var form UpdatePasswordForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } if form.NewPassword != form.ConfirmNewPassword { base.RespondError(c, i18n.T(c, "password_mismatch")) return } // 验证密码复杂度 if valid, errKey := utils.ValidatePassword(form.NewPassword); !valid { base.RespondError(c, i18n.T(c, errKey)) return } userModel := new(models.User) _, err := userModel.UpdatePassword(id, form.NewPassword) if err != nil { base.RespondError(c, i18n.T(c, "update_failed")) } else { base.RespondSuccess(c, i18n.T(c, "update_success"), nil) } } // UpdateMyPassword 更新我的密码 func UpdateMyPassword(c *gin.Context) { var form UpdateMyPasswordForm if err := c.ShouldBind(&form); err != nil { base.RespondValidationError(c, err) return } if form.NewPassword != form.ConfirmNewPassword { base.RespondError(c, i18n.T(c, "password_mismatch")) return } if form.OldPassword == form.NewPassword { base.RespondError(c, i18n.T(c, "password_same_as_old")) return } // 验证密码复杂度 if valid, errKey := utils.ValidatePassword(form.NewPassword); !valid { base.RespondError(c, i18n.T(c, errKey)) return } userModel := new(models.User) if !userModel.Match(Username(c), form.OldPassword) { base.RespondError(c, i18n.T(c, "old_password_error")) return } _, err := userModel.UpdatePassword(Uid(c), form.NewPassword) if err != nil { base.RespondError(c, i18n.T(c, "update_failed")) } else { base.RespondSuccess(c, i18n.T(c, "update_success"), nil) } } // ValidateLogin 验证用户登录 func ValidateLogin(c *gin.Context) { username := strings.TrimSpace(c.PostForm("username")) password := strings.TrimSpace(c.PostForm("password")) twoFactorCode := strings.TrimSpace(c.PostForm("two_factor_code")) if username == "" || password == "" { base.RespondError(c, i18n.T(c, "username_password_empty")) return } // 获取登录限制器 limiter := utils.GetLoginLimiter() // 检查账户是否被锁定 if locked, lockTime := limiter.IsLocked(username); locked { remainingTime := int(time.Until(lockTime).Minutes()) if remainingTime < 1 { remainingTime = 1 } base.RespondError(c, fmt.Sprintf(i18n.T(c, "account_locked"), remainingTime)) return } userModel := new(models.User) if !userModel.Match(username, password) { // 记录登录失败 limiter.RecordFailure(username) remaining := limiter.GetRemainingAttempts(username) if remaining > 0 { base.RespondError(c, fmt.Sprintf(i18n.T(c, "login_failed_with_attempts"), remaining)) } else { base.RespondError(c, i18n.T(c, "username_password_error")) } return } // 检查是否启用2FA if userModel.TwoFactorOn == 1 { if twoFactorCode == "" { base.RespondSuccess(c, i18n.T(c, "2fa_code_required"), map[string]interface{}{ "require_2fa": true, }) return } // 验证TOTP码 valid := totp.Validate(twoFactorCode, userModel.TwoFactorKey) if !valid { // 2FA验证失败也记录失败次数 limiter.RecordFailure(username) base.RespondError(c, i18n.T(c, "2fa_code_error")) return } } // 登录成功，清除失败记录 limiter.RecordSuccess(username) loginLogModel := new(models.LoginLog) loginLogModel.Username = userModel.Name ip := c.ClientIP() if ip == "::1" { ip = "127.0.0.1" } loginLogModel.Ip = ip _, err := loginLogModel.Create() if err != nil { logger.Error("记录用户登录日志失败", err) } token, err := generateToken(userModel) if err != nil { logger.Errorf("生成jwt失败: %s", err) base.RespondAuthError(c, i18n.T(c, "auth_failed")) return } base.RespondSuccess(c, utils.SuccessContent, map[string]interface{}{ "token": token, "uid": userModel.Id, "username": userModel.Name, "is_admin": userModel.IsAdmin, }) } // Username 获取session中的用户名 func Username(c *gin.Context) string { usernameInterface, ok := c.Get("username") if !ok { return "" } if username, ok := usernameInterface.(string); ok { return username } else { return "" } } // Uid 获取session中的Uid func Uid(c *gin.Context) int { uidInterface, ok := c.Get("uid") if !ok { return 0 } if uid, ok := uidInterface.(int); ok { return uid } else { return 0 } } // IsLogin 判断用户是否已登录 func IsLogin(c *gin.Context) bool { return Uid(c) > 0 } // IsAdmin 判断当前用户是否是管理员 func IsAdmin(c *gin.Context) bool { isAdmin, ok := c.Get("is_admin") if !ok { return false } if v, ok := isAdmin.(int); ok { return v > 0 } else { return false } } // 生成jwt func generateToken(user *models.User) (string, error) { claims := jwt.MapClaims{ "exp": time.Now().Add(tokenDuration).Unix(), "uid": user.Id, "iat": time.Now().Unix(), "issuer": "gocron", "username": user.Name, "is_admin": user.IsAdmin, } token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims) return token.SignedString([]byte(app.Setting.AuthSecret)) } // 还原jwt，如果 token 即将过期（小于1小时），则自动刷新 func RestoreToken(c *gin.Context) (string, error) { authToken := c.GetHeader("Auth-Token") if authToken == "" { return "", nil } token, err := jwt.Parse(authToken, func(token *jwt.Token) (interface{}, error) { if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok { return nil, errors.New("unexpected signing method") } return []byte(app.Setting.AuthSecret), nil }) if err != nil { return "", err } if !token.Valid { return "", errors.New("token is invalid") } claims, ok := token.Claims.(jwt.MapClaims) if !ok { return "", errors.New("invalid claims") } uidF, ok := claims["uid"].(float64) if !ok { return "", errors.New("invalid uid claim") } username, ok := claims["username"].(string) if !ok { return "", errors.New("invalid username claim") } isAdminF, ok := claims["is_admin"].(float64) if !ok { return "", errors.New("invalid is_admin claim") } expF, ok := claims["exp"].(float64) if !ok { return "", errors.New("invalid exp claim") } c.Set("uid", int(uidF)) c.Set("username", username) c.Set("is_admin", int(isAdminF)) // 检查 token 是否即将过期（小于 1 小时） exp := int64(expF) if time.Until(time.Unix(exp, 0)) < time.Hour { // 生成新 token userModel := &models.User{ Id: int(uidF), Name: username, IsAdmin: int8(isAdminF), } newToken, err := generateToken(userModel) if err != nil { logger.Warnf("刷新token失败: %v", err) return "", nil } logger.Infof("用户 %s 的 token 已自动刷新", userModel.Name) return newToken, nil } return "", nil } ================================================ FILE: internal/service/cron_preview.go ================================================ package service import ( "fmt" "strings" "time" "github.com/gocronx-team/cron" ) // Cron 预览：解析表达式，返回接下来 N 次执行时间 + 未来 7 天的执行分布热图。 // 设计要点： // - 复用后端 cron 库（和实际调度一致），避免前端重复实现造成语法漂移。 // - 迭代有硬上限，防 `* * * * * *` 这类极端表达式 DoS 服务。 // - 非法表达式不抛错，返回 Valid=false，让前端能平滑展示。 const ( maxHeatmapIterations = 2000 heatmapWindowHours = 24 * 7 maxNextRunsRequested = 20 defaultNextRuns = 10 maxSpecLength = 128 ) type CronRun struct { Unix int64 `json:"unix"` ISO string `json:"iso"` Weekday int `json:"weekday"` // 0=Sun..6=Sat } type HeatmapCell struct { Day int `json:"day"` Hour int `json:"hour"` Count int `json:"count"` } type CronPreviewResult struct { Valid bool `json:"valid"` Error string `json:"error,omitempty"` Timezone string `json:"timezone"` NowUnix int64 `json:"now_unix"` NextRuns []CronRun `json:"next_runs"` HeatmapCells []HeatmapCell `json:"heatmap_cells"` Truncated bool `json:"truncated,omitempty"` // heatmap 达到迭代上限提前截断 } // PreviewCron 以当前时间为基准计算预览。 func PreviewCron(spec, timezone string, count int) *CronPreviewResult { return previewCronAt(spec, timezone, count, time.Now()) } // previewCronAt 暴露 now 参数用于单测（时间可注入）。 func previewCronAt(spec, timezone string, count int, now time.Time) *CronPreviewResult { // 1. 输入清洗 spec = strings.TrimSpace(spec) result := &CronPreviewResult{ Timezone: timezone, NowUnix: now.Unix(), } if spec == "" { result.Error = "spec is required" return result } if strings.ContainsAny(spec, "\r\n") { result.Error = "spec must be single-line" return result } if len(spec) > maxSpecLength { result.Error = fmt.Sprintf("spec too long (max %d chars)", maxSpecLength) return result } if count <= 0 { count = defaultNextRuns } if count > maxNextRunsRequested { count = maxNextRunsRequested } // 2. 处理时区：如果显式传了 timezone，去掉 spec 里可能的前缀后再用它包 finalSpec, effectiveTZ, tzErr := resolveSpecTimezone(spec, timezone) if tzErr != "" { result.Error = tzErr return result } result.Timezone = effectiveTZ // 3. 解析 schedule, err := cron.ParseWithError(finalSpec) if err != nil { result.Error = err.Error() return result } result.Valid = true // 4. 接下来 N 次 t := now for i := 0; i < count; i++ { next := schedule.Next(t) // 防死循环：Next 返回零值或不推进说明永不触发 if next.IsZero() || !next.After(t) { break } result.NextRuns = append(result.NextRuns, CronRun{ Unix: next.Unix(), ISO: next.Format(time.RFC3339), Weekday: int(next.Weekday()), }) t = next } // 5. 未来 7 天分布（稀疏格式，只返 count>0 的格子） heatmapEnd := now.Add(time.Duration(heatmapWindowHours) * time.Hour) cells := make(map[[2]int]int) t = now for iter := 0; iter < maxHeatmapIterations; iter++ { next := schedule.Next(t) if next.IsZero() || !next.After(t) || next.After(heatmapEnd) { break } key := [2]int{int(next.Weekday()), next.Hour()} cells[key]++ t = next if iter == maxHeatmapIterations-1 { // 还有一次就触顶，再看一眼是否真到窗口末 if peek := schedule.Next(t); !peek.IsZero() && peek.After(t) && peek.Before(heatmapEnd) { result.Truncated = true } } } for key, c := range cells { result.HeatmapCells = append(result.HeatmapCells, HeatmapCell{ Day: key[0], Hour: key[1], Count: c, }) } return result } // resolveSpecTimezone 处理 spec 和 timezone 的组合： // - 显式 timezone != "" : 剥除 spec 里已有的 CRON_TZ=/TZ= 前缀后，用显式 timezone 重新包 // - 显式 timezone == "" 且 spec 带前缀：保留原样，effectiveTZ 返回前缀里的 tz // - 都没有：用服务器本地时区 func resolveSpecTimezone(spec, timezone string) (finalSpec, effectiveTZ, errMsg string) { bareSpec, prefixTZ := stripTimezonePrefix(spec) if timezone != "" { if _, err := time.LoadLocation(timezone); err != nil { return "", timezone, fmt.Sprintf("unknown timezone: %q", timezone) } return "CRON_TZ=" + timezone + " " + bareSpec, timezone, "" } if prefixTZ != "" { // spec 自带前缀，cron 库自己能解 if _, err := time.LoadLocation(prefixTZ); err != nil { return "", prefixTZ, fmt.Sprintf("unknown timezone: %q", prefixTZ) } return spec, prefixTZ, "" } // 都没指定，用服务器本地 return bareSpec, time.Local.String(), "" } // stripTimezonePrefix 返回 (去除前缀的 spec, 前缀中的 timezone) // 无前缀时 timezone 为空。 func stripTimezonePrefix(spec string) (bareSpec, timezone string) { if !strings.HasPrefix(spec, "CRON_TZ=") && !strings.HasPrefix(spec, "TZ=") { return spec, "" } eqIdx := strings.IndexByte(spec, '=') rest := spec[eqIdx+1:] spIdx := strings.IndexByte(rest, ' ') if spIdx < 0 { return spec, "" } return strings.TrimSpace(rest[spIdx+1:]), rest[:spIdx] } ================================================ FILE: internal/service/cron_preview_test.go ================================================ package service import ( "strings" "testing" "time" ) // 固定 now 让测试完全确定：2026-04-20 周一 12:00 UTC var testNow = time.Date(2026, 4, 20, 12, 0, 0, 0, time.UTC) func TestPreviewCron_ValidStandardExpressions(t *testing.T) { tests := []struct { name string spec string timezone string wantNextN int // 预期返回的 next_runs 条数 firstRunOK func(time.Time) bool }{ { name: "每分钟", spec: "0 * * * * *", wantNextN: 10, firstRunOK: func(tm time.Time) bool { // 下次执行应该在 now 的下一分钟 return tm.Second() == 0 && tm.After(testNow) && tm.Sub(testNow) <= time.Minute }, }, { name: "周一至周五 9:30", spec: "0 30 9 * * 1-5", wantNextN: 10, firstRunOK: func(tm time.Time) bool { wd := tm.Weekday() return tm.Hour() == 9 && tm.Minute() == 30 && wd >= time.Monday && wd <= time.Friday }, }, { name: "@daily 快捷", spec: "@daily", wantNextN: 10, firstRunOK: func(tm time.Time) bool { return tm.Hour() == 0 && tm.Minute() == 0 && tm.Second() == 0 }, }, { name: "@every 30s", spec: "@every 30s", wantNextN: 10, firstRunOK: func(tm time.Time) bool { return tm.Sub(testNow) <= 30*time.Second && tm.After(testNow) }, }, { name: "5 段格式（省略 dow，自动补 *）", spec: "0 30 9 * *", wantNextN: 10, firstRunOK: func(tm time.Time) bool { // second=0 minute=30 hour=9 day=* month=* dow=*（自动补） return tm.Hour() == 9 && tm.Minute() == 30 && tm.Second() == 0 }, }, { name: "自定义 count", spec: "0 * * * * *", wantNextN: 10, // count=0 走默认 10 }, } for _, tc := range tests { t.Run(tc.name, func(t *testing.T) { got := previewCronAt(tc.spec, tc.timezone, 0, testNow) if !got.Valid { t.Fatalf("expected valid=true, got error=%q", got.Error) } if len(got.NextRuns) != tc.wantNextN { t.Errorf("next_runs length=%d want=%d", len(got.NextRuns), tc.wantNextN) } // 单调递增 for i := 1; i < len(got.NextRuns); i++ { if got.NextRuns[i].Unix <= got.NextRuns[i-1].Unix { t.Errorf("next_runs not strictly increasing at index %d", i) } } if tc.firstRunOK != nil && len(got.NextRuns) > 0 { first := time.Unix(got.NextRuns[0].Unix, 0).UTC() if !tc.firstRunOK(first) { t.Errorf("first run %v failed validation", first) } } }) } } func TestPreviewCron_InvalidInput(t *testing.T) { tests := []struct { name string spec string timezone string wantErr string // 子串匹配 }{ {"空字符串", "", "", "required"}, {"只有空格", " ", "", "required"}, {"换行注入", "0 * * * *\n*", "", "single-line"}, {"回车注入", "0 * * * *\r*", "", "single-line"}, {"超长", strings.Repeat("0 ", 80), "", "too long"}, {"字段不够", "0 0", "", "expected 5 or 6 fields"}, {"字段过多", "0 0 0 0 0 0 0 0", "", "expected 5 or 6 fields"}, {"非法 @ 快捷", "@nonexistent", "", "unrecognized"}, {"非法时区", "0 * * * * *", "Mars/Phobos", "unknown timezone"}, } for _, tc := range tests { t.Run(tc.name, func(t *testing.T) { got := previewCronAt(tc.spec, tc.timezone, 0, testNow) if got.Valid { t.Fatalf("expected valid=false for %q", tc.spec) } if !strings.Contains(strings.ToLower(got.Error), strings.ToLower(tc.wantErr)) { t.Errorf("error %q does not contain %q", got.Error, tc.wantErr) } }) } } func TestPreviewCron_Timezone(t *testing.T) { t.Run("显式 timezone 生效", func(t *testing.T) { // Asia/Shanghai 09:30 = UTC 01:30；now=UTC 12:00 // 下次 SH 09:30 应该是次日 UTC 01:30 got := previewCronAt("0 30 9 * * *", "Asia/Shanghai", 3, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } if got.Timezone != "Asia/Shanghai" { t.Errorf("timezone=%q want=Asia/Shanghai", got.Timezone) } if len(got.NextRuns) == 0 { t.Fatal("no next runs") } // 验证首次执行换算到 UTC 是 01:30 first := time.Unix(got.NextRuns[0].Unix, 0).UTC() if first.Hour() != 1 || first.Minute() != 30 { t.Errorf("first run in UTC = %v, expected hour=1 min=30", first) } }) t.Run("spec 自带 CRON_TZ 前缀", func(t *testing.T) { got := previewCronAt("CRON_TZ=America/New_York 0 30 9 * * *", "", 3, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } if got.Timezone != "America/New_York" { t.Errorf("timezone=%q want=America/New_York", got.Timezone) } }) t.Run("显式 timezone 覆盖 spec 自带前缀", func(t *testing.T) { // spec 里是 NY，参数传 SH，应该以参数为准 got := previewCronAt("CRON_TZ=America/New_York 0 30 9 * * *", "Asia/Shanghai", 3, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } if got.Timezone != "Asia/Shanghai" { t.Errorf("timezone=%q want=Asia/Shanghai (explicit should override)", got.Timezone) } }) } func TestPreviewCron_Heatmap(t *testing.T) { t.Run("标准低频表达式", func(t *testing.T) { // 每天 09:30 一次，一周 7 次 got := previewCronAt("0 30 9 * * *", "", 0, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } // 应该只有一个 cell：9 点那列，每天都有 total := 0 for _, c := range got.HeatmapCells { total += c.Count if c.Hour != 9 { t.Errorf("unexpected cell hour=%d (expected only 9)", c.Hour) } } if total != 7 { t.Errorf("weekly total=%d want=7", total) } if got.Truncated { t.Errorf("should not be truncated for simple daily schedule") } }) t.Run("高频表达式触发迭代上限", func(t *testing.T) { // 每秒一次，一周 604800 次，会远超 maxHeatmapIterations=2000 got := previewCronAt("* * * * * *", "", 0, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } if !got.Truncated { t.Errorf("expected Truncated=true for * * * * * *") } // cell 数应 <= 24*7=168 if len(got.HeatmapCells) > 168 { t.Errorf("cells=%d should be <=168", len(got.HeatmapCells)) } }) t.Run("极低频（7 天内无触发）", func(t *testing.T) { // 每年 1 月 1 日 00:00，从 2026-04-20 算 7 天内肯定不到 got := previewCronAt("0 0 0 1 1 *", "", 0, testNow) if !got.Valid { t.Fatalf("expected valid, got %q", got.Error) } if len(got.HeatmapCells) != 0 { t.Errorf("cells=%d want=0 for yearly schedule", len(got.HeatmapCells)) } }) } func TestPreviewCron_CountClamp(t *testing.T) { t.Run("count<=0 用默认", func(t *testing.T) { got := previewCronAt("0 * * * * *", "", 0, testNow) if len(got.NextRuns) != defaultNextRuns { t.Errorf("count=%d want=%d", len(got.NextRuns), defaultNextRuns) } }) t.Run("count 超上限被钳制", func(t *testing.T) { got := previewCronAt("0 * * * * *", "", 1000, testNow) if len(got.NextRuns) != maxNextRunsRequested { t.Errorf("count=%d want=%d", len(got.NextRuns), maxNextRunsRequested) } }) t.Run("count=5", func(t *testing.T) { got := previewCronAt("0 * * * * *", "", 5, testNow) if len(got.NextRuns) != 5 { t.Errorf("count=%d want=5", len(got.NextRuns)) } }) } func TestStripTimezonePrefix(t *testing.T) { tests := []struct { in string wantBare string wantTZ string }{ {"0 * * * * *", "0 * * * * *", ""}, {"CRON_TZ=UTC 0 * * * * *", "0 * * * * *", "UTC"}, {"TZ=Asia/Shanghai 0 30 9 * * *", "0 30 9 * * *", "Asia/Shanghai"}, {"CRON_TZ=UTC", "CRON_TZ=UTC", ""}, // 没空格不算合法前缀 } for _, tc := range tests { gotBare, gotTZ := stripTimezonePrefix(tc.in) if gotBare != tc.wantBare || gotTZ != tc.wantTZ { t.Errorf("stripTimezonePrefix(%q) = (%q, %q), want (%q, %q)", tc.in, gotBare, gotTZ, tc.wantBare, tc.wantTZ) } } } // Benchmark：`* * * * * *` 应在毫秒级完成，防 DoS func BenchmarkPreviewCron_HighFrequency(b *testing.B) { b.ReportAllocs() for i := 0; i < b.N; i++ { _ = previewCronAt("* * * * * *", "", 10, testNow) } } func BenchmarkPreviewCron_Standard(b *testing.B) { b.ReportAllocs() for i := 0; i < b.N; i++ { _ = previewCronAt("0 30 9 * * 1-5", "Asia/Shanghai", 10, testNow) } } ================================================ FILE: internal/service/issue66_test.go ================================================ package service // https://github.com/gocronx-team/gocron/issues/66 import ( "sync" "testing" "time" "github.com/gocronx-team/gocron/internal/models" ) // TestIssue66RaceCondition 重现 Issue #66: 任务无法单实例运行 // 问题：beforeExecJob检查和createJob添加实例标记之间存在竞态条件 func TestIssue66RaceCondition(t *testing.T) { t.Run("旧实现-重现竞态条件bug", func(t *testing.T) { runInstance = Instance{} task := models.Task{ Id: 1, Name: "测试任务", Multi: 0, // 禁止并发 } var executedCount int var mu sync.Mutex var wg sync.WaitGroup // 模拟快速点击两次"手动执行" - 使用旧的实现方式 for i := 0; i < 2; i++ { wg.Add(1) go func(index int) { defer wg.Done() // 旧实现：分离的检查和添加 // 1. beforeExecJob 检查 taskLogId := int64(0) if task.Multi == 0 && runInstance.has(task.Id) { t.Logf("执行%d: 任务已在运行中，取消本次执行", index) return } taskLogId = int64(index + 1) // ⚠️ 竞态条件窗口：在检查和添加之间 time.Sleep(1 * time.Millisecond) // 2. 添加实例标记 if task.Multi == 0 { runInstance.add(task.Id) defer runInstance.done(task.Id) } // 3. 执行任务 mu.Lock() executedCount++ mu.Unlock() t.Logf("执行%d: 任务开始执行, taskLogId=%d", index, taskLogId) time.Sleep(10 * time.Millisecond) t.Logf("执行%d: 任务执行完成", index) }(i) } wg.Wait() if executedCount > 1 { t.Logf("✅ 成功重现Bug：期望执行1次，实际执行了%d次", executedCount) } }) t.Run("新实现-修复竞态条件", func(t *testing.T) { runInstance = Instance{} task := models.Task{ Id: 2, Name: "测试任务", Multi: 0, } var executedCount int var canceledCount int var mu sync.Mutex var wg sync.WaitGroup // 模拟快速点击两次"手动执行" - 使用新的原子实现 for i := 0; i < 2; i++ { wg.Add(1) go func(index int) { defer wg.Done() // 新实现：原子的检查和添加 if task.Multi == 0 { if !runInstance.tryAdd(task.Id) { mu.Lock() canceledCount++ mu.Unlock() t.Logf("执行%d: 任务已在运行中，取消本次执行", index) return } defer runInstance.done(task.Id) } // 执行任务 mu.Lock() executedCount++ mu.Unlock() t.Logf("执行%d: 任务开始执行", index) time.Sleep(10 * time.Millisecond) t.Logf("执行%d: 任务执行完成", index) }(i) } wg.Wait() if executedCount == 1 && canceledCount == 1 { t.Logf("✅ Bug已修复！执行%d次，取消%d次", executedCount, canceledCount) } else { t.Errorf("❌ 修复失败！执行%d次，取消%d次（期望：执行1次，取消1次）", executedCount, canceledCount) } }) } ================================================ FILE: internal/service/single_instance_test.go ================================================ package service import ( "sync" "testing" "time" "github.com/gocronx-team/gocron/internal/models" ) // TestSingleInstanceControl 测试单实例运行控制 func TestSingleInstanceControl(t *testing.T) { t.Run("Multi=0时阻止并发执行", func(t *testing.T) { instance := &Instance{} taskId := 100 // 第一次检查，应该不存在 if instance.has(taskId) { t.Error("任务不应该在运行中") } // 添加任务 instance.add(taskId) // 第二次检查，应该存在 if !instance.has(taskId) { t.Error("任务应该在运行中") } // 完成任务 instance.done(taskId) // 第三次检查，应该不存在 if instance.has(taskId) { t.Error("任务不应该在运行中") } }) t.Run("并发场景下的单实例控制", func(t *testing.T) { instance := &Instance{} taskId := 200 var wg sync.WaitGroup executionCount := 0 var mu sync.Mutex // 模拟10个并发请求 for i := 0; i < 10; i++ { wg.Add(1) go func() { defer wg.Done() // 使用互斥锁保护检查和添加操作的原子性 mu.Lock() if !instance.has(taskId) { instance.add(taskId) executionCount++ mu.Unlock() // 模拟任务执行 time.Sleep(10 * time.Millisecond) instance.done(taskId) } else { mu.Unlock() } }() } wg.Wait() // 只有第一个请求应该执行 if executionCount != 1 { t.Errorf("期望只有1次执行，实际执行了%d次", executionCount) } }) t.Run("不同任务ID互不影响", func(t *testing.T) { instance := &Instance{} instance.add(1) instance.add(2) instance.add(3) if !instance.has(1) || !instance.has(2) || !instance.has(3) { t.Error("所有任务都应该在运行中") } instance.done(2) if !instance.has(1) || instance.has(2) || !instance.has(3) { t.Error("只有任务2应该被移除") } }) } // TestBeforeExecJobSingleInstance 测试beforeExecJob中的单实例逻辑 func TestBeforeExecJobSingleInstance(t *testing.T) { t.Run("Multi=0且任务已运行时应该取消", func(t *testing.T) { // 重置runInstance runInstance = Instance{} task := models.Task{ Id: 1, Name: "测试任务", Multi: 0, // 不允许并发 } // 模拟任务已在运行 runInstance.add(task.Id) // 验证任务在运行中 if !runInstance.has(task.Id) { t.Error("任务应该在运行中") } // 模拟beforeExecJob的逻辑：如果Multi=0且任务已运行，应该取消 shouldCancel := task.Multi == 0 && runInstance.has(task.Id) if !shouldCancel { t.Error("任务已在运行，应该取消本次执行") } // 清理 runInstance.done(task.Id) }) t.Run("Multi=1时允许并发执行", func(t *testing.T) { // 重置runInstance runInstance = Instance{} task := models.Task{ Id: 2, Name: "测试任务", Multi: 1, // 允许并发 } // 模拟任务已在运行 runInstance.add(task.Id) // Multi=1时，不应该取消 shouldCancel := task.Multi == 0 && runInstance.has(task.Id) if shouldCancel { t.Error("Multi=1时应该允许并发执行") } // 清理 runInstance.done(task.Id) }) } // TestCreateJobSingleInstanceLogic 测试createJob中的单实例逻辑 func TestCreateJobSingleInstanceLogic(t *testing.T) { t.Run("Multi=0时应该添加和移除实例标记", func(t *testing.T) { // 重置runInstance runInstance = Instance{} taskId := 100 // 模拟createJob中的逻辑 if !runInstance.has(taskId) { // Multi=0时，添加实例标记 runInstance.add(taskId) // 验证已添加 if !runInstance.has(taskId) { t.Error("应该已添加实例标记") } // 模拟任务执行完成 runInstance.done(taskId) // 验证已移除 if runInstance.has(taskId) { t.Error("应该已移除实例标记") } } }) t.Run("Multi=1时不应该添加实例标记", func(t *testing.T) { // 重置runInstance runInstance = Instance{} taskId := 200 multi := 1 // Multi=1时，不添加实例标记 if multi == 0 { runInstance.add(taskId) } // 验证未添加 if runInstance.has(taskId) { t.Error("Multi=1时不应该添加实例标记") } }) } // TestInstanceThreadSafety 测试Instance的线程安全性 func TestInstanceThreadSafety(t *testing.T) { instance := &Instance{} var wg sync.WaitGroup // 并发添加和删除 for i := 0; i < 100; i++ { wg.Add(3) taskId := i // 添加 go func(id int) { defer wg.Done() instance.add(id) }(taskId) // 检查 go func(id int) { defer wg.Done() _ = instance.has(id) }(taskId) // 删除 go func(id int) { defer wg.Done() time.Sleep(1 * time.Millisecond) instance.done(id) }(taskId) } wg.Wait() // 测试通过表示没有发生竞态条件 t.Log("线程安全测试通过") } // TestSingleInstanceRealScenario 测试真实场景 func TestSingleInstanceRealScenario(t *testing.T) { t.Run("定时任务触发时上次未完成", func(t *testing.T) { runInstance = Instance{} task := models.Task{ Id: 1, Name: "慢速任务", Multi: 0, } var executionCount int var mu sync.Mutex // 模拟第一次执行（耗时较长） go func() { if task.Multi == 0 && runInstance.has(task.Id) { return } if task.Multi == 0 { runInstance.add(task.Id) defer runInstance.done(task.Id) } mu.Lock() executionCount++ mu.Unlock() time.Sleep(50 * time.Millisecond) }() // 等待第一次执行开始 time.Sleep(5 * time.Millisecond) // 模拟第二次触发（应该被阻止） if task.Multi == 0 && runInstance.has(task.Id) { t.Log("第二次执行被正确阻止") } else { if task.Multi == 0 { runInstance.add(task.Id) defer runInstance.done(task.Id) } mu.Lock() executionCount++ mu.Unlock() } // 等待第一次执行完成 time.Sleep(60 * time.Millisecond) mu.Lock() count := executionCount mu.Unlock() if count != 1 { t.Errorf("期望执行1次，实际执行%d次", count) } }) } ================================================ FILE: internal/service/task.go ================================================ package service import ( "bytes" "encoding/json" "errors" "fmt" "net/http" "os" "regexp" "strconv" "strings" "sync" "time" "github.com/gocronx-team/cron" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/app" "github.com/gocronx-team/gocron/internal/modules/httpclient" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/notify" rpcClient "github.com/gocronx-team/gocron/internal/modules/rpc/client" pb "github.com/gocronx-team/gocron/internal/modules/rpc/proto" "github.com/gocronx-team/gocron/internal/modules/utils" ) var ( ServiceTask Task ) var ( httpGetFunc = httpclient.Get httpPostParamsFunc = httpclient.PostParams httpPostJsonFunc = httpclient.PostJson httpGetWithHeadersFunc = httpclient.GetWithHeaders httpPostJsonWithHdrsFunc = httpclient.PostJsonWithHeaders httpPostParamsWithHdrs = httpclient.PostParamsWithHeaders notifyPushFunc = notify.Push sleepFunc = time.Sleep // 定时任务调度管理器 serviceCron *cron.Cron // 同一任务是否有实例处于运行中 runInstance Instance // 调度器运行状态 schedulerMu sync.Mutex schedulerRunning bool // 任务计数-正在运行的任务 taskCount TaskCount // 并发队列, 限制同时运行的任务数量 concurrencyQueue ConcurrencyQueue ) // 并发队列 type ConcurrencyQueue struct { queue chan struct{} } func (cq *ConcurrencyQueue) Add() { cq.queue <- struct{}{} } func (cq *ConcurrencyQueue) Done() { <-cq.queue } // 任务计数 type TaskCount struct { wg sync.WaitGroup exit chan struct{} } func (tc *TaskCount) Add() { tc.wg.Add(1) } func (tc *TaskCount) Done() { tc.wg.Done() } func (tc *TaskCount) Exit() { tc.wg.Done() <-tc.exit } func (tc *TaskCount) Wait() { tc.Add() tc.wg.Wait() close(tc.exit) } // 任务ID作为Key type Instance struct { m sync.Map } // 是否有任务处于运行中 func (i *Instance) has(key int) bool { _, ok := i.m.Load(key) return ok } func (i *Instance) add(key int) { i.m.Store(key, struct{}{}) } func (i *Instance) done(key int) { i.m.Delete(key) } // tryAdd 原子地尝试添加任务实例 // 返回 true 表示成功添加（任务未在运行），false 表示任务已在运行 func (i *Instance) tryAdd(key int) bool { _, loaded := i.m.LoadOrStore(key, struct{}{}) return !loaded } type Task struct{} type TaskResult struct { Result string Err error RetryTimes int8 } // Initialize 初始化调度器基础设施（不加载任务） // 任务加载由 StartScheduler 完成，配合 leader election 使用 func (task Task) Initialize() { concurrencyQueue = ConcurrencyQueue{queue: make(chan struct{}, app.Setting.ConcurrencyQueue)} taskCount = TaskCount{sync.WaitGroup{}, make(chan struct{})} go taskCount.Wait() logger.Info("Scheduler infrastructure initialized") } // StartScheduler 启动调度器并加载所有任务（当选 leader 时调用） func (task Task) StartScheduler() { schedulerMu.Lock() defer schedulerMu.Unlock() if schedulerRunning { return } serviceCron = cron.New() serviceCron.Start() logger.Info("Starting to load scheduled tasks (this node is leader)") taskModel := new(models.Task) taskNum := 0 page := 1 pageSize := 1000 maxPage := 1000 for page < maxPage { taskList, err := taskModel.ActiveList(page, pageSize) if err != nil { logger.Fatalf("Scheduled task initialization#Failed to get task list: %s", err) } if len(taskList) == 0 { break } for _, item := range taskList { logger.Infof("Adding task to scheduler#ID-%d#Name-%s#Protocol-%d#Host count-%d", item.Id, item.Name, item.Protocol, len(item.Hosts)) task.Add(item) taskNum++ } page++ } logger.Infof("Scheduled task initialization completed, %d tasks added to scheduler", taskNum) task.initLogCleanupTask() schedulerRunning = true } // StopScheduler 停止调度器（失去 leader 时调用） func (task Task) StopScheduler() { schedulerMu.Lock() defer schedulerMu.Unlock() if !schedulerRunning { return } logger.Info("Stopping scheduler (this node lost leadership)") serviceCron.Stop() serviceCron = nil schedulerRunning = false } // IsSchedulerRunning 返回调度器是否正在运行 func (task Task) IsSchedulerRunning() bool { schedulerMu.Lock() defer schedulerMu.Unlock() return schedulerRunning } // 初始化日志清理任务 func (task Task) initLogCleanupTask() { if serviceCron == nil { return } settingModel := new(models.Setting) cleanupTime := settingModel.GetLogCleanupTime() // 解析时间 HH:MM var hour, minute int if n, err := fmt.Sscanf(cleanupTime, "%d:%d", &hour, &minute); err != nil || n != 2 || hour < 0 || hour > 23 || minute < 0 || minute > 59 { logger.Warnf("日志清理时间解析失败，使用默认值 00:00 (cleanupTime=%q)", cleanupTime) hour, minute = 0, 0 } // 生成cron表达式: 秒 分 时 日 月 周 cronSpec := fmt.Sprintf("0 %d %d * * *", minute, hour) serviceCron.AddFunc(cronSpec, func() { // 1. Task-level log retention: clean logs for tasks with custom retention days taskLogModel := new(models.TaskLog) page := 1 pageSize := 1000 for { var tasks []models.Task err := models.Db.Where("log_retention_days > 0"). Limit(pageSize).Offset((page - 1) * pageSize). Find(&tasks).Error if err != nil { logger.Errorf("Failed to query tasks with custom log retention: %s", err) break } if len(tasks) == 0 { break } for _, t := range tasks { count, err := taskLogModel.RemoveByTaskIdAndDays(t.Id, t.LogRetentionDays) if err != nil { logger.Errorf("Failed to cleanup logs for task %d: %s", t.Id, err) } else if count > 0 { logger.Infof("Task %d: cleaned up %d logs older than %d days", t.Id, count, t.LogRetentionDays) } } page++ } // 2. Global log retention: clean remaining logs settingModel := new(models.Setting) days := settingModel.GetLogRetentionDays() if days > 0 { count, err := taskLogModel.RemoveByDaysExcludingCustomRetention(days) if err != nil { logger.Errorf("Failed to auto-cleanup database logs: %s", err) } else { logger.Infof("Auto-cleanup database logs older than %d days, deleted %d records", days, count) } // 清理日志文件 cleanupLogFiles() } }, "log-cleanup") logger.Infof("Log auto-cleanup task added, execution time: %s", cleanupTime) } // 重新加载日志清理任务 func (task Task) ReloadLogCleanupTask() { if serviceCron == nil { return // follower node or scheduler not started, skip } // 先移除旧任务 serviceCron.RemoveJob("log-cleanup") // 重新添加任务 task.initLogCleanupTask() logger.Info("Log cleanup task reloaded") } // 批量添加任务 func (task Task) BatchAdd(tasks []models.Task) { for _, item := range tasks { task.RemoveAndAdd(item) } } // 删除任务后添加 func (task Task) RemoveAndAdd(taskModel models.Task) { task.Remove(taskModel.Id) task.Add(taskModel) } // 添加任务 func (task Task) Add(taskModel models.Task) { if taskModel.Level == models.TaskLevelChild { logger.Errorf("Failed to add task#Child tasks cannot be added to scheduler#Task ID-%d", taskModel.Id) return } if serviceCron == nil { return // follower node, skip } taskFunc := createJob(taskModel) if taskFunc == nil { logger.Error("Failed to create task job#Unsupported task protocol#", taskModel.Protocol) return } cronName := strconv.Itoa(taskModel.Id) err := utils.PanicToError(func() { serviceCron.AddFunc(taskModel.Spec, taskFunc, cronName) }) if err != nil { logger.Error("Failed to add task to scheduler#", err) } } func (task Task) NextRunTime(taskModel models.Task) time.Time { if serviceCron == nil { return time.Time{} } if taskModel.Level != models.TaskLevelParent || taskModel.Status != models.Enabled { return time.Time{} } entries := serviceCron.Entries() taskName := strconv.Itoa(taskModel.Id) for _, item := range entries { if item.Name == taskName { return item.Next } } return time.Time{} } // 停止运行中的任务 func (task Task) Stop(ip string, port int, id int64) { rpcClient.Stop(ip, port, id) } func (task Task) Remove(id int) { if serviceCron == nil { return } serviceCron.RemoveJob(strconv.Itoa(id)) } // 等待所有任务结束后退出 func (task Task) WaitAndExit() { schedulerMu.Lock() if schedulerRunning && serviceCron != nil { serviceCron.Stop() schedulerRunning = false } schedulerMu.Unlock() taskCount.Exit() } // 直接运行任务 func (task Task) Run(taskModel models.Task) { go createJob(taskModel)() } type Handler interface { Run(taskModel models.Task, taskUniqueId int64) (string, error) } // HTTP任务 type HTTPHandler struct{} // HttpDefaultTimeout HTTP 任务默认超时（秒），用户未设置时使用 const HttpDefaultTimeout = 300 func (h *HTTPHandler) Run(taskModel models.Task, taskUniqueId int64) (result string, err error) { if taskModel.Timeout <= 0 { taskModel.Timeout = HttpDefaultTimeout } headers := strings.TrimSpace(taskModel.HttpHeaders) var resp httpclient.ResponseWrapper if taskModel.HttpMethod == models.TaskHTTPMethodGet { if headers != "" { resp = httpGetWithHeadersFunc(taskModel.Command, headers, taskModel.Timeout) } else { resp = httpGetFunc(taskModel.Command, taskModel.Timeout) } } else { // POST: 优先使用 HttpBody (JSON)，否则回退到 URL query 参数 if strings.TrimSpace(taskModel.HttpBody) != "" { if headers != "" { resp = httpPostJsonWithHdrsFunc(taskModel.Command, taskModel.HttpBody, headers, taskModel.Timeout) } else { resp = httpPostJsonFunc(taskModel.Command, taskModel.HttpBody, taskModel.Timeout) } } else { urlFields := strings.Split(taskModel.Command, "?") url := urlFields[0] var params string if len(urlFields) >= 2 { params = urlFields[1] } if headers != "" { resp = httpPostParamsWithHdrs(url, params, headers, taskModel.Timeout) } else { resp = httpPostParamsFunc(url, params, taskModel.Timeout) } } } // 返回状态码非200，均为失败 if resp.StatusCode != http.StatusOK { return resp.Body, fmt.Errorf("HTTP status code is not 200-->%d", resp.StatusCode) } // 响应内容断言 if taskModel.SuccessPattern != "" { re, regexErr := regexp.Compile(taskModel.SuccessPattern) if regexErr != nil { return resp.Body, fmt.Errorf("invalid success_pattern regex: %v", regexErr) } // 先匹配原始响应体，不匹配再尝试压缩 JSON 后匹配（兼容格式化空白差异） if !re.MatchString(resp.Body) { compacted := compactJSON(resp.Body) if compacted == resp.Body || !re.MatchString(compacted) { return resp.Body, fmt.Errorf("response body does not match success_pattern: %s", taskModel.SuccessPattern) } } } return resp.Body, err } // compactJSON 压缩 JSON 字符串，去掉格式化空白。非 JSON 则原样返回。 func compactJSON(s string) string { var buf bytes.Buffer if err := json.Compact(&buf, []byte(s)); err != nil { return s } return buf.String() } // RPC调用执行任务 type RPCHandler struct{} func (h *RPCHandler) Run(taskModel models.Task, taskUniqueId int64) (result string, err error) { logger.Infof("RPC task execution started#Task ID-%d#Host count-%d", taskModel.Id, len(taskModel.Hosts)) if len(taskModel.Hosts) == 0 { return "", fmt.Errorf("task is not associated with any host") } taskRequest := new(pb.TaskRequest) taskRequest.Timeout = int32(taskModel.Timeout) taskRequest.Command = taskModel.Command taskRequest.Id = taskUniqueId resultChan := make(chan TaskResult, len(taskModel.Hosts)) for _, taskHost := range taskModel.Hosts { logger.Infof("Preparing RPC call#Host-%s:%d#Command-%s", taskHost.Name, taskHost.Port, taskModel.Command) go func(th models.TaskHostDetail) { output, err := rpcClient.Exec(th.Name, th.Port, taskRequest) errorMessage := "" if err != nil { // 如果是手动停止错误，保留原始错误以便后续判断，但显示翻译后的文本 if errors.Is(err, rpcClient.ErrManualStop) { errorMessage = "Manually stopped" } else { errorMessage = err.Error() } } output = strings.TrimSpace(output) if errorMessage != "" { errorMessage = strings.TrimSpace(errorMessage) + "\n" } outputMessage := fmt.Sprintf("Host: [%s-%s:%d]\n%s%s", th.Alias, th.Name, th.Port, errorMessage, output, ) logger.Infof("RPC call completed#Host-%s:%d#Output length-%d#Error-%v", th.Name, th.Port, len(output), err) resultChan <- TaskResult{Err: err, Result: outputMessage} }(taskHost) } var aggregationErr error var resultBuilder strings.Builder for i := 0; i < len(taskModel.Hosts); i++ { taskResult := <-resultChan resultBuilder.WriteString(taskResult.Result) if taskResult.Err != nil { aggregationErr = taskResult.Err } } return resultBuilder.String(), aggregationErr } // 创建任务日志 func createTaskLog(taskModel models.Task, status models.Status) (int64, error) { taskLogModel := new(models.TaskLog) taskLogModel.TaskId = taskModel.Id taskLogModel.Name = taskModel.Name taskLogModel.Spec = taskModel.Spec taskLogModel.Protocol = taskModel.Protocol taskLogModel.Command = taskModel.Command taskLogModel.Timeout = taskModel.Timeout if taskModel.Protocol == models.TaskRPC { var hostBuilder strings.Builder for _, host := range taskModel.Hosts { hostBuilder.WriteString(host.Alias) hostBuilder.WriteString(" - ") hostBuilder.WriteString(host.Name) hostBuilder.WriteString("
") } taskLogModel.Hostname = hostBuilder.String() } taskLogModel.StartTime = models.LocalTime(time.Now()) taskLogModel.Status = status insertId, err := taskLogModel.Create() return insertId, err } // 更新任务日志 func updateTaskLog(taskLogId int64, taskResult TaskResult) (int64, error) { taskLogModel := new(models.TaskLog) var status models.Status result := taskResult.Result // 根据错误类型设置状态 if taskResult.Err != nil { // 检查是否是手动停止 if errors.Is(taskResult.Err, rpcClient.ErrManualStop) { status = models.Cancel } else { status = models.Failure } } else { status = models.Finish } return taskLogModel.Update(taskLogId, models.CommonMap{ "retry_times": taskResult.RetryTimes, "status": status, "result": result, "end_time": time.Now(), }) } func createJob(taskModel models.Task) cron.FuncJob { handler := createHandler(taskModel) if handler == nil { return nil } taskFunc := func() { taskCount.Add() defer taskCount.Done() taskLogId := beforeExecJob(taskModel) if taskLogId <= 0 { return } // Multi=0 时，确保清理实例标记 // 注意：beforeExecJob 已经添加了实例标记，这里只需要清理 if taskModel.Multi == 0 { defer runInstance.done(taskModel.Id) } concurrencyQueue.Add() defer concurrencyQueue.Done() logger.Infof("Starting task execution#%s#Command-%s", taskModel.Name, taskModel.Command) taskResult := execJob(handler, taskModel, taskLogId) logger.Infof("Task completed#%s#Command-%s", taskModel.Name, taskModel.Command) afterExecJob(taskModel, taskResult, taskLogId) } return taskFunc } func createHandler(taskModel models.Task) Handler { var handler Handler = nil switch taskModel.Protocol { case models.TaskHTTP: handler = new(HTTPHandler) case models.TaskRPC: handler = new(RPCHandler) } return handler } // 任务前置操作 func beforeExecJob(taskModel models.Task) (taskLogId int64) { // Multi=0 时，原子地检查并添加实例标记 if taskModel.Multi == 0 { if !runInstance.tryAdd(taskModel.Id) { logger.Infof("Task already running, canceling this execution#ID-%d", taskModel.Id) taskLogId, _ = createTaskLog(taskModel, models.Cancel) return } } taskLogId, err := createTaskLog(taskModel, models.Running) if err != nil { logger.Error("Task execution started#Failed to write task log-", err) // 如果创建日志失败，需要回滚实例标记 if taskModel.Multi == 0 { runInstance.done(taskModel.Id) } return } return taskLogId } // 任务执行后置操作 func afterExecJob(taskModel models.Task, taskResult TaskResult, taskLogId int64) { _, err := updateTaskLog(taskLogId, taskResult) if err != nil { logger.Error("Task ended#Failed to update task log-", err) } // 发送邮件 go SendNotification(taskModel, taskResult) // 执行依赖任务 go execDependencyTask(taskModel, taskResult) } // 执行依赖任务, 多个任务并发执行 func execDependencyTask(taskModel models.Task, taskResult TaskResult) { // 父任务才能执行子任务 if taskModel.Level != models.TaskLevelParent { return } // 是否存在子任务 dependencyTaskId := strings.TrimSpace(taskModel.DependencyTaskId) if dependencyTaskId == "" { return } // 父子任务关系为强依赖, 父任务执行失败, 不执行依赖任务 if taskModel.DependencyStatus == models.TaskDependencyStatusStrong && taskResult.Err != nil { logger.Infof("Parent-child tasks have strong dependency, parent task failed, dependency tasks will not run#Parent task ID-%d", taskModel.Id) return } // 获取子任务 model := new(models.Task) tasks, err := model.GetDependencyTaskList(dependencyTaskId) if err != nil { logger.Errorf("Failed to get dependency tasks#Parent task ID-%d#%s", taskModel.Id, err.Error()) return } if len(tasks) == 0 { logger.Warnf("Dependency task list is empty or tasks are disabled#Parent task ID-%d#Dependency task ID-%s", taskModel.Id, dependencyTaskId) return } logger.Infof("Starting dependency tasks execution#Parent task ID-%d#Dependency task count-%d", taskModel.Id, len(tasks)) for _, task := range tasks { logger.Infof("Executing dependency task#Parent task ID-%d#Dependency task ID-%d#Dependency task name-%s", taskModel.Id, task.Id, task.Name) task.Spec = fmt.Sprintf("Dependency task (Parent task ID-%d)", taskModel.Id) ServiceTask.Run(task) } } // 发送任务结果通知 func SendNotification(taskModel models.Task, taskResult TaskResult) { var statusName string // 未开启通知 if taskModel.NotifyStatus == 0 { return } if taskModel.NotifyStatus == 1 && taskResult.Err == nil { // 执行失败才发送通知 return } if taskModel.NotifyStatus == 3 { // 关键字匹配通知 if !strings.Contains(taskResult.Result, taskModel.NotifyKeyword) { return } } // NotifyType: 0=邮件, 1=Slack, 2=WebHook // WebHook(type=2)不需要receiver_id，其他类型需要 if taskModel.NotifyType != 2 && taskModel.NotifyReceiverId == "" { return } if taskResult.Err != nil { statusName = "Failed" } else { statusName = "Success" } // 发送通知 msg := notify.Message{ "task_type": taskModel.NotifyType, "task_receiver_id": taskModel.NotifyReceiverId, "name": taskModel.Name, "output": taskResult.Result, "status": statusName, "task_id": taskModel.Id, "remark": taskModel.Remark, } notifyPushFunc(msg) } // 执行具体任务 func execJob(handler Handler, taskModel models.Task, taskUniqueId int64) (result TaskResult) { defer func() { if err := recover(); err != nil { logger.Error("panic#service/task.go:execJob#", err) // 确保 panic 不会被误判为成功：返回失败结果 result = TaskResult{Err: fmt.Errorf("panic: %v", err)} } }() // 默认只运行任务一次 var execTimes int8 = 1 if taskModel.RetryTimes > 0 { execTimes += taskModel.RetryTimes } var i int8 = 0 var output string var err error for i < execTimes { output, err = handler.Run(taskModel, taskUniqueId) if err == nil { return TaskResult{Result: output, Err: err, RetryTimes: i} } i++ if i < execTimes { logger.Warnf("Task execution failed#Task ID-%d#Retry attempt %d#Output-%s#Error-%s", taskModel.Id, i, output, err.Error()) if taskModel.RetryInterval > 0 { sleepFunc(time.Duration(taskModel.RetryInterval) * time.Second) } else { // 默认重试间隔时间，每次递增1分钟 sleepFunc(time.Duration(i) * time.Minute) } } } return TaskResult{Result: output, Err: err, RetryTimes: taskModel.RetryTimes} } // 清理日志文件 func cleanupLogFiles() { settingModel := new(models.Setting) fileSizeLimit := settingModel.GetLogFileSizeLimit() // 如果设置为0，不清理日志文件 if fileSizeLimit <= 0 { return } logDir := "log" logFile := "cron.log" // 检查日志文件是否存在 logPath := fmt.Sprintf("%s/%s", logDir, logFile) fileInfo, err := os.Stat(logPath) if err != nil { if !os.IsNotExist(err) { logger.Errorf("Failed to check log file: %s", err) } return } // 如果文件大小超过限制，则清空 maxSize := int64(fileSizeLimit) * 1024 * 1024 // 转换为MB if fileInfo.Size() > maxSize { err := os.Truncate(logPath, 0) if err != nil { logger.Errorf("Failed to truncate log file: %s", err) } else { logger.Infof("Log file exceeded %dMB, truncated: %s", fileSizeLimit, logPath) } } } ================================================ FILE: internal/service/task_cleanup_test.go ================================================ package service import ( "testing" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/ncruces/go-sqlite3/gormlite" "gorm.io/gorm" "gorm.io/gorm/logger" ) func setupCleanupTestDB(t *testing.T) func() { t.Helper() originalDb := models.Db originalPrefix := models.TablePrefix db, err := gorm.Open(gormlite.Open(":memory:"), &gorm.Config{ Logger: logger.Default.LogMode(logger.Silent), }) if err != nil { t.Fatalf("failed to open test db: %v", err) } models.TablePrefix = "" models.Db = db err = db.AutoMigrate(&models.Task{}, &models.TaskLog{}) if err != nil { t.Fatalf("failed to migrate: %v", err) } return func() { models.Db = originalDb models.TablePrefix = originalPrefix } } // TestTaskLevelRetentionBeforeGlobal verifies that tasks with custom retention // days have their logs cleaned according to their own policy, and that the // global policy applies to remaining logs. func TestTaskLevelRetentionBeforeGlobal(t *testing.T) { cleanup := setupCleanupTestDB(t) defer cleanup() now := time.Now() // Task 1: custom retention of 3 days models.Db.Create(&models.Task{ Name: "task-custom", Level: 1, Spec: "* * * * *", Protocol: 1, Command: "echo 1", LogRetentionDays: 3, Status: models.Enabled, }) // Task 2: no custom retention (uses global) models.Db.Create(&models.Task{ Name: "task-global", Level: 1, Spec: "* * * * *", Protocol: 1, Command: "echo 2", LogRetentionDays: 0, Status: models.Enabled, }) oldTime5Days := models.LocalTime(now.AddDate(0, 0, -5)) oldTime2Days := models.LocalTime(now.AddDate(0, 0, -2)) // Task 1 logs: one 5-day old, one 2-day old models.Db.Create(&models.TaskLog{TaskId: 1, Name: "task-custom", Spec: "* * * * *", Protocol: 1, Command: "echo 1", Result: "ok", StartTime: oldTime5Days, Status: models.Finish}) models.Db.Create(&models.TaskLog{TaskId: 1, Name: "task-custom", Spec: "* * * * *", Protocol: 1, Command: "echo 1", Result: "ok", StartTime: oldTime2Days, Status: models.Finish}) // Task 2 logs: one 5-day old, one 2-day old models.Db.Create(&models.TaskLog{TaskId: 2, Name: "task-global", Spec: "* * * * *", Protocol: 1, Command: "echo 2", Result: "ok", StartTime: oldTime5Days, Status: models.Finish}) models.Db.Create(&models.TaskLog{TaskId: 2, Name: "task-global", Spec: "* * * * *", Protocol: 1, Command: "echo 2", Result: "ok", StartTime: oldTime2Days, Status: models.Finish}) // Step 1: Simulate task-level cleanup for tasks with custom retention var tasks []models.Task err := models.Db.Where("log_retention_days > 0").Find(&tasks).Error if err != nil { t.Fatalf("failed to query tasks: %v", err) } if len(tasks) != 1 { t.Fatalf("expected 1 task with custom retention, got %d", len(tasks)) } if tasks[0].Name != "task-custom" { t.Errorf("expected task-custom, got %s", tasks[0].Name) } taskLogModel := new(models.TaskLog) for _, task := range tasks { count, err := taskLogModel.RemoveByTaskIdAndDays(task.Id, task.LogRetentionDays) if err != nil { t.Fatalf("failed to cleanup task %d: %v", task.Id, err) } // Task 1 with 3-day retention should delete the 5-day old log if count != 1 { t.Errorf("expected 1 deleted for task %d, got %d", task.Id, count) } } // Verify task 1 has 1 remaining log (the 2-day old one) var task1Count int64 models.Db.Model(&models.TaskLog{}).Where("task_id = ?", 1).Count(&task1Count) if task1Count != 1 { t.Errorf("expected 1 remaining log for task 1, got %d", task1Count) } // Task 2 should still have both logs (no custom cleanup was applied) var task2Count int64 models.Db.Model(&models.TaskLog{}).Where("task_id = ?", 2).Count(&task2Count) if task2Count != 2 { t.Errorf("expected 2 remaining logs for task 2, got %d", task2Count) } // Step 2: Simulate global cleanup with 4-day retention globalDays := 4 globalCount, err := taskLogModel.RemoveByDays(globalDays) if err != nil { t.Fatalf("global cleanup failed: %v", err) } // Only task 2's 5-day old log should be deleted (task 1's 5-day old log was already removed) if globalCount != 1 { t.Errorf("expected 1 deleted by global cleanup, got %d", globalCount) } // Final state: task 1 has 1 log, task 2 has 1 log var finalTask1 int64 models.Db.Model(&models.TaskLog{}).Where("task_id = ?", 1).Count(&finalTask1) if finalTask1 != 1 { t.Errorf("expected 1 final log for task 1, got %d", finalTask1) } var finalTask2 int64 models.Db.Model(&models.TaskLog{}).Where("task_id = ?", 2).Count(&finalTask2) if finalTask2 != 1 { t.Errorf("expected 1 final log for task 2, got %d", finalTask2) } } // TestTaskWithoutCustomRetentionUsesGlobal verifies that tasks without // custom retention (log_retention_days=0) are not affected by task-level // cleanup and only cleaned by global policy. func TestTaskWithoutCustomRetentionUsesGlobal(t *testing.T) { cleanup := setupCleanupTestDB(t) defer cleanup() now := time.Now() oldTime := models.LocalTime(now.AddDate(0, 0, -10)) // Task with no custom retention models.Db.Create(&models.Task{ Name: "task-no-custom", Level: 1, Spec: "* * * * *", Protocol: 1, Command: "echo 1", LogRetentionDays: 0, Status: models.Enabled, }) // Create old logs models.Db.Create(&models.TaskLog{TaskId: 1, Name: "task-no-custom", Spec: "* * * * *", Protocol: 1, Command: "echo 1", Result: "ok", StartTime: oldTime, Status: models.Finish}) // Query tasks with custom retention - should find none var tasks []models.Task models.Db.Where("log_retention_days > 0").Find(&tasks) if len(tasks) != 0 { t.Errorf("expected 0 tasks with custom retention, got %d", len(tasks)) } // Logs should still exist var count int64 models.Db.Model(&models.TaskLog{}).Where("task_id = ?", 1).Count(&count) if count != 1 { t.Errorf("expected 1 log before global cleanup, got %d", count) } // Global cleanup with 5-day retention should remove it taskLogModel := new(models.TaskLog) deleted, err := taskLogModel.RemoveByDays(5) if err != nil { t.Fatalf("global cleanup failed: %v", err) } if deleted != 1 { t.Errorf("expected 1 deleted by global cleanup, got %d", deleted) } } ================================================ FILE: internal/service/task_partial_output_test.go ================================================ package service import ( "fmt" "strings" "testing" "time" "github.com/gocronx-team/gocron/internal/models" ) // 模拟RPC处理器，用于测试部分输出功能 type mockRPCHandlerWithPartialOutput struct { partialOutput string errorType string // "timeout", "manual_stop", "normal_error", "success" } func (m *mockRPCHandlerWithPartialOutput) Run(taskModel models.Task, taskUniqueId int64) (string, error) { switch m.errorType { case "timeout": // 模拟超时情况，返回部分输出和超时错误 return m.partialOutput + "\n\n[执行超时，强制结束]", fmt.Errorf("执行超时, 强制结束") case "manual_stop": // 模拟手动停止情况，返回部分输出和手动停止错误 return m.partialOutput + "\n\n[手动停止]", fmt.Errorf("手动停止") case "normal_error": // 模拟普通错误，返回完整输出 return m.partialOutput, fmt.Errorf("command failed") case "success": // 模拟成功情况 return m.partialOutput, nil default: return "", fmt.Errorf("unknown error type") } } func TestExecJobWithPartialOutput_Timeout(t *testing.T) { handler := &mockRPCHandlerWithPartialOutput{ partialOutput: "Task started\nProcessing data...\nPartial result: 50%", errorType: "timeout", } task := models.Task{ Id: 1, Name: "timeout-test", RetryTimes: 0, // 不重试，直接测试超时 } result := execJob(handler, task, 1) if result.Err == nil { t.Fatal("Expected timeout error") } if result.Err.Error() != "执行超时, 强制结束" { t.Fatalf("Expected timeout error, got: %v", result.Err) } if !strings.Contains(result.Result, "Task started") { t.Fatalf("Expected partial output to contain 'Task started', got: %s", result.Result) } if !strings.Contains(result.Result, "Partial result: 50%") { t.Fatalf("Expected partial output to contain 'Partial result: 50%%', got: %s", result.Result) } if !strings.Contains(result.Result, "[执行超时，强制结束]") { t.Fatalf("Expected timeout marker in output, got: %s", result.Result) } } func TestExecJobWithPartialOutput_ManualStop(t *testing.T) { handler := &mockRPCHandlerWithPartialOutput{ partialOutput: "Task started\nProcessing batch 1\nProcessing batch 2", errorType: "manual_stop", } task := models.Task{ Id: 2, Name: "manual-stop-test", RetryTimes: 0, } result := execJob(handler, task, 2) if result.Err == nil { t.Fatal("Expected manual stop error") } if result.Err.Error() != "手动停止" { t.Fatalf("Expected manual stop error, got: %v", result.Err) } if !strings.Contains(result.Result, "Processing batch 1") { t.Fatalf("Expected partial output to contain 'Processing batch 1', got: %s", result.Result) } if !strings.Contains(result.Result, "[手动停止]") { t.Fatalf("Expected manual stop marker in output, got: %s", result.Result) } } func TestExecJobWithPartialOutput_NormalError(t *testing.T) { handler := &mockRPCHandlerWithPartialOutput{ partialOutput: "Task started\nError occurred: file not found", errorType: "normal_error", } task := models.Task{ Id: 3, Name: "error-test", RetryTimes: 0, } result := execJob(handler, task, 3) if result.Err == nil { t.Fatal("Expected normal error") } if result.Err.Error() != "command failed" { t.Fatalf("Expected 'command failed' error, got: %v", result.Err) } // 对于普通错误，应该返回完整输出，不添加特殊标记 if !strings.Contains(result.Result, "Error occurred: file not found") { t.Fatalf("Expected full output for normal error, got: %s", result.Result) } if strings.Contains(result.Result, "[执行超时，强制结束]") || strings.Contains(result.Result, "[手动停止]") { t.Fatalf("Should not contain timeout/stop markers for normal error, got: %s", result.Result) } } func TestExecJobWithPartialOutput_Success(t *testing.T) { handler := &mockRPCHandlerWithPartialOutput{ partialOutput: "Task started\nProcessing completed\nResult: success", errorType: "success", } task := models.Task{ Id: 4, Name: "success-test", RetryTimes: 0, } result := execJob(handler, task, 4) if result.Err != nil { t.Fatalf("Expected no error for success, got: %v", result.Err) } if !strings.Contains(result.Result, "Result: success") { t.Fatalf("Expected success output, got: %s", result.Result) } if strings.Contains(result.Result, "[执行超时，强制结束]") || strings.Contains(result.Result, "[手动停止]") { t.Fatalf("Should not contain error markers for success, got: %s", result.Result) } } // 可重写的假处理器，用于测试 type overridableHandler struct { runFunc func(models.Task, int64) (string, error) } func (h *overridableHandler) Run(taskModel models.Task, taskUniqueId int64) (string, error) { return h.runFunc(taskModel, taskUniqueId) } func TestExecJobWithPartialOutput_RetryWithTimeout(t *testing.T) { // 测试重试机制与部分输出的结合 callCount := 0 results := []handlerResponse{ {result: "First attempt\nPartial output\n\n[执行超时，强制结束]", err: fmt.Errorf("执行超时, 强制结束")}, {result: "Second attempt\nSuccess!", err: nil}, } handler := &overridableHandler{ runFunc: func(taskModel models.Task, taskUniqueId int64) (string, error) { result := results[callCount] callCount++ return result.result, result.err }, } task := models.Task{ Id: 5, Name: "retry-test", RetryTimes: 1, RetryInterval: 0, // 不等待，加快测试 } // 模拟sleep函数，避免实际等待 originalSleep := sleepFunc sleepFunc = func(d time.Duration) { // 不实际睡眠 } defer func() { sleepFunc = originalSleep }() result := execJob(handler, task, 5) if result.Err != nil { t.Fatalf("Expected success after retry, got: %v", result.Err) } if !strings.Contains(result.Result, "Second attempt") { t.Fatalf("Expected second attempt output, got: %s", result.Result) } if result.RetryTimes != 1 { t.Fatalf("Expected 1 retry, got: %d", result.RetryTimes) } if callCount != 2 { t.Fatalf("Expected 2 handler calls, got: %d", callCount) } } ================================================ FILE: internal/service/task_test.go ================================================ package service import ( "errors" "net/http" "os" "strings" "testing" "time" "github.com/gocronx-team/gocron/internal/models" "github.com/gocronx-team/gocron/internal/modules/httpclient" "github.com/gocronx-team/gocron/internal/modules/logger" "github.com/gocronx-team/gocron/internal/modules/notify" ) func TestMain(m *testing.M) { _ = os.MkdirAll("log", 0o755) logger.InitLogger() os.Exit(m.Run()) } func TestHTTPHandlerRunGetUsesCustomTimeout(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() var capturedTimeout int httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { if url != "http://example.com" { t.Fatalf("unexpected url %s", url) } capturedTimeout = timeout return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", Timeout: 1000, HttpMethod: models.TaskHTTPMethodGet, } result, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if result != "ok" { t.Fatalf("unexpected result %s", result) } // Custom timeout should be preserved (no longer capped at 300) if capturedTimeout != 1000 { t.Fatalf("expected timeout 1000, got %d", capturedTimeout) } } func TestHTTPHandlerRunGetDefaultTimeout(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() var capturedTimeout int httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { capturedTimeout = timeout return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", Timeout: 0, // not set HttpMethod: models.TaskHTTPMethodGet, } handler.Run(task, 1) if capturedTimeout != HttpDefaultTimeout { t.Fatalf("expected default timeout %d, got %d", HttpDefaultTimeout, capturedTimeout) } } func TestHTTPHandlerRunPostParsesParams(t *testing.T) { original := httpPostParamsFunc defer func() { httpPostParamsFunc = original }() var capturedURL, capturedParams string var capturedTimeout int httpPostParamsFunc = func(url, params string, timeout int) httpclient.ResponseWrapper { capturedURL = url capturedParams = params capturedTimeout = timeout return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "posted"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com/cmd?foo=bar", Timeout: 10, HttpMethod: models.TaskHttpMethodPost, } result, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if result != "posted" { t.Fatalf("unexpected result %s", result) } if capturedURL != "http://example.com/cmd" || capturedParams != "foo=bar" { t.Fatalf("unexpected url/params %s %s", capturedURL, capturedParams) } if capturedTimeout != 10 { t.Fatalf("expected timeout 10, got %d", capturedTimeout) } } func TestHTTPHandlerRunReturnsErrorForNon200(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{StatusCode: http.StatusInternalServerError, Body: "bad"} } handler := &HTTPHandler{} task := models.Task{Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet} result, err := handler.Run(task, 1) if err == nil { t.Fatal("expected error for non-200 response") } if result != "bad" { t.Fatalf("unexpected result %s", result) } } func TestHTTPHandlerRunPostJsonBody(t *testing.T) { original := httpPostJsonFunc defer func() { httpPostJsonFunc = original }() var capturedBody string httpPostJsonFunc = func(url, body string, timeout int) httpclient.ResponseWrapper { capturedBody = body return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com/api", HttpMethod: models.TaskHttpMethodPost, HttpBody: `{"key":"value"}`, Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if capturedBody != `{"key":"value"}` { t.Fatalf("expected JSON body, got %s", capturedBody) } } func TestHTTPHandlerRunPostFallbackToParams(t *testing.T) { original := httpPostParamsFunc defer func() { httpPostParamsFunc = original }() var capturedParams string httpPostParamsFunc = func(url, params string, timeout int) httpclient.ResponseWrapper { capturedParams = params return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com/api?a=1", HttpMethod: models.TaskHttpMethodPost, HttpBody: "", // empty — should fallback to URL params Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if capturedParams != "a=1" { t.Fatalf("expected params 'a=1', got %s", capturedParams) } } func TestHTTPHandlerRunSuccessPatternMatch(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: `{"status":"ok","code":0}`} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, SuccessPattern: `"code":0`, Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("expected success, got error: %v", err) } } func TestHTTPHandlerRunSuccessPatternNoMatch(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: `{"status":"error","code":1}`} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, SuccessPattern: `"code":0`, Timeout: 10, } _, err := handler.Run(task, 1) if err == nil { t.Fatal("expected error for non-matching pattern") } if !strings.Contains(err.Error(), "does not match success_pattern") { t.Fatalf("unexpected error: %v", err) } } func TestHTTPHandlerRunSuccessPatternInvalidRegex(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, SuccessPattern: `[invalid`, Timeout: 10, } _, err := handler.Run(task, 1) if err == nil { t.Fatal("expected error for invalid regex") } if !strings.Contains(err.Error(), "invalid success_pattern regex") { t.Fatalf("unexpected error: %v", err) } } func TestHTTPHandlerRunSuccessPatternMatchCompactJSON(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() // 模拟 pretty-printed JSON 响应（如 httpbin.org） httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{ StatusCode: http.StatusOK, Body: `{ "json": { "key": "value" }, "data": "{\"key\":\"value\"}" }`, } } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, SuccessPattern: `"key":"value"`, Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("expected success with compacted JSON matching, got error: %v", err) } } func TestHTTPHandlerRunEmptyPatternSkipsCheck(t *testing.T) { original := httpGetFunc defer func() { httpGetFunc = original }() httpGetFunc = func(url string, timeout int) httpclient.ResponseWrapper { return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "anything"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, SuccessPattern: "", // empty — should skip assertion Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("expected success with empty pattern, got: %v", err) } } func TestHTTPHandlerRunGetWithHeaders(t *testing.T) { original := httpGetWithHeadersFunc defer func() { httpGetWithHeadersFunc = original }() var capturedHeaders string httpGetWithHeadersFunc = func(url, headers string, timeout int) httpclient.ResponseWrapper { capturedHeaders = headers return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com", HttpMethod: models.TaskHTTPMethodGet, HttpHeaders: `{"Authorization":"Bearer abc"}`, Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if capturedHeaders != `{"Authorization":"Bearer abc"}` { t.Fatalf("expected headers passed through, got %s", capturedHeaders) } } func TestHTTPHandlerRunPostJsonWithHeaders(t *testing.T) { original := httpPostJsonWithHdrsFunc defer func() { httpPostJsonWithHdrsFunc = original }() var capturedBody, capturedHeaders string httpPostJsonWithHdrsFunc = func(url, body, headers string, timeout int) httpclient.ResponseWrapper { capturedBody = body capturedHeaders = headers return httpclient.ResponseWrapper{StatusCode: http.StatusOK, Body: "ok"} } handler := &HTTPHandler{} task := models.Task{ Command: "http://example.com/api", HttpMethod: models.TaskHttpMethodPost, HttpBody: `{"key":"val"}`, HttpHeaders: `{"X-Token":"secret"}`, Timeout: 10, } _, err := handler.Run(task, 1) if err != nil { t.Fatalf("unexpected error: %v", err) } if capturedBody != `{"key":"val"}` { t.Fatalf("expected body, got %s", capturedBody) } if capturedHeaders != `{"X-Token":"secret"}` { t.Fatalf("expected headers, got %s", capturedHeaders) } } type fakeHandler struct { results []handlerResponse callCount int } type handlerResponse struct { result string err error } func (f *fakeHandler) Run(taskModel models.Task, taskUniqueId int64) (string, error) { res := f.results[f.callCount] f.callCount++ return res.result, res.err } func TestExecJobRetriesUntilSuccess(t *testing.T) { originalSleep := sleepFunc defer func() { sleepFunc = originalSleep }() sleepCalls := 0 sleepFunc = func(d time.Duration) { sleepCalls++ } handler := &fakeHandler{ results: []handlerResponse{ {result: "first", err: errors.New("fail1")}, {result: "second", err: nil}, }, } task := models.Task{Id: 1, RetryTimes: 1, RetryInterval: 1} result := execJob(handler, task, 1) if result.Result != "second" || result.Err != nil { t.Fatalf("unexpected result: %+v", result) } if result.RetryTimes != 1 { t.Fatalf("expected RetryTimes 1, got %d", result.RetryTimes) } if handler.callCount != 2 { t.Fatalf("expected 2 handler calls, got %d", handler.callCount) } if sleepCalls != 1 { t.Fatalf("expected 1 sleep call, got %d", sleepCalls) } } func TestExecJobReturnsErrorAfterRetriesExhausted(t *testing.T) { originalSleep := sleepFunc defer func() { sleepFunc = originalSleep }() sleepCount := 0 sleepFunc = func(d time.Duration) { sleepCount++ } handler := &fakeHandler{ results: []handlerResponse{ {result: "first", err: errors.New("fail1")}, {result: "second", err: errors.New("fail2")}, {result: "third", err: errors.New("fail3")}, }, } task := models.Task{Id: 2, RetryTimes: 2, RetryInterval: 1} result := execJob(handler, task, 1) if result.Err == nil { t.Fatal("expected error") } if result.RetryTimes != task.RetryTimes { t.Fatalf("expected retryTimes %d, got %d", task.RetryTimes, result.RetryTimes) } if handler.callCount != 3 { t.Fatalf("expected 3 handler calls, got %d", handler.callCount) } if sleepCount != 2 { t.Fatalf("expected 2 sleep calls, got %d", sleepCount) } } func TestSendNotificationBehavior(t *testing.T) { type expectation struct { name string task models.Task result TaskResult count int check func(t *testing.T, msg notify.Message) } tests := []expectation{ { name: "disabled", task: models.Task{NotifyStatus: 0}, count: 0, }, { name: "failOnlySuccess", task: models.Task{NotifyStatus: 1, NotifyType: 1, NotifyReceiverId: "user"}, result: TaskResult{Result: "ok", Err: nil}, count: 0, }, { name: "failOnlyTriggered", task: models.Task{Name: "job", NotifyStatus: 1, NotifyType: 1, NotifyReceiverId: "user"}, result: TaskResult{Result: "bad", Err: errors.New("boom")}, count: 1, }, { name: "keywordMismatch", task: models.Task{NotifyStatus: 3, NotifyType: 2, NotifyKeyword: "ERROR"}, result: TaskResult{Result: "all good"}, count: 0, }, { name: "keywordMatch", task: models.Task{Name: "job", NotifyStatus: 3, NotifyType: 2, NotifyKeyword: "ERROR"}, result: TaskResult{Result: "found ERROR", Err: nil}, count: 1, check: func(t *testing.T, msg notify.Message) { if msg["status"] != "Success" { t.Fatalf("expected status Success, got %v", msg["status"]) } }, }, { name: "missingReceiverForMail", task: models.Task{NotifyStatus: 2, NotifyType: 1, NotifyReceiverId: ""}, result: TaskResult{Result: "any"}, count: 0, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { captured := stubNotifyPush(t) SendNotification(tt.task, tt.result) if len(*captured) != tt.count { t.Fatalf("expected %d notifications, got %d", tt.count, len(*captured)) } if tt.count > 0 && tt.check != nil { tt.check(t, (*captured)[0]) } }) } } func stubNotifyPush(t *testing.T) *[]notify.Message { t.Helper() var captured []notify.Message original := notifyPushFunc notifyPushFunc = func(msg notify.Message) { msgCopy := notify.Message{} for k, v := range msg { msgCopy[k] = v } captured = append(captured, msgCopy) } t.Cleanup(func() { notifyPushFunc = original }) return &captured } // 测试依赖任务执行逻辑 - 简化版本，直接测试逻辑分支 func TestExecDependencyTaskLogic(t *testing.T) { tests := []struct { name string parentTask models.Task taskResult TaskResult shouldExit bool // 是否应该提前退出（不查询数据库） reason string }{ { name: "子任务不应该触发依赖任务", parentTask: models.Task{ Id: 1, Level: models.TaskLevelChild, DependencyTaskId: "2,3", }, taskResult: TaskResult{Err: nil}, shouldExit: true, reason: "Level is Child", }, { name: "没有依赖任务ID", parentTask: models.Task{ Id: 1, Level: models.TaskLevelParent, DependencyTaskId: "", }, taskResult: TaskResult{Err: nil}, shouldExit: true, reason: "Empty DependencyTaskId", }, { name: "强依赖且父任务失败", parentTask: models.Task{ Id: 1, Level: models.TaskLevelParent, DependencyTaskId: "2,3", DependencyStatus: models.TaskDependencyStatusStrong, }, taskResult: TaskResult{Err: errors.New("parent failed")}, shouldExit: true, reason: "Strong dependency and parent failed", }, { name: "弱依赖且父任务失败应该继续", parentTask: models.Task{ Id: 1, Level: models.TaskLevelParent, DependencyTaskId: "2", DependencyStatus: models.TaskDependencyStatusWeak, }, taskResult: TaskResult{Err: errors.New("parent failed")}, shouldExit: false, reason: "Weak dependency, should continue", }, { name: "父任务成功应该继续", parentTask: models.Task{ Id: 1, Level: models.TaskLevelParent, DependencyTaskId: "2,3", DependencyStatus: models.TaskDependencyStatusStrong, }, taskResult: TaskResult{Err: nil}, shouldExit: false, reason: "Parent success, should continue", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { // 测试逻辑分支 if tt.parentTask.Level != models.TaskLevelParent { if !tt.shouldExit { t.Errorf("Expected to exit for child task, but shouldExit is false") } t.Logf("✓ Correctly exits for: %s", tt.reason) return } if tt.parentTask.DependencyTaskId == "" { if !tt.shouldExit { t.Errorf("Expected to exit for empty dependency ID, but shouldExit is false") } t.Logf("✓ Correctly exits for: %s", tt.reason) return } if tt.parentTask.DependencyStatus == models.TaskDependencyStatusStrong && tt.taskResult.Err != nil { if !tt.shouldExit { t.Errorf("Expected to exit for strong dependency failure, but shouldExit is false") } t.Logf("✓ Correctly exits for: %s", tt.reason) return } // 如果到这里，说明应该继续执行 if tt.shouldExit { t.Errorf("Should have exited but didn't for: %s", tt.reason) } else { t.Logf("✓ Correctly continues for: %s (would query DB and execute)", tt.reason) } }) } } ================================================ FILE: makefile ================================================ GO111MODULE=on # 版本信息 VERSION ?= $(shell git describe --tags --always --dirty) GIT_COMMIT ?= $(shell git rev-parse --short HEAD) BUILD_DATE ?= $(shell date '+%Y-%m-%d %H:%M:%S') LDFLAGS = -w -X 'main.AppVersion=$(VERSION)' -X 'main.BuildDate=$(BUILD_DATE)' -X 'main.GitCommit=$(GIT_COMMIT)' # 构建目录 BIN_DIR = bin PACKAGE_DIR = packages # 默认目标 .DEFAULT_GOAL := build # 本地构建 .PHONY: build build: gocron node .PHONY: build-race build-race: enable-race build .PHONY: run run: build kill ./$(BIN_DIR)/gocron-node & ./$(BIN_DIR)/gocron web -e dev .PHONY: run-with-packages run-with-packages: build-web package-all kill ./$(BIN_DIR)/gocron-node & ./$(BIN_DIR)/gocron web -e dev .PHONY: run-race run-race: enable-race run .PHONY: kill kill: -killall gocron-node .PHONY: gocron gocron: @mkdir -p $(BIN_DIR) go build $(RACE) -ldflags "$(LDFLAGS)" -o $(BIN_DIR)/gocron ./cmd/gocron .PHONY: node node: @mkdir -p $(BIN_DIR) CGO_ENABLED=0 go build $(RACE) -ldflags "$(LDFLAGS)" -o $(BIN_DIR)/gocron-node ./cmd/node .PHONY: test test: go test $(RACE) ./... .PHONY: test-race test-race: enable-race test .PHONY: enable-race enable-race: $(eval RACE = -race) # 多平台打包 .PHONY: package package: build-web @echo "Building packages for current platform..." bash ./package.sh .PHONY: package-linux package-linux: build-web @echo "Building packages for Linux..." bash ./package.sh -p linux -a "amd64,arm64" .PHONY: package-linux-nosqlite package-linux-nosqlite: build-web @echo "Building Linux packages without SQLite..." CGO_ENABLED=0 bash ./package.sh -p linux -a "amd64,arm64" .PHONY: package-darwin package-darwin: build-web @echo "Building packages for macOS..." bash ./package.sh -p darwin -a "amd64,arm64" .PHONY: package-windows package-windows: build-web @echo "Building packages for Windows..." bash ./package.sh -p windows -a "amd64" .PHONY: package-all package-all: build-web @echo "Building packages for all platforms..." bash ./package.sh -p "linux,darwin" -a "amd64,arm64" bash ./package.sh -p "windows" -a "amd64" # 前端构建 .PHONY: build-vue build-vue: @echo "Installing Vue dependencies..." @if [ -f web/vue/pnpm-lock.yaml ]; then \ echo "Using pnpm..."; \ cd web/vue && pnpm install; \ elif [ -f web/vue/yarn.lock ]; then \ echo "Using yarn..."; \ cd web/vue && yarn install; \ else \ echo "Using npm..."; \ cd web/vue && npm install; \ fi @echo "Building Vue frontend..." @if [ -f web/vue/pnpm-lock.yaml ]; then \ cd web/vue && pnpm run build; \ elif [ -f web/vue/yarn.lock ]; then \ cd web/vue && yarn run build; \ else \ cd web/vue && npm run build; \ fi @echo "✅ Vue build complete! Files will be embedded during Go build." .PHONY: install-vue install-vue: @echo "Installing Vue dependencies..." @if [ -f web/vue/pnpm-lock.yaml ]; then \ cd web/vue && pnpm install; \ elif [ -f web/vue/yarn.lock ]; then \ cd web/vue && yarn install; \ else \ cd web/vue && npm install; \ fi .PHONY: run-vue run-vue: @echo "Starting Vue dev server..." @if [ -f web/vue/pnpm-lock.yaml ]; then \ cd web/vue && pnpm run dev; \ elif [ -f web/vue/yarn.lock ]; then \ cd web/vue && yarn run dev; \ else \ cd web/vue && npm run dev; \ fi .PHONY: build-web build-web: build-vue @echo "Web build complete!" # 代码质量检查 .PHONY: check check: fmt vet test @echo "✅ All checks passed!" .PHONY: lint lint: @echo "Running linter..." @if command -v golangci-lint >/dev/null 2>&1; then \ golangci-lint run || echo "⚠️ Linter found issues (non-blocking)"; \ else \ echo "⚠️ golangci-lint not installed, skipping..."; \ echo " Install: go install github.com/golangci/golangci-lint/cmd/golangci-lint@latest"; \ fi .PHONY: fmt fmt: @echo "Formatting code..." @find . -name '*.go' -exec gofmt -w {} \; .PHONY: fmt-check fmt-check: @echo "Checking code formatting..." @unformatted=$$(gofmt -l .); \ if [ -n "$$unformatted" ]; then \ echo "❌ Code not formatted:" && echo "$$unformatted" && echo "Run 'make fmt' to fix" && exit 1; \ fi @echo "✅ Code formatting OK" .PHONY: vet vet: @echo "Running go vet..." @go vet ./... @echo "✅ go vet passed" .PHONY: test-coverage test-coverage: @echo "Running tests with coverage..." @go test -cover -coverprofile=coverage.out ./... @go tool cover -html=coverage.out -o coverage.html @echo "✅ Coverage report generated: coverage.html" .PHONY: security security: @echo "Running security checks..." @if command -v gosec >/dev/null 2>&1; then \ gosec ./... || echo "⚠️ Security issues found (non-blocking)"; \ else \ echo "⚠️ gosec not installed, skipping..."; \ echo " Install: go install github.com/securego/gosec/v2/cmd/gosec@latest"; \ fi # 预发布检查（完整检查） .PHONY: pre-release pre-release: clean @echo "==========================================" @echo "Running pre-release checks..." @echo "==========================================" @$(MAKE) fmt-check @$(MAKE) vet @$(MAKE) test @echo "" @echo "==========================================" @echo "✅ All pre-release checks passed!" @echo "==========================================" @echo "" @echo "Optional checks (run manually if needed):" @echo " make lint - Code quality linter" @echo " make security - Security vulnerability scan" # 清理 .PHONY: clean clean: @echo "Cleaning build artifacts..." -rm -rf $(BIN_DIR) -rm -rf $(PACKAGE_DIR) -rm -rf gocron-package -rm -rf gocron-node-package -rm -rf gocron-build -rm -rf gocron-node-build -rm -f coverage.out coverage.html .PHONY: clean-web clean-web: @echo "Cleaning web build artifacts..." -rm -rf web/vue/dist -rm -rf web/public/static -rm -f web/public/index.html # 开发工具 .PHONY: dev-deps dev-deps: @echo "Installing development dependencies..." go install github.com/cosmtrek/air@latest go install github.com/golangci/golangci-lint/cmd/golangci-lint@latest go install github.com/securego/gosec/v2/cmd/gosec@latest # 版本管理 .PHONY: version version: @echo "Current version: $(VERSION)" @echo "Recent releases:" @git tag -l "v*.*.*" | grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$$' | sort -V | tail -5 .PHONY: release release: @if [ -z "$(VERSION)" ]; then \ echo "Error: VERSION is required. Usage: make release VERSION=v1.3.18"; \ exit 1; \ fi @echo "Creating release $(VERSION)..." @if git rev-parse $(VERSION) >/dev/null 2>&1; then \ echo "Error: Tag $(VERSION) already exists!"; \ exit 1; \ fi @git tag -a $(VERSION) -m "Release $(VERSION)" @git push origin $(VERSION) @echo "✅ Release $(VERSION) created and pushed successfully!" .PHONY: release-patch release-patch: @echo "Creating patch release..." @$(MAKE) release VERSION=$$($(MAKE) next-patch) .PHONY: release-minor release-minor: @echo "Creating minor release..." @$(MAKE) release VERSION=$$($(MAKE) next-minor) .PHONY: release-major release-major: @echo "Creating major release..." @$(MAKE) release VERSION=$$($(MAKE) next-major) .PHONY: next-patch next-patch: @latest=$$(git tag -l "v*.*.*" | grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$$' | sort -V | tail -1); \ if [ -z "$$latest" ]; then \ echo "v1.0.0"; \ else \ echo $$latest | sed 's/^v//' | awk -F. '{printf "v%d.%d.%d", $$1, $$2, $$3+1}'; \ fi .PHONY: next-minor next-minor: @latest=$$(git tag -l "v*.*.*" | grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$$' | sort -V | tail -1); \ if [ -z "$$latest" ]; then \ echo "v1.0.0"; \ else \ echo $$latest | sed 's/^v//' | awk -F. '{printf "v%d.%d.0", $$1, $$2+1}'; \ fi .PHONY: next-major next-major: @latest=$$(git tag -l "v*.*.*" | grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$$' | sort -V | tail -1); \ if [ -z "$$latest" ]; then \ echo "v1.0.0"; \ else \ echo $$latest | sed 's/^v//' | awk -F. '{printf "v%d.0.0", $$1+1}'; \ fi .PHONY: delete-tag delete-tag: @if [ -z "$(VERSION)" ]; then \ echo "Error: VERSION is required. Usage: make delete-tag VERSION=v1.3.18"; \ exit 1; \ fi @echo "Deleting tag $(VERSION)..." @git tag -d $(VERSION) @git push origin :refs/tags/$(VERSION) @echo "✅ Tag $(VERSION) deleted locally and remotely" # 帮助信息 .PHONY: help help: @echo "Available targets:" @echo "" @echo "Build:" @echo " build - Build gocron and gocron-node for current platform" @echo " run - Build and run in development mode" @echo " test - Run tests" @echo " package-all - Build packages for all platforms" @echo "" @echo "Code Quality:" @echo " check - Run fmt + vet + test" @echo " pre-release - Run all checks before release" @echo " fmt - Format code" @echo " fmt-check - Check code formatting" @echo " vet - Run go vet" @echo " lint - Run linter (golangci-lint)" @echo " test-coverage - Run tests with coverage report" @echo " security - Run security checks (gosec)" @echo "" @echo "Version Management:" @echo " version - Show current version" @echo " release - Create release tag (VERSION=v1.3.18)" @echo " release-patch - Auto increment patch version" @echo "" @echo "Development:" @echo " dev-deps - Install development dependencies" @echo " clean - Clean build artifacts" @echo " help - Show this help message" ================================================ FILE: package.json ================================================ { "name": "gocron", "version": "2.0.0", "description": "定时任务管理系统", "author": "gocronx ", "private": true, "type": "module", "scripts": { "prepare": "husky", "commit": "git-cz", "lint:lint-staged": "lint-staged" }, "config": { "commitizen": { "path": "node_modules/cz-git" } }, "lint-staged": { "web/vue/**/*.{js,ts,vue}": [ "cd web/vue && pnpm exec eslint --fix", "cd web/vue && pnpm exec prettier --write" ], "*.{json,md,yml,yaml}": [ "cd web/vue && pnpm exec prettier --write" ] }, "pnpm": { "overrides": { "lodash": ">=4.18.0", "tmp@<0.2.4": "0.2.4", "flatted": ">=3.4.2" } }, "devDependencies": { "@commitlint/cli": "^20.5.0", "@commitlint/config-conventional": "^20.5.0", "commitizen": "^4.3.1", "cz-git": "^1.12.0", "eslint": "^10.1.0", "husky": "^9.1.7", "lint-staged": "^16.4.0", "prettier": "^3.8.1" } } ================================================ FILE: package.sh ================================================ #!/usr/bin/env bash # 生成压缩包 xx.tar.gz或xx.zip # 使用 ./package.sh -a amd664 -p linux -v v2.0.0 # 任何命令返回非0值退出 set -o errexit # 使用未定义的变量退出 set -o nounset # 管道中任一命令执行失败退出 set -o pipefail # 获取 Go 环境变量 GOHOSTOS=$(go env GOHOSTOS) GOHOSTARCH=$(go env GOHOSTARCH) # 二进制文件名 BINARY_NAME='' # main函数所在文件 MAIN_FILE="" # 提取git最新tag作为应用版本 VERSION='' # 最新git commit id GIT_COMMIT_ID='' # 外部输入的系统 INPUT_OS=() # 外部输入的架构 INPUT_ARCH=() # 未指定OS，默认值 DEFAULT_OS=${GOHOSTOS} # 未指定ARCH,默认值 DEFAULT_ARCH=${GOHOSTARCH} # 支持的系统 SUPPORT_OS=(linux darwin windows) # 支持的架构 SUPPORT_ARCH=(386 amd64 arm64) # 编译参数 LDFLAGS='' # 需要打包的文件 INCLUDE_FILE=() # 打包文件生成目录 PACKAGE_DIR='' # 编译文件生成目录 BUILD_DIR='' # 获取git 最新tag name git_latest_tag() { local COMMIT_ID="" local TAG_NAME="" COMMIT_ID=`git rev-list --tags --max-count=1` TAG_NAME=`git describe --tags "${COMMIT_ID}"` echo ${TAG_NAME} } # 获取git 最新commit id git_latest_commit() { echo "$(git rev-parse --short HEAD)" } # 打印信息 print_message() { echo "$1" } # 打印信息后推出 print_message_and_exit() { if [[ -n $1 ]]; then print_message "$1" fi exit 1 } # 设置系统、CPU架构 set_os_arch() { if [[ ${#INPUT_OS[@]} = 0 ]];then INPUT_OS=("${DEFAULT_OS}") fi if [[ ${#INPUT_ARCH[@]} = 0 ]];then INPUT_ARCH=("${DEFAULT_ARCH}") fi for OS in "${INPUT_OS[@]}"; do if [[ ! "${SUPPORT_OS[*]}" =~ ${OS} ]]; then print_message_and_exit "不支持的系统${OS}" fi done for ARCH in "${INPUT_ARCH[@]}";do if [[ ! "${SUPPORT_ARCH[*]}" =~ ${ARCH} ]]; then print_message_and_exit "不支持的CPU架构${ARCH}" fi done } # 初始化 init() { set_os_arch if [[ -z "${VERSION}" ]];then VERSION=`git_latest_tag` fi GIT_COMMIT_ID=`git_latest_commit` LDFLAGS="-w -X 'main.AppVersion=${VERSION}' -X 'main.BuildDate=`date '+%Y-%m-%d %H:%M:%S'`' -X 'main.GitCommit=${GIT_COMMIT_ID}'" PACKAGE_DIR=${BINARY_NAME}-package BUILD_DIR=${BINARY_NAME}-build # 只清理 BUILD_DIR，保留 PACKAGE_DIR 以支持增量构建 if [[ -d ${BUILD_DIR} ]];then rm -rf ${BUILD_DIR} fi mkdir -p ${BUILD_DIR} mkdir -p ${PACKAGE_DIR} } # 编译 build() { local FILENAME='' for OS in "${INPUT_OS[@]}";do for ARCH in "${INPUT_ARCH[@]}";do if [[ "${OS}" = "windows" ]];then FILENAME=${BINARY_NAME}.exe else FILENAME=${BINARY_NAME} fi print_message "编译 ${BINARY_NAME} ${OS}-${ARCH} 版本" env CGO_ENABLED=0 GOOS=${OS} GOARCH=${ARCH} go build -ldflags "${LDFLAGS}" -o ${BUILD_DIR}/${BINARY_NAME}-${OS}-${ARCH}/${FILENAME} ${MAIN_FILE} done done } # 打包 package_binary() { cd ${BUILD_DIR} for OS in "${INPUT_OS[@]}";do for ARCH in "${INPUT_ARCH[@]}";do package_file ${BINARY_NAME}-${OS}-${ARCH} # gocron-node 不使用版本号 if [[ "${BINARY_NAME}" = "gocron-node" ]]; then if [[ "${OS}" = "windows" ]];then zip -rq ../${PACKAGE_DIR}/${BINARY_NAME}-${OS}-${ARCH}.zip ${BINARY_NAME}-${OS}-${ARCH} else tar czf ../${PACKAGE_DIR}/${BINARY_NAME}-${OS}-${ARCH}.tar.gz ${BINARY_NAME}-${OS}-${ARCH} fi elif [[ -z "${VERSION}" ]]; then if [[ "${OS}" = "windows" ]];then zip -rq ../${PACKAGE_DIR}/${BINARY_NAME}-${OS}-${ARCH}.zip ${BINARY_NAME}-${OS}-${ARCH} else tar czf ../${PACKAGE_DIR}/${BINARY_NAME}-${OS}-${ARCH}.tar.gz ${BINARY_NAME}-${OS}-${ARCH} fi else if [[ "${OS}" = "windows" ]];then zip -rq ../${PACKAGE_DIR}/${BINARY_NAME}-${VERSION}-${OS}-${ARCH}.zip ${BINARY_NAME}-${OS}-${ARCH} else tar czf ../${PACKAGE_DIR}/${BINARY_NAME}-${VERSION}-${OS}-${ARCH}.tar.gz ${BINARY_NAME}-${OS}-${ARCH} fi fi done done cd ${OLDPWD} } # 打包文件 package_file() { if [[ "${#INCLUDE_FILE[@]}" = "0" ]];then return fi for item in "${INCLUDE_FILE[@]}"; do cp -r ../${item} $1 done } # 清理 clean() { if [[ -d ${BUILD_DIR} ]];then rm -rf ${BUILD_DIR} fi } # 运行 run() { init build package_binary clean } package_gocron() { BINARY_NAME='gocron' MAIN_FILE="./cmd/gocron/gocron.go" INCLUDE_FILE=() run } package_gocron_node() { BINARY_NAME='gocron-node' MAIN_FILE="./cmd/node/node.go" INCLUDE_FILE=() run } # p 平台 linux darwin windows # a 架构 386 amd64 arm64 # v 版本号 默认取git最新tag # t 类型 all(默认), gocron, node BUILD_TYPE="all" while getopts "p:a:v:t:" OPT; do case ${OPT} in p) IFS=',' read -r -a INPUT_OS <<< "${OPTARG}" ;; a) IFS=',' read -r -a INPUT_ARCH <<< "${OPTARG}" ;; v) VERSION=$OPTARG ;; t) BUILD_TYPE=$OPTARG ;; *) ;; esac done # 默认构建所有 if [[ -z "${BUILD_TYPE}" ]]; then BUILD_TYPE="all" fi if [[ "${BUILD_TYPE}" = "all" ]] || [[ "${BUILD_TYPE}" = "gocron" ]]; then package_gocron fi if [[ "${BUILD_TYPE}" = "all" ]] || [[ "${BUILD_TYPE}" = "node" ]]; then package_gocron_node fi ================================================ FILE: release.sh ================================================ #!/bin/bash # 本地构建并发布到 GitHub Release set -e VERSION="" PRERELEASE=false SKIP_CHECKS=false # 解析参数 while [[ $# -gt 0 ]]; do case $1 in -v|--version) VERSION="$2" shift 2 ;; --prerelease) PRERELEASE=true shift ;; --skip-checks) SKIP_CHECKS=true shift ;; *) echo "Unknown option: $1" echo "Usage: $0 -v [--prerelease] [--skip-checks]" echo "Example: $0 -v v1.3.21" exit 1 ;; esac done if [ -z "$VERSION" ]; then echo "Error: Version is required" echo "Usage: $0 -v [--prerelease] [--skip-checks]" exit 1 fi echo "==========================================" echo "Local Build and Release to GitHub" echo "==========================================" echo "Version: $VERSION" echo "Prerelease: $PRERELEASE" echo "Skip Checks: $SKIP_CHECKS" echo "" # 0. 代码质量检查 if [ "$SKIP_CHECKS" = false ]; then echo "0. Running code quality checks..." echo "" # 格式检查 echo " → Checking code formatting..." if ! make fmt-check 2>/dev/null; then echo "❌ Code formatting check failed!" echo " Run 'make fmt' to fix formatting issues" exit 1 fi # go vet 检查 echo " → Running go vet..." if ! make vet 2>/dev/null; then echo "❌ go vet check failed!" exit 1 fi # 运行测试 echo " → Running tests..." if ! make test 2>/dev/null; then echo "❌ Tests failed!" exit 1 fi # 可选：linter 检查 echo " → Running linter (optional)..." make lint 2>/dev/null || echo "⚠️ Linter check skipped" echo "" echo "✅ All code quality checks passed!" echo "" else echo "⚠️ Skipping code quality checks (--skip-checks flag)" echo "" fi # 1. 检查是否需要清理 echo "1. Checking existing builds..." if [ -d "gocron-package" ] && [ -n "$(ls -A gocron-package 2>/dev/null)" ]; then echo "Found existing packages. Clean and rebuild? (y/N): " read -r CLEAN_RESPONSE if [[ $CLEAN_RESPONSE =~ ^[Yy]$ ]]; then rm -rf gocron-package gocron-node-package gocron-build gocron-node-build echo "✓ Cleaned" else echo "✓ Keeping existing packages" fi else echo "✓ No existing packages" fi echo "" # 2. 构建前端 echo "2. Building frontend..." cd web/vue # 检测使用哪个包管理器 if [ -f "pnpm-lock.yaml" ]; then echo "Using pnpm..." pnpm install --frozen-lockfile pnpm run build elif [ -f "yarn.lock" ]; then echo "Using yarn..." yarn install --frozen-lockfile yarn run build else echo "Using npm..." npm ci npm run build fi cd ../.. echo "✓ Frontend built (output: web/vue/dist/)" echo "" # 3. 构建所有平台的包 echo "3. Building packages for all platforms..." MISSING_PACKAGES=false # 检查 Linux/macOS gocron 包 for os in linux darwin; do for arch in amd64 arm64; do if [ ! -f "gocron-package/gocron-${VERSION}-${os}-${arch}.tar.gz" ] || \ [ ! -f "gocron-node-package/gocron-node-${os}-${arch}.tar.gz" ]; then MISSING_PACKAGES=true break 2 fi done done if [ "$MISSING_PACKAGES" = true ]; then echo "Building Linux and macOS packages..." ./package.sh -p "linux,darwin" -a "amd64,arm64" -v "$VERSION" else echo "Linux/macOS packages already exist, skipping..." fi # 检查 Windows 包 if [ ! -f "gocron-package/gocron-${VERSION}-windows-amd64.zip" ] || \ [ ! -f "gocron-node-package/gocron-node-windows-amd64.zip" ]; then echo "Building Windows packages..." ./package.sh -p "windows" -a "amd64" -v "$VERSION" else echo "Windows packages already exist, skipping..." fi echo "✓ All packages built" echo "" # 4. 显示构建结果 echo "4. Build summary:" echo "" echo "gocron packages:" ls -lh gocron-package/ echo "" echo "gocron-node packages:" ls -lh gocron-node-package/ echo "" # 5. 验证包内容 echo "5. Verifying package contents..." SAMPLE_PACKAGE=$(ls gocron-package/*.tar.gz 2>/dev/null | head -1) if [ -n "$SAMPLE_PACKAGE" ]; then echo "Checking: $SAMPLE_PACKAGE" tar tzf "$SAMPLE_PACKAGE" | head -5 echo "✓ Package verified" else SAMPLE_PACKAGE=$(ls gocron-package/*.zip 2>/dev/null | head -1) if [ -n "$SAMPLE_PACKAGE" ]; then echo "Checking: $SAMPLE_PACKAGE" unzip -l "$SAMPLE_PACKAGE" | head -5 echo "✓ Package verified" fi fi echo "" # 6. 创建 Git tag echo "6. Creating Git tag..." if git rev-parse "$VERSION" >/dev/null 2>&1; then echo "Tag $VERSION already exists" read -p "Delete and recreate? (y/N): " -n 1 -r echo if [[ $REPLY =~ ^[Yy]$ ]]; then git tag -d "$VERSION" git push origin ":refs/tags/$VERSION" 2>/dev/null || true else echo "Skipping tag creation" fi fi if ! git rev-parse "$VERSION" >/dev/null 2>&1; then git tag -a "$VERSION" -m "Release $VERSION" git push origin "$VERSION" echo "✓ Tag created and pushed" else echo "✓ Using existing tag" fi echo "" # 7. 创建 GitHub Release echo "7. Creating GitHub Release..." echo "" PRERELEASE_FLAG="" if [ "$PRERELEASE" = true ]; then PRERELEASE_FLAG="--prerelease" fi # 生成 release notes cat > /tmp/release_notes.md < chore: bump to v1.5.8 with migration 158 EOF # 检查 gh CLI 是否安装 if ! command -v gh &> /dev/null; then echo "Error: GitHub CLI (gh) is not installed" echo "Install it from: https://cli.github.com/" echo "" echo "Packages are ready in:" echo " - gocron-package/" echo " - gocron-node-package/" echo "" echo "You can manually create a release on GitHub and upload these files." exit 1 fi # 创建 release gh release create "$VERSION" \ --title "Release $VERSION" \ --notes-file /tmp/release_notes.md \ $PRERELEASE_FLAG \ gocron-package/*.tar.gz \ gocron-package/*.zip \ gocron-node-package/*.tar.gz \ gocron-node-package/*.zip echo "" echo "==========================================" echo "✅ Release $VERSION created successfully!" echo "==========================================" echo "" echo "View release: https://github.com/$(git config --get remote.origin.url | sed 's/.*github.com[:/]\(.*\)\.git/\1/')/releases/tag/$VERSION" ================================================ FILE: test_windows_cmd.go ================================================ //go:build ignore // +build ignore package main import ( "fmt" "os/exec" ) func main() { // 测试不同的命令构造方式 command := `dir "C:\Program Files (x86)"` fmt.Println("原始命令:", command) fmt.Println() // 方式1: 直接传递 fmt.Println("方式1: cmd /C command") cmd1 := exec.Command("cmd", "/C", command) fmt.Printf(" Args: %#v\n", cmd1.Args) fmt.Printf(" String: %s\n\n", cmd1.String()) // 方式2: 使用 /S /C "command" fmt.Println("方式2: cmd /S /C \"command\"") wrappedCommand := `"` + command + `"` cmd2 := exec.Command("cmd", "/S", "/C", wrappedCommand) fmt.Printf(" Args: %#v\n", cmd2.Args) fmt.Printf(" String: %s\n\n", cmd2.String()) // 方式3: 使用 /c 和完整引号包裹 fmt.Println("方式3: cmd /c \"command\"") cmd3 := exec.Command("cmd", "/c", `"`+command+`"`) fmt.Printf(" Args: %#v\n", cmd3.Args) fmt.Printf(" String: %s\n\n", cmd3.String()) } ================================================ FILE: web/vue/.editorconfig ================================================ root = true [*] charset = utf-8 indent_style = space indent_size = 2 end_of_line = lf insert_final_newline = true trim_trailing_whitespace = true ================================================ FILE: web/vue/.gitattributes ================================================ # Auto detect text files and perform LF normalization * text=auto # Source code *.js text eol=lf *.vue text eol=lf *.json text eol=lf *.css text eol=lf *.scss text eol=lf *.md text eol=lf # Binary files *.png binary *.jpg binary *.jpeg binary *.gif binary *.ico binary *.woff binary *.woff2 binary *.ttf binary *.eot binary ================================================ FILE: web/vue/.gitignore ================================================ .DS_Store node_modules/ /dist/ .vite/ npm-debug.log* yarn-debug.log* yarn-error.log* # Editor directories and files .idea .vscode *.suo *.ntvs* *.njsproj *.sln ================================================ FILE: web/vue/.prettierrc.json ================================================ { "semi": false, "singleQuote": true, "printWidth": 100, "trailingComma": "none", "arrowParens": "avoid" } ================================================ FILE: web/vue/README.md ================================================ # gocron > 分布式定时任务管理系统 ## Build Setup ``` bash # install dependencies yarn install # serve with hot reload at localhost:8080 yarn run dev # build for production with minification yarn run build # build for production and view the bundle analyzer report yarn run build --report ``` For a detailed explanation on how things work, check out the [guide](http://vuejs-templates.github.io/webpack/) and [docs for vue-loader](http://vuejs.github.io/vue-loader). ================================================ FILE: web/vue/eslint.config.js ================================================ import js from '@eslint/js' import pluginVue from 'eslint-plugin-vue' export default [ js.configs.recommended, ...pluginVue.configs['flat/recommended'], { languageOptions: { ecmaVersion: 'latest', sourceType: 'module', globals: { window: 'readonly', document: 'readonly', navigator: 'readonly', console: 'readonly', setTimeout: 'readonly', clearTimeout: 'readonly', setInterval: 'readonly', clearInterval: 'readonly', fetch: 'readonly', URL: 'readonly', URLSearchParams: 'readonly', FormData: 'readonly', Blob: 'readonly', File: 'readonly', process: 'readonly', performance: 'readonly', AbortController: 'readonly', __dirname: 'readonly', alert: 'readonly', confirm: 'readonly', localStorage: 'readonly', sessionStorage: 'readonly', history: 'readonly', location: 'readonly', btoa: 'readonly', atob: 'readonly', crypto: 'readonly', structuredClone: 'readonly' } }, rules: { 'vue/multi-word-component-names': 'off', 'no-unused-vars': 'warn', 'no-console': process.env.NODE_ENV === 'production' ? 'warn' : 'off', 'no-debugger': process.env.NODE_ENV === 'production' ? 'error' : 'off', 'vue/no-unused-components': 'warn', 'vue/require-default-prop': 'off' } }, { ignores: ['dist/**', 'node_modules/**'] } ] ================================================ FILE: web/vue/index.html ================================================ gocron - 分布式定时任务系统
================================================ FILE: web/vue/jsconfig.json ================================================ { "compilerOptions": { "target": "es6", "module": "es6", "allowSyntheticDefaultImports": true, "baseUrl": "./", "paths": { "@/*": ["src/*"] } }, "exclude": ["node_modules", "dist"], "include": ["src/**/*"] } ================================================ FILE: web/vue/package.json ================================================ { "name": "gocron", "version": "2.0.0", "description": "定时任务管理系统", "author": "gocronx ", "private": true, "type": "module", "scripts": { "dev": "vite", "build": "vite build", "build:analyze": "vite build --mode analyze", "preview": "vite preview", "lint": "eslint . --fix", "lint:check": "eslint .", "test": "vitest", "test:ui": "vitest --ui" }, "dependencies": { "@element-plus/icons-vue": "^2.3.2", "@vueuse/core": "^14.2.1", "axios": "^1.15.0", "dayjs": "^1.11.20", "element-plus": "^2.13.5", "nprogress": "^0.2.0", "pinia": "^3.0.4", "pinia-plugin-persistedstate": "^4.7.1", "qs": "^6.15.0", "vue": "^3.5.30", "vue-i18n": "^9.14.5", "vue-router": "^4.6.4" }, "devDependencies": { "@eslint/js": "^9.39.4", "@vitejs/plugin-vue": "^6.0.4", "@vitest/ui": "^3.2.4", "@vue/test-utils": "^2.4.6", "eslint": "^9.39.4", "eslint-plugin-vue": "^10.8.0", "jsdom": "^27.4.0", "prettier": "^3.8.1", "unplugin-auto-import": "^20.3.0", "unplugin-vue-components": "^30.0.0", "vite": "^7.3.2", "vite-plugin-compression": "^0.5.1", "vitest": "^3.2.4" }, "pnpm": { "overrides": { "flatted": ">=3.4.2", "picomatch": ">=4.0.4", "brace-expansion": "2.0.3", "lodash": ">=4.18.0", "lodash-es": ">=4.18.0", "defu": ">=6.1.5", "follow-redirects": ">=1.16.0" } } } ================================================ FILE: web/vue/src/App.vue ================================================ ================================================ FILE: web/vue/src/api/agent.js ================================================ import httpClient from '../utils/httpClient' export default { generateToken (callback) { httpClient.post('/agent/generate-token', {}, callback) } } ================================================ FILE: web/vue/src/api/audit.js ================================================ import httpClient from '../utils/httpClient' export default { list (query, callback) { httpClient.get('/audit', query, callback) } } ================================================ FILE: web/vue/src/api/host.js ================================================ import httpClient from '../utils/httpClient' export default { // 任务列表 list (query, callback) { httpClient.get('/host', query, callback) }, all (query, callback) { httpClient.get('/host/all', {}, callback) }, detail (id, callback) { httpClient.get(`/host/${id}`, {}, callback) }, update (data, callback) { httpClient.post('/host/store', data, callback) }, remove (id, callback) { httpClient.post(`/host/remove/${id}`, {}, callback) }, ping (id, callback) { httpClient.get(`/host/ping/${id}`, {}, callback) } } ================================================ FILE: web/vue/src/api/install.js ================================================ import httpClient from '../utils/httpClient' export default { store (data, callback) { httpClient.post('/install/store', data, callback) }, status (callback) { httpClient.get('/install/status', {}, callback) } } ================================================ FILE: web/vue/src/api/notification.js ================================================ import httpClient from '../utils/httpClient' export default { slack(callback) { httpClient.get('/system/slack', {}, callback) }, updateSlack(data, callback) { httpClient.post('/system/slack/update', data, callback) }, createSlackChannel(channel, callback) { httpClient.post('/system/slack/channel', { channel }, callback) }, removeSlackChannel(channelId, callback) { httpClient.post(`/system/slack/channel/remove/${channelId}`, {}, callback) }, mail(callback) { httpClient.get('/system/mail', {}, callback) }, updateMail(data, callback) { httpClient.post('/system/mail/update', data, callback) }, createMailUser(data, callback) { httpClient.post('/system/mail/user', data, callback) }, removeMailUser(userId, callback) { httpClient.post(`/system/mail/user/remove/${userId}`, {}, callback) }, webhook(callback) { httpClient.get('/system/webhook', {}, callback) }, updateWebHook(data, callback) { httpClient.post('/system/webhook/update', data, callback) }, createWebhookUrl(data, callback) { httpClient.post('/system/webhook/url', data, callback) }, removeWebhookUrl(urlId, callback) { httpClient.post(`/system/webhook/url/remove/${urlId}`, {}, callback) } } ================================================ FILE: web/vue/src/api/statistics.js ================================================ import httpClient from '../utils/httpClient' export default { getOverview (callback) { httpClient.get('/statistics/overview', {}, callback) } } ================================================ FILE: web/vue/src/api/system.js ================================================ import httpClient from '../utils/httpClient' export default { loginLogList (query, callback) { httpClient.get('/system/login-log', query, callback) } } ================================================ FILE: web/vue/src/api/task.js ================================================ import httpClient from '../utils/httpClient' export default { list(query, callback) { httpClient.batchGet([{ uri: '/task', params: query }, { uri: '/host/all' }], callback) }, detail(id, callback) { if (!id) { httpClient.get('/host/all', {}, hosts => { callback(null, hosts) }) return } httpClient.batchGet([{ uri: `/task/${id}` }, { uri: '/host/all' }], callback) }, update(data, callback) { httpClient.post('/task/store', data, callback) }, remove(id, callback) { httpClient.post(`/task/remove/${id}`, {}, callback) }, enable(id, callback) { httpClient.post(`/task/enable/${id}`, {}, callback) }, disable(id, callback) { httpClient.post(`/task/disable/${id}`, {}, callback) }, run(id, callback) { httpClient.get(`/task/run/${id}`, { _t: Date.now() }, callback) }, allTags(callback) { httpClient.get('/task/tags', {}, callback) }, batchEnable(ids, callback) { httpClient.postJson('/task/batch-enable', { ids }, callback) }, batchDisable(ids, callback) { httpClient.postJson('/task/batch-disable', { ids }, callback) }, batchRemove(ids, callback) { httpClient.postJson('/task/batch-remove', { ids }, callback) }, /** * 预览 cron 表达式：接下来 N 次执行时间 + 未来 7 天分布热图 * @param {{spec: string, timezone?: string, count?: number}} params */ cronPreview(params, callback) { httpClient.postJson('/task/cron-preview', params, callback) }, versions(taskId, params, callback) { httpClient.get(`/task/versions/${taskId}`, params, callback) }, versionDetail(taskId, versionId, callback) { httpClient.get(`/task/versions/${taskId}/${versionId}`, {}, callback) }, versionRollback(taskId, versionId, callback) { httpClient.post(`/task/versions/${taskId}/${versionId}/rollback`, {}, callback) } } ================================================ FILE: web/vue/src/api/taskLog.js ================================================ import httpClient from '../utils/httpClient' export default { list (query, callback) { httpClient.get('/task/log', query, callback) }, clear (callback) { httpClient.post('/task/log/clear', {}, callback) }, stop (id, taskId, callback) { httpClient.post('/task/log/stop', {id, task_id: taskId}, callback) }, clearByTaskId (taskId, callback) { httpClient.post(`/task/log/clear/${taskId}`, {}, callback) } } ================================================ FILE: web/vue/src/api/template.js ================================================ import httpClient from '../utils/httpClient' export default { list(query, callback) { httpClient.get('/template', query, callback) }, categories(callback) { httpClient.get('/template/categories', {}, callback) }, detail(id, callback) { httpClient.get(`/template/${id}`, {}, callback) }, store(data, callback) { httpClient.post('/template/store', data, callback) }, remove(id, callback) { httpClient.post(`/template/remove/${id}`, {}, callback) }, apply(id, callback) { httpClient.post(`/template/apply/${id}`, {}, callback) }, saveFromTask(data, callback) { httpClient.post('/template/save-from-task', data, callback) }, } ================================================ FILE: web/vue/src/api/user.js ================================================ import httpClient from '../utils/httpClient' export default { list (query, callback) { httpClient.get('/user', {}, callback) }, detail (id, callback) { httpClient.get(`/user/${id}`, {}, callback) }, update (data, callback) { httpClient.post('/user/store', data, callback) }, login (username, password, twoFactorCode, callback, errorCallback) { const data = { username, password } if (twoFactorCode) { data.two_factor_code = twoFactorCode } httpClient.post('/user/login', data, callback, errorCallback) }, enable (id, callback) { httpClient.post(`/user/enable/${id}`, {}, callback) }, disable (id, callback) { httpClient.post(`/user/disable/${id}`, {}, callback) }, remove (id, callback) { httpClient.post(`/user/remove/${id}`, {}, callback) }, editPassword (data, callback) { httpClient.post(`/user/editPassword/${data.id}`, { 'new_password': data.new_password, 'confirm_new_password': data.confirm_new_password }, callback) }, editMyPassword (data, callback) { httpClient.post(`/user/editMyPassword`, data, callback) }, get2FAStatus (callback) { httpClient.get('/user/2fa/status', {}, callback) }, setup2FA (callback) { httpClient.get('/user/2fa/setup', {}, callback) }, enable2FA (secret, code, callback) { httpClient.post('/user/2fa/enable', { secret, code }, callback) }, disable2FA (code, callback, errorCallback) { httpClient.post('/user/2fa/disable', { code }, callback, errorCallback) } } ================================================ FILE: web/vue/src/components/common/CronInput.vue ================================================ ================================================ FILE: web/vue/src/components/common/CronPreview.vue ================================================ ================================================ FILE: web/vue/src/components/common/HeatmapSvg.vue ================================================ ================================================ FILE: web/vue/src/components/common/LanguageSwitcher.vue ================================================ ================================================ FILE: web/vue/src/components/common/MonacoEditor.vue ================================================ ================================================ FILE: web/vue/src/components/common/footer.vue ================================================ ================================================ FILE: web/vue/src/components/common/header.vue ================================================ ================================================ FILE: web/vue/src/components/common/navMenu.vue ================================================ ================================================ FILE: web/vue/src/components/common/notFound.vue ================================================ ================================================ FILE: web/vue/src/components/common/sidebar.vue ================================================ ================================================ FILE: web/vue/src/composables/__tests__/useDebounce.spec.js ================================================ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest' import { useDebounceFn } from '../useDebounce' describe('useDebounce', () => { beforeEach(() => { vi.useFakeTimers() }) afterEach(() => { vi.restoreAllMocks() }) it('should debounce function calls', () => { const mockFn = vi.fn() const debouncedFn = useDebounceFn(mockFn, 300) // 快速调用多次 debouncedFn('call 1') debouncedFn('call 2') debouncedFn('call 3') // 还没到延迟时间，不应该被调用 expect(mockFn).not.toHaveBeenCalled() // 快进时间 vi.advanceTimersByTime(300) // 应该只被调用一次，使用最后一次的参数 expect(mockFn).toHaveBeenCalledTimes(1) expect(mockFn).toHaveBeenCalledWith('call 3') }) it('should use custom delay', () => { const mockFn = vi.fn() const debouncedFn = useDebounceFn(mockFn, 500) debouncedFn('test') vi.advanceTimersByTime(300) expect(mockFn).not.toHaveBeenCalled() vi.advanceTimersByTime(200) expect(mockFn).toHaveBeenCalledWith('test') }) }) ================================================ FILE: web/vue/src/composables/__tests__/useLoading.spec.js ================================================ import { describe, it, expect } from 'vitest' import { useLoading } from '../useLoading' describe('useLoading', () => { it('should initialize with false', () => { const { loading } = useLoading() expect(loading.value).toBe(false) }) it('should set loading during async operation', async () => { const { loading, withLoading } = useLoading() const promise = withLoading(async () => { expect(loading.value).toBe(true) return 'result' }) const result = await promise expect(loading.value).toBe(false) expect(result).toBe('result') }) }) ================================================ FILE: web/vue/src/composables/__tests__/useMessage.spec.js ================================================ import { describe, it, expect, vi } from 'vitest' import { useMessage } from '../useMessage' import { ElMessage, ElMessageBox } from 'element-plus' vi.mock('element-plus', () => ({ ElMessage: { success: vi.fn(), error: vi.fn(), warning: vi.fn(), info: vi.fn() }, ElMessageBox: { confirm: vi.fn() } })) vi.mock('vue-i18n', () => ({ useI18n: () => ({ t: (key) => key }) })) describe('useMessage', () => { it('should call ElMessage.success', () => { const { success } = useMessage() success('test message') expect(ElMessage.success).toHaveBeenCalledWith('test message') }) it('should call ElMessage.error', () => { const { error } = useMessage() error('error message') expect(ElMessage.error).toHaveBeenCalledWith('error message') }) it('should call ElMessageBox.confirm with default options', () => { const { confirm } = useMessage() confirm('Are you sure?') expect(ElMessageBox.confirm).toHaveBeenCalledWith( 'Are you sure?', 'common.tip', expect.objectContaining({ confirmButtonText: 'common.confirm', cancelButtonText: 'common.cancel', type: 'warning', center: true }) ) }) }) ================================================ FILE: web/vue/src/composables/useDebounce.js ================================================ import { ref, watch, onUnmounted } from 'vue' export function useDebounce(value, delay = 300) { const debouncedValue = ref(value.value) let timeout = null watch(value, (newValue) => { if (timeout) clearTimeout(timeout) timeout = setTimeout(() => { debouncedValue.value = newValue }, delay) }) // 组件卸载时清理 onUnmounted(() => { if (timeout) clearTimeout(timeout) }) return debouncedValue } export function useDebounceFn(fn, delay = 300) { let timeout = null return (...args) => { if (timeout) clearTimeout(timeout) timeout = setTimeout(() => { fn(...args) }, delay) } } ================================================ FILE: web/vue/src/composables/useLoading.js ================================================ import { ref } from 'vue' import { ElLoading } from 'element-plus' export function useLoading(initialState = false) { const loading = ref(initialState) const withLoading = async (fn) => { loading.value = true try { return await fn() } finally { loading.value = false } } return { loading, withLoading } } // 全屏 loading export function useFullScreenLoading() { let loadingInstance = null const show = (text = '加载中...') => { loadingInstance = ElLoading.service({ lock: true, text, background: 'rgba(0, 0, 0, 0.7)' }) } const hide = () => { loadingInstance?.close() } const withLoading = async (fn, text) => { show(text) try { return await fn() } finally { hide() } } return { show, hide, withLoading } } ================================================ FILE: web/vue/src/composables/useMessage.js ================================================ import { ElMessage, ElMessageBox } from 'element-plus' import { useI18n } from 'vue-i18n' export function useMessage() { const { t } = useI18n() const success = (message) => { ElMessage.success(message) } const error = (message) => { ElMessage.error(message) } const warning = (message) => { ElMessage.warning(message) } const info = (message) => { ElMessage.info(message) } const confirm = (message, title, options = {}) => { return ElMessageBox.confirm( message, title || t('common.tip'), { confirmButtonText: t('common.confirm'), cancelButtonText: t('common.cancel'), type: 'warning', center: true, ...options } ) } return { success, error, warning, info, confirm } } ================================================ FILE: web/vue/src/const/index.js ================================================ export * from './lang' ================================================ FILE: web/vue/src/const/lang.js ================================================ export const availableLanguages = { zhCN: { value: 'zh-CN', label: '简体中文' }, enUS: { value: 'en-US', label: 'English' } } ================================================ FILE: web/vue/src/locales/en-US.js ================================================ export default { select: 'Please select', cronValidator: { required: 'Please enter a cron expression', everyFormatError: 'Invalid @every format, e.g.: @every 30s, @every 1m20s, @every 3h5m10s', shortcutError: 'Invalid shortcut, click "Examples" to see valid ones', sixFieldsRequired: 'Cron expression must have 6 fields (second minute hour day month weekday)', fieldSecond: 'second', fieldMinute: 'minute', fieldHour: 'hour', fieldDay: 'day', fieldMonth: 'month', fieldWeek: 'weekday', illegalChar: '{field} field contains illegal characters', valueOutOfRange: '{field} field value {value} is out of range [{min}-{max}]', formatError: '{field} field format error', rangeFormatError: '{field} field range format error', rangeNotNumber: '{field} field range must be numeric', rangeInvalid: '{field} field range [{start}-{end}] is invalid', stepFormatError: '{field} field step format error', stepNotPositive: '{field} field step must be a positive integer' }, cronPreview: { waitingInput: 'Enter a cron expression to preview upcoming executions', computing: 'Computing...', nextRuns: 'Next {count} executions', weeklyDistribution: 'Next 7-day distribution', truncated: 'Truncated (high frequency)', noUpcomingRuns: 'No executions in next 7 days', noRuns: 'No runs', runs: 'run(s)', heatmapEmpty: 'No executions in next 7 days', heatmapAria: 'Weekly execution distribution heatmap', requestFailed: 'Preview request failed', invalidSyntax: 'Invalid cron expression', inSeconds: 'in {n}s', inMinutes: 'in {n}m', inHours: 'in {n}h', inHoursMinutes: 'in {h}h {m}m', inDays: 'in {n}d', inDaysHours: 'in {d}d {h}h', sun: 'Sun', mon: 'Mon', tue: 'Tue', wed: 'Wed', thu: 'Thu', fri: 'Fri', sat: 'Sat', sunAbbr: 'Sun', monAbbr: 'Mon', tueAbbr: 'Tue', wedAbbr: 'Wed', thuAbbr: 'Thu', friAbbr: 'Fri', satAbbr: 'Sat' }, common: { confirm: 'Confirm', cancel: 'Cancel', save: 'Save', delete: 'Delete', edit: 'Edit', search: 'Search', reset: 'Reset', add: 'Add', refresh: 'Refresh', tip: 'Tip', confirmOperation: 'Are you sure to perform this operation?', operation: 'Operation', status: 'Status', enabled: 'Enabled', disabled: 'Disabled', yes: 'Yes', no: 'No', total: 'Total', items: 'items', date: 'Date' }, nav: { taskManage: 'Tasks', taskNode: 'Nodes', userManage: 'Users', systemManage: 'System', statistics: 'Statistics', logout: 'Logout', changePassword: 'Change Password', twoFactor: 'Two-Factor Authentication' }, login: { title: 'User Login', username: 'Username', password: 'Password', verifyCode: '2FA Code', login: 'Login', usernamePlaceholder: 'Please enter username or email', passwordPlaceholder: 'Please enter password', verifyCodePlaceholder: 'Please enter 6-digit code', usernameRequired: 'Please enter username', passwordRequired: 'Please enter password', verifyCodeRequired: 'Please enter 2FA code' }, task: { list: 'Task List', log: 'Task Log', id: 'Task ID', name: 'Task Name', tag: 'Tag', tagPlaceholder: 'Select or enter tags', type: 'Task Type', mainTask: 'Main Task', childTask: 'Child Task', dependency: 'Dependency', strongDependency: 'Strong Dependency', weakDependency: 'Weak Dependency', childTaskId: 'Child Task ID', childTaskIdPlaceholder: 'Multiple IDs separated by comma', cronExpression: 'Cron Expression', cronPlaceholder: 'Second Minute Hour Day Month Week', cronExample: 'Examples', timezone: 'Timezone', timezoneServer: 'Server Timezone', protocol: 'Method', httpMethod: 'HTTP Method', httpBody: 'Request Body', httpBodyPlaceholder: 'JSON body for POST, e.g. {\'{\'}"key": "value"{\'}\'}', httpHeaders: 'Custom Headers', httpHeadersPlaceholder: 'JSON format, e.g. {\'{\'}"Authorization": "Bearer token"{\'}\'}', successPattern: 'Response Assertion', successPatternPlaceholder: 'Regex to match response body, leave empty to skip', taskNode: 'Task Node', taskNodePlaceholder: 'Please select task node', command: 'Command', timeout: 'Task Timeout', singleInstance: 'Single Instance', retryTimes: 'Retry Times on Failure', retryTimesPlaceholder: '0 - 10, default 0, no retry', retryInterval: 'Retry Interval on Failure', retryIntervalPlaceholder: '0 - 3600 (seconds), default 0, use system default', notification: 'Task Notification', notifyType: 'Notification Type', notifyReceiver: 'Receiver', notifyReceiverPlaceholder: 'Please select', notifyChannel: 'Channel', notifyKeyword: 'Task Output Keyword', notifyKeywordPlaceholder: 'Notification will be triggered if task output contains this keyword', logRetentionDays: 'Log Retention Days', logRetentionDaysTip: '0 = use global setting', clearTaskLog: 'Clear Task Logs', confirmClearTaskLog: 'Confirm clearing all logs for task ID {taskId}?', remark: 'Remark', status: 'Status', nextRunTime: 'Next Run', operation: 'Operation', manualRun: 'Manual Run', viewLog: 'View Log', enable: 'Enable', disable: 'Disable', mainTaskTip: 'Main task can configure multiple child tasks. Child tasks will be executed automatically after main task completes.
Task type cannot be changed after creation.', dependencyTip: 'Strong Dependency: Child tasks run only when main task succeeds
Weak Dependency: Child tasks run regardless of main task result', timeoutTip: 'Force terminate task on timeout, range 0-86400 (seconds), default 3600, 0 means no limit', singleInstanceTip: 'Single instance mode: whether to execute next scheduled task if previous task is still running', cronStandard: 'Standard Syntax (Second Minute Hour Day Month Week)', cronShortcut: 'Shortcut Syntax', notifyDisabled: 'Disabled', notifyOnFailure: 'On Failure', notifyAlways: 'Always', notifyKeywordMatch: 'Keyword Match', notifyEmail: 'Email', notifySlack: 'Slack', notifyWebhook: 'WebHook', createNew: 'Create Task', versionHistory: 'Version History', version: 'Version', versionRemark: 'Change Note', versionUser: 'Modified By', versionTime: 'Modified Time', versionRollback: 'Rollback', versionRollbackConfirm: 'Are you sure you want to rollback to version {version}?', versionRollbackSuccess: 'Rollback successful', versionCommand: 'Command Content' }, host: { list: 'Task Nodes', name: 'Host Name', alias: 'Alias', port: 'Port', remark: 'Remark', createTime: 'Create Time', createNew: 'Add Node', namePlaceholder: 'Please enter host name', aliasPlaceholder: 'Please enter alias', portPlaceholder: 'Please enter port', nameRequired: 'Please enter host name', portRequired: 'Please enter port', aliasRequired: 'Please enter node name', portInvalid: 'Invalid port', autoRegister: 'Auto Register', agentInstall: 'Agent Installation', installCommand: 'Install Command', installTip: 'Run the corresponding command on the target server to automatically install and register the Agent node. Note: Must be executed by a non-root user', tokenExpires: 'Token Expires', tokenUsage: 'Usage', tokenReusable: 'This token can be reused within the validity period for batch installation', bashCommand: 'Run the following command in terminal (non-root user):', powershellCommand: 'Run in PowerShell (Administrator):', windowsManualInstall: 'Windows Manual Installation', windowsManualInstallTip: 'For security reasons, manual installation of gocron-node is recommended for Windows systems', windowsStep1: 'Download Package', windowsStep1Desc: 'Download gocron-node-windows-amd64.zip from GitHub Releases', windowsStep2: 'Extract and Configure', windowsStep2Desc: 'Extract to target directory and manually add node configuration in Web UI', windowsStep3: 'Start Service', windowsStep3Desc: 'Run gocron-node.exe or create a Windows service' }, user: { list: 'User Management', username: 'Username', email: 'Email', role: 'Role', admin: 'Administrator', normalUser: 'Normal User', password: 'Password', confirmPassword: 'Confirm Password', oldPassword: 'Old Password', newPassword: 'New Password', confirmNewPassword: 'Confirm New Password', createNew: 'Add User', changePassword: 'Change Password', usernamePlaceholder: 'Please enter username', emailPlaceholder: 'Please enter email', passwordPlaceholder: 'At least 8 characters, letters and digits', usernameRequired: 'Please enter username', emailRequired: 'Please enter valid email address', passwordRequired: 'Please enter password', confirmPasswordRequired: 'Please enter password again', oldPasswordRequired: 'Please enter old password', newPasswordRequired: 'Please enter new password' }, system: { manage: 'System Management', loginLog: 'Login Log', logRetention: 'Log Retention', notification: 'Notifications', email: 'Email Notification', slack: 'Slack Notification', webhook: 'WebHook Notification', loginTime: 'Login Time', loginIp: 'Login IP', retentionDays: 'Retention Days', retentionDaysPlaceholder: 'Please enter retention days', smtpHost: 'SMTP Host', smtpPort: 'SMTP Port', smtpUser: 'SMTP User', smtpPassword: 'SMTP Password', mailFrom: 'Mail From', slackUrl: 'Slack Webhook URL', webhookUrl: 'Webhook URL', testSend: 'Test Send', logRetentionSettings: 'Log Auto-Cleanup Settings', dbLogRetentionDays: 'Database Log Retention Days', dbLogRetentionTip: 'Set to 0 to disable automatic database log cleanup', cleanupTime: 'Cleanup Time', cleanupTimeTip: 'Automatically execute log cleanup at this time every day, takes effect immediately after modification', selectTime: 'Select Time', logFileSizeLimit: 'Log File Size Limit', logFileSizeLimitTip: 'Set to 0 to disable log file cleanup, greater than 0 will automatically clear when log file exceeds this size', logRetentionSaveSuccess: 'Saved successfully, cleanup task has been reloaded', emailServerConfig: 'Email Server Configuration', templateSupportsHtml: 'Notification template supports HTML', template: 'Template', addUser: 'Add User', notificationUsers: 'Notification Users', emailAddress: 'Email Address', pleaseEnterEmailServer: 'Please enter email server address', pleaseEnterValidPort: 'Please enter valid port', pleaseEnterUserEmail: 'Please enter user email', pleaseEnterTemplate: 'Please enter notification template content', incompleteParameters: 'Incomplete parameters', channel: 'Channel', channels: 'Channels', addChannel: 'Add Channel', channelName: 'Channel Name', pleaseEnterChannelName: 'Please enter channel name', pleaseEnterValidUrl: 'Please enter valid notification URL', webhookTip: 'POST request, set Header[Content-Type: application/json]', addWebhookUrl: 'Add Webhook URL', webhookUrls: 'Webhook URLs', webhookName: 'Webhook Name', logCleanup: 'Log Cleanup', templateVariables: 'Template Variables', taskIdVar: 'Task ID', taskNameVar: 'Task Name', statusVar: 'Task Execution Result Status', resultVar: 'Task Execution Output', emailTemplatePlaceholder: 'Task ID: {{.TaskId}}\nTask Name: {{.TaskName}}\nStatus: {{.Status}}\nResult: {{.Result}}\nRemark: {{.Remark}}', slackTemplatePlaceholder: 'Task ID: {{.TaskId}}\nTask Name: {{.TaskName}}\nStatus: {{.Status}}\nResult: {{.Result}}\nRemark: {{.Remark}}', webhookTemplatePlaceholder: '{"task_id": "{{.TaskId}}", "task_name": "{{.TaskName}}", "status": "{{.Status}}", "result": "{{.Result}}", "remark": "{{.Remark}}"}' }, taskLog: { list: 'Task Log', taskName: 'Task Name', startTime: 'Start Time', endTime: 'End Time', duration: 'Duration', result: 'Result', host: 'Host', output: 'Output', success: 'Success', failed: 'Failed', viewOutput: 'View Output' }, twoFactor: { title: 'Two-Factor Authentication (2FA)', status: 'Status', enabled: 'Enabled', disabled: 'Disabled', enable: 'Enable 2FA', disable: 'Disable 2FA', setup: 'Enable Two-Factor Authentication', qrCode: 'QR Code', secret: 'Secret Key', scanQR: '1. Scan the QR code below with your authenticator app:', manualEntry: '2. Or manually enter the secret key:', verifyCode: 'Verification Code', verifyCodePlaceholder: 'Please enter 6-digit code', verifyCodeStep: '3. Enter the 6-digit code displayed in your app:', confirm: 'Confirm', confirmDisable: 'Confirm Disable', confirmDisableMsg: 'Are you sure you want to disable two-factor authentication?', enableSuccess: '2FA enabled', disableSuccess: '2FA disabled', verifyFailed: 'Verification code is incorrect', alertTitle: 'Notice', alertDescription: 'Enabling two-factor authentication greatly enhances account security. It is recommended for all users, especially administrators.', enabledAlertTitle: '2FA Enabled', enabledAlertDescription: 'Your account is protected by two-factor authentication.', disableDialogTitle: 'Disable Two-Factor Authentication', disableDialogDescription: 'Please enter the 6-digit code displayed in your authenticator app to disable 2FA:', copySecret: 'Copy', secretCopied: 'Secret key copied to clipboard', verifyCodeLength: 'Please enter 6-digit code', disableFailed: 'Failed to disable 2FA' }, install: { title: 'System Installation', welcome: 'Welcome to Gocron', dbConfig: 'Database Configuration', dbType: 'Database Type', dbHost: 'Host', dbPort: 'Port', dbName: 'Database Name', dbFilePath: 'Database File Path', dbUser: 'Username', dbPassword: 'Password', dbTablePrefix: 'Table Prefix', adminConfig: 'Administrator Account', adminUsername: 'Username', adminPassword: 'Password', confirmPassword: 'Confirm Password', adminEmail: 'Email', install: 'Install', installing: 'Installing...', installSuccess: 'Installation Successful', installFailed: 'Installation Failed', dbNamePlaceholder: 'Create DB first if needed', dbFilePathPlaceholder: './data/gocron.db', passwordPlaceholder: '8+ chars with letters & digits', selectDb: 'Please select database', enterDbName: 'Please enter database name', enterDbHost: 'Please enter host', enterDbPort: 'Please enter port', enterDbUser: 'Please enter username', enterDbPassword: 'Please enter password', enterAdminUsername: 'Please enter username', enterAdminEmail: 'Please enter email', enterAdminPassword: 'Please enter password', confirmAdminPassword: 'Please confirm password', passwordMinLength: 'At least 8 characters' }, message: { saveSuccess: 'Saved successfully', saveFailed: 'Save failed', deleteSuccess: 'Deleted successfully', deleteFailed: 'Delete failed', updateSuccess: 'Updated successfully', updateFailed: 'Update failed', operationSuccess: 'Operation successful', operationFailed: 'Operation failed', refreshSuccess: 'Refreshed successfully', loadFailed: 'Load failed', requestTimeout: 'Request timeout, please try again later', authExpired: 'Login expired, please login again', requestFailed: 'Request failed', networkError: 'Network error', serverError: 'Server error', dataNotFound: 'Data not found', confirmDelete: 'Are you sure you want to delete?', confirmDeleteTask: 'Are you sure you want to delete task "{name}"?', confirmRunTask: 'Are you sure you want to manually run task "{name}"?', taskStarted: 'Task has started executing', selectTaskNode: 'Please select task node', selectMailReceiver: 'Please select email receiver', selectSlackChannel: 'Please select Slack channel', selectWebhookUrl: 'Please select Webhook URL', passwordMismatch: 'Passwords do not match', oldPasswordError: 'Old password is incorrect', passwordSameAsOld: 'New password cannot be the same as old password', usernameExists: 'Username already exists', emailExists: 'Email already exists', formValidationFailed: 'Form validation failed, please check your input', pleaseEnterPassword: 'Please enter password', pleaseEnterPasswordAgain: 'Please enter password again', pleaseEnterTaskName: 'Please enter task name', pleaseEnterCronExpression: 'Please enter crontab expression', pleaseEnterCommand: 'Please enter command', pleaseEnterValidTimeout: 'Please enter valid task timeout', pleaseEnterValidRetryTimes: 'Please enter valid retry times', pleaseEnterValidRetryInterval: 'Please enter valid retry interval', pleaseEnterNotifyKeyword: 'Please enter notification keyword', pleaseEnterUrl: 'Please enter URL', pleaseEnterShellCommand: 'Please enter shell command', selected: 'Selected', tasks: 'tasks', batchEnable: 'Batch Enable', batchDisable: 'Batch Disable', batchDelete: 'Batch Delete', confirmBatchEnable: 'Are you sure you want to enable {count} selected tasks?', confirmBatchDisable: 'Are you sure you want to disable {count} selected tasks?', confirmBatchDelete: 'Are you sure you want to delete {count} selected tasks? This operation cannot be undone!', batchEnableSuccess: 'Batch enable successful', batchDisableSuccess: 'Batch disable successful', batchDeleteSuccess: 'Batch delete successful', pleaseSelectTask: 'Please select tasks to {action}', manualRunTask: 'Manual Run Task', confirmExecute: 'Confirm Execute', confirmDeleteTitle: 'Tip', confirmDeleteButton: 'Confirm Delete', taskCreatedTime: 'Task Created Time', taskType: 'Task Type', singleInstanceRun: 'Single Instance', timeoutTime: 'Timeout', retryCount: 'Retry Times', retryIntervalTime: 'Retry Interval', taskNodeLabel: 'Task Node', commandLabel: 'Command', remarkLabel: 'Remark', noLimit: 'No Limit', systemDefault: 'System Default', seconds: 'seconds', activated: 'Activated', stopped: 'Stopped', confirmDeleteNode: 'Are you sure you want to delete this node?', confirmDeleteUser: 'Are you sure you want to delete this user?', connectionSuccess: 'Connection successful', copySuccess: 'Copied successfully', copyFailed: 'Copy failed', all: 'All', clearLog: 'Clear Log', confirmClearLog: 'Are you sure you want to clear all logs?', running: 'Running', cancelled: 'Cancelled', stopTask: 'Stop Task', taskExecutionResult: 'Task Execution Result', cronExamples: 'Cron Expression Examples', everyMinute: 'Run at 0 seconds of every minute', every20Seconds: 'Run every 20 seconds', everyDay21_30: 'Run at 21:30:00 every day', everySaturday23: 'Run at 23:00:00 every Saturday', reboot: 'Run only once at application startup', yearly: 'Run once a year', monthly: 'Run once a month', weekly: 'Run once a week', daily: 'Run once a day', hourly: 'Run once an hour', every30s: 'Run every 30 seconds', every1m20s: 'Run every 1 minute and 20 seconds' }, template: { list: 'Templates', name: 'Template Name', description: 'Description', category: 'Category', protocol: 'Method', command: 'Command', timeout: 'Timeout', usageCount: 'Usage', builtin: 'Built-in', custom: 'Custom', createNew: 'Create Template', useTemplate: 'Use Template', saveAsTemplate: 'Save as Template', templateVarTip: 'Use {{variable_name}} syntax for template variables. Users will fill values when applying.', applyTemplate: 'Apply Template', applySuccess: 'Template applied', fillVariables: 'Fill Template Variables', variableName: 'Variable', variableValue: 'Value', noTemplates: 'No templates', confirmDelete: 'Are you sure you want to delete template "{name}"?', category_all: 'All', category_backup: 'Backup', category_cleanup: 'Cleanup', category_monitor: 'Monitor', category_deploy: 'Deploy', category_api: 'API Call', category_custom: 'Custom', preview: 'Preview', templateNamePlaceholder: 'Enter template name', templateDescPlaceholder: 'Enter template description', selectCategory: 'Select category', saveAsTemplateName: 'Template Name', saveAsTemplateDesc: 'Description', saveAsTemplateCategory: 'Category', securityWarning: 'Variable values will be stored in plaintext in the task command. Avoid entering passwords directly. Use environment variable references (e.g. $DB_PASS) instead.', saveAsTemplateWarning: 'The command will be saved as-is into the template (visible to all users). Remove any passwords or secrets first and replace them with {{variable}} placeholders.' }, audit: { log: 'Audit Log', module: 'Module', action: 'Action', target: 'Target', detail: 'Detail', module_task: 'Task', module_host: 'Host', module_user: 'User', module_system: 'System', action_create: 'Create', action_update: 'Update', action_delete: 'Delete', action_enable: 'Enable', action_disable: 'Disable', action_run: 'Manual Run', action_batch_enable: 'Batch Enable', action_batch_disable: 'Batch Disable', action_batch_remove: 'Batch Delete', action_change_password: 'Change Password', action_reset_password: 'Reset Password' }, statistics: { title: 'Statistics', totalTasks: 'Total Tasks', todayExecutions: 'Today Executions', last7DaysExecutions: '7-Day Executions', successRate: '7-Day Success Rate', failedCount: '7-Day Failed Tasks', last7DaysTrend: 'Last 7 Days Trend', success: 'Success', failed: 'Failed', total: 'Total', executionCount: 'Execution Count', date: 'Date', detailedData: 'Detailed Data' } } ================================================ FILE: web/vue/src/locales/index.js ================================================ import { createI18n } from 'vue-i18n' import zhCN from './zh-CN' import enUS from './en-US' import { availableLanguages } from '@/const/index' const getDefaultLocale = () => { const savedLocale = localStorage.getItem('locale') return savedLocale || availableLanguages.zhCN.value } const i18n = createI18n({ legacy: false, locale: getDefaultLocale(), fallbackLocale: availableLanguages.zhCN.value, messages: { [availableLanguages.zhCN.value]: zhCN, [availableLanguages.enUS.value]: enUS } }) export default i18n ================================================ FILE: web/vue/src/locales/zh-CN.js ================================================ export default { select: '请选择', cronValidator: { required: '请输入cron表达式', everyFormatError: '@every 格式错误，示例：@every 30s, @every 1m20s, @every 3h5m10s', shortcutError: '快捷语法错误，请点击"示例"查看', sixFieldsRequired: 'cron表达式需包含6段（秒 分 时 天 月 周）', fieldSecond: '秒', fieldMinute: '分', fieldHour: '时', fieldDay: '天', fieldMonth: '月', fieldWeek: '周', illegalChar: '{field}字段包含非法字符', valueOutOfRange: '{field}字段值{value}超出范围[{min}-{max}]', formatError: '{field}字段格式错误', rangeFormatError: '{field}字段范围格式错误', rangeNotNumber: '{field}字段范围必须是数字', rangeInvalid: '{field}字段范围[{start}-{end}]无效', stepFormatError: '{field}字段步长格式错误', stepNotPositive: '{field}字段步长必须是正整数' }, cronPreview: { waitingInput: '输入 cron 表达式后在此实时查看接下来的执行时间', computing: '计算中...', nextRuns: '接下来 {count} 次执行', weeklyDistribution: '未来 7 天执行分布', truncated: '已截断 (高频表达式)', noUpcomingRuns: '未来 7 天无触发', noRuns: '无执行', runs: '次', heatmapEmpty: '未来 7 天无触发', heatmapAria: 'cron 一周执行分布热图', requestFailed: '预览请求失败', invalidSyntax: 'cron 表达式格式错误', inSeconds: '{n} 秒后', inMinutes: '{n} 分后', inHours: '{n} 小时后', inHoursMinutes: '{h} 小时 {m} 分后', inDays: '{n} 天后', inDaysHours: '{d} 天 {h} 小时后', sun: '周日', mon: '周一', tue: '周二', wed: '周三', thu: '周四', fri: '周五', sat: '周六', sunAbbr: '日', monAbbr: '一', tueAbbr: '二', wedAbbr: '三', thuAbbr: '四', friAbbr: '五', satAbbr: '六' }, common: { confirm: '确定', cancel: '取消', save: '保存', delete: '删除', edit: '编辑', search: '搜索', reset: '重置', add: '新增', refresh: '刷新', tip: '提示', confirmOperation: '确定执行此操作?', operation: '操作', status: '状态', enabled: '启用', disabled: '禁用', yes: '是', no: '否', total: '共', items: '条', date: '日期' }, nav: { taskManage: '任务管理', taskNode: '任务节点', userManage: '用户管理', systemManage: '系统管理', statistics: '数据统计', logout: '退出', changePassword: '修改密码', twoFactor: '双因素认证' }, login: { title: '用户登录', username: '用户名', password: '密码', verifyCode: '验证码', login: '登录', usernamePlaceholder: '请输入用户名或邮箱', passwordPlaceholder: '请输入密码', verifyCodePlaceholder: '请输入6位验证码', usernameRequired: '请输入用户名', passwordRequired: '请输入密码', verifyCodeRequired: '请输入验证码' }, task: { list: '定时任务', log: '任务日志', id: '任务ID', name: '任务名称', tag: '标签', tagPlaceholder: '选择或输入标签', type: '任务类型', mainTask: '主任务', childTask: '子任务', dependency: '依赖关系', strongDependency: '强依赖', weakDependency: '弱依赖', childTaskId: '子任务ID', childTaskIdPlaceholder: '多个ID逗号分隔', cronExpression: 'crontab表达式', cronPlaceholder: '秒 分 时 天 月 周', cronExample: '示例', timezone: '时区', timezoneServer: '服务器时区', protocol: '执行方式', httpMethod: '请求方法', httpBody: '请求 Body', httpBodyPlaceholder: 'POST 请求的 JSON Body，例如 {\'{\'}"key": "value"{\'}\'}', httpHeaders: '自定义 Header', httpHeadersPlaceholder: 'JSON 格式，例如 {\'{\'}"Authorization": "Bearer token"{\'}\'}', successPattern: '响应断言', successPatternPlaceholder: '正则表达式匹配响应内容，为空则不校验', taskNode: '任务节点', taskNodePlaceholder: '请选择任务节点', command: '命令', timeout: '任务超时时间', singleInstance: '单实例运行', retryTimes: '任务失败重试次数', retryTimesPlaceholder: '0 - 10, 默认0，不重试', retryInterval: '任务失败重试间隔时间', retryIntervalPlaceholder: '0 - 3600 (秒), 默认0，执行系统默认策略', notification: '任务通知', notifyType: '通知类型', notifyReceiver: '接收用户', notifyReceiverPlaceholder: '请选择', notifyChannel: '发送Channel', notifyKeyword: '任务执行输出关键字', notifyKeywordPlaceholder: '任务执行输出中包含此关键字将触发通知', logRetentionDays: '日志保留天数', logRetentionDaysTip: '0 = 使用全局设置', clearTaskLog: '清空该任务日志', confirmClearTaskLog: '确认清空任务ID为 {taskId} 的所有日志？', remark: '备注', status: '状态', nextRunTime: '下次执行时间', operation: '操作', manualRun: '手动执行', viewLog: '查看日志', enable: '启用', disable: '禁用', mainTaskTip: '主任务可以配置多个子任务, 当主任务执行完成后，自动执行子任务
任务类型新增后不能变更', dependencyTip: '强依赖: 主任务执行成功，才会运行子任务
弱依赖: 无论主任务执行是否成功，都会运行子任务', timeoutTip: '任务执行超时强制结束, 取值0-86400(秒), 默认3600, 0表示不限制', singleInstanceTip: '单实例运行, 前次任务未执行完成，下次任务调度时间到了是否要执行, 即是否允许多进程执行同一任务', cronStandard: '标准语法（秒 分 时 天 月 周）', cronShortcut: '快捷语法', notifyDisabled: '不通知', notifyOnFailure: '失败通知', notifyAlways: '总是通知', notifyKeywordMatch: '关键字匹配通知', notifyEmail: '邮件', notifySlack: 'Slack', notifyWebhook: 'WebHook', createNew: '新增任务', versionHistory: '版本历史', version: '版本', versionRemark: '变更说明', versionUser: '修改人', versionTime: '修改时间', versionRollback: '回滚', versionRollbackConfirm: '确定要回滚到版本 {version} 吗？', versionRollbackSuccess: '回滚成功', versionCommand: '命令内容' }, host: { list: '任务节点', name: '主机名', alias: '别名', port: '端口', remark: '备注', createTime: '创建时间', createNew: '新增节点', namePlaceholder: '请输入主机名', aliasPlaceholder: '请输入别名', portPlaceholder: '请输入端口', nameRequired: '请输入主机名', portRequired: '请输入端口', aliasRequired: '请输入节点名称', portInvalid: '端口无效', autoRegister: '自动注册', agentInstall: 'Agent安装', installCommand: '安装命令', installTip: '在目标服务器上执行对应的命令，将自动安装并注册Agent节点。注意：必须使用非root用户执行安装脚本', tokenExpires: 'Token有效期', tokenUsage: '使用说明', tokenReusable: '此Token可在有效期内重复使用，适用于批量安装', bashCommand: '在终端（非root用户）执行以下命令：', powershellCommand: '在PowerShell（管理员权限）中执行：', windowsManualInstall: 'Windows 手动安装', windowsManualInstallTip: '出于安全考虑，Windows 系统建议手动安装 gocron-node', windowsStep1: '下载安装包', windowsStep1Desc: '从 GitHub Releases 下载对应版本的 gocron-node-windows-amd64.zip', windowsStep2: '解压并配置', windowsStep2Desc: '解压到目标目录，在 Web 界面手动添加节点配置', windowsStep3: '启动服务', windowsStep3Desc: '运行 gocron-node.exe 或创建 Windows 服务' }, user: { list: '用户管理', username: '用户名', email: '邮箱', role: '角色', admin: '管理员', normalUser: '普通用户', password: '密码', confirmPassword: '确认密码', oldPassword: '旧密码', newPassword: '新密码', confirmNewPassword: '确认新密码', createNew: '新增用户', changePassword: '修改密码', usernamePlaceholder: '请输入用户名', emailPlaceholder: '请输入邮箱', passwordPlaceholder: '至少8位，包含字母和数字', usernameRequired: '请输入用户名', emailRequired: '请输入有效邮箱地址', passwordRequired: '请输入密码', confirmPasswordRequired: '请再次输入密码', oldPasswordRequired: '请输入旧密码', newPasswordRequired: '请输入新密码' }, system: { manage: '系统管理', loginLog: '登录日志', logRetention: '日志保留', notification: '通知设置', email: '邮件通知', slack: 'Slack通知', webhook: 'WebHook通知', loginTime: '登录时间', loginIp: '登录IP', retentionDays: '保留天数', retentionDaysPlaceholder: '请输入保留天数', smtpHost: 'SMTP主机', smtpPort: 'SMTP端口', smtpUser: 'SMTP用户', smtpPassword: 'SMTP密码', mailFrom: '发件人', slackUrl: 'Slack Webhook URL', webhookUrl: 'Webhook URL', testSend: '测试发送', logRetentionSettings: '日志自动清理设置', dbLogRetentionDays: '数据库日志保留天数', dbLogRetentionTip: '设置为0表示不自动清理数据库日志', cleanupTime: '清理时间', cleanupTimeTip: '每天在此时间自动执行日志清理，修改后立即生效', selectTime: '选择时间', logFileSizeLimit: '日志文件大小限制', logFileSizeLimitTip: '设置为0表示不清理日志文件，大于0则当日志文件超过此大小时自动清空', logRetentionSaveSuccess: '保存成功，清理任务已重新加载', emailServerConfig: '邮件服务器配置', templateSupportsHtml: '通知模板支持html', template: '模板', addUser: '新增用户', notificationUsers: '通知用户', emailAddress: '邮箱地址', pleaseEnterEmailServer: '请输入邮件服务器地址', pleaseEnterValidPort: '请输入有效的端口', pleaseEnterUserEmail: '请输入用户email', pleaseEnterTemplate: '请输入通知模板内容', incompleteParameters: '参数不完整', channel: 'Channel', channels: 'Channels', addChannel: '新增Channel', channelName: 'Channel名称', pleaseEnterChannelName: '请输入Channel名称', pleaseEnterValidUrl: '请输入有效的通知URL', webhookTip: 'POST请求，设置Header[Content-Type: application/json]', addWebhookUrl: '新增Webhook地址', webhookUrls: 'Webhook地址列表', webhookName: 'Webhook名称', logCleanup: '日志清理', templateVariables: '通知模板支持的变量', taskIdVar: '任务ID', taskNameVar: '任务名称', statusVar: '任务执行结果状态', resultVar: '任务执行输出', emailTemplatePlaceholder: function () { return `${this.taskIdVar}: {{.TaskId}}\\n${this.taskNameVar}: {{.TaskName}}\\n${this.statusVar}: {{.Status}}\\n${this.resultVar}: {{.Result}}` }, slackTemplatePlaceholder: function () { return `${this.taskIdVar}: {{.TaskId}}\\n${this.taskNameVar}: {{.TaskName}}\\n${this.statusVar}: {{.Status}}\\n${this.resultVar}: {{.Result}}` }, webhookTemplatePlaceholder: '{"task_id": "{{.TaskId}}", "task_name": "{{.TaskName}}", "status": "{{.Status}}", "result": "{{.Result}}", "remark": "{{.Remark}}"}' }, taskLog: { list: '任务日志', taskName: '任务名称', startTime: '开始时间', endTime: '结束时间', duration: '执行时长', result: '执行结果', host: '主机', output: '执行输出', success: '成功', failed: '失败', viewOutput: '查看输出' }, twoFactor: { title: '双因素认证 (2FA)', status: '状态', enabled: '已启用', disabled: '未启用', enable: '启用2FA', disable: '禁用2FA', setup: '启用双因素认证', qrCode: '二维码', secret: '密钥', scanQR: '1. 使用认证APP扫描下方二维码：', manualEntry: '2. 或手动输入密钥：', verifyCode: '验证码', verifyCodePlaceholder: '请输入6位验证码', verifyCodeStep: '3. 输入APP显示的6位验证码：', confirm: '确定', confirmDisable: '确定禁用', confirmDisableMsg: '确定要禁用双因素认证吗？', enableSuccess: '2FA已启用', disableSuccess: '2FA已禁用', verifyFailed: '验证码错误', alertTitle: '提示', alertDescription: '启用双因素认证可以大大提高账户安全性。建议所有用户特别是管理员启用此功能。', enabledAlertTitle: '2FA已启用', enabledAlertDescription: '您的账户已启用双因素认证保护。', disableDialogTitle: '禁用双因素认证', disableDialogDescription: '请输入认证APP显示的6位验证码以禁用2FA：', copySecret: '复制', secretCopied: '密钥已复制到剪贴板', verifyCodeLength: '请输入6位验证码', disableFailed: '禁用2FA失败' }, install: { title: '系统安装', welcome: '欢迎使用 Gocron', dbConfig: '数据库配置', dbType: '数据库选择', dbHost: '主机名', dbPort: '端口', dbName: '数据库名称', dbFilePath: '数据库文件路径', dbUser: '用户名', dbPassword: '密码', dbTablePrefix: '表前缀', adminConfig: '管理员账号配置', adminUsername: '账号', adminPassword: '密码', confirmPassword: '确认密码', adminEmail: '邮箱', install: '安装', installing: '安装中...', installSuccess: '安装成功', installFailed: '安装失败', dbNamePlaceholder: '数据库不存在需提前创建', dbFilePathPlaceholder: './data/gocron.db', passwordPlaceholder: '至少8位，含字母和数字', selectDb: '请选择数据库', enterDbName: '请输入数据库名称', enterDbHost: '请输入主机名', enterDbPort: '请输入端口', enterDbUser: '请输入用户名', enterDbPassword: '请输入密码', enterAdminUsername: '请输入账号', enterAdminEmail: '请输入邮箱', enterAdminPassword: '请输入密码', confirmAdminPassword: '请再次输入密码', passwordMinLength: '长度至少8个字符' }, message: { saveSuccess: '保存成功', saveFailed: '保存失败', deleteSuccess: '删除成功', deleteFailed: '删除失败', updateSuccess: '更新成功', updateFailed: '更新失败', operationSuccess: '操作成功', operationFailed: '操作失败', refreshSuccess: '刷新成功', loadFailed: '加载失败', requestTimeout: '请求超时，请稍后重试', authExpired: '登录已过期，请重新登录', requestFailed: '请求失败', networkError: '网络错误', serverError: '服务器错误', dataNotFound: '数据不存在', confirmDelete: '确定要删除吗？', confirmDeleteTask: '确定要删除任务 "{name}" 吗？', confirmRunTask: '确定要手动执行任务 "{name}" 吗？', taskStarted: '任务已开始执行', selectTaskNode: '请选择任务节点', selectMailReceiver: '请选择邮件接收用户', selectSlackChannel: '请选择Slack Channel', selectWebhookUrl: '请选择Webhook地址', passwordMismatch: '两次密码输入不一致', oldPasswordError: '原密码输入错误', passwordSameAsOld: '原密码与新密码不能相同', usernameExists: '用户名已存在', emailExists: '邮箱已存在', formValidationFailed: '表单验证失败，请检查输入', pleaseEnterPassword: '请输入密码', pleaseEnterPasswordAgain: '请再次输入密码', pleaseEnterTaskName: '请输入任务名称', pleaseEnterCronExpression: '请输入crontab表达式', pleaseEnterCommand: '请输入命令', pleaseEnterValidTimeout: '请输入有效的任务超时时间', pleaseEnterValidRetryTimes: '请输入有效的任务执行失败重试次数', pleaseEnterValidRetryInterval: '请输入有效的任务执行失败，重试间隔时间', pleaseEnterNotifyKeyword: '请输入要匹配的任务执行输出关键字', pleaseEnterUrl: '请输入URL地址', pleaseEnterShellCommand: '请输入shell命令', selected: '已选择', tasks: '个任务', batchEnable: '批量启用', batchDisable: '批量禁用', batchDelete: '批量删除', confirmBatchEnable: '确定要启用选中的 {count} 个任务吗？', confirmBatchDisable: '确定要禁用选中的 {count} 个任务吗？', confirmBatchDelete: '确定要删除选中的 {count} 个任务吗？此操作不可恢复！', batchEnableSuccess: '批量启用成功', batchDisableSuccess: '批量禁用成功', batchDeleteSuccess: '批量删除成功', pleaseSelectTask: '请选择要{action}的任务', manualRunTask: '手动执行任务', confirmExecute: '确定执行', confirmDeleteTitle: '提示', confirmDeleteButton: '确定删除', taskCreatedTime: '任务创建时间', taskType: '任务类型', singleInstanceRun: '单实例运行', timeoutTime: '超时时间', retryCount: '重试次数', retryIntervalTime: '重试间隔', taskNodeLabel: '任务节点', commandLabel: '命令', remarkLabel: '备注', noLimit: '不限制', systemDefault: '系统默认', seconds: '秒', activated: '激活', stopped: '停止', confirmDeleteNode: '确定删除此节点?', confirmDeleteUser: '确定删除此用户?', connectionSuccess: '连接成功', copySuccess: '复制成功', copyFailed: '复制失败', all: '全部', clearLog: '清空日志', confirmClearLog: '确定清空所有日志?', running: '执行中', cancelled: '取消', stopTask: '停止任务', taskExecutionResult: '任务执行结果', cronExamples: 'Cron表达式示例', everyMinute: '每分钟第0秒运行', every20Seconds: '每隔20秒运行一次', everyDay21_30: '每天晚上21:30:00运行', everySaturday23: '每周六晚上23:00:00运行', reboot: '仅在应用启动时执行一次', yearly: '每年运行一次', monthly: '每月运行一次', weekly: '每周运行一次', daily: '每天运行一次', hourly: '每小时运行一次', every30s: '每隔30秒运行一次', every1m20s: '每隔1分钟20秒运行一次' }, template: { list: '任务模板', name: '模板名称', description: '描述', category: '分类', protocol: '执行方式', command: '命令', timeout: '超时时间', usageCount: '使用次数', builtin: '内置', custom: '自定义', createNew: '新建模板', useTemplate: '使用模板', saveAsTemplate: '保存为模板', templateVarTip: '使用 {{变量名}} 语法定义模板变量，应用时用户将填写变量值', applyTemplate: '应用模板', applySuccess: '模板已应用', fillVariables: '填写模板变量', variableName: '变量名', variableValue: '值', noTemplates: '暂无模板', confirmDelete: '确定删除模板 "{name}" 吗？', category_all: '全部', category_backup: '备份', category_cleanup: '清理', category_monitor: '监控', category_deploy: '部署', category_api: 'API调用', category_custom: '自定义', preview: '预览', templateNamePlaceholder: '请输入模板名称', templateDescPlaceholder: '请输入模板描述', selectCategory: '请选择分类', saveAsTemplateName: '模板名称', saveAsTemplateDesc: '模板描述', saveAsTemplateCategory: '分类', securityWarning: '变量值将明文存储在任务命令中。请勿填入密码等敏感信息，建议使用环境变量引用（如 $DB_PASS）。', saveAsTemplateWarning: '命令内容将原样保存为模板（所有用户可见）。请先移除密码、密钥等敏感信息，用 {{变量名}} 占位符替代。' }, audit: { log: '操作审计', module: '模块', action: '操作', target: '操作对象', detail: '变更详情', module_task: '任务', module_host: '节点', module_user: '用户', module_system: '系统', action_create: '创建', action_update: '修改', action_delete: '删除', action_enable: '启用', action_disable: '禁用', action_run: '手动执行', action_batch_enable: '批量启用', action_batch_disable: '批量禁用', action_batch_remove: '批量删除', action_change_password: '修改密码', action_reset_password: '重置密码' }, statistics: { title: '数据统计', totalTasks: '任务总数', todayExecutions: '今日执行', last7DaysExecutions: '7天执行次数', successRate: '7天成功率', failedCount: '7天失败任务', last7DaysTrend: '最近7天趋势', success: '成功', failed: '失败', total: '总计', executionCount: '执行次数', date: '日期', detailedData: '详细数据' } } ================================================ FILE: web/vue/src/main.js ================================================ import { createApp } from 'vue' import { createPinia } from 'pinia' import piniaPluginPersistedstate from 'pinia-plugin-persistedstate' import { ElMessageBox, ElMessage } from 'element-plus' import 'element-plus/dist/index.css' import 'nprogress/nprogress.css' import dayjs from 'dayjs' import utc from 'dayjs/plugin/utc' import timezone from 'dayjs/plugin/timezone' import App from './App.vue' import router from './router' import i18n from './locales' dayjs.extend(utc) dayjs.extend(timezone) const app = createApp(App) const pinia = createPinia() pinia.use(piniaPluginPersistedstate) app.use(pinia) app.use(router) app.use(i18n) app.directive('focus', { mounted(el) { el.focus() } }) app.config.globalProperties.$appConfirm = function (callback) { ElMessageBox.confirm(i18n.global.t('common.confirmOperation'), i18n.global.t('common.tip'), { confirmButtonText: i18n.global.t('common.confirm'), cancelButtonText: i18n.global.t('common.cancel'), type: 'warning', center: true, customClass: 'custom-message-box' }) .then(() => { callback() }) .catch(() => {}) } app.config.globalProperties.$message = ElMessage app.config.globalProperties.$filters = { formatTime(time) { if (!time) return '' return dayjs(time).format('YYYY-MM-DD HH:mm:ss') } } // 全局错误处理 app.config.errorHandler = (err, instance, info) => { if (import.meta.env.DEV) { console.error('[Global Error]', err, info) } ElMessage.error('系统错误，请刷新页面重试') } app.config.warnHandler = (msg, instance, trace) => { if (import.meta.env.DEV) { console.warn('[Vue Warn]', msg, trace) } } // 开发环境性能监控 if (import.meta.env.DEV) { app.config.performance = true } // 生产环境禁用 devtools if (import.meta.env.PROD) { app.config.devtools = false } app.mount('#app') ================================================ FILE: web/vue/src/pages/host/edit.vue ================================================ ================================================ FILE: web/vue/src/pages/host/list.vue ================================================ ================================================ FILE: web/vue/src/pages/install/index.vue ================================================ ================================================ FILE: web/vue/src/pages/statistics/index.vue ================================================ ================================================ FILE: web/vue/src/pages/system/auditLog.vue ================================================ ================================================ FILE: web/vue/src/pages/system/logRetention.vue ================================================ ================================================ FILE: web/vue/src/pages/system/loginLog.vue ================================================ ================================================ FILE: web/vue/src/pages/system/notification/email.vue ================================================ ================================================ FILE: web/vue/src/pages/system/notification/slack.vue ================================================ ================================================ FILE: web/vue/src/pages/system/notification/tab.vue ================================================ ================================================ FILE: web/vue/src/pages/system/notification/webhook.vue ================================================ ================================================ FILE: web/vue/src/pages/system/sidebar.vue ================================================ ================================================ FILE: web/vue/src/pages/task/edit.vue ================================================ ================================================ FILE: web/vue/src/pages/task/list.vue ================================================ ================================================ FILE: web/vue/src/pages/task/sidebar.vue ================================================ ================================================ FILE: web/vue/src/pages/taskLog/list.vue ================================================ ================================================ FILE: web/vue/src/pages/template/edit.vue ================================================ ================================================ FILE: web/vue/src/pages/template/list.vue ================================================ ================================================ FILE: web/vue/src/pages/user/edit.vue ================================================ ================================================ FILE: web/vue/src/pages/user/editMyPassword.vue ================================================ ================================================ FILE: web/vue/src/pages/user/editPassword.vue ================================================ ================================================ FILE: web/vue/src/pages/user/list.vue ================================================ ================================================ FILE: web/vue/src/pages/user/login.vue ================================================ ================================================ FILE: web/vue/src/pages/user/twoFactor.vue ================================================ ================================================ FILE: web/vue/src/router/index.js ================================================ import { createRouter, createWebHashHistory } from 'vue-router' import { useUserStore } from '../stores/user' const routes = [ { path: '/', redirect: '/task' }, { path: '/install', name: 'install', component: () => import('../pages/install/index.vue'), meta: { noLogin: true, noNeedAdmin: true } }, { path: '/task', name: 'task-list', component: () => import('../pages/task/list.vue'), meta: { noNeedAdmin: true } }, { path: '/task/create', name: 'task-create', component: () => import('../pages/task/edit.vue') }, { path: '/task/edit/:id', name: 'task-edit', component: () => import('../pages/task/edit.vue') }, { path: '/task/log', name: 'task-log', component: () => import('../pages/taskLog/list.vue'), meta: { noNeedAdmin: true } }, { path: '/template', name: 'template-list', component: () => import('../pages/template/list.vue'), meta: { noNeedAdmin: true } }, { path: '/template/create', name: 'template-create', component: () => import('../pages/template/edit.vue') }, { path: '/template/edit/:id', name: 'template-edit', component: () => import('../pages/template/edit.vue') }, { path: '/host', name: 'host-list', component: () => import('../pages/host/list.vue'), meta: { noNeedAdmin: true } }, { path: '/host/create', name: 'host-create', component: () => import('../pages/host/edit.vue') }, { path: '/host/edit/:id', name: 'host-edit', component: () => import('../pages/host/edit.vue') }, { path: '/user', name: 'user-list', component: () => import('../pages/user/list.vue') }, { path: '/user/create', name: 'user-create', component: () => import('../pages/user/edit.vue') }, { path: '/user/edit/:id', name: 'user-edit', component: () => import('../pages/user/edit.vue') }, { path: '/user/login', name: 'user-login', component: () => import('../pages/user/login.vue'), meta: { noLogin: true } }, { path: '/user/edit-password/:id', name: 'user-edit-password', component: () => import('../pages/user/editPassword.vue') }, { path: '/user/edit-my-password', name: 'user-edit-my-password', component: () => import('../pages/user/editMyPassword.vue'), meta: { noNeedAdmin: true } }, { path: '/user/two-factor', name: 'user-two-factor', component: () => import('../pages/user/twoFactor.vue'), meta: { noNeedAdmin: true } }, { path: '/system', redirect: '/system/notification/email' }, { path: '/system/notification/email', name: 'system-notification-email', component: () => import('../pages/system/notification/email.vue') }, { path: '/system/notification/slack', name: 'system-notification-slack', component: () => import('../pages/system/notification/slack.vue') }, { path: '/system/notification/webhook', name: 'system-notification-webhook', component: () => import('../pages/system/notification/webhook.vue') }, { path: '/system/login-log', name: 'login-log', component: () => import('../pages/system/loginLog.vue') }, { path: '/system/log-retention', name: 'log-retention', component: () => import('../pages/system/logRetention.vue') }, { path: '/system/audit-log', name: 'system-audit-log', component: () => import('../pages/system/auditLog.vue') }, { path: '/statistics', name: 'statistics', component: () => import('../pages/statistics/index.vue'), meta: { noNeedAdmin: true } }, { path: '/:pathMatch(.*)*', component: () => import('../components/common/notFound.vue'), meta: { noLogin: true, noNeedAdmin: true } } ] const router = createRouter({ history: createWebHashHistory(), routes }) router.beforeEach((to, from, next) => { // 防止登录后访问登录页 if (to.path === '/user/login') { const userStore = useUserStore() if (userStore.token) { next({ path: '/' }) return } } if (to.meta.noLogin) { next() return } const userStore = useUserStore() if (userStore.token) { if (userStore.isAdmin || to.meta.noNeedAdmin) { next() return } next({ path: '/404.html' }) return } next({ path: '/user/login', query: { redirect: to.fullPath } }) }) export default router ================================================ FILE: web/vue/src/storage/user.js ================================================ class User { get () { return { 'token': this.getToken(), 'uid': this.getUid(), 'username': this.getUsername(), 'isAdmin': this.getIsAdmin() } } getToken () { return localStorage.getItem('token') || '' } setToken (token) { localStorage.setItem('token', token) return this } clear () { localStorage.clear() } getUid () { return localStorage.getItem('uid') || '' } setUid (uid) { localStorage.setItem('uid', uid) return this } getUsername () { return localStorage.getItem('username') || '' } setUsername (username) { localStorage.setItem('username', username) return this } getIsAdmin () { let isAdmin = localStorage.getItem('is_admin') return isAdmin === '1' } setIsAdmin (isAdmin) { localStorage.setItem('is_admin', isAdmin) return this } } export default new User() ================================================ FILE: web/vue/src/stores/user.js ================================================ import { defineStore } from 'pinia' export const useUserStore = defineStore('user', { state: () => ({ token: '', uid: '', username: '', isAdmin: false }), getters: { isLogin: (state) => state.token !== '' }, actions: { setUser(user) { this.token = user.token || '' this.uid = user.uid || '' this.username = user.username || '' this.isAdmin = user.isAdmin || false }, logout() { this.token = '' this.uid = '' this.username = '' this.isAdmin = false } }, persist: { key: 'gocron-user', storage: localStorage, paths: ['token', 'uid', 'username', 'isAdmin'] } }) ================================================ FILE: web/vue/src/utils/__tests__/cronValidator.spec.js ================================================ import { describe, it, expect } from 'vitest' import { extractTimezone, validateCronSpec } from '../cronValidator' describe('extractTimezone', () => { it('extracts CRON_TZ= prefix', () => { const result = extractTimezone('CRON_TZ=Asia/Shanghai 0 30 8 * * *') expect(result.timezone).toBe('Asia/Shanghai') expect(result.spec).toBe('0 30 8 * * *') }) it('extracts TZ= prefix', () => { const result = extractTimezone('TZ=America/New_York 0 0 9 * * *') expect(result.timezone).toBe('America/New_York') expect(result.spec).toBe('0 0 9 * * *') }) it('extracts TZ= prefix with descriptor', () => { const result = extractTimezone('CRON_TZ=UTC @daily') expect(result.timezone).toBe('UTC') expect(result.spec).toBe('@daily') }) it('returns empty timezone when no prefix', () => { const result = extractTimezone('0 30 8 * * *') expect(result.timezone).toBe('') expect(result.spec).toBe('0 30 8 * * *') }) it('returns empty timezone for descriptors without prefix', () => { const result = extractTimezone('@daily') expect(result.timezone).toBe('') expect(result.spec).toBe('@daily') }) it('handles empty string', () => { const result = extractTimezone('') expect(result.timezone).toBe('') expect(result.spec).toBe('') }) it('handles null/undefined', () => { expect(extractTimezone(null).timezone).toBe('') expect(extractTimezone(undefined).timezone).toBe('') }) it('preserves spec with extra spaces', () => { const result = extractTimezone('CRON_TZ=Asia/Tokyo @every 30s') expect(result.timezone).toBe('Asia/Tokyo') expect(result.spec).toBe('@every 30s') }) }) describe('validateCronSpec with timezone prefix', () => { it('validates spec with CRON_TZ= prefix', () => { const result = validateCronSpec('CRON_TZ=Asia/Shanghai 0 30 8 * * *') expect(result.valid).toBe(true) }) it('validates spec with TZ= prefix', () => { const result = validateCronSpec('TZ=UTC @daily') expect(result.valid).toBe(true) }) it('rejects invalid cron after stripping timezone', () => { const result = validateCronSpec('CRON_TZ=Asia/Shanghai invalid') expect(result.valid).toBe(false) }) it('validates spec without prefix (backward compatible)', () => { expect(validateCronSpec('0 30 8 * * *').valid).toBe(true) expect(validateCronSpec('@daily').valid).toBe(true) expect(validateCronSpec('@every 30s').valid).toBe(true) }) it('rejects empty spec', () => { expect(validateCronSpec('').valid).toBe(false) expect(validateCronSpec(null).valid).toBe(false) }) }) ================================================ FILE: web/vue/src/utils/__tests__/env.spec.js ================================================ import { describe, it, expect, vi } from 'vitest' import { env, devLog } from '../env' describe('env utilities', () => { it('should have correct env properties', () => { expect(env).toHaveProperty('isDev') expect(env).toHaveProperty('isProd') expect(env).toHaveProperty('apiBaseUrl') }) it('should have default apiBaseUrl', () => { expect(env.apiBaseUrl).toBe('/api') }) it('devLog should only log in dev mode', () => { const consoleSpy = vi.spyOn(console, 'log').mockImplementation(() => {}) devLog('test message') if (env.isDev) { expect(consoleSpy).toHaveBeenCalledWith('[Dev]', 'test message') } else { expect(consoleSpy).not.toHaveBeenCalled() } consoleSpy.mockRestore() }) }) ================================================ FILE: web/vue/src/utils/cronValidator.js ================================================ /** * Cron表达式验证器 * 支持格式：秒 分 时 天 月 周 * 支持快捷语法：@yearly, @monthly, @weekly, @daily, @midnight, @hourly, @every */ import i18n from '@/locales' const t = (key, params) => i18n.global.t(key, params) // 快捷语法列表 const SHORTCUTS = [ '@reboot', '@yearly', '@annually', '@monthly', '@weekly', '@daily', '@midnight', '@hourly' ] // @every 语法正则 const EVERY_PATTERN = /^@every\s+(\d+[smh])+$/ /** * 从 spec 中提取 CRON_TZ=/TZ= 前缀，返回 { timezone, spec } * 无前缀时 timezone 为空字符串 */ export function extractTimezone(spec) { if (!spec || typeof spec !== 'string') { return { timezone: '', spec: spec || '' } } const trimmed = spec.trim() const match = trimmed.match(/^(?:CRON_TZ|TZ)=(\S+)\s+(.+)$/) if (match) { return { timezone: match[1], spec: match[2] } } return { timezone: '', spec: trimmed } } /** * 验证cron表达式 * @param {string} spec - cron表达式（可带 CRON_TZ= 前缀） * @returns {{valid: boolean, message: string}} */ export function validateCronSpec(spec) { if (!spec || typeof spec !== 'string') { return { valid: false, message: t('cronValidator.required') } } // 剥离 CRON_TZ=/TZ= 前缀后再验证 const { spec: cronExpr } = extractTimezone(spec) const trimmed = cronExpr.trim() if (!trimmed) { return { valid: false, message: t('cronValidator.required') } } // 检查快捷语法 if (trimmed.startsWith('@')) { return validateShortcut(trimmed) } // 检查标准cron表达式 return validateStandardCron(trimmed) } /** * 验证快捷语法 */ function validateShortcut(spec) { const lower = spec.toLowerCase() // 检查固定快捷语法 if (SHORTCUTS.includes(lower)) { return { valid: true, message: '' } } // 检查 @every 语法 if (lower.startsWith('@every')) { if (!EVERY_PATTERN.test(lower)) { return { valid: false, message: t('cronValidator.everyFormatError') } } return { valid: true, message: '' } } return { valid: false, message: t('cronValidator.shortcutError') } } /** * 验证标准cron表达式（6段式） */ function validateStandardCron(spec) { const segments = spec.split(/\s+/) // 必须是6段 if (segments.length !== 6) { return { valid: false, message: t('cronValidator.sixFieldsRequired') } } // 字段范围定义 const ranges = [ { name: t('cronValidator.fieldSecond'), min: 0, max: 59 }, { name: t('cronValidator.fieldMinute'), min: 0, max: 59 }, { name: t('cronValidator.fieldHour'), min: 0, max: 23 }, { name: t('cronValidator.fieldDay'), min: 1, max: 31 }, { name: t('cronValidator.fieldMonth'), min: 1, max: 12 }, { name: t('cronValidator.fieldWeek'), min: 0, max: 7 } ] // 验证每一段 for (let i = 0; i < segments.length; i++) { const result = validateSegment(segments[i], ranges[i]) if (!result.valid) { return result } } return { valid: true, message: '' } } /** * 验证单个字段 */ function validateSegment(segment, range) { // 允许的字符 if (!/^[0-9*/,\-?LW#]+$/.test(segment)) { return { valid: false, message: t('cronValidator.illegalChar', { field: range.name }) } } // * 通配符 if (segment === '*') { return { valid: true } } // ? 占位符（用于天和周） if (segment === '?') { return { valid: true } } // 范围：1-5 if (segment.includes('-')) { return validateRange(segment, range) } // 步长：*/5 或 1-10/2 if (segment.includes('/')) { return validateStep(segment, range) } // 列表：1,2,3 if (segment.includes(',')) { return validateList(segment, range) } // 单个数字 if (/^\d+$/.test(segment)) { const num = parseInt(segment, 10) if (num < range.min || num > range.max) { return { valid: false, message: t('cronValidator.valueOutOfRange', { field: range.name, value: num, min: range.min, max: range.max }) } } return { valid: true } } // L, W, # 等特殊字符（简单验证） if (/^[LW#]/.test(segment)) { return { valid: true } } return { valid: false, message: t('cronValidator.formatError', { field: range.name }) } } /** * 验证范围表达式：1-5 */ function validateRange(segment, range) { const parts = segment.split('-') if (parts.length !== 2) { return { valid: false, message: t('cronValidator.rangeFormatError', { field: range.name }) } } const start = parseInt(parts[0], 10) const end = parseInt(parts[1], 10) if (isNaN(start) || isNaN(end)) { return { valid: false, message: t('cronValidator.rangeNotNumber', { field: range.name }) } } if (start < range.min || end > range.max || start > end) { return { valid: false, message: t('cronValidator.rangeInvalid', { field: range.name, start, end }) } } return { valid: true } } /** * 验证步长表达式：星号/5 或 1-10/2 */ function validateStep(segment, range) { const parts = segment.split('/') if (parts.length !== 2) { return { valid: false, message: t('cronValidator.stepFormatError', { field: range.name }) } } const step = parseInt(parts[1], 10) if (isNaN(step) || step <= 0) { return { valid: false, message: t('cronValidator.stepNotPositive', { field: range.name }) } } // 验证基础部分 if (parts[0] !== '*') { return validateSegment(parts[0], range) } return { valid: true } } /** * 验证列表表达式：1,2,3 */ function validateList(segment, range) { const parts = segment.split(',') for (const part of parts) { const result = validateSegment(part.trim(), range) if (!result.valid) { return result } } return { valid: true } } /** * 获取cron表达式示例 */ export function getCronExamples() { return [ { expr: '0 * * * * *', desc: '每分钟第0秒运行' }, { expr: '*/20 * * * * *', desc: '每隔20秒运行一次' }, { expr: '0 30 21 * * *', desc: '每天晚上21:30:00运行' }, { expr: '0 0 23 * * 6', desc: '每周六晚上23:00:00运行' }, { expr: '0 0 1 1 * *', desc: '每月1号凌晨1点运行' }, { expr: '@hourly', desc: '每小时运行一次' }, { expr: '@daily', desc: '每天运行一次' }, { expr: '@every 30s', desc: '每隔30秒运行一次' }, { expr: '@every 1m20s', desc: '每隔1分钟20秒运行一次' } ] } ================================================ FILE: web/vue/src/utils/env.js ================================================ // 环境变量验证和获取 export const env = { isDev: import.meta.env.DEV, isProd: import.meta.env.PROD, apiBaseUrl: import.meta.env.VITE_API_BASE_URL || '/api' } // 开发环境日志 export const devLog = (...args) => { if (env.isDev) { console.log('[Dev]', ...args) } } export const devWarn = (...args) => { if (env.isDev) { console.warn('[Dev]', ...args) } } export const devError = (...args) => { if (env.isDev) { console.error('[Dev]', ...args) } } ================================================ FILE: web/vue/src/utils/httpClient.js ================================================ import axios from 'axios' import { ElMessage } from 'element-plus' import router from '../router/index' import { useUserStore } from '../stores/user' import qs from 'qs' import NProgress from '@/utils/progress' const getLocale = () => localStorage.getItem('locale') || 'zh-CN' const messages = { 'zh-CN': { loadFailed: '加载失败, 请稍后再试', requestTimeout: '请求超时，请稍后重试', authExpired: '登录已过期，请重新登录', requestFailed: '请求失败' }, 'en-US': { loadFailed: 'Load failed, please try again later', requestTimeout: 'Request timeout, please try again later', authExpired: 'Login expired, please login again', requestFailed: 'Request failed' } } const t = key => { const locale = getLocale() return messages[locale]?.[key] || messages['zh-CN'][key] } // 成功状态码 const SUCCESS_CODE = 0 // 认证失败 const AUTH_ERROR_CODE = 401 // 应用未安装 const APP_NOT_INSTALL_CODE = 801 axios.defaults.baseURL = '/api' axios.defaults.timeout = 30000 axios.defaults.responseType = 'json' axios.interceptors.request.use( config => { NProgress.start() const userStore = useUserStore() config.headers['Auth-Token'] = userStore.token config.headers['Accept-Language'] = localStorage.getItem('locale') || 'zh-CN' return config }, error => { NProgress.done() ElMessage.error({ message: t('loadFailed') }) return Promise.reject(error) } ) axios.interceptors.response.use( data => { NProgress.done() // 检查是否有新的 token const newToken = data.headers['new-auth-token'] if (newToken) { const userStore = useUserStore() userStore.token = newToken } return data }, error => { NProgress.done() // 处理超时 if (error.code === 'ECONNABORTED' && error.message.includes('timeout')) { ElMessage.error({ message: t('requestTimeout') }) return Promise.reject(error) } // 处理认证失败 if (error.response && error.response.status === 401) { const userStore = useUserStore() userStore.token = '' ElMessage.warning({ message: t('authExpired') }) setTimeout(() => { window.location.href = '/' }, 500) return Promise.reject(error) } ElMessage.error({ message: t('loadFailed') }) return Promise.reject(error) } ) function handle(promise, next, errorCallback) { promise .then(res => successCallback(res, next, errorCallback)) .catch(error => failureCallback(error)) } function checkResponseCode(code, msg) { switch (code) { // 应用未安装 case APP_NOT_INSTALL_CODE: router.push('/install') return false // 认证失败 case AUTH_ERROR_CODE: { const userStore = useUserStore() userStore.token = '' ElMessage.warning({ message: t('authExpired') }) setTimeout(() => { window.location.href = '/' }, 500) return false } } if (code !== SUCCESS_CODE) { ElMessage.error({ message: msg }) return false } return true } function successCallback(res, next, errorCallback) { if (res.data.code !== SUCCESS_CODE) { if (errorCallback) { errorCallback(res.data.code, res.data.message) return } if (!checkResponseCode(res.data.code, res.data.message)) { return } } if (!next) { return } next(res.data.data, res.data.code, res.data.message) } function failureCallback(error) { // 避免重复提示（已在 interceptor 中处理） if (error.response && error.response.status === 401) { return } if (error.code === 'ECONNABORTED') { return } ElMessage.error({ message: t('requestFailed') + ' - ' + error.message }) } export default { get(uri, params, next) { const promise = axios.get(uri, { params }) handle(promise, next) }, batchGet(uriGroup, next) { const requests = [] for (let item of uriGroup) { let params = {} if (item.params !== undefined) { params = item.params } requests.push(axios.get(item.uri, { params })) } Promise.all(requests) .then(function (res) { const result = [] for (let item of res) { if (!checkResponseCode(item.data.code, item.data.message)) { return } result.push(item.data.data) } next(...result) }) .catch(error => failureCallback(error)) }, post(uri, data, next, errorCallback) { const promise = axios.post(uri, qs.stringify(data), { headers: { 'Content-Type': 'application/x-www-form-urlencoded' } }) handle(promise, next, errorCallback) }, postJson(uri, data, next, errorCallback) { const promise = axios.post(uri, data, { headers: { 'Content-Type': 'application/json' } }) handle(promise, next, errorCallback) } } ================================================ FILE: web/vue/src/utils/performance.js ================================================ // 性能监控工具 export const measurePerformance = (name, fn) => { if (import.meta.env.DEV) { const start = performance.now() const result = fn() const end = performance.now() console.log(`[Performance] ${name}: ${(end - start).toFixed(2)}ms`) return result } return fn() } // 监控路由切换性能 export const measureRouteChange = (to, from) => { if (import.meta.env.DEV && performance.mark) { performance.mark(`route-${to.path}-start`) } } export const measureRouteChangeEnd = (to) => { if (import.meta.env.DEV && performance.mark && performance.measure) { performance.mark(`route-${to.path}-end`) try { performance.measure( `route-${to.path}`, `route-${to.path}-start`, `route-${to.path}-end` ) const measure = performance.getEntriesByName(`route-${to.path}`)[0] console.log(`[Route Performance] ${to.path}: ${measure.duration.toFixed(2)}ms`) } catch (e) { // ignore } } } ================================================ FILE: web/vue/src/utils/progress/index.js ================================================ //@ts-ignore import NProgress from 'nprogress' NProgress.configure({ // 动画方式 easing: 'ease', // 递增进度条的速度 speed: 500, // 是否显示加载ico showSpinner: false, // 自动递增间隔 trickleSpeed: 200, // 初始化时的最小百分比 minimum: 0.3 }) let activeRequests = 0 export function start() { if (activeRequests === 0) { NProgress.start() } activeRequests++ } export function done() { activeRequests = Math.max(0, activeRequests - 1) if (activeRequests === 0) { NProgress.done() } } export default { start, done } ================================================ FILE: web/vue/src/utils/request.js ================================================ import axios from 'axios' import { ElMessage } from 'element-plus' import router from '../router' import { useUserStore } from '../stores/user' const SUCCESS_CODE = 0 const AUTH_ERROR_CODE = 401 const APP_NOT_INSTALL_CODE = 801 // 请求取消管理 const pendingRequests = new Map() const request = axios.create({ baseURL: '/api', timeout: 10000, withCredentials: false, headers: { 'X-Requested-With': 'XMLHttpRequest' } }) request.interceptors.request.use( config => { const userStore = useUserStore() if (userStore.token) { config.headers['Auth-Token'] = userStore.token } // 取消重复请求 const requestKey = `${config.method}_${config.url}` if (pendingRequests.has(requestKey)) { const controller = pendingRequests.get(requestKey) controller.abort() } const controller = new AbortController() config.signal = controller.signal pendingRequests.set(requestKey, controller) return config }, error => { ElMessage.error('请求失败') return Promise.reject(error) } ) request.interceptors.response.use( response => { // 清除已完成的请求 const requestKey = `${response.config.method}_${response.config.url}` pendingRequests.delete(requestKey) const { code, message, data } = response.data if (code === APP_NOT_INSTALL_CODE) { router.push('/install') return Promise.reject(new Error(message)) } if (code === AUTH_ERROR_CODE) { const userStore = useUserStore() userStore.logout() router.push('/user/login') return Promise.reject(new Error(message)) } if (code !== SUCCESS_CODE) { ElMessage.error(message || '请求失败') return Promise.reject(new Error(message)) } return data }, error => { // 清除失败的请求 if (error.config) { const requestKey = `${error.config.method}_${error.config.url}` pendingRequests.delete(requestKey) } // 忽略取消的请求 if (axios.isCancel(error)) { return Promise.reject(error) } // 网络错误或超时 if (error.code === 'ECONNABORTED') { ElMessage.error('请求超时，请稍后重试') } else if (!error.response) { ElMessage.error('网络连接失败，请检查网络') } else { ElMessage.error(error.message || '请求失败') } return Promise.reject(error) } ) export default request ================================================ FILE: web/vue/static/.gitkeep ================================================ ================================================ FILE: web/vue/static/robots.txt ================================================ User-agent: * Disallow: / ================================================ FILE: web/vue/verify.sh ================================================ #!/bin/bash set -e echo "==========================================" echo "前端改动验证脚本" echo "==========================================" echo "" # 1. 依赖检查 echo "1. 检查依赖..." yarn install --frozen-lockfile echo "✅ 依赖检查完成" echo "" # 2. 运行测试 echo "2. 运行单元测试..." yarn test --run echo "✅ 测试通过" echo "" # 3. Lint 检查 echo "3. 运行 Lint 检查..." yarn lint || echo "⚠️ Lint 有警告（非阻塞）" echo "" # 4. 构建验证 echo "4. 构建生产版本..." yarn build echo "✅ 构建成功" echo "" # 5. 检查构建产物 echo "5. 检查构建产物大小..." echo "" echo "主要 JS 文件:" ls -lh dist/static/*.js 2>/dev/null | head -5 || echo "无 JS 文件" echo "" echo "主要 CSS 文件:" ls -lh dist/static/*.css 2>/dev/null | head -5 || echo "无 CSS 文件" echo "" # 6. 总结 echo "==========================================" echo "✅ 所有验证通过！" echo "==========================================" echo "" echo "下一步:" echo " 1. 启动开发服务器: yarn dev" echo " 2. 手动测试功能" echo "" ================================================ FILE: web/vue/vite.config.js ================================================ import { defineConfig } from 'vite' import vue from '@vitejs/plugin-vue' import { resolve } from 'path' import AutoImport from 'unplugin-auto-import/vite' import Components from 'unplugin-vue-components/vite' import { ElementPlusResolver } from 'unplugin-vue-components/resolvers' import viteCompression from 'vite-plugin-compression' export default defineConfig({ plugins: [ vue(), AutoImport({ resolvers: [ElementPlusResolver()], imports: ['vue', 'vue-router', 'pinia', '@vueuse/core'] }), Components({ resolvers: [ElementPlusResolver()] }), viteCompression({ algorithm: 'gzip', ext: '.gz' }) ], resolve: { alias: { '@': resolve(__dirname, 'src') } }, server: { port: 8080, proxy: { '/api': { target: 'http://localhost:5920', changeOrigin: true } } }, build: { outDir: 'dist', assetsDir: 'static', sourcemap: false, minify: 'esbuild', cssCodeSplit: true, reportCompressedSize: false, rollupOptions: { output: { manualChunks: { 'vue-vendor': ['vue', 'vue-router', 'pinia'], 'element-plus': ['element-plus', '@element-plus/icons-vue'], 'utils': ['axios', 'dayjs', 'qs'] }, chunkFileNames: 'static/js/[name]-[hash].js', entryFileNames: 'static/js/[name]-[hash].js', assetFileNames: 'static/[ext]/[name]-[hash].[ext]' } }, chunkSizeWarningLimit: 1000 } }) ================================================ FILE: web/vue/vitest.config.js ================================================ import { defineConfig } from 'vitest/config' import vue from '@vitejs/plugin-vue' import { resolve } from 'path' export default defineConfig({ plugins: [vue()], test: { environment: 'jsdom', globals: true }, resolve: { alias: { '@': resolve(__dirname, 'src') } } }) ================================================ FILE: webhook-test/go.mod ================================================ module webhook-test go 1.21 ================================================ FILE: webhook-test/go.sum ================================================ ================================================ FILE: webhook-test/start-webhook-server.sh ================================================ #!/bin/bash echo "🚀 启动Webhook测试服务..." # 检查Go是否安装 if ! command -v go &> /dev/null; then echo "❌ 未找到Go，请先安装Go语言环境" exit 1 fi # 进入webhook-test目录 cd "$(dirname "$0")" # 启动服务 echo "📡 启动服务在端口8080..." go run webhook-test-server.go ================================================ FILE: webhook-test/test-webhook.sh ================================================ #!/bin/bash echo "🧪 测试Webhook服务..." # 测试数据 test_data='{ "task_id": 123, "task_name": "测试任务", "status": "成功", "output": "任务执行完成", "remark": "这是一个测试webhook" }' echo "📤 发送测试数据到webhook服务..." echo "数据: $test_data" # 发送POST请求 curl -X POST \ -H "Content-Type: application/json" \ -d "$test_data" \ http://localhost:8080/webhook echo -e "\n\n✅ 测试完成" ================================================ FILE: webhook-test/webhook-test-server.go ================================================ package main import ( "encoding/json" "fmt" "io" "log" "net/http" "time" ) // WebhookPayload webhook接收的数据结构 type WebhookPayload struct { TaskID int `json:"task_id"` TaskName string `json:"task_name"` Status string `json:"status"` Output string `json:"output"` Remark string `json:"remark"` } func main() { http.HandleFunc("/webhook", handleWebhook) http.HandleFunc("/health", handleHealth) fmt.Println("🚀 Webhook测试服务启动") fmt.Println("📡 监听地址: http://localhost:8080/webhook") fmt.Println("💚 健康检查: http://localhost:8080/health") log.Fatal(http.ListenAndServe(":8080", nil)) } func handleWebhook(w http.ResponseWriter, r *http.Request) { // 记录请求时间 timestamp := time.Now().Format("2006-01-02 15:04:05") fmt.Printf("\n=== [%s] 收到Webhook请求 ===\n", timestamp) fmt.Printf("方法: %s\n", r.Method) fmt.Printf("路径: %s\n", r.URL.Path) // 打印请求头 fmt.Println("请求头:") for name, values := range r.Header { for _, value := range values { fmt.Printf(" %s: %s\n", name, value) } } // 读取请求体 body, err := io.ReadAll(r.Body) if err != nil { fmt.Printf("❌ 读取请求体失败: %v\n", err) http.Error(w, "读取请求体失败", http.StatusBadRequest) return } fmt.Printf("请求体: %s\n", string(body)) // 尝试解析JSON var payload WebhookPayload if err := json.Unmarshal(body, &payload); err != nil { fmt.Printf("⚠️ JSON解析失败: %v\n", err) fmt.Println("将作为纯文本处理") } else { fmt.Println("✅ JSON解析成功:") fmt.Printf(" 任务ID: %d\n", payload.TaskID) fmt.Printf(" 任务名称: %s\n", payload.TaskName) fmt.Printf(" 状态: %s\n", payload.Status) fmt.Printf(" 输出: %s\n", payload.Output) fmt.Printf(" 备注: %s\n", payload.Remark) } // 返回成功响应 w.Header().Set("Content-Type", "application/json") w.WriteHeader(http.StatusOK) response := map[string]interface{}{ "success": true, "message": "webhook接收成功", "timestamp": timestamp, "received": len(body) > 0, } json.NewEncoder(w).Encode(response) fmt.Println("✅ 响应已发送") } func handleHealth(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") w.WriteHeader(http.StatusOK) response := map[string]string{ "status": "ok", "time": time.Now().Format("2006-01-02 15:04:05"), } json.NewEncoder(w).Encode(response) }