resize img of readme

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,12 @@
+---
+name: Bug Report
+about: Report a bug encountered while operating Nightingale
+labels: kind/bug
+---
+
+**夜莺版本**:
+
+
+**问题和复现方法**:
+
+

.github/ISSUE_TEMPLATE/config.yml

@@ -1,5 +1,5 @@
 blank_issues_enabled: false
 contact_links:
-  - name: Nightingale docs
-    url: https://n9e.github.io/
-    about: You may want to read through the document before asking questions.
+  - name: Nightingale community
+    url: https://n9e.didiyun.com/community/
+    about: List of communication channels for the Nightingale community.

.github/ISSUE_TEMPLATE/question.yml

@@ -1,33 +0,0 @@
-name: Bug Report & Usage Question
-description: Reporting a bug or asking a question about how to use Nightingale 
-labels: []
-
-body:
-  - type: markdown
-    attributes:
-      value: |
-        The more detailed the form is filled in, the easier the problem will be solved.
-        提供的信息越详细，问题解决的可能性就越大。另外, 提问之前请先搜索历史 issue (包括 close 的), 以免重复提问。
-  - type: textarea
-    id: question
-    attributes:
-      label: Question and Steps to reproduce
-      description: Describe your question and steps to reproduce the bug. 描述问题以及复现步骤
-    validations:
-      required: true
-  - type: textarea
-    id: logs
-    attributes:
-      label: Relevant logs and configurations
-      description: Relevant logs and configurations. 报错日志（[查看方法](https://flashcat.cloud/docs/content/flashcat-monitor/nightingale-v6/faq/how-to-check-logs/)）以及各个相关组件的配置信息
-      render: text
-    validations:
-      required: true
-  - type: textarea
-    id: system-info
-    attributes:
-      label: Version
-      description: Include nightingale version, operating system, and other relevant details. 请告知夜莺的版本、操作系统的版本、CPU架构等信息
-    validations:
-      required: true
- 

.github/workflows/n9e.yml

@@ -1,33 +1,26 @@
-name: Release
+name: Go
 
 on:
   push:
-    tags:
-      - 'v*'
-env:
-  GO_VERSION: 1.23
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
 
 jobs:
-  goreleaser:
+
+  build:
+    name: Build
     runs-on: ubuntu-latest
     steps:
-      - name: Checkout Source Code
-        uses: actions/checkout@v3
-        with:
-          fetch-depth: 0
-      - name: Setup Go Environment
-        uses: actions/setup-go@v3
-        with:
-          go-version: ${{ env.GO_VERSION }}
-      - uses: docker/login-action@v2
-        with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_TOKEN }}
-      - name: Run GoReleaser
-        uses: goreleaser/goreleaser-action@v3
-        with:
-          distribution: goreleaser
-          version: '~> v1'
-          args: release --rm-dist
-        env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Set up Go 1.17
+      uses: actions/setup-go@v1
+      with:
+        go-version: 1.17
+      id: go
+
+    - name: Check out code into the Go module directory
+      uses: actions/checkout@v2
+
+    - name: Build
+      run: make

.gitignore

@@ -9,7 +9,6 @@
 *.o
 *.a
 *.so
-*.db
 *.sw[po]
 *.tar.gz
 *.[568vq]
@@ -31,11 +30,7 @@ _test
 /dist
 /etc/*.local.yml
 /etc/*.local.conf
-/etc/rsa/*
 /etc/plugins/*.local.yml
-/etc/script/rules.yaml
-/etc/script/alert-rules.json
-/etc/script/record-rules.json
 /data*
 /tarball
 /run
@@ -45,14 +40,7 @@ _test
 /n9e
 /docker/pub
 /docker/n9e
-/docker/compose-bridge/mysqldata
-/docker/compose-host-network/mysqldata
-/docker/compose-host-network-metric-log/mysqldata
-/docker/compose-host-network-metric-log/n9e-logs
-/docker/compose-postgres/pgdata
-/etc.local*
-/front/statik/statik.go
-/docker/compose-bridge/etc-nightingale/rsa/
+/docker/mysqldata
 
 .alerts
 .idea
@@ -64,8 +52,4 @@ _test
 queries.active
 
 /n9e-*
-n9e.sql
 
-!/datasource
-
-.env.json

.goreleaser.yaml

@@ -1,122 +0,0 @@
-before:
-  hooks:
-    # You may remove this if you don't use go modules.
-    - go mod tidy
-    - go install github.com/rakyll/statik
-
-snapshot:
-  name_template: '{{ .Tag }}'
-checksum:
-  name_template: 'checksums.txt'
-changelog:
-  skip: true
-
-builds:
-  - id: build
-    hooks:
-      pre:
-      - cmd: sh -x ./fe.sh
-        output: true
-    main: ./cmd/center/
-    binary: n9e
-    env:
-      - CGO_ENABLED=0
-    goos:
-      - linux
-    goarch:
-      - amd64
-      - arm64
-    ldflags:
-      - -s -w
-      - -X github.com/ccfos/nightingale/v6/pkg/version.Version={{ .Tag }}-{{.Commit}}
-  - id: build-cli
-    main: ./cmd/cli/
-    binary: n9e-cli
-    env:
-      - CGO_ENABLED=0
-    goos:
-      - linux
-    goarch:
-      - amd64
-      - arm64
-    ldflags:
-      - -s -w
-      - -X github.com/ccfos/nightingale/v6/pkg/version.Version={{ .Tag }}-{{.Commit}}
-  - id: build-edge
-    main: ./cmd/edge/
-    binary: n9e-edge
-    env:
-      - CGO_ENABLED=0
-    goos:
-      - linux
-    goarch:
-      - amd64
-      - arm64
-    ldflags:
-      - -s -w
-      - -X github.com/ccfos/nightingale/v6/pkg/version.Version={{ .Tag }}-{{.Commit}}
-
-archives:
-  - id: n9e
-    builds:
-      - build
-      - build-cli
-      - build-edge
-    format: tar.gz
-    format_overrides:
-      - goos: windows
-        format: zip
-    name_template: "n9e-v{{ .Version }}-{{ .Os }}-{{ .Arch }}"
-    wrap_in_directory: false
-    files:
-      - docker/*
-      - etc/*
-      - integrations/*
-      - cli/*
-      - n9e.sql
-
-release:
-  github:
-    owner: ccfos
-    name: nightingale
-  name_template: "v{{ .Version }}"
-
-dockers:
-  - image_templates:
-      - flashcatcloud/nightingale:{{ .Version }}-amd64
-    goos: linux
-    goarch: amd64
-    ids:
-      - build
-    dockerfile: docker/Dockerfile.goreleaser
-    extra_files:
-      - etc
-      - integrations
-    use: buildx
-    build_flag_templates:
-      - "--platform=linux/amd64"
-  - image_templates:
-      - flashcatcloud/nightingale:{{ .Version }}-arm64v8
-    goos: linux
-    goarch: arm64
-    ids:
-      - build
-    dockerfile: docker/Dockerfile.goreleaser.arm64
-    extra_files:
-      - etc
-      - integrations
-    use: buildx
-    build_flag_templates:
-      - "--platform=linux/arm64/v8"
-
-docker_manifests:
-
-  - name_template: flashcatcloud/nightingale:{{ .Version }}
-    image_templates:
-      - flashcatcloud/nightingale:{{ .Version }}-amd64
-      - flashcatcloud/nightingale:{{ .Version }}-arm64v8
-
-  - name_template: flashcatcloud/nightingale:latest
-    image_templates:
-      - flashcatcloud/nightingale:{{ .Version }}-amd64
-      - flashcatcloud/nightingale:{{ .Version }}-arm64v8

LICENSE

@@ -1,201 +1,433 @@
-                                 Apache License
-                           Version 2.0, January 2004
-                        http://www.apache.org/licenses/
-
-   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
-
-   1. Definitions.
-
-      "License" shall mean the terms and conditions for use, reproduction,
-      and distribution as defined by Sections 1 through 9 of this document.
-
-      "Licensor" shall mean the copyright owner or entity authorized by
-      the copyright owner that is granting the License.
-
-      "Legal Entity" shall mean the union of the acting entity and all
-      other entities that control, are controlled by, or are under common
-      control with that entity. For the purposes of this definition,
-      "control" means (i) the power, direct or indirect, to cause the
-      direction or management of such entity, whether by contract or
-      otherwise, or (ii) ownership of fifty percent (50%) or more of the
-      outstanding shares, or (iii) beneficial ownership of such entity.
-
-      "You" (or "Your") shall mean an individual or Legal Entity
-      exercising permissions granted by this License.
-
-      "Source" form shall mean the preferred form for making modifications,
-      including but not limited to software source code, documentation
-      source, and configuration files.
-
-      "Object" form shall mean any form resulting from mechanical
-      transformation or translation of a Source form, including but
-      not limited to compiled object code, generated documentation,
-      and conversions to other media types.
-
-      "Work" shall mean the work of authorship, whether in Source or
-      Object form, made available under the License, as indicated by a
-      copyright notice that is included in or attached to the work
-      (an example is provided in the Appendix below).
-
-      "Derivative Works" shall mean any work, whether in Source or Object
-      form, that is based on (or derived from) the Work and for which the
-      editorial revisions, annotations, elaborations, or other modifications
-      represent, as a whole, an original work of authorship. For the purposes
-      of this License, Derivative Works shall not include works that remain
-      separable from, or merely link (or bind by name) to the interfaces of,
-      the Work and Derivative Works thereof.
-
-      "Contribution" shall mean any work of authorship, including
-      the original version of the Work and any modifications or additions
-      to that Work or Derivative Works thereof, that is intentionally
-      submitted to Licensor for inclusion in the Work by the copyright owner
-      or by an individual or Legal Entity authorized to submit on behalf of
-      the copyright owner. For the purposes of this definition, "submitted"
-      means any form of electronic, verbal, or written communication sent
-      to the Licensor or its representatives, including but not limited to
-      communication on electronic mailing lists, source code control systems,
-      and issue tracking systems that are managed by, or on behalf of, the
-      Licensor for the purpose of discussing and improving the Work, but
-      excluding communication that is conspicuously marked or otherwise
-      designated in writing by the copyright owner as "Not a Contribution."
-
-      "Contributor" shall mean Licensor and any individual or Legal Entity
-      on behalf of whom a Contribution has been received by Licensor and
-      subsequently incorporated within the Work.
-
-   2. Grant of Copyright License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      copyright license to reproduce, prepare Derivative Works of,
-      publicly display, publicly perform, sublicense, and distribute the
-      Work and such Derivative Works in Source or Object form.
-
-   3. Grant of Patent License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      (except as stated in this section) patent license to make, have made,
-      use, offer to sell, sell, import, and otherwise transfer the Work,
-      where such license applies only to those patent claims licensable
-      by such Contributor that are necessarily infringed by their
-      Contribution(s) alone or by combination of their Contribution(s)
-      with the Work to which such Contribution(s) was submitted. If You
-      institute patent litigation against any entity (including a
-      cross-claim or counterclaim in a lawsuit) alleging that the Work
-      or a Contribution incorporated within the Work constitutes direct
-      or contributory patent infringement, then any patent licenses
-      granted to You under this License for that Work shall terminate
-      as of the date such litigation is filed.
-
-   4. Redistribution. You may reproduce and distribute copies of the
-      Work or Derivative Works thereof in any medium, with or without
-      modifications, and in Source or Object form, provided that You
-      meet the following conditions:
-
-      (a) You must give any other recipients of the Work or
-          Derivative Works a copy of this License; and
-
-      (b) You must cause any modified files to carry prominent notices
-          stating that You changed the files; and
-
-      (c) You must retain, in the Source form of any Derivative Works
-          that You distribute, all copyright, patent, trademark, and
-          attribution notices from the Source form of the Work,
-          excluding those notices that do not pertain to any part of
-          the Derivative Works; and
-
-      (d) If the Work includes a "NOTICE" text file as part of its
-          distribution, then any Derivative Works that You distribute must
-          include a readable copy of the attribution notices contained
-          within such NOTICE file, excluding those notices that do not
-          pertain to any part of the Derivative Works, in at least one
-          of the following places: within a NOTICE text file distributed
-          as part of the Derivative Works; within the Source form or
-          documentation, if provided along with the Derivative Works; or,
-          within a display generated by the Derivative Works, if and
-          wherever such third-party notices normally appear. The contents
-          of the NOTICE file are for informational purposes only and
-          do not modify the License. You may add Your own attribution
-          notices within Derivative Works that You distribute, alongside
-          or as an addendum to the NOTICE text from the Work, provided
-          that such additional attribution notices cannot be construed
-          as modifying the License.
-
-      You may add Your own copyright statement to Your modifications and
-      may provide additional or different license terms and conditions
-      for use, reproduction, or distribution of Your modifications, or
-      for any such Derivative Works as a whole, provided Your use,
-      reproduction, and distribution of the Work otherwise complies with
-      the conditions stated in this License.
-
-   5. Submission of Contributions. Unless You explicitly state otherwise,
-      any Contribution intentionally submitted for inclusion in the Work
-      by You to the Licensor shall be under the terms and conditions of
-      this License, without any additional terms or conditions.
-      Notwithstanding the above, nothing herein shall supersede or modify
-      the terms of any separate license agreement you may have executed
-      with Licensor regarding such Contributions.
-
-   6. Trademarks. This License does not grant permission to use the trade
-      names, trademarks, service marks, or product names of the Licensor,
-      except as required for reasonable and customary use in describing the
-      origin of the Work and reproducing the content of the NOTICE file.
-
-   7. Disclaimer of Warranty. Unless required by applicable law or
-      agreed to in writing, Licensor provides the Work (and each
-      Contributor provides its Contributions) on an "AS IS" BASIS,
-      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
-      implied, including, without limitation, any warranties or conditions
-      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
-      PARTICULAR PURPOSE. You are solely responsible for determining the
-      appropriateness of using or redistributing the Work and assume any
-      risks associated with Your exercise of permissions under this License.
-
-   8. Limitation of Liability. In no event and under no legal theory,
-      whether in tort (including negligence), contract, or otherwise,
-      unless required by applicable law (such as deliberate and grossly
-      negligent acts) or agreed to in writing, shall any Contributor be
-      liable to You for damages, including any direct, indirect, special,
-      incidental, or consequential damages of any character arising as a
-      result of this License or out of the use or inability to use the
-      Work (including but not limited to damages for loss of goodwill,
-      work stoppage, computer failure or malfunction, or any and all
-      other commercial damages or losses), even if such Contributor
-      has been advised of the possibility of such damages.
-
-   9. Accepting Warranty or Additional Liability. While redistributing
-      the Work or Derivative Works thereof, You may choose to offer,
-      and charge a fee for, acceptance of support, warranty, indemnity,
-      or other liability obligations and/or rights consistent with this
-      License. However, in accepting such obligations, You may act only
-      on Your own behalf and on Your sole responsibility, not on behalf
-      of any other Contributor, and only if You agree to indemnify,
-      defend, and hold each Contributor harmless for any liability
-      incurred by, or claims asserted against, such Contributor by reason
-      of your accepting any such warranty or additional liability.
-
-   END OF TERMS AND CONDITIONS
-
-   APPENDIX: How to apply the Apache License to your work.
-
-      To apply the Apache License to your work, attach the following
-      boilerplate notice, with the fields enclosed by brackets "[]"
-      replaced with your own identifying information. (Don't include
-      the brackets!)  The text should be enclosed in the appropriate
-      comment syntax for the file format. We also recommend that a
-      file or class name and description of purpose be included on the
-      same "printed page" as the copyright notice for easier
-      identification within third-party archives.
-
-   Copyright CCF ODC.
-
-   Licensed under the Apache License, Version 2.0 (the "License");
-   you may not use this file except in compliance with the License.
-   You may obtain a copy of the License at
-
-       http://www.apache.org/licenses/LICENSE-2.0
-
-   Unless required by applicable law or agreed to in writing, software
-   distributed under the License is distributed on an "AS IS" BASIS,
-   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-   See the License for the specific language governing permissions and
-   limitations under the License.
+                              Apache License
+ 
+                        Version 2.0, January 2004
+ 
+                     http://www.apache.org/licenses/
+ 
+ 
+ 
+ 
+TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+ 
+ 
+ 
+ 
+1. Definitions.
+ 
+ 
+ 
+ 
+   "License" shall mean the terms and conditions for use, reproduction,
+ 
+   and distribution as defined by Sections 1 through 9 of this document.
+ 
+ 
+ 
+ 
+   "Licensor" shall mean the copyright owner or entity authorized by
+ 
+   the copyright owner that is granting the License.
+ 
+ 
+ 
+ 
+   "Legal Entity" shall mean the union of the acting entity and all
+ 
+   other entities that control, are controlled by, or are under common
+ 
+   control with that entity. For the purposes of this definition,
+ 
+   "control" means (i) the power, direct or indirect, to cause the
+ 
+   direction or management of such entity, whether by contract or
+ 
+   otherwise, or (ii) ownership of fifty percent (50%) or more of the
+ 
+   outstanding shares, or (iii) beneficial ownership of such entity.
+ 
+ 
+ 
+ 
+   "You" (or "Your") shall mean an individual or Legal Entity
+ 
+   exercising permissions granted by this License.
+ 
+ 
+ 
+ 
+   "Source" form shall mean the preferred form for making modifications,
+ 
+   including but not limited to software source code, documentation
+ 
+   source, and configuration files.
+ 
+ 
+ 
+ 
+   "Object" form shall mean any form resulting from mechanical
+ 
+   transformation or translation of a Source form, including but
+ 
+   not limited to compiled object code, generated documentation,
+ 
+   and conversions to other media types.
+ 
+ 
+ 
+ 
+   "Work" shall mean the work of authorship, whether in Source or
+ 
+   Object form, made available under the License, as indicated by a
+ 
+   copyright notice that is included in or attached to the work
+ 
+   (an example is provided in the Appendix below).
+ 
+ 
+ 
+ 
+   "Derivative Works" shall mean any work, whether in Source or Object
+ 
+   form, that is based on (or derived from) the Work and for which the
+ 
+   editorial revisions, annotations, elaborations, or other modifications
+ 
+   represent, as a whole, an original work of authorship. For the purposes
+ 
+   of this License, Derivative Works shall not include works that remain
+ 
+   separable from, or merely link (or bind by name) to the interfaces of,
+ 
+   the Work and Derivative Works thereof.
+ 
+ 
+ 
+ 
+   "Contribution" shall mean any work of authorship, including
+ 
+   the original version of the Work and any modifications or additions
+ 
+   to that Work or Derivative Works thereof, that is intentionally
+ 
+   submitted to Licensor for inclusion in the Work by the copyright owner
+ 
+   or by an individual or Legal Entity authorized to submit on behalf of
+ 
+   the copyright owner. For the purposes of this definition, "submitted"
+ 
+   means any form of electronic, verbal, or written communication sent
+ 
+   to the Licensor or its representatives, including but not limited to
+ 
+   communication on electronic mailing lists, source code control systems,
+ 
+   and issue tracking systems that are managed by, or on behalf of, the
+ 
+   Licensor for the purpose of discussing and improving the Work, but
+ 
+   excluding communication that is conspicuously marked or otherwise
+ 
+   designated in writing by the copyright owner as "Not a Contribution."
+ 
+ 
+ 
+ 
+   "Contributor" shall mean Licensor and any individual or Legal Entity
+ 
+   on behalf of whom a Contribution has been received by Licensor and
+ 
+   subsequently incorporated within the Work.
+ 
+ 
+ 
+ 
+2. Grant of Copyright License. Subject to the terms and conditions of
+ 
+   this License, each Contributor hereby grants to You a perpetual,
+ 
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ 
+   copyright license to reproduce, prepare Derivative Works of,
+ 
+   publicly display, publicly perform, sublicense, and distribute the
+ 
+   Work and such Derivative Works in Source or Object form.
+ 
+ 
+ 
+ 
+3. Grant of Patent License. Subject to the terms and conditions of
+ 
+   this License, each Contributor hereby grants to You a perpetual,
+ 
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ 
+   (except as stated in this section) patent license to make, have made,
+ 
+   use, offer to sell, sell, import, and otherwise transfer the Work,
+ 
+   where such license applies only to those patent claims licensable
+ 
+   by such Contributor that are necessarily infringed by their
+ 
+   Contribution(s) alone or by combination of their Contribution(s)
+ 
+   with the Work to which such Contribution(s) was submitted. If You
+ 
+   institute patent litigation against any entity (including a
+ 
+   cross-claim or counterclaim in a lawsuit) alleging that the Work
+ 
+   or a Contribution incorporated within the Work constitutes direct
+ 
+   or contributory patent infringement, then any patent licenses
+ 
+   granted to You under this License for that Work shall terminate
+ 
+   as of the date such litigation is filed.
+ 
+ 
+ 
+ 
+4. Redistribution. You may reproduce and distribute copies of the
+ 
+   Work or Derivative Works thereof in any medium, with or without
+ 
+   modifications, and in Source or Object form, provided that You
+ 
+   meet the following conditions:
+ 
+ 
+ 
+ 
+   (a) You must give any other recipients of the Work or
+ 
+       Derivative Works a copy of this License; and
+ 
+ 
+ 
+ 
+   (b) You must cause any modified files to carry prominent notices
+ 
+       stating that You changed the files; and
+ 
+ 
+ 
+ 
+   (c) You must retain, in the Source form of any Derivative Works
+ 
+       that You distribute, all copyright, patent, trademark, and
+ 
+       attribution notices from the Source form of the Work,
+ 
+       excluding those notices that do not pertain to any part of
+ 
+       the Derivative Works; and
+ 
+ 
+ 
+ 
+   (d) If the Work includes a "NOTICE" text file as part of its
+ 
+       distribution, then any Derivative Works that You distribute must
+ 
+       include a readable copy of the attribution notices contained
+ 
+       within such NOTICE file, excluding those notices that do not
+ 
+       pertain to any part of the Derivative Works, in at least one
+ 
+       of the following places: within a NOTICE text file distributed
+ 
+       as part of the Derivative Works; within the Source form or
+ 
+       documentation, if provided along with the Derivative Works; or,
+ 
+       within a display generated by the Derivative Works, if and
+ 
+       wherever such third-party notices normally appear. The contents
+ 
+       of the NOTICE file are for informational purposes only and
+ 
+       do not modify the License. You may add Your own attribution
+ 
+       notices within Derivative Works that You distribute, alongside
+ 
+       or as an addendum to the NOTICE text from the Work, provided
+ 
+       that such additional attribution notices cannot be construed
+ 
+       as modifying the License.
+ 
+ 
+ 
+ 
+   You may add Your own copyright statement to Your modifications and
+ 
+   may provide additional or different license terms and conditions
+ 
+   for use, reproduction, or distribution of Your modifications, or
+ 
+   for any such Derivative Works as a whole, provided Your use,
+ 
+   reproduction, and distribution of the Work otherwise complies with
+ 
+   the conditions stated in this License.
+ 
+ 
+ 
+ 
+5. Submission of Contributions. Unless You explicitly state otherwise,
+ 
+   any Contribution intentionally submitted for inclusion in the Work
+ 
+   by You to the Licensor shall be under the terms and conditions of
+ 
+   this License, without any additional terms or conditions.
+ 
+   Notwithstanding the above, nothing herein shall supersede or modify
+ 
+   the terms of any separate license agreement you may have executed
+ 
+   with Licensor regarding such Contributions.
+ 
+ 
+ 
+ 
+6. Trademarks. This License does not grant permission to use the trade
+ 
+   names, trademarks, service marks, or product names of the Licensor,
+ 
+   except as required for reasonable and customary use in describing the
+ 
+   origin of the Work and reproducing the content of the NOTICE file.
+ 
+ 
+ 
+ 
+7. Disclaimer of Warranty. Unless required by applicable law or
+ 
+   agreed to in writing, Licensor provides the Work (and each
+ 
+   Contributor provides its Contributions) on an "AS IS" BASIS,
+ 
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ 
+   implied, including, without limitation, any warranties or conditions
+ 
+   of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+ 
+   PARTICULAR PURPOSE. You are solely responsible for determining the
+ 
+   appropriateness of using or redistributing the Work and assume any
+ 
+   risks associated with Your exercise of permissions under this License.
+ 
+ 
+ 
+ 
+8. Limitation of Liability. In no event and under no legal theory,
+ 
+   whether in tort (including negligence), contract, or otherwise,
+ 
+   unless required by applicable law (such as deliberate and grossly
+ 
+   negligent acts) or agreed to in writing, shall any Contributor be
+ 
+   liable to You for damages, including any direct, indirect, special,
+ 
+   incidental, or consequential damages of any character arising as a
+ 
+   result of this License or out of the use or inability to use the
+ 
+   Work (including but not limited to damages for loss of goodwill,
+ 
+   work stoppage, computer failure or malfunction, or any and all
+ 
+   other commercial damages or losses), even if such Contributor
+ 
+   has been advised of the possibility of such damages.
+ 
+ 
+ 
+ 
+9. Accepting Warranty or Additional Liability. While redistributing
+ 
+   the Work or Derivative Works thereof, You may choose to offer,
+ 
+   and charge a fee for, acceptance of support, warranty, indemnity,
+ 
+   or other liability obligations and/or rights consistent with this
+ 
+   License. However, in accepting such obligations, You may act only
+ 
+   on Your own behalf and on Your sole responsibility, not on behalf
+ 
+   of any other Contributor, and only if You agree to indemnify,
+ 
+   defend, and hold each Contributor harmless for any liability
+ 
+   incurred by, or claims asserted against, such Contributor by reason
+ 
+   of your accepting any such warranty or additional liability.
+ 
+ 
+ 
+ 
+END OF TERMS AND CONDITIONS
+ 
+ 
+ 
+ 
+APPENDIX: How to apply the Apache License to your work.
+ 
+ 
+ 
+ 
+   To apply the Apache License to your work, attach the following
+ 
+   boilerplate notice, with the fields enclosed by brackets "{}"
+ 
+   replaced with your own identifying information. (Don't include
+ 
+   the brackets!)  The text should be enclosed in the appropriate
+ 
+   comment syntax for the file format. We also recommend that a
+ 
+   file or class name and description of purpose be included on the
+ 
+   same "printed page" as the copyright notice for easier
+ 
+   identification within third-party archives.
+ 
+ 
+ 
+ 
+Copyright (C) 2017 Beijing Didi Infinity Technology and Development Co.,Ltd. All rights reserved.
+ 
+ 
+ 
+ 
+Licensed under the Apache License, Version 2.0 (the "License");
+ 
+you may not use this file except in compliance with the License.
+ 
+You may obtain a copy of the License at
+ 
+ 
+ 
+ 
+    http://www.apache.org/licenses/LICENSE-2.0
+ 
+ 
+ 
+ 
+Unless required by applicable law or agreed to in writing, software
+ 
+distributed under the License is distributed on an "AS IS" BASIS,
+ 
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ 
+See the License for the specific language governing permissions and
+ 
+limitations under the License.

Makefile

@@ -1,41 +1,42 @@
-.PHONY: prebuild build
+.PHONY: start build
 
-ROOT:=$(shell pwd -P)
-GIT_COMMIT:=$(shell git --work-tree ${ROOT}  rev-parse 'HEAD^{commit}')
-_GIT_VERSION:=$(shell git --work-tree ${ROOT} describe --tags --abbrev=14 "${GIT_COMMIT}^{commit}" 2>/dev/null)
-TAG=$(shell echo "${_GIT_VERSION}" |  awk -F"-" '{print $$1}')
-RELEASE_VERSION:="$(TAG)-$(GIT_COMMIT)"
+NOW = $(shell date -u '+%Y%m%d%I%M%S')
 
-all: prebuild build
+RELEASE_VERSION = 5.5.0
 
-prebuild:
-	echo "begin download and embed the front-end file..."
-	sh fe.sh
-	echo "front-end file download and embedding completed."
+APP 			= n9e
+SERVER_BIN  	= $(APP)
+# RELEASE_ROOT 	= release
+# RELEASE_SERVER 	= release/${APP}
+# GIT_COUNT 		= $(shell git rev-list --all --count)
+# GIT_HASH        = $(shell git rev-parse --short HEAD)
+# RELEASE_TAG     = $(RELEASE_VERSION).$(GIT_COUNT).$(GIT_HASH)
+
+all: build
 
 build:
-	go build -ldflags "-w -s -X github.com/ccfos/nightingale/v6/pkg/version.Version=$(RELEASE_VERSION)" -o n9e ./cmd/center/main.go
+	go build -ldflags "-w -s -X main.VERSION=$(RELEASE_VERSION)" -o $(SERVER_BIN) ./src
 
-build-edge:
-	go build -ldflags "-w -s -X github.com/ccfos/nightingale/v6/pkg/version.Version=$(RELEASE_VERSION)" -o n9e-edge ./cmd/edge/
+# start:
+# 	@go run -ldflags "-X main.VERSION=$(RELEASE_TAG)" ./cmd/${APP}/main.go web -c ./configs/config.toml -m ./configs/model.conf --menu ./configs/menu.yaml
+run_webapi:
+	nohup ./n9e webapi > webapi.log 2>&1 &
 
-build-alert:
-	go build -ldflags "-w -s -X github.com/ccfos/nightingale/v6/pkg/version.Version=$(RELEASE_VERSION)" -o n9e-alert ./cmd/alert/main.go
+run_server:
+	nohup ./n9e server > server.log 2>&1 &
 
-build-pushgw:
-	go build -ldflags "-w -s -X github.com/ccfos/nightingale/v6/pkg/version.Version=$(RELEASE_VERSION)" -o n9e-pushgw ./cmd/pushgw/main.go
+# swagger:
+# 	@swag init --parseDependency --generalInfo ./cmd/${APP}/main.go --output ./internal/app/swagger
 
-build-cli: 
-	go build -ldflags "-w -s -X github.com/ccfos/nightingale/v6/pkg/version.Version=$(RELEASE_VERSION)" -o n9e-cli ./cmd/cli/main.go
+# wire:
+# 	@wire gen ./internal/app
 
-run:
-	nohup ./n9e > n9e.log 2>&1 &
+# test:
+# 	cd ./internal/app/test && go test -v
 
-run-alert:
-	nohup ./n9e-alert > n9e-alert.log 2>&1 &
+# clean:
+# 	rm -rf data release $(SERVER_BIN) internal/app/test/data cmd/${APP}/data
 
-run-pushgw:
-	nohup ./n9e-pushgw > n9e-pushgw.log 2>&1 &
-
-release:
-	goreleaser --skip-validate --skip-publish --snapshot
+pack: build
+	rm -rf $(APP)-$(RELEASE_VERSION).tar.gz
+	tar -zcvf $(APP)-$(RELEASE_VERSION).tar.gz docker etc $(SERVER_BIN) pub/font pub/index.html pub/assets pub/image

README.md

@@ -1,118 +1,84 @@
-<p align="center">
-  <a href="https://github.com/ccfos/nightingale">
-    <img src="doc/img/Nightingale_L_V.png" alt="nightingale - cloud native monitoring" width="100" /></a>
-</p>
-<p align="center">
-  <b>开源告警管理专家</b>
-</p>
+## 介绍
 
-<p align="center">
-<a href="https://flashcat.cloud/docs/">
-  <img alt="Docs" src="https://img.shields.io/badge/docs-get%20started-brightgreen"/></a>
-<a href="https://hub.docker.com/u/flashcatcloud">
-  <img alt="Docker pulls" src="https://img.shields.io/docker/pulls/flashcatcloud/nightingale"/></a>
-<a href="https://github.com/ccfos/nightingale/graphs/contributors">
-  <img alt="GitHub contributors" src="https://img.shields.io/github/contributors-anon/ccfos/nightingale"/></a>
-<img alt="GitHub Repo stars" src="https://img.shields.io/github/stars/ccfos/nightingale">
-<img alt="GitHub forks" src="https://img.shields.io/github/forks/ccfos/nightingale">
-<br/><img alt="GitHub Repo issues" src="https://img.shields.io/github/issues/ccfos/nightingale">
-<img alt="GitHub Repo issues closed" src="https://img.shields.io/github/issues-closed/ccfos/nightingale">
-<img alt="GitHub latest release" src="https://img.shields.io/github/v/release/ccfos/nightingale"/>
-<img alt="License" src="https://img.shields.io/badge/license-Apache--2.0-blue"/>
-<a href="https://n9e-talk.slack.com/">
-  <img alt="GitHub contributors" src="https://img.shields.io/badge/join%20slack-%23n9e-brightgreen.svg"/></a>
-</p>
+> Nightingale is an enterprise-level cloud-native monitoring system, which can be used as drop-in replacement of Prometheus for alerting and management.
+>
+>夜莺是一款开源的云原生监控系统，采用 All-In-One 的设计，提供企业级的功能特性，开箱即用的产品体验。推荐升级您的 `Prometheus` + `AlertManager` + `Grafana` 组合方案到夜莺。
+
+- 内置丰富的Dashboard、好用实用的告警管理、自定义视图、故障自愈；
+- Dashboard和告警策略支持一键导入，详细的指标分类和解释；
+- 支持多 Prometheus 数据源管理，以一个集中的视图来管理所有的告警和dashboard；
+- 支持 Prometheus、M3DB、VictoriaMetrics、Influxdb、TDEngine 等多种时序库作为存储方案；
+- 原生支持 PromQL；
+- 支持 Exporter 作为数据采集方案；
+- 支持 Telegraf 作为监控数据采集方案；
+- 支持对接 Grafana 作为补充可视化方案；
 
 
+#### 如果您在使用 Prometheus 过程中，有以下的一个或者多个需求场景，推荐您升级到夜莺：
 
-[English](./README_en.md) | [中文](./README.md)
+- Prometheus、Alertmanager、Grafana 等多个系统较为割裂，缺乏统一视图，无法开箱即用;
+- 通过修改配置文件来管理 Prometheus、Alertmanager 的方式，学习曲线大，协同有难度;
+- 数据量过大而无法扩展您的 Prometheus 集群；
+- 生产环境运行多套 Prometheus 集群，面临管理和使用成本高的问题；
 
-## 夜莺是什么
+#### 如果您在使用Zabbix，有以下的场景，推荐您升级到夜莺：
 
-夜莺监控（Nightingale）是一款侧重告警的监控类开源项目。类似 Grafana 的数据源集成方式，夜莺也是对接多种既有的数据源，不过 Grafana 侧重在可视化，夜莺是侧重在告警引擎、告警事件的处理和分发。
-
-夜莺监控项目，最初由滴滴开发和开源，并于 2022 年 5 月 11 日，捐赠予中国计算机学会开源发展委员会（CCF ODC），为 CCF ODC 成立后接受捐赠的第一个开源项目。
-
-## 夜莺的工作逻辑
-
-很多用户已经自行采集了指标、日志数据，此时就把存储库（VictoriaMetrics、ElasticSearch等）作为数据源接入夜莺，即可在夜莺里配置告警规则、通知规则，完成告警事件的生成和派发。
-
-![夜莺产品架构](doc/img/readme/20240221152601.png)
-
-夜莺项目本身不提供监控数据采集能力。推荐您使用 [Categraf](https://github.com/flashcatcloud/categraf) 作为采集器，可以和夜莺丝滑对接。
-
-[Categraf](https://github.com/flashcatcloud/categraf) 可以采集操作系统、网络设备、各类中间件、数据库的监控数据，通过 Remote Write 协议推送给夜莺，夜莺把监控数据转存到时序库（如 Prometheus、VictoriaMetrics 等），并提供告警和可视化能力。
-
-对于个别边缘机房，如果和中心夜莺服务端网络链路不好，希望提升告警可用性，夜莺也提供边缘机房告警引擎下沉部署模式，这个模式下，即便边缘和中心端网络割裂，告警功能也不受影响。
-
-![边缘部署模式](doc/img/readme/20240222102119.png)
-
-> 上图中，机房A和中心机房的网络链路很好，所以直接由中心端的夜莺进程做告警引擎，机房B和中心机房的网络链路不好，所以在机房B部署了 `n9e-edge` 做告警引擎，对机房B的数据源做告警判定。
-
-## 告警降噪、升级、协同
-
-夜莺的侧重点是做告警引擎，即负责产生告警事件，并根据规则做灵活派发，内置支持 20 种通知媒介（电话、短信、邮件、钉钉、飞书、企微、Slack 等）。
-
-如果您有更高级的需求，比如：
-
-- 想要把公司的多套监控系统产生的事件聚拢到一个平台，统一做收敛降噪、响应处理、数据分析
-- 想要支持人员的排班，践行 On-call 文化，想要支持告警认领、升级（避免遗漏）、协同处理
-
-那夜莺是不合适的，推荐您选用 [FlashDuty](https://flashcat.cloud/product/flashcat-duty/) 这样的 On-call 产品，产品简单易用，也有免费套餐。
+- 监控的数据量太大，希望有更好的扩展解决方案；
+- 学习曲线高，多人多团队模式下，希望有更好的协同使用效率；
+- 微服务和云原生架构下，监控数据的生命周期多变、监控数据维度基数高，Zabbix数据模型不易适配；
 
 
-## 相关资料 & 交流渠道
-- 📚 [夜莺介绍PPT](https://mp.weixin.qq.com/s/Mkwx_46xrltSq8NLqAIYow) 对您了解夜莺各项关键特性会有帮助（PPT链接在文末）
-- 👉 [文档中心](https://flashcat.cloud/docs/) 为了更快的访问速度，站点托管在 [FlashcatCloud](https://flashcat.cloud)
-- ❤️ [报告 Bug](https://github.com/ccfos/nightingale/issues/new?assignees=&labels=&projects=&template=question.yml) 写清楚问题描述、复现步骤、截图等信息，更容易得到答案
-- 💡 前后端代码分离，前端代码仓库：[https://github.com/n9e/fe](https://github.com/n9e/fe)
-- 🎯 关注[这个公众号](https://gitlink.org.cn/UlricQin)了解更多夜莺动态和知识
-- 🌟 加我微信：`picobyte`（我已关闭好友验证）拉入微信群，备注：`夜莺互助群`，如果已经把夜莺上到生产环境，可联系我拉入资深监控用户群
+#### 如果您在使用[open-falcon](https://github.com/open-falcon/falcon-plus)，我们更推荐您升级到夜莺：
+- 关于open-falcon和夜莺的详细介绍，请参考阅读[云原生监控的十个特点和趋势](https://mp.weixin.qq.com/s?__biz=MzkzNjI5OTM5Nw==&mid=2247483738&idx=1&sn=e8bdbb974a2cd003c1abcc2b5405dd18&chksm=c2a19fb0f5d616a63185cd79277a79a6b80118ef2185890d0683d2bb20451bd9303c78d083c5#rd)。
+
+## 快速安装部署
+- [n9e.github.io/quickstart](https://n9e.github.io/quickstart/)
+
+## 详细文档
+- [n9e.github.io](https://n9e.github.io/)
+
+## 产品演示
+#### 您可以直接导入并生成 MySQL 相关的告警策略：
+<img src="doc/img/mysql-alerts.png" width="680">
+
+#### 您可以直接导入并生成主机相关的 dashboard：
+<img src="doc/img/n9e-node-dashboard.png" width="680">
+
+#### 您也可以在夜莺中方便的查看所有活跃的告警以及历史告警：
+<img src="https://n9e.github.io/intro/alert-events.png" width="680">
+
+## 系统架构
+#### 一个典型的 Nightingale 部署架构:
+<img src="https://n9e.github.io/intro/arch-system.png" width="680">
+
+#### 使用 VictoriaMetrics 作为时序数据库的典型部署架构:
+<img src="https://n9e.github.io/fc-monitoring-vm.png" width="680">
 
 
-## 关键特性简介
+## 联系我们和反馈问题
+- 我们推荐您优先使用[github issue](https://github.com/didi/nightingale/issues)作为首选问题反馈和需求提交的通道；
+- 您可以加入我们的微信群组——[Nightingale 微信群组](https://s3-gz01.didistatic.com/n9e-pub/image/n9e-wx.png)；
+- 当然，推荐您关注夜莺监控公众号，及时获取相关产品动态 
 
-![夜莺告警规则](doc/img/readme/2025-05-23_18-43-37.png)
-
-- 夜莺支持告警规则、屏蔽规则、订阅规则、通知规则，内置支持 20 种通知媒介，支持消息模板自定义
-- 支持事件管道，对告警事件做 Pipeline 处理，方便和自有系统做自动化整合，比如给告警事件附加一些元信息，对事件做 relabel
-- 支持业务组概念，引入权限体系，分门别类管理各类规则
-- 很多数据库、中间件内置了告警规则，可以直接导入使用，也可以直接导入 Prometheus 的告警规则
-- 支持告警自愈，即告警之后自动触发一个脚本执行一些预定义的逻辑，比如清理一下磁盘、抓一下现场等
-
-![夜莺事件大盘](doc/img/readme/2025-05-30_08-49-28.png)
-
-- 夜莺存档了历史告警事件，支持多维度的查询和统计
-- 支持灵活的聚合分组，一目了然看到公司的告警事件分布情况
-
-![夜莺集成中心](doc/img/readme/2025-05-23_18-46-06.png)
-
-- 夜莺内置常用操作系统、中间件、数据库的的指标说明、仪表盘、告警规则，不过都是社区贡献的，整体也是参差不齐
-- 夜莺直接接收 Remote Write、OpenTSDB、Datadog、Falcon 等多种协议的数据，故而可以和各类 Agent 对接
-- 夜莺支持 Prometheus、ElasticSearch、Loki、TDEngine 等多种数据源，可以对其中的数据做告警
-- 夜莺可以很方便内嵌企业内部系统，比如 Grafana、CMDB 等，甚至可以配置这些内嵌系统的菜单可见性
+<img src="https://n9e.github.io/cloudmon.png" width="180">
 
 
-![夜莺仪表盘](doc/img/readme/2025-05-23_18-49-02.png)
+## 参与到夜莺开源项目和社区
+我们欢迎您以各种方式参与到夜莺开源项目和开源社区中来，工作包括不限于：
+- 反馈使用中遇到的问题和Bug => [github issue](https://github.com/didi/nightingale/issues)
+- 补充和完善文档 => [n9e.github.io](https://n9e.github.io/)
+- 分享您在使用夜莺监控过程中的最佳实践和经验心得 => [夜莺User Story](https://github.com/didi/nightingale/issues/897)
+- 参与我们的社区活动 => [Nightingale 微信群组](https://s3-gz01.didistatic.com/n9e-pub/image/n9e-wx.png)
+- 提交代码，让夜莺监控更快、更稳、更好用 =>[github PR](https://github.com/didi/nightingale/pulls)
 
-- 夜莺支持仪表盘功能，支持常见的图表类型，也内置了一些仪表盘，上图是其中一个仪表盘的截图。
-- 如果你已经习惯了 Grafana，建议仍然使用 Grafana 看图。Grafana 在看图方面道行更深。
-- 机器相关的监控数据，如果是 Categraf 采集的，建议使用夜莺自带的仪表盘查看，因为 Categraf 的指标命名 Follow 的是 Telegraf 的命名方式，和 Node Exporter 不同
-- 因为夜莺有个业务组的概念，机器可以归属不同的业务组，有时在仪表盘里只想查看当前所属业务组的机器，所以夜莺的仪表盘可以和业务组联动
-
-## 广受关注
-[![Stargazers over time](https://api.star-history.com/svg?repos=ccfos/nightingale&type=Date)](https://star-history.com/#ccfos/nightingale&Date)
-
-## 感谢众多企业的信赖
-
-![夜莺客户](doc/img/readme/logos.png)
-
-## 社区共建
-- ❇️ 请阅读浏览[夜莺开源项目和社区治理架构草案](./doc/community-governance.md)，真诚欢迎每一位用户、开发者、公司以及组织，使用夜莺监控、积极反馈 Bug、提交功能需求、分享最佳实践，共建专业、活跃的夜莺开源社区。
-- ❤️ 夜莺贡献者
-<a href="https://github.com/ccfos/nightingale/graphs/contributors">
-  <img src="https://contrib.rocks/image?repo=ccfos/nightingale" />
-</a>
+## TODO
+- [x] deploy nightingale in docker
+- [x] export /metrics endpoint
+- [x] notify.py support feishu
+- [ ] notify.py support sms
+- [ ] notify.py support voice
+- [x] support remote write api
+- [ ] support pushgateway api
 
 ## License
-- [Apache License V2.0](https://github.com/didi/nightingale/blob/main/LICENSE)
+夜莺监控，采用[Apache License V2.0](https://github.com/didi/nightingale/blob/main/LICENSE)开源许可证。

README_en.md

@@ -1,113 +0,0 @@
-<p align="center">
-  <a href="https://github.com/ccfos/nightingale">
-    <img src="doc/img/Nightingale_L_V.png" alt="nightingale - cloud native monitoring" width="100" /></a>
-</p>
-<p align="center">
-  <b>Open-source Alert Management Expert, an Integrated Observability Platform</b>
-</p>
-
-<p align="center">
-<a href="https://flashcat.cloud/docs/">
-  <img alt="Docs" src="https://img.shields.io/badge/docs-get%20started-brightgreen"/></a>
-<a href="https://hub.docker.com/u/flashcatcloud">
-  <img alt="Docker pulls" src="https://img.shields.io/docker/pulls/flashcatcloud/nightingale"/></a>
-<a href="https://github.com/ccfos/nightingale/graphs/contributors">
-  <img alt="GitHub contributors" src="https://img.shields.io/github/contributors-anon/ccfos/nightingale"/></a>
-<img alt="GitHub Repo stars" src="https://img.shields.io/github/stars/ccfos/nightingale">
-<img alt="GitHub forks" src="https://img.shields.io/github/forks/ccfos/nightingale">
-<br/><img alt="GitHub Repo issues" src="https://img.shields.io/github/issues/ccfos/nightingale">
-<img alt="GitHub Repo issues closed" src="https://img.shields.io/github/issues-closed/ccfos/nightingale">
-<img alt="GitHub latest release" src="https://img.shields.io/github/v/release/ccfos/nightingale"/>
-<img alt="License" src="https://img.shields.io/badge/license-Apache--2.0-blue"/>
-<a href="https://n9e-talk.slack.com/">
-  <img alt="GitHub contributors" src="https://img.shields.io/badge/join%20slack-%23n9e-brightgreen.svg"/></a>
-</p>
-
-
-
-[English](./README_en.md) | [中文](./README.md)
-
-## What is Nightingale
-
-Nightingale is an open-source project focused on alerting. Similar to Grafana's data source integration approach, Nightingale also connects with various existing data sources. However, while Grafana focuses on visualization, Nightingale focuses on alerting engines.
-
-Originally developed and open-sourced by Didi, Nightingale was donated to the China Computer Federation Open Source Development Committee (CCF ODC) on May 11, 2022, becoming the first open-source project accepted by the CCF ODC after its establishment. 
-
-
-## Quick Start
-
-- 👉 [Documentation](https://flashcat.cloud/docs/) | [Download](https://flashcat.cloud/download/nightingale/)
-- ❤️ [Report a Bug](https://github.com/ccfos/nightingale/issues/new?assignees=&labels=&projects=&template=question.yml)
-- ℹ️ For faster access, the above documentation and download sites are hosted on [FlashcatCloud](https://flashcat.cloud).
-
-## Features
-
-- **Integration with Multiple Time-Series Databases:** Supports integration with various time-series databases such as Prometheus, VictoriaMetrics, Thanos, Mimir, M3DB, and TDengine, enabling unified alert management.
-- **Advanced Alerting Capabilities:** Comes with built-in support for multiple alerting rules, extensible to common notification channels. It also supports alert suppression, silencing, subscription, self-healing, and alert event management.
-- **High-Performance Visualization Engine:** Offers various chart styles with numerous built-in dashboard templates and the ability to import Grafana templates. Ready to use with a business-friendly open-source license.
-- **Support for Common Collectors:** Compatible with [Categraf](https://flashcat.cloud/product/categraf), Telegraf, Grafana-agent, Datadog-agent, and various exporters as collectors—there's no data that can't be monitored.
-- **Seamless Integration with [Flashduty](https://flashcat.cloud/product/flashcat-duty/):** Enables alert aggregation, acknowledgment, escalation, scheduling, and IM integration, ensuring no alerts are missed, reducing unnecessary interruptions, and enhancing efficient collaboration.
-
-
-## Screenshots
-
-You can switch languages and themes in the top right corner. We now support English, Simplified Chinese, and Traditional Chinese. 
-
-![18n switch](doc/img/readme/n9e-switch-i18n.png)
-
-### Instant Query
-
-Similar to the built-in query analysis page in Prometheus, Nightingale offers an ad-hoc query feature with UI enhancements. It also provides built-in PromQL metrics, allowing users unfamiliar with PromQL to quickly perform queries.
-
-![Instant Query](doc/img/readme/20240513103305.png)
-
-### Metric View
-
-Alternatively, you can use the Metric View to access data. With this feature, Instant Query becomes less necessary, as it caters more to advanced users. Regular users can easily perform queries using the Metric View.
-
-![Metric View](doc/img/readme/20240513103530.png)
-
-### Built-in Dashboards
-
-Nightingale includes commonly used dashboards that can be imported and used directly. You can also import Grafana dashboards, although compatibility is limited to basic Grafana charts. If you’re accustomed to Grafana, it’s recommended to continue using it for visualization, with Nightingale serving as an alerting engine.
-
-![Built-in Dashboards](doc/img/readme/20240513103628.png)
-
-### Built-in Alert Rules
-
-In addition to the built-in dashboards, Nightingale also comes with numerous alert rules that are ready to use out of the box.
-
-![Built-in Alert Rules](doc/img/readme/20240513103825.png)
-
-
-
-## Architecture
-
-In most community scenarios, Nightingale is primarily used as an alert engine, integrating with multiple time-series databases to unify alert rule management. Grafana remains the preferred tool for visualization. As an alert engine, the product architecture of Nightingale is as follows:
-
-![Product Architecture](doc/img/readme/20240221152601.png)
-
-For certain edge data centers with poor network connectivity to the central Nightingale server, we offer a distributed deployment mode for the alert engine. In this mode, even if the network is disconnected, the alerting functionality remains unaffected.
-
-![Edge Deployment Mode](doc/img/readme/20240222102119.png)
-
-
-## Communication Channels
-
-- **Report Bugs:** It is highly recommended to submit issues via the [Nightingale GitHub Issue tracker](https://github.com/ccfos/nightingale/issues/new?assignees=&labels=kind%2Fbug&projects=&template=bug_report.yml).
-- **Documentation:** For more information, we recommend thoroughly browsing the [Nightingale Documentation Site](https://flashcat.cloud/docs/content/flashcat-monitor/nightingale-v7/introduction/).
-
-## Stargazers over time
-
-[![Stargazers over time](https://api.star-history.com/svg?repos=ccfos/nightingale&type=Date)](https://star-history.com/#ccfos/nightingale&Date)
-
-## Community Co-Building
-
-- ❇️ Please read the [Nightingale Open Source Project and Community Governance Draft](./doc/community-governance.md). We sincerely welcome every user, developer, company, and organization to use Nightingale, actively report bugs, submit feature requests, share best practices, and help build a professional and active open-source community.
--  ❤️ Nightingale Contributors
-<a href="https://github.com/ccfos/nightingale/graphs/contributors">
-  <img src="https://contrib.rocks/image?repo=ccfos/nightingale" />
-</a>
-
-## License
-- [Apache License V2.0](https://github.com/didi/nightingale/blob/main/LICENSE)

alert/aconf/conf.go

@@ -1,66 +0,0 @@
-package aconf
-
-import (
-	"path"
-)
-
-type Alert struct {
-	Disable     bool
-	EngineDelay int64
-	Heartbeat   HeartbeatConfig
-	Alerting    Alerting
-}
-
-type SMTPConfig struct {
-	Host               string
-	Port               int
-	User               string
-	Pass               string
-	From               string
-	InsecureSkipVerify bool
-	Batch              int
-}
-
-type HeartbeatConfig struct {
-	IP         string
-	Interval   int64
-	Endpoint   string
-	EngineName string
-}
-
-type Alerting struct {
-	Timeout           int64
-	TemplatesDir      string
-	NotifyConcurrency int
-	WebhookBatchSend  bool
-}
-
-type CallPlugin struct {
-	Enable     bool
-	PluginPath string
-	Caller     string
-}
-
-type RedisPub struct {
-	Enable        bool
-	ChannelPrefix string
-	ChannelKey    string
-}
-
-func (a *Alert) PreCheck(configDir string) {
-	if a.Alerting.TemplatesDir == "" {
-		a.Alerting.TemplatesDir = path.Join(configDir, "template")
-	}
-
-	if a.Alerting.NotifyConcurrency == 0 {
-		a.Alerting.NotifyConcurrency = 10
-	}
-
-	if a.Heartbeat.Interval == 0 {
-		a.Heartbeat.Interval = 1000
-	}
-
-	if a.EngineDelay == 0 {
-		a.EngineDelay = 30
-	}
-}

alert/alert.go

@@ -1,132 +0,0 @@
-package alert
-
-import (
-	"context"
-	"fmt"
-
-	"github.com/ccfos/nightingale/v6/dscache"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/dispatch"
-	"github.com/ccfos/nightingale/v6/alert/eval"
-	"github.com/ccfos/nightingale/v6/alert/naming"
-	"github.com/ccfos/nightingale/v6/alert/process"
-	"github.com/ccfos/nightingale/v6/alert/queue"
-	"github.com/ccfos/nightingale/v6/alert/record"
-	"github.com/ccfos/nightingale/v6/alert/router"
-	"github.com/ccfos/nightingale/v6/alert/sender"
-	"github.com/ccfos/nightingale/v6/conf"
-	"github.com/ccfos/nightingale/v6/dumper"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/httpx"
-	"github.com/ccfos/nightingale/v6/pkg/logx"
-	"github.com/ccfos/nightingale/v6/pkg/macros"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/ccfos/nightingale/v6/pushgw/pconf"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-	"github.com/ccfos/nightingale/v6/storage"
-	"github.com/flashcatcloud/ibex/src/cmd/ibex"
-)
-
-func Initialize(configDir string, cryptoKey string) (func(), error) {
-	config, err := conf.InitConfig(configDir, cryptoKey)
-	if err != nil {
-		return nil, fmt.Errorf("failed to init config: %v", err)
-	}
-
-	logxClean, err := logx.Init(config.Log)
-	if err != nil {
-		return nil, err
-	}
-
-	ctx := ctx.NewContext(context.Background(), nil, false, config.CenterApi)
-
-	var redis storage.Redis
-	redis, err = storage.NewRedis(config.Redis)
-	if err != nil {
-		return nil, err
-	}
-
-	syncStats := memsto.NewSyncStats()
-	alertStats := astats.NewSyncStats()
-
-	configCache := memsto.NewConfigCache(ctx, syncStats, nil, "")
-	targetCache := memsto.NewTargetCache(ctx, syncStats, redis)
-	busiGroupCache := memsto.NewBusiGroupCache(ctx, syncStats)
-	alertMuteCache := memsto.NewAlertMuteCache(ctx, syncStats)
-	alertRuleCache := memsto.NewAlertRuleCache(ctx, syncStats)
-	notifyConfigCache := memsto.NewNotifyConfigCache(ctx, configCache)
-	dsCache := memsto.NewDatasourceCache(ctx, syncStats)
-	userCache := memsto.NewUserCache(ctx, syncStats)
-	userGroupCache := memsto.NewUserGroupCache(ctx, syncStats)
-	taskTplsCache := memsto.NewTaskTplCache(ctx)
-	configCvalCache := memsto.NewCvalCache(ctx, syncStats)
-	notifyRuleCache := memsto.NewNotifyRuleCache(ctx, syncStats)
-	notifyChannelCache := memsto.NewNotifyChannelCache(ctx, syncStats)
-	messageTemplateCache := memsto.NewMessageTemplateCache(ctx, syncStats)
-
-	promClients := prom.NewPromClient(ctx)
-	dispatch.InitRegisterQueryFunc(promClients)
-
-	externalProcessors := process.NewExternalProcessors()
-
-	macros.RegisterMacro(macros.MacroInVain)
-	dscache.Init(ctx, false)
-	Start(config.Alert, config.Pushgw, syncStats, alertStats, externalProcessors, targetCache, busiGroupCache, alertMuteCache, alertRuleCache, notifyConfigCache, taskTplsCache, dsCache, ctx, promClients, userCache, userGroupCache, notifyRuleCache, notifyChannelCache, messageTemplateCache)
-
-	r := httpx.GinEngine(config.Global.RunMode, config.HTTP,
-		configCvalCache.PrintBodyPaths, configCvalCache.PrintAccessLog)
-	rt := router.New(config.HTTP, config.Alert, alertMuteCache, targetCache, busiGroupCache, alertStats, ctx, externalProcessors)
-
-	if config.Ibex.Enable {
-		ibex.ServerStart(false, nil, redis, config.HTTP.APIForService.BasicAuth, config.Alert.Heartbeat, &config.CenterApi, r, nil, config.Ibex, config.HTTP.Port)
-	}
-
-	rt.Config(r)
-	dumper.ConfigRouter(r)
-
-	httpClean := httpx.Init(config.HTTP, r)
-
-	return func() {
-		logxClean()
-		httpClean()
-	}, nil
-}
-
-func Start(alertc aconf.Alert, pushgwc pconf.Pushgw, syncStats *memsto.Stats, alertStats *astats.Stats, externalProcessors *process.ExternalProcessorsType, targetCache *memsto.TargetCacheType, busiGroupCache *memsto.BusiGroupCacheType,
-	alertMuteCache *memsto.AlertMuteCacheType, alertRuleCache *memsto.AlertRuleCacheType, notifyConfigCache *memsto.NotifyConfigCacheType, taskTplsCache *memsto.TaskTplCache, datasourceCache *memsto.DatasourceCacheType, ctx *ctx.Context,
-	promClients *prom.PromClientMap, userCache *memsto.UserCacheType, userGroupCache *memsto.UserGroupCacheType, notifyRuleCache *memsto.NotifyRuleCacheType, notifyChannelCache *memsto.NotifyChannelCacheType, messageTemplateCache *memsto.MessageTemplateCacheType) {
-	alertSubscribeCache := memsto.NewAlertSubscribeCache(ctx, syncStats)
-	recordingRuleCache := memsto.NewRecordingRuleCache(ctx, syncStats)
-	targetsOfAlertRulesCache := memsto.NewTargetOfAlertRuleCache(ctx, alertc.Heartbeat.EngineName, syncStats)
-
-	go models.InitNotifyConfig(ctx, alertc.Alerting.TemplatesDir)
-	go models.InitNotifyChannel(ctx)
-	go models.InitMessageTemplate(ctx)
-
-	naming := naming.NewNaming(ctx, alertc.Heartbeat, alertStats)
-
-	writers := writer.NewWriters(pushgwc)
-	record.NewScheduler(alertc, recordingRuleCache, promClients, writers, alertStats, datasourceCache)
-
-	eval.NewScheduler(alertc, externalProcessors, alertRuleCache, targetCache, targetsOfAlertRulesCache,
-		busiGroupCache, alertMuteCache, datasourceCache, promClients, naming, ctx, alertStats)
-
-	eventProcessorCache := memsto.NewEventProcessorCache(ctx, syncStats)
-
-	dp := dispatch.NewDispatch(alertRuleCache, userCache, userGroupCache, alertSubscribeCache, targetCache, notifyConfigCache, taskTplsCache, notifyRuleCache, notifyChannelCache, messageTemplateCache, eventProcessorCache, alertc.Alerting, ctx, alertStats)
-	consumer := dispatch.NewConsumer(alertc.Alerting, ctx, dp, promClients)
-
-	notifyRecordComsumer := sender.NewNotifyRecordConsumer(ctx)
-
-	go dp.ReloadTpls()
-	go consumer.LoopConsume()
-	go notifyRecordComsumer.LoopConsume()
-
-	go queue.ReportQueueSize(alertStats)
-	go sender.ReportNotifyRecordQueueSize(alertStats)
-	go sender.InitEmailSender(ctx, notifyConfigCache)
-}

alert/astats/stats.go

@@ -1,193 +0,0 @@
-package astats
-
-import (
-	"github.com/prometheus/client_golang/prometheus"
-)
-
-const (
-	namespace = "n9e"
-	subsystem = "alert"
-)
-
-type Stats struct {
-	AlertNotifyTotal            *prometheus.CounterVec
-	AlertNotifyErrorTotal       *prometheus.CounterVec
-	CounterAlertsTotal          *prometheus.CounterVec
-	GaugeAlertQueueSize         prometheus.Gauge
-	CounterRuleEval             *prometheus.CounterVec
-	CounterQueryDataErrorTotal  *prometheus.CounterVec
-	CounterQueryDataTotal       *prometheus.CounterVec
-	CounterVarFillingQuery      *prometheus.CounterVec
-	CounterRecordEval           *prometheus.CounterVec
-	CounterRecordEvalErrorTotal *prometheus.CounterVec
-	CounterMuteTotal            *prometheus.CounterVec
-	CounterRuleEvalErrorTotal   *prometheus.CounterVec
-	CounterHeartbeatErrorTotal  *prometheus.CounterVec
-	CounterSubEventTotal        *prometheus.CounterVec
-	GaugeQuerySeriesCount       *prometheus.GaugeVec
-	GaugeRuleEvalDuration       *prometheus.GaugeVec
-	GaugeNotifyRecordQueueSize  prometheus.Gauge
-}
-
-func NewSyncStats() *Stats {
-	CounterRuleEval := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "rule_eval_total",
-		Help:      "Number of rule eval.",
-	}, []string{})
-
-	CounterRuleEvalErrorTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "rule_eval_error_total",
-		Help:      "Number of rule eval error.",
-	}, []string{"datasource", "stage", "busi_group", "rule_id"})
-
-	CounterQueryDataErrorTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "query_data_error_total",
-		Help:      "Number of rule eval query data error.",
-	}, []string{"datasource"})
-
-	CounterQueryDataTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "query_data_total",
-		Help:      "Number of rule eval query data.",
-	}, []string{"datasource", "rule_id"})
-
-	CounterRecordEval := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "record_eval_total",
-		Help:      "Number of record eval.",
-	}, []string{"datasource"})
-
-	CounterRecordEvalErrorTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "record_eval_error_total",
-		Help:      "Number of record eval error.",
-	}, []string{"datasource"})
-
-	AlertNotifyTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "alert_notify_total",
-		Help:      "Number of send msg.",
-	}, []string{"channel"})
-
-	AlertNotifyErrorTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "alert_notify_error_total",
-		Help:      "Number of send msg.",
-	}, []string{"channel"})
-
-	// 产生的告警总量
-	CounterAlertsTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "alerts_total",
-		Help:      "Total number alert events.",
-	}, []string{"cluster", "type", "busi_group"})
-
-	// 内存中的告警事件队列的长度
-	GaugeAlertQueueSize := prometheus.NewGauge(prometheus.GaugeOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "alert_queue_size",
-		Help:      "The size of alert queue.",
-	})
-
-	CounterMuteTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "mute_total",
-		Help:      "Number of mute.",
-	}, []string{"group", "rule_id", "mute_rule_id", "datasource_id"})
-
-	CounterSubEventTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "sub_event_total",
-		Help:      "Number of sub event.",
-	}, []string{"group"})
-
-	CounterHeartbeatErrorTotal := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "heartbeat_error_count",
-		Help:      "Number of heartbeat error.",
-	}, []string{})
-
-	GaugeQuerySeriesCount := prometheus.NewGaugeVec(prometheus.GaugeOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "eval_query_series_count",
-		Help:      "Number of series retrieved from data source after query.",
-	}, []string{"rule_id", "datasource_id", "ref"})
-	// 通知记录队列的长度
-	GaugeNotifyRecordQueueSize := prometheus.NewGauge(prometheus.GaugeOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "notify_record_queue_size",
-		Help:      "The size of notify record queue.",
-	})
-
-	GaugeRuleEvalDuration := prometheus.NewGaugeVec(prometheus.GaugeOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "rule_eval_duration_ms",
-		Help:      "Duration of rule eval in milliseconds.",
-	}, []string{"rule_id", "datasource_id"})
-
-	CounterVarFillingQuery := prometheus.NewCounterVec(prometheus.CounterOpts{
-		Namespace: namespace,
-		Subsystem: subsystem,
-		Name:      "var_filling_query_total",
-		Help:      "Number of var filling query.",
-	}, []string{"rule_id", "datasource_id", "ref", "typ"})
-
-	prometheus.MustRegister(
-		CounterAlertsTotal,
-		GaugeAlertQueueSize,
-		AlertNotifyTotal,
-		AlertNotifyErrorTotal,
-		CounterRuleEval,
-		CounterQueryDataTotal,
-		CounterQueryDataErrorTotal,
-		CounterRecordEval,
-		CounterRecordEvalErrorTotal,
-		CounterMuteTotal,
-		CounterRuleEvalErrorTotal,
-		CounterHeartbeatErrorTotal,
-		CounterSubEventTotal,
-		GaugeQuerySeriesCount,
-		GaugeRuleEvalDuration,
-		GaugeNotifyRecordQueueSize,
-		CounterVarFillingQuery,
-	)
-
-	return &Stats{
-		CounterAlertsTotal:          CounterAlertsTotal,
-		GaugeAlertQueueSize:         GaugeAlertQueueSize,
-		AlertNotifyTotal:            AlertNotifyTotal,
-		AlertNotifyErrorTotal:       AlertNotifyErrorTotal,
-		CounterRuleEval:             CounterRuleEval,
-		CounterQueryDataTotal:       CounterQueryDataTotal,
-		CounterQueryDataErrorTotal:  CounterQueryDataErrorTotal,
-		CounterRecordEval:           CounterRecordEval,
-		CounterRecordEvalErrorTotal: CounterRecordEvalErrorTotal,
-		CounterMuteTotal:            CounterMuteTotal,
-		CounterRuleEvalErrorTotal:   CounterRuleEvalErrorTotal,
-		CounterHeartbeatErrorTotal:  CounterHeartbeatErrorTotal,
-		CounterSubEventTotal:        CounterSubEventTotal,
-		GaugeQuerySeriesCount:       GaugeQuerySeriesCount,
-		GaugeRuleEvalDuration:       GaugeRuleEvalDuration,
-		GaugeNotifyRecordQueueSize:  GaugeNotifyRecordQueueSize,
-		CounterVarFillingQuery:      CounterVarFillingQuery,
-	}
-}

alert/common/key.go

@@ -1,54 +0,0 @@
-package common
-
-import (
-	"fmt"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-func RuleKey(datasourceId, id int64) string {
-	return fmt.Sprintf("alert-%d-%d", datasourceId, id)
-}
-
-func MatchTags(eventTagsMap map[string]string, itags []models.TagFilter) bool {
-	for _, filter := range itags {
-		value, has := eventTagsMap[filter.Key]
-		if !has {
-			return false
-		}
-		if !matchTag(value, filter) {
-			return false
-		}
-	}
-	return true
-}
-func MatchGroupsName(groupName string, groupFilter []models.TagFilter) bool {
-	for _, filter := range groupFilter {
-		if !matchTag(groupName, filter) {
-			return false
-		}
-	}
-	return true
-}
-
-func matchTag(value string, filter models.TagFilter) bool {
-	switch filter.Func {
-	case "==":
-		return strings.TrimSpace(filter.Value) == strings.TrimSpace(value)
-	case "!=":
-		return strings.TrimSpace(filter.Value) != strings.TrimSpace(value)
-	case "in":
-		_, has := filter.Vset[value]
-		return has
-	case "not in":
-		_, has := filter.Vset[value]
-		return !has
-	case "=~":
-		return filter.Regexp.MatchString(value)
-	case "!~":
-		return !filter.Regexp.MatchString(value)
-	}
-	// unexpect func
-	return false
-}

alert/dispatch/consume.go

@@ -1,206 +0,0 @@
-package dispatch
-
-import (
-	"context"
-	"encoding/json"
-	"fmt"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/common"
-	"github.com/ccfos/nightingale/v6/alert/queue"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/poster"
-	promsdk "github.com/ccfos/nightingale/v6/pkg/prom"
-	"github.com/ccfos/nightingale/v6/pkg/tplx"
-	"github.com/ccfos/nightingale/v6/prom"
-
-	"github.com/prometheus/common/model"
-	"github.com/toolkits/pkg/concurrent/semaphore"
-	"github.com/toolkits/pkg/logger"
-)
-
-type Consumer struct {
-	alerting aconf.Alerting
-	ctx      *ctx.Context
-
-	dispatch    *Dispatch
-	promClients *prom.PromClientMap
-}
-
-func InitRegisterQueryFunc(promClients *prom.PromClientMap) {
-	tplx.RegisterQueryFunc(func(datasourceID int64, promql string) model.Value {
-		if promClients.IsNil(datasourceID) {
-			return nil
-		}
-
-		readerClient := promClients.GetCli(datasourceID)
-		value, _, _ := readerClient.Query(context.Background(), promql, time.Now())
-		return value
-	})
-}
-
-// 创建一个 Consumer 实例
-func NewConsumer(alerting aconf.Alerting, ctx *ctx.Context, dispatch *Dispatch, promClients *prom.PromClientMap) *Consumer {
-	return &Consumer{
-		alerting:    alerting,
-		ctx:         ctx,
-		dispatch:    dispatch,
-		promClients: promClients,
-	}
-}
-
-func (e *Consumer) LoopConsume() {
-	sema := semaphore.NewSemaphore(e.alerting.NotifyConcurrency)
-	duration := time.Duration(100) * time.Millisecond
-	for {
-		events := queue.EventQueue.PopBackBy(100)
-		if len(events) == 0 {
-			time.Sleep(duration)
-			continue
-		}
-		e.consume(events, sema)
-	}
-}
-
-func (e *Consumer) consume(events []interface{}, sema *semaphore.Semaphore) {
-	for i := range events {
-		if events[i] == nil {
-			continue
-		}
-
-		event := events[i].(*models.AlertCurEvent)
-		sema.Acquire()
-		go func(event *models.AlertCurEvent) {
-			defer sema.Release()
-			e.consumeOne(event)
-		}(event)
-	}
-}
-
-func (e *Consumer) consumeOne(event *models.AlertCurEvent) {
-	LogEvent(event, "consume")
-
-	eventType := "alert"
-	if event.IsRecovered {
-		eventType = "recovery"
-	}
-
-	e.dispatch.Astats.CounterAlertsTotal.WithLabelValues(event.Cluster, eventType, event.GroupName).Inc()
-
-	if err := event.ParseRule("rule_name"); err != nil {
-		logger.Warningf("ruleid:%d failed to parse rule name: %v", event.RuleId, err)
-		event.RuleName = fmt.Sprintf("failed to parse rule name: %v", err)
-	}
-
-	if err := event.ParseRule("annotations"); err != nil {
-		logger.Warningf("ruleid:%d failed to parse annotations: %v", event.RuleId, err)
-		event.Annotations = fmt.Sprintf("failed to parse annotations: %v", err)
-		event.AnnotationsJSON["error"] = event.Annotations
-	}
-
-	e.queryRecoveryVal(event)
-
-	if err := event.ParseRule("rule_note"); err != nil {
-		logger.Warningf("ruleid:%d failed to parse rule note: %v", event.RuleId, err)
-		event.RuleNote = fmt.Sprintf("failed to parse rule note: %v", err)
-	}
-
-	e.persist(event)
-
-	if event.IsRecovered && event.NotifyRecovered == 0 {
-		return
-	}
-
-	e.dispatch.HandleEventNotify(event, false)
-}
-
-func (e *Consumer) persist(event *models.AlertCurEvent) {
-	if event.Status != 0 {
-		return
-	}
-
-	if !e.ctx.IsCenter {
-		event.DB2FE()
-		var err error
-		event.Id, err = poster.PostByUrlsWithResp[int64](e.ctx, "/v1/n9e/event-persist", event)
-		if err != nil {
-			logger.Errorf("event:%+v persist err:%v", event, err)
-			e.dispatch.Astats.CounterRuleEvalErrorTotal.WithLabelValues(fmt.Sprintf("%v", event.DatasourceId), "persist_event", event.GroupName, fmt.Sprintf("%v", event.RuleId)).Inc()
-		}
-		return
-	}
-
-	err := models.EventPersist(e.ctx, event)
-	if err != nil {
-		logger.Errorf("event%+v persist err:%v", event, err)
-		e.dispatch.Astats.CounterRuleEvalErrorTotal.WithLabelValues(fmt.Sprintf("%v", event.DatasourceId), "persist_event", event.GroupName, fmt.Sprintf("%v", event.RuleId)).Inc()
-	}
-}
-
-func (e *Consumer) queryRecoveryVal(event *models.AlertCurEvent) {
-	if !event.IsRecovered {
-		return
-	}
-
-	// If the event is a recovery event, execute the recovery_promql query
-	promql, ok := event.AnnotationsJSON["recovery_promql"]
-	if !ok {
-		return
-	}
-
-	promql = strings.TrimSpace(promql)
-	if promql == "" {
-		logger.Warningf("rule_eval:%s promql is blank", getKey(event))
-		return
-	}
-
-	if e.promClients.IsNil(event.DatasourceId) {
-		logger.Warningf("rule_eval:%s error reader client is nil", getKey(event))
-		return
-	}
-
-	readerClient := e.promClients.GetCli(event.DatasourceId)
-
-	var warnings promsdk.Warnings
-	value, warnings, err := readerClient.Query(e.ctx.Ctx, promql, time.Now())
-	if err != nil {
-		logger.Errorf("rule_eval:%s promql:%s, error:%v", getKey(event), promql, err)
-		event.AnnotationsJSON["recovery_promql_error"] = fmt.Sprintf("promql:%s error:%v", promql, err)
-
-		b, err := json.Marshal(event.AnnotationsJSON)
-		if err != nil {
-			event.AnnotationsJSON = make(map[string]string)
-			event.AnnotationsJSON["error"] = fmt.Sprintf("failed to parse annotations: %v", err)
-		} else {
-			event.Annotations = string(b)
-		}
-		return
-	}
-
-	if len(warnings) > 0 {
-		logger.Errorf("rule_eval:%s promql:%s, warnings:%v", getKey(event), promql, warnings)
-	}
-
-	anomalyPoints := models.ConvertAnomalyPoints(value)
-	if len(anomalyPoints) == 0 {
-		logger.Warningf("rule_eval:%s promql:%s, result is empty", getKey(event), promql)
-		event.AnnotationsJSON["recovery_promql_error"] = fmt.Sprintf("promql:%s error:%s", promql, "result is empty")
-	} else {
-		event.AnnotationsJSON["recovery_value"] = fmt.Sprintf("%v", anomalyPoints[0].Value)
-	}
-
-	b, err := json.Marshal(event.AnnotationsJSON)
-	if err != nil {
-		event.AnnotationsJSON = make(map[string]string)
-		event.AnnotationsJSON["error"] = fmt.Sprintf("failed to parse annotations: %v", err)
-	} else {
-		event.Annotations = string(b)
-	}
-}
-
-func getKey(event *models.AlertCurEvent) string {
-	return common.RuleKey(event.DatasourceId, event.RuleId)
-}

alert/dispatch/dispatch.go

@@ -1,812 +0,0 @@
-package dispatch
-
-import (
-	"bytes"
-	"encoding/json"
-	"errors"
-	"fmt"
-	"html/template"
-	"net/url"
-	"strconv"
-	"strings"
-	"sync"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/common"
-	"github.com/ccfos/nightingale/v6/alert/pipeline"
-	"github.com/ccfos/nightingale/v6/alert/sender"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type Dispatch struct {
-	alertRuleCache      *memsto.AlertRuleCacheType
-	userCache           *memsto.UserCacheType
-	userGroupCache      *memsto.UserGroupCacheType
-	alertSubscribeCache *memsto.AlertSubscribeCacheType
-	targetCache         *memsto.TargetCacheType
-	notifyConfigCache   *memsto.NotifyConfigCacheType
-	taskTplsCache       *memsto.TaskTplCache
-
-	notifyRuleCache      *memsto.NotifyRuleCacheType
-	notifyChannelCache   *memsto.NotifyChannelCacheType
-	messageTemplateCache *memsto.MessageTemplateCacheType
-	eventProcessorCache  *memsto.EventProcessorCacheType
-
-	alerting aconf.Alerting
-
-	Senders          map[string]sender.Sender
-	CallBacks        map[string]sender.CallBacker
-	tpls             map[string]*template.Template
-	ExtraSenders     map[string]sender.Sender
-	BeforeSenderHook func(*models.AlertCurEvent) bool
-
-	ctx    *ctx.Context
-	Astats *astats.Stats
-
-	RwLock sync.RWMutex
-}
-
-// 创建一个 Notify 实例
-func NewDispatch(alertRuleCache *memsto.AlertRuleCacheType, userCache *memsto.UserCacheType, userGroupCache *memsto.UserGroupCacheType,
-	alertSubscribeCache *memsto.AlertSubscribeCacheType, targetCache *memsto.TargetCacheType, notifyConfigCache *memsto.NotifyConfigCacheType,
-	taskTplsCache *memsto.TaskTplCache, notifyRuleCache *memsto.NotifyRuleCacheType, notifyChannelCache *memsto.NotifyChannelCacheType,
-	messageTemplateCache *memsto.MessageTemplateCacheType, eventProcessorCache *memsto.EventProcessorCacheType, alerting aconf.Alerting, ctx *ctx.Context, astats *astats.Stats) *Dispatch {
-	notify := &Dispatch{
-		alertRuleCache:       alertRuleCache,
-		userCache:            userCache,
-		userGroupCache:       userGroupCache,
-		alertSubscribeCache:  alertSubscribeCache,
-		targetCache:          targetCache,
-		notifyConfigCache:    notifyConfigCache,
-		taskTplsCache:        taskTplsCache,
-		notifyRuleCache:      notifyRuleCache,
-		notifyChannelCache:   notifyChannelCache,
-		messageTemplateCache: messageTemplateCache,
-		eventProcessorCache:  eventProcessorCache,
-
-		alerting: alerting,
-
-		Senders:          make(map[string]sender.Sender),
-		tpls:             make(map[string]*template.Template),
-		ExtraSenders:     make(map[string]sender.Sender),
-		BeforeSenderHook: func(*models.AlertCurEvent) bool { return true },
-
-		ctx:    ctx,
-		Astats: astats,
-	}
-
-	pipeline.Init()
-	return notify
-}
-
-func (e *Dispatch) ReloadTpls() error {
-	err := e.reloadTpls()
-	if err != nil {
-		logger.Errorf("failed to reload tpls: %v", err)
-	}
-
-	duration := time.Duration(9000) * time.Millisecond
-	for {
-		time.Sleep(duration)
-		if err := e.reloadTpls(); err != nil {
-			logger.Warning("failed to reload tpls:", err)
-		}
-	}
-}
-
-func (e *Dispatch) reloadTpls() error {
-	tmpTpls, err := models.ListTpls(e.ctx)
-	if err != nil {
-		return err
-	}
-	smtp := e.notifyConfigCache.GetSMTP()
-
-	senders := map[string]sender.Sender{
-		models.Email:      sender.NewSender(models.Email, tmpTpls, smtp),
-		models.Dingtalk:   sender.NewSender(models.Dingtalk, tmpTpls),
-		models.Wecom:      sender.NewSender(models.Wecom, tmpTpls),
-		models.Feishu:     sender.NewSender(models.Feishu, tmpTpls),
-		models.Mm:         sender.NewSender(models.Mm, tmpTpls),
-		models.Telegram:   sender.NewSender(models.Telegram, tmpTpls),
-		models.FeishuCard: sender.NewSender(models.FeishuCard, tmpTpls),
-		models.Lark:       sender.NewSender(models.Lark, tmpTpls),
-		models.LarkCard:   sender.NewSender(models.LarkCard, tmpTpls),
-	}
-
-	// domain -> Callback()
-	callbacks := map[string]sender.CallBacker{
-		models.DingtalkDomain:   sender.NewCallBacker(models.DingtalkDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.WecomDomain:      sender.NewCallBacker(models.WecomDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.FeishuDomain:     sender.NewCallBacker(models.FeishuDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.TelegramDomain:   sender.NewCallBacker(models.TelegramDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.FeishuCardDomain: sender.NewCallBacker(models.FeishuCardDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.IbexDomain:       sender.NewCallBacker(models.IbexDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.LarkDomain:       sender.NewCallBacker(models.LarkDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.DefaultDomain:    sender.NewCallBacker(models.DefaultDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-		models.LarkCardDomain:   sender.NewCallBacker(models.LarkCardDomain, e.targetCache, e.userCache, e.taskTplsCache, tmpTpls),
-	}
-
-	e.RwLock.RLock()
-	for channelName, extraSender := range e.ExtraSenders {
-		senders[channelName] = extraSender
-	}
-	e.RwLock.RUnlock()
-
-	e.RwLock.Lock()
-	e.tpls = tmpTpls
-	e.Senders = senders
-	e.CallBacks = callbacks
-	e.RwLock.Unlock()
-	return nil
-}
-
-func (e *Dispatch) HandleEventWithNotifyRule(eventOrigin *models.AlertCurEvent) {
-
-	if len(eventOrigin.NotifyRuleIds) > 0 {
-		for _, notifyRuleId := range eventOrigin.NotifyRuleIds {
-			// 深拷贝新的 event，避免并发修改 event 冲突
-			eventCopy := eventOrigin.DeepCopy()
-
-			logger.Infof("notify rule ids: %v, event: %+v", notifyRuleId, eventCopy)
-			notifyRule := e.notifyRuleCache.Get(notifyRuleId)
-			if notifyRule == nil {
-				continue
-			}
-
-			if !notifyRule.Enable {
-				continue
-			}
-
-			var processors []models.Processor
-			for _, pipelineConfig := range notifyRule.PipelineConfigs {
-				if !pipelineConfig.Enable {
-					continue
-				}
-
-				eventPipeline := e.eventProcessorCache.Get(pipelineConfig.PipelineId)
-				if eventPipeline == nil {
-					logger.Warningf("notify_id: %d, event:%+v, processor not found", notifyRuleId, eventCopy)
-					continue
-				}
-
-				if !pipelineApplicable(eventPipeline, eventCopy) {
-					logger.Debugf("notify_id: %d, event:%+v, pipeline_id: %d, not applicable", notifyRuleId, eventCopy, pipelineConfig.PipelineId)
-					continue
-				}
-
-				processors = append(processors, e.eventProcessorCache.GetProcessorsById(pipelineConfig.PipelineId)...)
-			}
-
-			for _, processor := range processors {
-				logger.Infof("before processor notify_id: %d, event:%+v, processor:%+v", notifyRuleId, eventCopy, processor)
-				eventCopy, res, err := processor.Process(e.ctx, eventCopy)
-				logger.Infof("after processor notify_id: %d, event:%+v, processor:%+v, res:%v, err:%v", notifyRuleId, eventCopy, processor, res, err)
-				if eventCopy == nil {
-					logger.Warningf("notify_id: %d, event:%+v, processor:%+v, event is nil", notifyRuleId, eventCopy, processor)
-					break
-				}
-			}
-
-			if eventCopy == nil {
-				// 如果 eventCopy 为 nil，说明 eventCopy 被 processor drop 掉了, 不再发送通知
-				continue
-			}
-
-			// notify
-			for i := range notifyRule.NotifyConfigs {
-				if !NotifyRuleApplicable(&notifyRule.NotifyConfigs[i], eventCopy) {
-					continue
-				}
-				notifyChannel := e.notifyChannelCache.Get(notifyRule.NotifyConfigs[i].ChannelID)
-				messageTemplate := e.messageTemplateCache.Get(notifyRule.NotifyConfigs[i].TemplateID)
-				if notifyChannel == nil {
-					sender.NotifyRecord(e.ctx, []*models.AlertCurEvent{eventCopy}, notifyRuleId, fmt.Sprintf("notify_channel_id:%d", notifyRule.NotifyConfigs[i].ChannelID), "", "", errors.New("notify_channel not found"))
-					logger.Warningf("notify_id: %d, event:%+v, channel_id:%d, template_id: %d, notify_channel not found", notifyRuleId, eventCopy, notifyRule.NotifyConfigs[i].ChannelID, notifyRule.NotifyConfigs[i].TemplateID)
-					continue
-				}
-
-				if notifyChannel.RequestType != "flashduty" && messageTemplate == nil {
-					logger.Warningf("notify_id: %d, channel_name: %v, event:%+v, template_id: %d, message_template not found", notifyRuleId, notifyChannel.Ident, eventCopy, notifyRule.NotifyConfigs[i].TemplateID)
-					sender.NotifyRecord(e.ctx, []*models.AlertCurEvent{eventCopy}, notifyRuleId, notifyChannel.Name, "", "", errors.New("message_template not found"))
-
-					continue
-				}
-
-				// todo go send
-				// todo 聚合 event
-				go e.sendV2([]*models.AlertCurEvent{eventCopy}, notifyRuleId, &notifyRule.NotifyConfigs[i], notifyChannel, messageTemplate)
-			}
-		}
-	}
-}
-
-func pipelineApplicable(pipeline *models.EventPipeline, event *models.AlertCurEvent) bool {
-	if pipeline == nil {
-		return true
-	}
-
-	if !pipeline.FilterEnable {
-		return true
-	}
-
-	tagMatch := true
-	if len(pipeline.LabelFilters) > 0 {
-		for i := range pipeline.LabelFilters {
-			if pipeline.LabelFilters[i].Func == "" {
-				pipeline.LabelFilters[i].Func = pipeline.LabelFilters[i].Op
-			}
-		}
-
-		tagFilters, err := models.ParseTagFilter(pipeline.LabelFilters)
-		if err != nil {
-			logger.Errorf("pipeline applicable failed to parse tag filter: %v event:%+v pipeline:%+v", err, event, pipeline)
-			return false
-		}
-		tagMatch = common.MatchTags(event.TagsMap, tagFilters)
-	}
-
-	attributesMatch := true
-	if len(pipeline.AttrFilters) > 0 {
-		tagFilters, err := models.ParseTagFilter(pipeline.AttrFilters)
-		if err != nil {
-			logger.Errorf("pipeline applicable failed to parse tag filter: %v event:%+v pipeline:%+v err:%v", tagFilters, event, pipeline, err)
-			return false
-		}
-
-		attributesMatch = common.MatchTags(event.JsonTagsAndValue(), tagFilters)
-	}
-
-	return tagMatch && attributesMatch
-}
-
-func NotifyRuleApplicable(notifyConfig *models.NotifyConfig, event *models.AlertCurEvent) bool {
-	tm := time.Unix(event.TriggerTime, 0)
-	triggerTime := tm.Format("15:04")
-	triggerWeek := int(tm.Weekday())
-
-	timeMatch := false
-
-	if len(notifyConfig.TimeRanges) == 0 {
-		timeMatch = true
-	}
-	for j := range notifyConfig.TimeRanges {
-		if timeMatch {
-			break
-		}
-		enableStime := notifyConfig.TimeRanges[j].Start
-		enableEtime := notifyConfig.TimeRanges[j].End
-		enableDaysOfWeek := notifyConfig.TimeRanges[j].Week
-		length := len(enableDaysOfWeek)
-		// enableStime,enableEtime,enableDaysOfWeek三者长度肯定相同，这里循环一个即可
-		for i := 0; i < length; i++ {
-			if enableDaysOfWeek[i] != triggerWeek {
-				continue
-			}
-
-			if enableStime < enableEtime {
-				if enableEtime == "23:59" {
-					// 02:00-23:59，这种情况做个特殊处理，相当于左闭右闭区间了
-					if triggerTime < enableStime {
-						// mute, 即没生效
-						continue
-					}
-				} else {
-					// 02:00-04:00 或者 02:00-24:00
-					if triggerTime < enableStime || triggerTime >= enableEtime {
-						// mute, 即没生效
-						continue
-					}
-				}
-			} else if enableStime > enableEtime {
-				// 21:00-09:00
-				if triggerTime < enableStime && triggerTime >= enableEtime {
-					// mute, 即没生效
-					continue
-				}
-			}
-
-			// 到这里说明当前时刻在告警规则的某组生效时间范围内，即没有 mute，直接返回 false
-			timeMatch = true
-			break
-		}
-	}
-
-	severityMatch := false
-	for i := range notifyConfig.Severities {
-		if notifyConfig.Severities[i] == event.Severity {
-			severityMatch = true
-		}
-	}
-
-	tagMatch := true
-	if len(notifyConfig.LabelKeys) > 0 {
-		for i := range notifyConfig.LabelKeys {
-			if notifyConfig.LabelKeys[i].Func == "" {
-				notifyConfig.LabelKeys[i].Func = notifyConfig.LabelKeys[i].Op
-			}
-		}
-
-		tagFilters, err := models.ParseTagFilter(notifyConfig.LabelKeys)
-		if err != nil {
-			logger.Errorf("notify send failed to parse tag filter: %v event:%+v notify_config:%+v", err, event, notifyConfig)
-			return false
-		}
-		tagMatch = common.MatchTags(event.TagsMap, tagFilters)
-	}
-
-	attributesMatch := true
-	if len(notifyConfig.Attributes) > 0 {
-		tagFilters, err := models.ParseTagFilter(notifyConfig.Attributes)
-		if err != nil {
-			logger.Errorf("notify send failed to parse tag filter: %v event:%+v notify_config:%+v err:%v", tagFilters, event, notifyConfig, err)
-			return false
-		}
-
-		attributesMatch = common.MatchTags(event.JsonTagsAndValue(), tagFilters)
-	}
-	logger.Infof("notify send timeMatch:%v severityMatch:%v tagMatch:%v attributesMatch:%v event:%+v notify_config:%+v", timeMatch, severityMatch, tagMatch, attributesMatch, event, notifyConfig)
-	return timeMatch && severityMatch && tagMatch && attributesMatch
-}
-
-func GetNotifyConfigParams(notifyConfig *models.NotifyConfig, contactKey string, userCache *memsto.UserCacheType, userGroupCache *memsto.UserGroupCacheType) ([]string, []int64, map[string]string) {
-	customParams := make(map[string]string)
-	var flashDutyChannelIDs []int64
-	var userInfoParams models.CustomParams
-
-	for key, value := range notifyConfig.Params {
-		switch key {
-		case "user_ids", "user_group_ids", "ids":
-			if data, err := json.Marshal(value); err == nil {
-				var ids []int64
-				if json.Unmarshal(data, &ids) == nil {
-					if key == "user_ids" {
-						userInfoParams.UserIDs = ids
-					} else if key == "user_group_ids" {
-						userInfoParams.UserGroupIDs = ids
-					} else if key == "ids" {
-						flashDutyChannelIDs = ids
-					}
-				}
-			}
-		default:
-			customParams[key] = value.(string)
-		}
-	}
-
-	if len(userInfoParams.UserIDs) == 0 && len(userInfoParams.UserGroupIDs) == 0 {
-		return []string{}, flashDutyChannelIDs, customParams
-	}
-
-	userIds := make([]int64, 0)
-	userIds = append(userIds, userInfoParams.UserIDs...)
-
-	if len(userInfoParams.UserGroupIDs) > 0 {
-		userGroups := userGroupCache.GetByUserGroupIds(userInfoParams.UserGroupIDs)
-		for _, userGroup := range userGroups {
-			userIds = append(userIds, userGroup.UserIds...)
-		}
-	}
-
-	users := userCache.GetByUserIds(userIds)
-	visited := make(map[int64]bool)
-	sendtos := make([]string, 0)
-	for _, user := range users {
-		if visited[user.Id] {
-			continue
-		}
-		var sendto string
-		if contactKey == "phone" {
-			sendto = user.Phone
-		} else if contactKey == "email" {
-			sendto = user.Email
-		} else {
-			sendto, _ = user.ExtractToken(contactKey)
-		}
-
-		if sendto == "" {
-			continue
-		}
-		sendtos = append(sendtos, sendto)
-		visited[user.Id] = true
-	}
-
-	return sendtos, flashDutyChannelIDs, customParams
-}
-
-func (e *Dispatch) sendV2(events []*models.AlertCurEvent, notifyRuleId int64, notifyConfig *models.NotifyConfig, notifyChannel *models.NotifyChannelConfig, messageTemplate *models.MessageTemplate) {
-	if len(events) == 0 {
-		logger.Errorf("notify_id: %d events is empty", notifyRuleId)
-		return
-	}
-
-	tplContent := make(map[string]interface{})
-	if notifyChannel.RequestType != "flashduty" {
-		tplContent = messageTemplate.RenderEvent(events)
-	}
-
-	var contactKey string
-	if notifyChannel.ParamConfig != nil && notifyChannel.ParamConfig.UserInfo != nil {
-		contactKey = notifyChannel.ParamConfig.UserInfo.ContactKey
-	}
-
-	sendtos, flashDutyChannelIDs, customParams := GetNotifyConfigParams(notifyConfig, contactKey, e.userCache, e.userGroupCache)
-
-	e.Astats.GaugeNotifyRecordQueueSize.Inc()
-	defer e.Astats.GaugeNotifyRecordQueueSize.Dec()
-
-	switch notifyChannel.RequestType {
-	case "flashduty":
-		if len(flashDutyChannelIDs) == 0 {
-			flashDutyChannelIDs = []int64{0} // 如果 flashduty 通道没有配置，则使用 0, 给 SendFlashDuty 判断使用, 不给 flashduty 传 channel_id 参数
-		}
-
-		for i := range flashDutyChannelIDs {
-			respBody, err := notifyChannel.SendFlashDuty(events, flashDutyChannelIDs[i], e.notifyChannelCache.GetHttpClient(notifyChannel.ID))
-			logger.Infof("notify_id: %d, channel_name: %v, event:%+v, IntegrationUrl: %v dutychannel_id: %v, respBody: %v, err: %v", notifyRuleId, notifyChannel.Name, events[0], notifyChannel.RequestConfig.FlashDutyRequestConfig.IntegrationUrl, flashDutyChannelIDs[i], respBody, err)
-			sender.NotifyRecord(e.ctx, events, notifyRuleId, notifyChannel.Name, strconv.FormatInt(flashDutyChannelIDs[i], 10), respBody, err)
-		}
-		return
-	case "http":
-		if e.notifyChannelCache.HttpConcurrencyAdd(notifyChannel.ID) {
-			defer e.notifyChannelCache.HttpConcurrencyDone(notifyChannel.ID)
-		}
-		if notifyChannel.RequestConfig == nil {
-			logger.Warningf("notify_id: %d, channel_name: %v, event:%+v, request config not found", notifyRuleId, notifyChannel.Name, events[0])
-		}
-
-		if notifyChannel.RequestConfig.HTTPRequestConfig == nil {
-			logger.Warningf("notify_id: %d, channel_name: %v, event:%+v, http request config not found", notifyRuleId, notifyChannel.Name, events[0])
-		}
-
-		if NeedBatchContacts(notifyChannel.RequestConfig.HTTPRequestConfig) || len(sendtos) == 0 {
-			resp, err := notifyChannel.SendHTTP(events, tplContent, customParams, sendtos, e.notifyChannelCache.GetHttpClient(notifyChannel.ID))
-			logger.Infof("notify_id: %d, channel_name: %v, event:%+v, tplContent:%s, customParams:%v, userInfo:%+v, respBody: %v, err: %v", notifyRuleId, notifyChannel.Name, events[0], tplContent, customParams, sendtos, resp, err)
-
-			sender.NotifyRecord(e.ctx, events, notifyRuleId, notifyChannel.Name, getSendTarget(customParams, sendtos), resp, err)
-		} else {
-			for i := range sendtos {
-				resp, err := notifyChannel.SendHTTP(events, tplContent, customParams, []string{sendtos[i]}, e.notifyChannelCache.GetHttpClient(notifyChannel.ID))
-				logger.Infof("notify_id: %d, channel_name: %v, event:%+v, tplContent:%s, customParams:%v, userInfo:%+v, respBody: %v, err: %v", notifyRuleId, notifyChannel.Name, events[0], tplContent, customParams, sendtos[i], resp, err)
-				sender.NotifyRecord(e.ctx, events, notifyRuleId, notifyChannel.Name, getSendTarget(customParams, []string{sendtos[i]}), resp, err)
-			}
-		}
-
-	case "smtp":
-		notifyChannel.SendEmail(notifyRuleId, events, tplContent, sendtos, e.notifyChannelCache.GetSmtpClient(notifyChannel.ID))
-
-	case "script":
-		target, res, err := notifyChannel.SendScript(events, tplContent, customParams, sendtos)
-		logger.Infof("notify_id: %d, channel_name: %v, event:%+v, tplContent:%s, customParams:%v, target:%s, res:%s, err:%v", notifyRuleId, notifyChannel.Name, events[0], tplContent, customParams, target, res, err)
-		sender.NotifyRecord(e.ctx, events, notifyRuleId, notifyChannel.Name, target, res, err)
-	default:
-		logger.Warningf("notify_id: %d, channel_name: %v, event:%+v send type not found", notifyRuleId, notifyChannel.Name, events[0])
-	}
-}
-
-func NeedBatchContacts(requestConfig *models.HTTPRequestConfig) bool {
-	b, _ := json.Marshal(requestConfig)
-	return strings.Contains(string(b), "$sendtos")
-}
-
-// HandleEventNotify 处理event事件的主逻辑
-// event: 告警/恢复事件
-// isSubscribe: 告警事件是否由subscribe的配置产生
-func (e *Dispatch) HandleEventNotify(event *models.AlertCurEvent, isSubscribe bool) {
-	rule := e.alertRuleCache.Get(event.RuleId)
-	if rule == nil {
-		return
-	}
-
-	fillUsers(event, e.userCache, e.userGroupCache)
-
-	var (
-		// 处理事件到 notifyTarget 关系,处理的notifyTarget用OrMerge进行合并
-		handlers []NotifyTargetDispatch
-
-		// 额外去掉一些订阅,处理的notifyTarget用AndMerge进行合并, 如设置 channel=false,合并后不通过这个channel发送
-		// 如果实现了相关 Dispatch,可以添加到interceptors中
-		interceptorHandlers []NotifyTargetDispatch
-	)
-	if isSubscribe {
-		handlers = []NotifyTargetDispatch{NotifyGroupDispatch, EventCallbacksDispatch}
-	} else {
-		handlers = []NotifyTargetDispatch{NotifyGroupDispatch, GlobalWebhookDispatch, EventCallbacksDispatch}
-	}
-
-	notifyTarget := NewNotifyTarget()
-	// 处理订阅关系使用OrMerge
-	for _, handler := range handlers {
-		notifyTarget.OrMerge(handler(rule, event, notifyTarget, e))
-	}
-
-	// 处理移除订阅关系的逻辑,比如员工离职，临时静默某个通道的策略等
-	for _, handler := range interceptorHandlers {
-		notifyTarget.AndMerge(handler(rule, event, notifyTarget, e))
-	}
-
-	go e.HandleEventWithNotifyRule(event)
-	go e.Send(rule, event, notifyTarget, isSubscribe)
-
-	// 如果是不是订阅规则出现的event, 则需要处理订阅规则的event
-	if !isSubscribe {
-		e.handleSubs(event)
-	}
-}
-
-func (e *Dispatch) handleSubs(event *models.AlertCurEvent) {
-	// handle alert subscribes
-	subscribes := make([]*models.AlertSubscribe, 0)
-	// rule specific subscribes
-	if subs, has := e.alertSubscribeCache.Get(event.RuleId); has {
-		subscribes = append(subscribes, subs...)
-	}
-	// global subscribes
-	if subs, has := e.alertSubscribeCache.Get(0); has {
-		subscribes = append(subscribes, subs...)
-	}
-
-	for _, sub := range subscribes {
-		e.handleSub(sub, *event)
-	}
-}
-
-// handleSub 处理订阅规则的event,注意这里event要使用值传递,因为后面会修改event的状态
-func (e *Dispatch) handleSub(sub *models.AlertSubscribe, event models.AlertCurEvent) {
-	if sub.IsDisabled() {
-		return
-	}
-
-	if !sub.MatchCluster(event.DatasourceId) {
-		return
-	}
-
-	if !sub.MatchProd(event.RuleProd) {
-		return
-	}
-
-	if !common.MatchTags(event.TagsMap, sub.ITags) {
-		return
-	}
-	// event BusiGroups filter
-	if !common.MatchGroupsName(event.GroupName, sub.IBusiGroups) {
-		return
-	}
-	if sub.ForDuration > (event.TriggerTime - event.FirstTriggerTime) {
-		return
-	}
-
-	if len(sub.SeveritiesJson) != 0 {
-		match := false
-		for _, s := range sub.SeveritiesJson {
-			if s == event.Severity || s == 0 {
-				match = true
-				break
-			}
-		}
-		if !match {
-			return
-		}
-	}
-
-	e.Astats.CounterSubEventTotal.WithLabelValues(event.GroupName).Inc()
-	sub.ModifyEvent(&event)
-	event.SubRuleId = sub.Id
-
-	LogEvent(&event, "subscribe")
-	e.HandleEventNotify(&event, true)
-}
-
-func (e *Dispatch) Send(rule *models.AlertRule, event *models.AlertCurEvent, notifyTarget *NotifyTarget, isSubscribe bool) {
-	needSend := e.BeforeSenderHook(event)
-	if needSend {
-		for channel, uids := range notifyTarget.ToChannelUserMap() {
-			msgCtx := sender.BuildMessageContext(e.ctx, rule, []*models.AlertCurEvent{event},
-				uids, e.userCache, e.Astats)
-			e.RwLock.RLock()
-			s := e.Senders[channel]
-			e.RwLock.RUnlock()
-			if s == nil {
-				logger.Debugf("no sender for channel: %s", channel)
-				continue
-			}
-
-			var event *models.AlertCurEvent
-			if len(msgCtx.Events) > 0 {
-				event = msgCtx.Events[0]
-			}
-
-			logger.Debugf("send to channel:%s event:%+v users:%+v", channel, event, msgCtx.Users)
-			s.Send(msgCtx)
-		}
-	}
-
-	// handle event callbacks
-	e.SendCallbacks(rule, notifyTarget, event)
-
-	// handle global webhooks
-	if !event.OverrideGlobalWebhook() {
-		if e.alerting.WebhookBatchSend {
-			sender.BatchSendWebhooks(e.ctx, notifyTarget.ToWebhookMap(), event, e.Astats)
-		} else {
-			sender.SingleSendWebhooks(e.ctx, notifyTarget.ToWebhookMap(), event, e.Astats)
-		}
-	}
-
-	// handle plugin call
-	go sender.MayPluginNotify(e.ctx, e.genNoticeBytes(event), e.notifyConfigCache.
-		GetNotifyScript(), e.Astats, event)
-
-	if !isSubscribe {
-		// handle ibex callbacks
-		e.HandleIbex(rule, event)
-	}
-}
-
-func (e *Dispatch) SendCallbacks(rule *models.AlertRule, notifyTarget *NotifyTarget, event *models.AlertCurEvent) {
-	uids := notifyTarget.ToUidList()
-	urls := notifyTarget.ToCallbackList()
-	whMap := notifyTarget.ToWebhookMap()
-	ogw := event.OverrideGlobalWebhook()
-	for _, urlStr := range urls {
-		if len(urlStr) == 0 {
-			continue
-		}
-
-		cbCtx := sender.BuildCallBackContext(e.ctx, urlStr, rule, []*models.AlertCurEvent{event}, uids, e.userCache, e.alerting.WebhookBatchSend, e.Astats)
-
-		if wh, ok := whMap[cbCtx.CallBackURL]; !ogw && ok && wh.Enable {
-			logger.Debugf("SendCallbacks: webhook[%s] is in global conf.", cbCtx.CallBackURL)
-			continue
-		}
-
-		if strings.HasPrefix(urlStr, "${ibex}") {
-			e.CallBacks[models.IbexDomain].CallBack(cbCtx)
-			continue
-		}
-
-		if !(strings.HasPrefix(urlStr, "http://") || strings.HasPrefix(urlStr, "https://")) {
-			cbCtx.CallBackURL = "http://" + urlStr
-		}
-
-		parsedURL, err := url.Parse(urlStr)
-		if err != nil {
-			logger.Errorf("SendCallbacks: failed to url.Parse(urlStr=%s): %v", urlStr, err)
-			continue
-		}
-
-		// process feishu card
-		if parsedURL.Host == models.FeishuDomain && parsedURL.Query().Get("card") == "1" {
-			e.CallBacks[models.FeishuCardDomain].CallBack(cbCtx)
-			continue
-		}
-
-		// process lark card
-		if parsedURL.Host == models.LarkDomain && parsedURL.Query().Get("card") == "1" {
-			e.CallBacks[models.LarkCardDomain].CallBack(cbCtx)
-			continue
-		}
-
-		callBacker, ok := e.CallBacks[parsedURL.Host]
-		if ok {
-			callBacker.CallBack(cbCtx)
-		} else {
-			e.CallBacks[models.DefaultDomain].CallBack(cbCtx)
-		}
-	}
-}
-
-func (e *Dispatch) HandleIbex(rule *models.AlertRule, event *models.AlertCurEvent) {
-	// 解析 RuleConfig 字段
-	var ruleConfig struct {
-		TaskTpls []*models.Tpl `json:"task_tpls"`
-	}
-	json.Unmarshal([]byte(rule.RuleConfig), &ruleConfig)
-
-	if event.IsRecovered {
-		// 恢复事件不需要走故障自愈的逻辑
-		return
-	}
-
-	for _, t := range ruleConfig.TaskTpls {
-		if t.TplId == 0 {
-			continue
-		}
-
-		if len(t.Host) == 0 {
-			sender.CallIbex(e.ctx, t.TplId, event.TargetIdent,
-				e.taskTplsCache, e.targetCache, e.userCache, event)
-			continue
-		}
-		for _, host := range t.Host {
-			sender.CallIbex(e.ctx, t.TplId, host,
-				e.taskTplsCache, e.targetCache, e.userCache, event)
-		}
-	}
-}
-
-type Notice struct {
-	Event *models.AlertCurEvent `json:"event"`
-	Tpls  map[string]string     `json:"tpls"`
-}
-
-func (e *Dispatch) genNoticeBytes(event *models.AlertCurEvent) []byte {
-	// build notice body with templates
-	ntpls := make(map[string]string)
-
-	e.RwLock.RLock()
-	defer e.RwLock.RUnlock()
-	for filename, tpl := range e.tpls {
-		var body bytes.Buffer
-		if err := tpl.Execute(&body, event); err != nil {
-			ntpls[filename] = err.Error()
-		} else {
-			ntpls[filename] = body.String()
-		}
-	}
-
-	notice := Notice{Event: event, Tpls: ntpls}
-	stdinBytes, err := json.Marshal(notice)
-	if err != nil {
-		logger.Errorf("event_notify: failed to marshal notice: %v", err)
-		return nil
-	}
-
-	return stdinBytes
-}
-
-// for alerting
-func fillUsers(ce *models.AlertCurEvent, uc *memsto.UserCacheType, ugc *memsto.UserGroupCacheType) {
-	gids := make([]int64, 0, len(ce.NotifyGroupsJSON))
-	for i := 0; i < len(ce.NotifyGroupsJSON); i++ {
-		gid, err := strconv.ParseInt(ce.NotifyGroupsJSON[i], 10, 64)
-		if err != nil {
-			continue
-		}
-		gids = append(gids, gid)
-	}
-
-	ce.NotifyGroupsObj = ugc.GetByUserGroupIds(gids)
-
-	uids := make(map[int64]struct{})
-	for i := 0; i < len(ce.NotifyGroupsObj); i++ {
-		ug := ce.NotifyGroupsObj[i]
-		for j := 0; j < len(ug.UserIds); j++ {
-			uids[ug.UserIds[j]] = struct{}{}
-		}
-	}
-
-	ce.NotifyUsersObj = uc.GetByUserIds(mapKeys(uids))
-}
-
-func mapKeys(m map[int64]struct{}) []int64 {
-	lst := make([]int64, 0, len(m))
-	for k := range m {
-		lst = append(lst, k)
-	}
-	return lst
-}
-
-func getSendTarget(customParams map[string]string, sendtos []string) string {
-	if len(customParams) == 0 {
-		return strings.Join(sendtos, ",")
-	}
-
-	values := make([]string, 0)
-	for _, value := range customParams {
-		runes := []rune(value)
-		if len(runes) <= 4 {
-			values = append(values, value)
-		} else {
-			maskedValue := string(runes[:len(runes)-4]) + "****"
-			values = append(values, maskedValue)
-		}
-	}
-
-	return strings.Join(values, ",")
-}

alert/dispatch/notify_channel.go

@@ -1,33 +0,0 @@
-package dispatch
-
-// NotifyChannels channelKey -> bool
-type NotifyChannels map[string]bool
-
-func NewNotifyChannels(channels []string) NotifyChannels {
-	nc := make(NotifyChannels)
-	for _, ch := range channels {
-		nc[ch] = true
-	}
-	return nc
-}
-
-func (nc NotifyChannels) OrMerge(other NotifyChannels) {
-	nc.merge(other, func(a, b bool) bool { return a || b })
-}
-
-func (nc NotifyChannels) AndMerge(other NotifyChannels) {
-	nc.merge(other, func(a, b bool) bool { return a && b })
-}
-
-func (nc NotifyChannels) merge(other NotifyChannels, f func(bool, bool) bool) {
-	if other == nil {
-		return
-	}
-	for k, v := range other {
-		if curV, has := nc[k]; has {
-			nc[k] = f(curV, v)
-		} else {
-			nc[k] = v
-		}
-	}
-}

alert/dispatch/notify_target.go

@@ -1,138 +0,0 @@
-package dispatch
-
-import (
-	"strconv"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-// NotifyTarget 维护所有需要发送的目标 用户-通道/回调/钩子信息,用map维护的数据结构具有去重功能
-type NotifyTarget struct {
-	userMap   map[int64]NotifyChannels
-	webhooks  map[string]*models.Webhook
-	callbacks map[string]struct{}
-}
-
-func NewNotifyTarget() *NotifyTarget {
-	return &NotifyTarget{
-		userMap:   make(map[int64]NotifyChannels),
-		webhooks:  make(map[string]*models.Webhook),
-		callbacks: make(map[string]struct{}),
-	}
-}
-
-// OrMerge 将 channelMap 按照 or 的方式合并,方便实现多种组合的策略,比如根据某个 tag 进行路由等
-func (s *NotifyTarget) OrMerge(other *NotifyTarget) {
-	s.merge(other, NotifyChannels.OrMerge)
-}
-
-// AndMerge 将 channelMap 中的 bool 值按照 and 的逻辑进行合并,可以单独将人/通道维度的通知移除
-// 常用的场景有:
-// 1. 人员离职了不需要发送告警了
-// 2. 某个告警通道进行维护,暂时不需要发送告警了
-// 3. 业务值班的重定向逻辑，将高等级的告警额外发送给应急人员等
-// 可以结合业务需求自己实现router
-func (s *NotifyTarget) AndMerge(other *NotifyTarget) {
-	s.merge(other, NotifyChannels.AndMerge)
-}
-
-func (s *NotifyTarget) merge(other *NotifyTarget, f func(NotifyChannels, NotifyChannels)) {
-	if other == nil {
-		return
-	}
-	for k, v := range other.userMap {
-		if curV, has := s.userMap[k]; has {
-			f(curV, v)
-		} else {
-			s.userMap[k] = v
-		}
-	}
-	for k, v := range other.webhooks {
-		s.webhooks[k] = v
-	}
-	for k, v := range other.callbacks {
-		s.callbacks[k] = v
-	}
-}
-
-// ToChannelUserMap userMap(map[uid][channel]bool) 转换为 map[channel][]uid 的结构
-func (s *NotifyTarget) ToChannelUserMap() map[string][]int64 {
-	m := make(map[string][]int64)
-	for uid, nc := range s.userMap {
-		for ch, send := range nc {
-			if send {
-				m[ch] = append(m[ch], uid)
-			}
-		}
-	}
-	return m
-}
-
-func (s *NotifyTarget) ToCallbackList() []string {
-	callbacks := make([]string, 0, len(s.callbacks))
-	for cb := range s.callbacks {
-		callbacks = append(callbacks, cb)
-	}
-	return callbacks
-}
-
-func (s *NotifyTarget) ToWebhookMap() map[string]*models.Webhook {
-	return s.webhooks
-}
-
-func (s *NotifyTarget) ToUidList() []int64 {
-	uids := make([]int64, 0, len(s.userMap))
-	for uid, _ := range s.userMap {
-		uids = append(uids, uid)
-	}
-	return uids
-}
-
-// Dispatch 抽象由告警事件到信息接收者的路由策略
-// rule: 告警规则
-// event: 告警事件
-// prev: 前一次路由结果, Dispatch 的实现可以直接修改 prev, 也可以返回一个新的 NotifyTarget 用于 AndMerge/OrMerge
-type NotifyTargetDispatch func(rule *models.AlertRule, event *models.AlertCurEvent, prev *NotifyTarget, dispatch *Dispatch) *NotifyTarget
-
-// GroupDispatch 处理告警规则的组订阅关系
-func NotifyGroupDispatch(rule *models.AlertRule, event *models.AlertCurEvent, prev *NotifyTarget, dispatch *Dispatch) *NotifyTarget {
-	groupIds := make([]int64, 0, len(event.NotifyGroupsJSON))
-	for _, groupId := range event.NotifyGroupsJSON {
-		gid, err := strconv.ParseInt(groupId, 10, 64)
-		if err != nil {
-			continue
-		}
-		groupIds = append(groupIds, gid)
-	}
-
-	groups := dispatch.userGroupCache.GetByUserGroupIds(groupIds)
-	NotifyTarget := NewNotifyTarget()
-	for _, group := range groups {
-		for _, userId := range group.UserIds {
-			NotifyTarget.userMap[userId] = NewNotifyChannels(event.NotifyChannelsJSON)
-		}
-	}
-	return NotifyTarget
-}
-
-func GlobalWebhookDispatch(rule *models.AlertRule, event *models.AlertCurEvent, prev *NotifyTarget, dispatch *Dispatch) *NotifyTarget {
-	webhooks := dispatch.notifyConfigCache.GetWebhooks()
-	NotifyTarget := NewNotifyTarget()
-	for _, webhook := range webhooks {
-		if !webhook.Enable {
-			continue
-		}
-		NotifyTarget.webhooks[webhook.Url] = webhook
-	}
-	return NotifyTarget
-}
-
-func EventCallbacksDispatch(rule *models.AlertRule, event *models.AlertCurEvent, prev *NotifyTarget, dispatch *Dispatch) *NotifyTarget {
-	for _, c := range event.CallbacksJSON {
-		if c == "" {
-			continue
-		}
-		prev.callbacks[c] = struct{}{}
-	}
-	return nil
-}

alert/eval/alert_rule.go

@@ -1,187 +0,0 @@
-package eval
-
-import (
-	"context"
-	"fmt"
-	"strconv"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/naming"
-	"github.com/ccfos/nightingale/v6/alert/process"
-	"github.com/ccfos/nightingale/v6/datasource/commons/eslike"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/toolkits/pkg/logger"
-)
-
-type Scheduler struct {
-	// key: hash
-	alertRules map[string]*AlertRuleWorker
-
-	ExternalProcessors *process.ExternalProcessorsType
-
-	aconf aconf.Alert
-
-	alertRuleCache          *memsto.AlertRuleCacheType
-	targetCache             *memsto.TargetCacheType
-	targetsOfAlertRuleCache *memsto.TargetsOfAlertRuleCacheType
-	busiGroupCache          *memsto.BusiGroupCacheType
-	alertMuteCache          *memsto.AlertMuteCacheType
-	datasourceCache         *memsto.DatasourceCacheType
-
-	promClients *prom.PromClientMap
-
-	naming *naming.Naming
-
-	ctx   *ctx.Context
-	stats *astats.Stats
-}
-
-func NewScheduler(aconf aconf.Alert, externalProcessors *process.ExternalProcessorsType, arc *memsto.AlertRuleCacheType,
-	targetCache *memsto.TargetCacheType, toarc *memsto.TargetsOfAlertRuleCacheType,
-	busiGroupCache *memsto.BusiGroupCacheType, alertMuteCache *memsto.AlertMuteCacheType, datasourceCache *memsto.DatasourceCacheType,
-	promClients *prom.PromClientMap, naming *naming.Naming, ctx *ctx.Context, stats *astats.Stats) *Scheduler {
-	scheduler := &Scheduler{
-		aconf:      aconf,
-		alertRules: make(map[string]*AlertRuleWorker),
-
-		ExternalProcessors: externalProcessors,
-
-		alertRuleCache:          arc,
-		targetCache:             targetCache,
-		targetsOfAlertRuleCache: toarc,
-		busiGroupCache:          busiGroupCache,
-		alertMuteCache:          alertMuteCache,
-		datasourceCache:         datasourceCache,
-
-		promClients: promClients,
-		naming:      naming,
-
-		ctx:   ctx,
-		stats: stats,
-	}
-	eslike.SetEsIndexPatternCacheType(memsto.NewEsIndexPatternCacheType(ctx))
-
-	go scheduler.LoopSyncRules(context.Background())
-	return scheduler
-}
-
-func (s *Scheduler) LoopSyncRules(ctx context.Context) {
-	time.Sleep(time.Duration(s.aconf.EngineDelay) * time.Second)
-	duration := 9000 * time.Millisecond
-	for {
-		select {
-		case <-ctx.Done():
-			return
-		case <-time.After(duration):
-			s.syncAlertRules()
-		}
-	}
-}
-
-func (s *Scheduler) syncAlertRules() {
-	ids := s.alertRuleCache.GetRuleIds()
-	alertRuleWorkers := make(map[string]*AlertRuleWorker)
-	externalRuleWorkers := make(map[string]*process.Processor)
-	for _, id := range ids {
-		rule := s.alertRuleCache.Get(id)
-		if rule == nil {
-			continue
-		}
-
-		ruleType := rule.GetRuleType()
-		if rule.IsPrometheusRule() || rule.IsInnerRule() {
-			datasourceIds := s.datasourceCache.GetIDsByDsCateAndQueries(rule.Cate, rule.DatasourceQueries)
-			for _, dsId := range datasourceIds {
-				if !naming.DatasourceHashRing.IsHit(strconv.FormatInt(dsId, 10), fmt.Sprintf("%d", rule.Id), s.aconf.Heartbeat.Endpoint) {
-					continue
-				}
-				ds := s.datasourceCache.GetById(dsId)
-				if ds == nil {
-					logger.Debugf("datasource %d not found", dsId)
-					continue
-				}
-
-				if ds.PluginType != ruleType {
-					logger.Debugf("datasource %d category is %s not %s", dsId, ds.PluginType, ruleType)
-					continue
-				}
-
-				if ds.Status != "enabled" {
-					logger.Debugf("datasource %d status is %s", dsId, ds.Status)
-					continue
-				}
-				processor := process.NewProcessor(s.aconf.Heartbeat.EngineName, rule, dsId, s.alertRuleCache, s.targetCache, s.targetsOfAlertRuleCache, s.busiGroupCache, s.alertMuteCache, s.datasourceCache, s.ctx, s.stats)
-
-				alertRule := NewAlertRuleWorker(rule, dsId, processor, s.promClients, s.ctx)
-				alertRuleWorkers[alertRule.Hash()] = alertRule
-			}
-		} else if rule.IsHostRule() {
-			// all host rule will be processed by center instance
-			if !naming.DatasourceHashRing.IsHit(s.aconf.Heartbeat.EngineName, strconv.FormatInt(rule.Id, 10), s.aconf.Heartbeat.Endpoint) {
-				continue
-			}
-			processor := process.NewProcessor(s.aconf.Heartbeat.EngineName, rule, 0, s.alertRuleCache, s.targetCache, s.targetsOfAlertRuleCache, s.busiGroupCache, s.alertMuteCache, s.datasourceCache, s.ctx, s.stats)
-			alertRule := NewAlertRuleWorker(rule, 0, processor, s.promClients, s.ctx)
-			alertRuleWorkers[alertRule.Hash()] = alertRule
-		} else {
-			// 如果 rule 不是通过 prometheus engine 来告警的，则创建为 externalRule
-			// if rule is not processed by prometheus engine, create it as externalRule
-			dsIds := s.datasourceCache.GetIDsByDsCateAndQueries(rule.Cate, rule.DatasourceQueries)
-			for _, dsId := range dsIds {
-				ds := s.datasourceCache.GetById(dsId)
-				if ds == nil {
-					logger.Debugf("datasource %d not found", dsId)
-					continue
-				}
-
-				if ds.Status != "enabled" {
-					logger.Debugf("datasource %d status is %s", dsId, ds.Status)
-					continue
-				}
-				processor := process.NewProcessor(s.aconf.Heartbeat.EngineName, rule, dsId, s.alertRuleCache, s.targetCache, s.targetsOfAlertRuleCache, s.busiGroupCache, s.alertMuteCache, s.datasourceCache, s.ctx, s.stats)
-				externalRuleWorkers[processor.Key()] = processor
-			}
-		}
-	}
-
-	for hash, rule := range alertRuleWorkers {
-		if _, has := s.alertRules[hash]; !has {
-			rule.Prepare()
-			time.Sleep(time.Duration(20) * time.Millisecond)
-			rule.Start()
-			s.alertRules[hash] = rule
-		}
-	}
-
-	for hash, rule := range s.alertRules {
-		if _, has := alertRuleWorkers[hash]; !has {
-			rule.Stop()
-			delete(s.alertRules, hash)
-		}
-	}
-
-	s.ExternalProcessors.ExternalLock.Lock()
-	for key, processor := range externalRuleWorkers {
-		if curProcessor, has := s.ExternalProcessors.Processors[key]; has {
-			// rule存在,且hash一致,认为没有变更,这里可以根据需求单独实现一个关联数据更多的hash函数
-			if processor.Hash() == curProcessor.Hash() {
-				continue
-			}
-		}
-
-		// 现有规则中没有rule以及有rule但hash不一致的场景，需要触发rule的update
-		processor.RecoverAlertCurEventFromDb()
-		s.ExternalProcessors.Processors[key] = processor
-	}
-
-	for key := range s.ExternalProcessors.Processors {
-		if _, has := externalRuleWorkers[key]; !has {
-			delete(s.ExternalProcessors.Processors, key)
-		}
-	}
-	s.ExternalProcessors.ExternalLock.Unlock()
-}

alert/eval/eval_test.go

@@ -1,458 +0,0 @@
-package eval
-
-import (
-	"reflect"
-	"testing"
-
-	"golang.org/x/exp/slices"
-)
-
-var (
-	reHashTagIndex1 = map[uint64][][]uint64{
-		1: {
-			{1, 2}, {3, 4},
-		},
-		2: {
-			{5, 6}, {7, 8},
-		},
-	}
-	reHashTagIndex2 = map[uint64][][]uint64{
-		1: {
-			{9, 10}, {11, 12},
-		},
-		3: {
-			{13, 14}, {15, 16},
-		},
-	}
-	seriesTagIndex1 = map[uint64][]uint64{
-		1: {1, 2, 3, 4},
-		2: {5, 6, 7, 8},
-	}
-	seriesTagIndex2 = map[uint64][]uint64{
-		1: {9, 10, 11, 12},
-		3: {13, 14, 15, 16},
-	}
-)
-
-func Test_originalJoin(t *testing.T) {
-	type args struct {
-		seriesTagIndex1 map[uint64][]uint64
-		seriesTagIndex2 map[uint64][]uint64
-	}
-	tests := []struct {
-		name string
-		args args
-		want map[uint64][]uint64
-	}{
-		{
-			name: "original join",
-			args: args{
-				seriesTagIndex1: map[uint64][]uint64{
-					1: {1, 2, 3, 4},
-					2: {5, 6, 7, 8},
-				},
-				seriesTagIndex2: map[uint64][]uint64{
-					1: {9, 10, 11, 12},
-					3: {13, 14, 15, 16},
-				},
-			},
-			want: map[uint64][]uint64{
-				1: {1, 2, 3, 4, 9, 10, 11, 12},
-				2: {5, 6, 7, 8},
-				3: {13, 14, 15, 16},
-			},
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := originalJoin(tt.args.seriesTagIndex1, tt.args.seriesTagIndex2); !reflect.DeepEqual(got, tt.want) {
-				t.Errorf("originalJoin() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-func Test_exclude(t *testing.T) {
-	type args struct {
-		reHashTagIndex1 map[uint64][][]uint64
-		reHashTagIndex2 map[uint64][][]uint64
-	}
-	tests := []struct {
-		name string
-		args args
-		want map[uint64][]uint64
-	}{
-		{
-			name: "left exclude",
-			args: args{
-				reHashTagIndex1: reHashTagIndex1,
-				reHashTagIndex2: reHashTagIndex2,
-			},
-			want: map[uint64][]uint64{
-				0: {5, 6},
-				1: {7, 8},
-			},
-		},
-		{
-			name: "right exclude",
-			args: args{
-				reHashTagIndex1: reHashTagIndex2,
-				reHashTagIndex2: reHashTagIndex1,
-			},
-			want: map[uint64][]uint64{
-				3: {13, 14},
-				4: {15, 16},
-			},
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := exclude(tt.args.reHashTagIndex1, tt.args.reHashTagIndex2); !allValueDeepEqual(flatten(got), tt.want) {
-				t.Errorf("exclude() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-func Test_noneJoin(t *testing.T) {
-	type args struct {
-		seriesTagIndex1 map[uint64][]uint64
-		seriesTagIndex2 map[uint64][]uint64
-	}
-	tests := []struct {
-		name string
-		args args
-		want map[uint64][]uint64
-	}{
-		{
-			name: "none join, direct splicing",
-			args: args{
-				seriesTagIndex1: seriesTagIndex1,
-				seriesTagIndex2: seriesTagIndex2,
-			},
-			want: map[uint64][]uint64{
-				0: {1, 2, 3, 4},
-				1: {5, 6, 7, 8},
-				2: {9, 10, 11, 12},
-				3: {13, 14, 15, 16},
-			},
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := noneJoin(tt.args.seriesTagIndex1, tt.args.seriesTagIndex2); !allValueDeepEqual(got, tt.want) {
-				t.Errorf("noneJoin() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-func Test_cartesianJoin(t *testing.T) {
-	type args struct {
-		seriesTagIndex1 map[uint64][]uint64
-		seriesTagIndex2 map[uint64][]uint64
-	}
-	tests := []struct {
-		name string
-		args args
-		want map[uint64][]uint64
-	}{
-		{
-			name: "cartesian join",
-			args: args{
-				seriesTagIndex1: seriesTagIndex1,
-				seriesTagIndex2: seriesTagIndex2,
-			},
-			want: map[uint64][]uint64{
-				0: {1, 2, 3, 4, 9, 10, 11, 12},
-				1: {5, 6, 7, 8, 9, 10, 11, 12},
-				2: {5, 6, 7, 8, 13, 14, 15, 16},
-				3: {1, 2, 3, 4, 13, 14, 15, 16},
-			},
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := cartesianJoin(tt.args.seriesTagIndex1, tt.args.seriesTagIndex2); !allValueDeepEqual(got, tt.want) {
-				t.Errorf("cartesianJoin() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-func Test_onJoin(t *testing.T) {
-	type args struct {
-		reHashTagIndex1 map[uint64][][]uint64
-		reHashTagIndex2 map[uint64][][]uint64
-		joinType        JoinType
-	}
-	tests := []struct {
-		name string
-		args args
-		want map[uint64][]uint64
-	}{
-		{
-			name: "left join",
-			args: args{
-				reHashTagIndex1: reHashTagIndex1,
-				reHashTagIndex2: reHashTagIndex2,
-				joinType:        Left,
-			},
-			want: map[uint64][]uint64{
-				1: {1, 2, 9, 10},
-				2: {3, 4, 9, 10},
-				3: {1, 2, 11, 12},
-				4: {3, 4, 11, 12},
-				5: {5, 6},
-				6: {7, 8},
-			},
-		},
-		{
-			name: "right join",
-			args: args{
-				reHashTagIndex1: reHashTagIndex2,
-				reHashTagIndex2: reHashTagIndex1,
-				joinType:        Right,
-			},
-			want: map[uint64][]uint64{
-				1: {1, 2, 9, 10},
-				2: {3, 4, 9, 10},
-				3: {1, 2, 11, 12},
-				4: {3, 4, 11, 12},
-				5: {13, 14},
-				6: {15, 16},
-			},
-		},
-
-		{
-			name: "inner join",
-			args: args{
-				reHashTagIndex1: reHashTagIndex1,
-				reHashTagIndex2: reHashTagIndex2,
-				joinType:        Inner,
-			},
-			want: map[uint64][]uint64{
-				1: {1, 2, 9, 10},
-				2: {3, 4, 9, 10},
-				3: {1, 2, 11, 12},
-				4: {3, 4, 11, 12},
-			},
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := onJoin(tt.args.reHashTagIndex1, tt.args.reHashTagIndex2, tt.args.joinType); !allValueDeepEqual(flatten(got), tt.want) {
-				t.Errorf("onJoin() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-// allValueDeepEqual 判断 map 的 value 是否相同，不考虑 key
-func allValueDeepEqual(got, want map[uint64][]uint64) bool {
-	if len(got) != len(want) {
-		return false
-	}
-	for _, v1 := range got {
-		curEqual := false
-		slices.Sort(v1)
-		for _, v2 := range want {
-			slices.Sort(v2)
-			if reflect.DeepEqual(v1, v2) {
-				curEqual = true
-				break
-			}
-		}
-		if !curEqual {
-			return false
-		}
-	}
-	return true
-}
-
-// allValueDeepEqualOmitOrder 判断两个字符串切片是否相等，不考虑顺序
-func allValueDeepEqualOmitOrder(got, want []string) bool {
-	if len(got) != len(want) {
-		return false
-	}
-	slices.Sort(got)
-	slices.Sort(want)
-	for i := range got {
-		if got[i] != want[i] {
-			return false
-		}
-	}
-	return true
-}
-
-func Test_removeVal(t *testing.T) {
-	type args struct {
-		promql string
-	}
-	tests := []struct {
-		name string
-		args args
-		want string
-	}{
-		// TODO: Add test cases.
-		{
-			name: "removeVal1",
-			args: args{
-				promql: "mem{test1=\"$test1\",test2=\"$test2\",test3=\"$test3\"} > $val",
-			},
-			want: "mem{} > $val",
-		},
-		{
-			name: "removeVal2",
-			args: args{
-				promql: "mem{test1=\"test1\",test2=\"$test2\",test3=\"$test3\"} > $val",
-			},
-			want: "mem{test1=\"test1\"} > $val",
-		},
-		{
-			name: "removeVal3",
-			args: args{
-				promql: "mem{test1=\"$test1\",test2=\"test2\",test3=\"$test3\"} > $val",
-			},
-			want: "mem{test2=\"test2\"} > $val",
-		},
-		{
-			name: "removeVal4",
-			args: args{
-				promql: "mem{test1=\"$test1\",test2=\"$test2\",test3=\"test3\"} > $val",
-			},
-			want: "mem{test3=\"test3\"} > $val",
-		},
-		{
-			name: "removeVal5",
-			args: args{
-				promql: "mem{test1=\"$test1\",test2=\"test2\",test3=\"test3\"} > $val",
-			},
-			want: "mem{test2=\"test2\",test3=\"test3\"} > $val",
-		},
-		{
-			name: "removeVal6",
-			args: args{
-				promql: "mem{test1=\"test1\",test2=\"$test2\",test3=\"test3\"} > $val",
-			},
-			want: "mem{test1=\"test1\",test3=\"test3\"} > $val",
-		},
-		{
-			name: "removeVal7",
-			args: args{
-				promql: "mem{test1=\"test1\",test2=\"test2\",test3='$test3'} > $val",
-			},
-			want: "mem{test1=\"test1\",test2=\"test2\"} > $val",
-		},
-		{
-			name: "removeVal8",
-			args: args{
-				promql: "mem{test1=\"test1\",test2=\"test2\",test3=\"test3\"} > $val",
-			},
-			want: "mem{test1=\"test1\",test2=\"test2\",test3=\"test3\"} > $val",
-		},
-		{
-			name: "removeVal9",
-			args: args{
-				promql: "mem{test1=\"$test1\",test2=\"test2\"} > $val1 and mem{test3=\"test3\",test4=\"test4\"} > $val2",
-			},
-			want: "mem{test2=\"test2\"} > $val1 and mem{test3=\"test3\",test4=\"test4\"} > $val2",
-		},
-		{
-			name: "removeVal10",
-			args: args{
-				promql: "mem{test1=\"test1\",test2='$test2'} > $val1 and mem{test3=\"test3\",test4=\"test4\"} > $val2",
-			},
-			want: "mem{test1=\"test1\"} > $val1 and mem{test3=\"test3\",test4=\"test4\"} > $val2",
-		},
-		{
-			name: "removeVal11",
-			args: args{
-				promql: "mem{test1='test1',test2=\"test2\"} > $val1 and mem{test3=\"$test3\",test4=\"test4\"} > $val2",
-			},
-			want: "mem{test1='test1',test2=\"test2\"} > $val1 and mem{test4=\"test4\"} > $val2",
-		},
-		{
-			name: "removeVal12",
-			args: args{
-				promql: "mem{test1=\"test1\",test2=\"test2\"} > $val1 and mem{test3=\"test3\",test4=\"$test4\"} > $val2",
-			},
-			want: "mem{test1=\"test1\",test2=\"test2\"} > $val1 and mem{test3=\"test3\"} > $val2",
-		},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			if got := removeVal(tt.args.promql); got != tt.want {
-				t.Errorf("removeVal() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}
-
-func TestExtractVarMapping(t *testing.T) {
-	tests := []struct {
-		name   string
-		promql string
-		want   map[string]string
-	}{
-		{
-			name:   "单个花括号单个变量",
-			promql: `mem_used_percent{host="$my_host"} > $val`,
-			want:   map[string]string{"my_host": "host"},
-		},
-		{
-			name:   "单个花括号多个变量",
-			promql: `mem_used_percent{host="$my_host",region="$region",env="prod"} > $val`,
-			want:   map[string]string{"my_host": "host", "region": "region"},
-		},
-		{
-			name:   "多个花括号多个变量",
-			promql: `sum(rate(mem_used_percent{host="$my_host"})) by (instance) + avg(node_load1{region="$region"}) > $val`,
-			want:   map[string]string{"my_host": "host", "region": "region"},
-		},
-		{
-			name:   "相同变量出现多次",
-			promql: `sum(rate(mem_used_percent{host="$my_host"})) + avg(node_load1{host="$my_host"}) > $val`,
-			want:   map[string]string{"my_host": "host"},
-		},
-		{
-			name:   "没有变量",
-			promql: `mem_used_percent{host="localhost",region="cn"} > 80`,
-			want:   map[string]string{},
-		},
-		{
-			name:   "没有花括号",
-			promql: `80 > $val`,
-			want:   map[string]string{},
-		},
-		{
-			name:   "格式不规范的标签",
-			promql: `mem_used_percent{host=$my_host,region = $region} > $val`,
-			want:   map[string]string{"my_host": "host", "region": "region"},
-		},
-		{
-			name:   "空花括号",
-			promql: `mem_used_percent{} > $val`,
-			want:   map[string]string{},
-		},
-		{
-			name:   "不完整的花括号",
-			promql: `mem_used_percent{host="$my_host"`,
-			want:   map[string]string{},
-		},
-		{
-			name:   "复杂表达式",
-			promql: `sum(rate(http_requests_total{handler="$handler",code="$code"}[5m])) by (handler) / sum(rate(http_requests_total{handler="$handler"}[5m])) by (handler) * 100 > $threshold`,
-			want:   map[string]string{"handler": "handler", "code": "code"},
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			got := ExtractVarMapping(tt.promql)
-			if !reflect.DeepEqual(got, tt.want) {
-				t.Errorf("ExtractVarMapping() = %v, want %v", got, tt.want)
-			}
-		})
-	}
-}

alert/mute/mute.go

@@ -1,204 +0,0 @@
-package mute
-
-import (
-	"strconv"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/common"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-func IsMuted(rule *models.AlertRule, event *models.AlertCurEvent, targetCache *memsto.TargetCacheType, alertMuteCache *memsto.AlertMuteCacheType) (bool, string, int64) {
-	if rule.Disabled == 1 {
-		return true, "rule disabled", 0
-	}
-
-	if TimeSpanMuteStrategy(rule, event) {
-		return true, "rule is not effective for period of time", 0
-	}
-
-	if IdentNotExistsMuteStrategy(rule, event, targetCache) {
-		return true, "ident not exists mute", 0
-	}
-
-	if BgNotMatchMuteStrategy(rule, event, targetCache) {
-		return true, "bg not match mute", 0
-	}
-
-	hit, muteId := EventMuteStrategy(event, alertMuteCache)
-	if hit {
-		return true, "match mute rule", muteId
-	}
-
-	return false, "", 0
-}
-
-// TimeSpanMuteStrategy 根据规则配置的告警生效时间段过滤,如果产生的告警不在规则配置的告警生效时间段内,则不告警,即被mute
-// 时间范围，左闭右开，默认范围：00:00-24:00
-func TimeSpanMuteStrategy(rule *models.AlertRule, event *models.AlertCurEvent) bool {
-	tm := time.Unix(event.TriggerTime, 0)
-	triggerTime := tm.Format("15:04")
-	triggerWeek := strconv.Itoa(int(tm.Weekday()))
-
-	if rule.EnableDaysOfWeek == "" {
-		// 如果规则没有配置生效时间，则默认全天生效
-
-		return false
-	}
-
-	enableStime := strings.Fields(rule.EnableStime)
-	enableEtime := strings.Fields(rule.EnableEtime)
-	enableDaysOfWeek := strings.Split(rule.EnableDaysOfWeek, ";")
-	length := len(enableDaysOfWeek)
-	// enableStime,enableEtime,enableDaysOfWeek三者长度肯定相同，这里循环一个即可
-	for i := 0; i < length; i++ {
-		enableDaysOfWeek[i] = strings.Replace(enableDaysOfWeek[i], "7", "0", 1)
-		if !strings.Contains(enableDaysOfWeek[i], triggerWeek) {
-			continue
-		}
-
-		if enableStime[i] < enableEtime[i] {
-			if enableEtime[i] == "23:59" {
-				// 02:00-23:59，这种情况做个特殊处理，相当于左闭右闭区间了
-				if triggerTime < enableStime[i] {
-					// mute, 即没生效
-					continue
-				}
-			} else {
-				// 02:00-04:00 或者 02:00-24:00
-				if triggerTime < enableStime[i] || triggerTime >= enableEtime[i] {
-					// mute, 即没生效
-					continue
-				}
-			}
-		} else if enableStime[i] > enableEtime[i] {
-			// 21:00-09:00
-			if triggerTime < enableStime[i] && triggerTime >= enableEtime[i] {
-				// mute, 即没生效
-				continue
-			}
-		}
-
-		// 到这里说明当前时刻在告警规则的某组生效时间范围内，即没有 mute，直接返回 false
-		return false
-	}
-
-	return true
-}
-
-// IdentNotExistsMuteStrategy 根据ident是否存在过滤,如果ident不存在,则target_up的告警直接过滤掉
-func IdentNotExistsMuteStrategy(rule *models.AlertRule, event *models.AlertCurEvent, targetCache *memsto.TargetCacheType) bool {
-	ident, has := event.TagsMap["ident"]
-	if !has {
-		return false
-	}
-	_, exists := targetCache.Get(ident)
-	// 如果是target_up的告警,且ident已经不存在了,直接过滤掉
-	// 这里的判断有点太粗暴了,但是目前没有更好的办法
-	if !exists && strings.Contains(rule.PromQl, "target_up") {
-		logger.Debugf("[%s] mute: rule_eval:%d cluster:%s ident:%s", "IdentNotExistsMuteStrategy", rule.Id, event.Cluster, ident)
-		return true
-	}
-	return false
-}
-
-// BgNotMatchMuteStrategy 当规则开启只在bg内部告警时,对于非bg内部的机器过滤
-func BgNotMatchMuteStrategy(rule *models.AlertRule, event *models.AlertCurEvent, targetCache *memsto.TargetCacheType) bool {
-	// 没有开启BG内部告警,直接不过滤
-	if rule.EnableInBG == 0 {
-		return false
-	}
-
-	ident, has := event.TagsMap["ident"]
-	if !has {
-		return false
-	}
-
-	target, exists := targetCache.Get(ident)
-	// 对于包含ident的告警事件，check一下ident所属bg和rule所属bg是否相同
-	// 如果告警规则选择了只在本BG生效，那其他BG的机器就不能因此规则产生告警
-	if exists && !target.MatchGroupId(rule.GroupId) {
-		logger.Debugf("[%s] mute: rule_eval:%d cluster:%s", "BgNotMatchMuteStrategy", rule.Id, event.Cluster)
-		return true
-	}
-	return false
-}
-
-func EventMuteStrategy(event *models.AlertCurEvent, alertMuteCache *memsto.AlertMuteCacheType) (bool, int64) {
-	mutes, has := alertMuteCache.Gets(event.GroupId)
-	if !has || len(mutes) == 0 {
-		return false, 0
-	}
-
-	for i := 0; i < len(mutes); i++ {
-		if MatchMute(event, mutes[i]) {
-			return true, mutes[i].Id
-		}
-	}
-
-	return false, 0
-}
-
-// MatchMute 如果传入了clock这个可选参数，就表示使用这个clock表示的时间，否则就从event的字段中取TriggerTime
-func MatchMute(event *models.AlertCurEvent, mute *models.AlertMute, clock ...int64) bool {
-	if mute.Disabled == 1 {
-		return false
-	}
-
-	// 如果不是全局的，判断 匹配的 datasource id
-	if len(mute.DatasourceIdsJson) != 0 && mute.DatasourceIdsJson[0] != 0 && event.DatasourceId != 0 {
-		idm := make(map[int64]struct{}, len(mute.DatasourceIdsJson))
-		for i := 0; i < len(mute.DatasourceIdsJson); i++ {
-			idm[mute.DatasourceIdsJson[i]] = struct{}{}
-		}
-
-		// 判断 event.datasourceId 是否包含在 idm 中
-		if _, has := idm[event.DatasourceId]; !has {
-			return false
-		}
-	}
-
-	if mute.MuteTimeType == models.TimeRange {
-		if !mute.IsWithinTimeRange(event.TriggerTime) {
-			return false
-		}
-	} else if mute.MuteTimeType == models.Periodic {
-		ts := event.TriggerTime
-		if len(clock) > 0 {
-			ts = clock[0]
-		}
-
-		if !mute.IsWithinPeriodicMute(ts) {
-			return false
-		}
-	} else {
-		logger.Warningf("mute time type invalid, %d", mute.MuteTimeType)
-		return false
-	}
-
-	var matchSeverity bool
-	if len(mute.SeveritiesJson) > 0 {
-		for _, s := range mute.SeveritiesJson {
-			if event.Severity == s || s == 0 {
-				matchSeverity = true
-				break
-			}
-		}
-	} else {
-		matchSeverity = true
-	}
-
-	if !matchSeverity {
-		return false
-	}
-
-	if mute.ITags == nil || len(mute.ITags) == 0 {
-		return true
-	}
-
-	return common.MatchTags(event.TagsMap, mute.ITags)
-}

alert/naming/hashring.go

@@ -1,85 +0,0 @@
-package naming
-
-import (
-	"errors"
-	"sync"
-
-	"github.com/toolkits/pkg/consistent"
-	"github.com/toolkits/pkg/logger"
-)
-
-const NodeReplicas = 500
-
-type DatasourceHashRingType struct {
-	sync.RWMutex
-	Rings map[string]*consistent.Consistent
-}
-
-// for alert_rule sharding
-var HostDatasource int64 = 99999999
-var DatasourceHashRing = DatasourceHashRingType{Rings: make(map[string]*consistent.Consistent)}
-
-func NewConsistentHashRing(replicas int32, nodes []string) *consistent.Consistent {
-	ret := consistent.New()
-	ret.NumberOfReplicas = int(replicas)
-	for i := 0; i < len(nodes); i++ {
-		ret.Add(nodes[i])
-	}
-	return ret
-}
-
-func RebuildConsistentHashRing(datasourceId string, nodes []string) {
-	r := consistent.New()
-	r.NumberOfReplicas = NodeReplicas
-	for i := 0; i < len(nodes); i++ {
-		r.Add(nodes[i])
-	}
-
-	DatasourceHashRing.Set(datasourceId, r)
-	logger.Infof("hash ring %s rebuild %+v", datasourceId, r.Members())
-}
-
-func (chr *DatasourceHashRingType) GetNode(datasourceId string, pk string) (string, error) {
-	chr.Lock()
-	defer chr.Unlock()
-	_, exists := chr.Rings[datasourceId]
-	if !exists {
-		chr.Rings[datasourceId] = NewConsistentHashRing(int32(NodeReplicas), []string{})
-	}
-
-	return chr.Rings[datasourceId].Get(pk)
-}
-
-func (chr *DatasourceHashRingType) IsHit(datasourceId string, pk string, currentNode string) bool {
-	node, err := chr.GetNode(datasourceId, pk)
-	if err != nil {
-		if !errors.Is(err, consistent.ErrEmptyCircle) {
-			logger.Errorf("rule id:%s is not work, datasource id:%s failed to get node from hashring:%v", pk, datasourceId, err)
-		}
-		return false
-	}
-	return node == currentNode
-}
-
-func (chr *DatasourceHashRingType) Set(datasourceId string, r *consistent.Consistent) {
-	chr.Lock()
-	defer chr.Unlock()
-	chr.Rings[datasourceId] = r
-}
-
-func (chr *DatasourceHashRingType) Del(datasourceId string) {
-	chr.Lock()
-	defer chr.Unlock()
-	delete(chr.Rings, datasourceId)
-}
-
-func (chr *DatasourceHashRingType) Clear(engineName string) {
-	chr.Lock()
-	defer chr.Unlock()
-	for id := range chr.Rings {
-		if id == engineName {
-			continue
-		}
-		delete(chr.Rings, id)
-	}
-}

alert/naming/heartbeat.go

@@ -1,192 +0,0 @@
-package naming
-
-import (
-	"fmt"
-	"sort"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/poster"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type Naming struct {
-	ctx             *ctx.Context
-	heartbeatConfig aconf.HeartbeatConfig
-	astats          *astats.Stats
-}
-
-func NewNaming(ctx *ctx.Context, heartbeat aconf.HeartbeatConfig, alertStats *astats.Stats) *Naming {
-	naming := &Naming{
-		ctx:             ctx,
-		heartbeatConfig: heartbeat,
-		astats:          alertStats,
-	}
-	naming.Heartbeats()
-	return naming
-}
-
-// local servers
-var localss map[int64]string
-var localHostServers map[string]string
-
-func (n *Naming) Heartbeats() error {
-	localss = make(map[int64]string)
-	localHostServers = make(map[string]string)
-	if err := n.heartbeat(); err != nil {
-		fmt.Println("failed to heartbeat:", err)
-		return err
-	}
-
-	go n.loopHeartbeat()
-	go n.loopDeleteInactiveInstances()
-	return nil
-}
-
-func (n *Naming) loopDeleteInactiveInstances() {
-	if !n.ctx.IsCenter {
-		return
-	}
-
-	interval := time.Duration(10) * time.Minute
-	for {
-		time.Sleep(interval)
-		n.DeleteInactiveInstances()
-	}
-}
-
-func (n *Naming) DeleteInactiveInstances() {
-	err := models.DB(n.ctx).Where("clock < ?", time.Now().Unix()-600).Delete(new(models.AlertingEngines)).Error
-	if err != nil {
-		logger.Errorf("delete inactive instances err:%v", err)
-	}
-}
-
-func (n *Naming) loopHeartbeat() {
-	interval := time.Duration(n.heartbeatConfig.Interval) * time.Millisecond
-	for {
-		time.Sleep(interval)
-		if err := n.heartbeat(); err != nil {
-			logger.Warning(err)
-		}
-	}
-}
-
-func (n *Naming) heartbeat() error {
-	var datasourceIds []int64
-	var err error
-
-	// 在页面上维护实例和集群的对应关系
-	datasourceIds, err = models.GetDatasourceIdsByEngineName(n.ctx, n.heartbeatConfig.EngineName)
-	if err != nil {
-		return err
-	}
-
-	if len(datasourceIds) == 0 {
-		err := models.AlertingEngineHeartbeatWithCluster(n.ctx, n.heartbeatConfig.Endpoint, n.heartbeatConfig.EngineName, 0)
-		if err != nil {
-			logger.Warningf("heartbeat with cluster %s err:%v", "", err)
-			n.astats.CounterHeartbeatErrorTotal.WithLabelValues().Inc()
-		}
-	} else {
-		for i := 0; i < len(datasourceIds); i++ {
-			err := models.AlertingEngineHeartbeatWithCluster(n.ctx, n.heartbeatConfig.Endpoint, n.heartbeatConfig.EngineName, datasourceIds[i])
-			if err != nil {
-				logger.Warningf("heartbeat with cluster %d err:%v", datasourceIds[i], err)
-				n.astats.CounterHeartbeatErrorTotal.WithLabelValues().Inc()
-			}
-		}
-	}
-
-	if len(datasourceIds) == 0 {
-		DatasourceHashRing.Clear(n.heartbeatConfig.EngineName)
-		for dsId := range localss {
-			delete(localss, dsId)
-		}
-	}
-
-	newDatasource := make(map[int64]struct{})
-	for i := 0; i < len(datasourceIds); i++ {
-		newDatasource[datasourceIds[i]] = struct{}{}
-		servers, err := n.ActiveServers(datasourceIds[i])
-		if err != nil {
-			logger.Warningf("hearbeat %d get active server err:%v", datasourceIds[i], err)
-			n.astats.CounterHeartbeatErrorTotal.WithLabelValues().Inc()
-			continue
-		}
-
-		sort.Strings(servers)
-		newss := strings.Join(servers, " ")
-
-		oldss, exists := localss[datasourceIds[i]]
-		if exists && oldss == newss {
-			continue
-		}
-
-		RebuildConsistentHashRing(fmt.Sprintf("%d", datasourceIds[i]), servers)
-		localss[datasourceIds[i]] = newss
-	}
-
-	for dsId := range localss {
-		if _, exists := newDatasource[dsId]; !exists {
-			delete(localss, dsId)
-			DatasourceHashRing.Del(fmt.Sprintf("%d", dsId))
-		}
-	}
-
-	// host 告警使用的是 hash ring
-	err = models.AlertingEngineHeartbeatWithCluster(n.ctx, n.heartbeatConfig.Endpoint, n.heartbeatConfig.EngineName, HostDatasource)
-	if err != nil {
-		logger.Warningf("heartbeat with cluster %s err:%v", "", err)
-		n.astats.CounterHeartbeatErrorTotal.WithLabelValues().Inc()
-	}
-
-	servers, err := n.ActiveServersByEngineName()
-	if err != nil {
-		logger.Warningf("hearbeat %d get active server err:%v", HostDatasource, err)
-		n.astats.CounterHeartbeatErrorTotal.WithLabelValues().Inc()
-		return nil
-	}
-
-	sort.Strings(servers)
-	newss := strings.Join(servers, " ")
-
-	oldss, exists := localHostServers[n.heartbeatConfig.EngineName]
-	if exists && oldss == newss {
-		return nil
-	}
-
-	RebuildConsistentHashRing(n.heartbeatConfig.EngineName, servers)
-	localHostServers[n.heartbeatConfig.EngineName] = newss
-
-	return nil
-}
-
-func (n *Naming) ActiveServers(datasourceId int64) ([]string, error) {
-	if datasourceId == -1 {
-		return nil, fmt.Errorf("cluster is empty")
-	}
-
-	if !n.ctx.IsCenter {
-		lst, err := poster.GetByUrls[[]string](n.ctx, "/v1/n9e/servers-active?dsid="+fmt.Sprintf("%d", datasourceId))
-		return lst, err
-	}
-
-	// 30秒内有心跳，就认为是活的
-	return models.AlertingEngineGetsInstances(n.ctx, "datasource_id = ? and clock > ?", datasourceId, time.Now().Unix()-30)
-}
-
-func (n *Naming) ActiveServersByEngineName() ([]string, error) {
-	if !n.ctx.IsCenter {
-		lst, err := poster.GetByUrls[[]string](n.ctx, "/v1/n9e/servers-active?engine_name="+n.heartbeatConfig.EngineName)
-		return lst, err
-	}
-
-	// 30秒内有心跳，就认为是活的
-	return models.AlertingEngineGetsInstances(n.ctx, "engine_cluster = ? and clock > ?", n.heartbeatConfig.EngineName, time.Now().Unix()-30)
-}

alert/naming/leader.go

@@ -1,28 +0,0 @@
-package naming
-
-import (
-	"sort"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-func (n *Naming) IamLeader() bool {
-	if !n.ctx.IsCenter {
-		return false
-	}
-
-	servers, err := n.ActiveServersByEngineName()
-	if err != nil {
-		logger.Errorf("failed to get active servers: %v", err)
-		return false
-	}
-
-	if len(servers) == 0 {
-		logger.Errorf("active servers empty")
-		return false
-	}
-
-	sort.Strings(servers)
-
-	return n.heartbeatConfig.Endpoint == servers[0]
-}

alert/pipeline/pipeline.go

@@ -1,12 +0,0 @@
-package pipeline
-
-import (
-	_ "github.com/ccfos/nightingale/v6/alert/pipeline/processor/aisummary"
-	_ "github.com/ccfos/nightingale/v6/alert/pipeline/processor/callback"
-	_ "github.com/ccfos/nightingale/v6/alert/pipeline/processor/eventdrop"
-	_ "github.com/ccfos/nightingale/v6/alert/pipeline/processor/eventupdate"
-	_ "github.com/ccfos/nightingale/v6/alert/pipeline/processor/relabel"
-)
-
-func Init() {
-}

alert/pipeline/processor/aisummary/ai_summary.go

@@ -1,198 +0,0 @@
-package aisummary
-
-import (
-	"bytes"
-	"crypto/tls"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-	"text/template"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/callback"
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/common"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/tplx"
-)
-
-const (
-	HTTP_STATUS_SUCCESS_MAX = 299
-)
-
-// AISummaryConfig 配置结构体
-type AISummaryConfig struct {
-	callback.HTTPConfig
-	ModelName      string                 `json:"model_name"`
-	APIKey         string                 `json:"api_key"`
-	PromptTemplate string                 `json:"prompt_template"`
-	CustomParams   map[string]interface{} `json:"custom_params"`
-}
-
-type Message struct {
-	Role    string `json:"role"`
-	Content string `json:"content"`
-}
-
-type ChatCompletionResponse struct {
-	Choices []struct {
-		Message struct {
-			Content string `json:"content"`
-		} `json:"message"`
-	} `json:"choices"`
-}
-
-func init() {
-	models.RegisterProcessor("ai_summary", &AISummaryConfig{})
-}
-
-func (c *AISummaryConfig) Init(settings interface{}) (models.Processor, error) {
-	result, err := common.InitProcessor[*AISummaryConfig](settings)
-	return result, err
-}
-
-func (c *AISummaryConfig) Process(ctx *ctx.Context, event *models.AlertCurEvent) (*models.AlertCurEvent, string, error) {
-	if c.Client == nil {
-		if err := c.initHTTPClient(); err != nil {
-			return event, "", fmt.Errorf("failed to initialize HTTP client: %v processor: %v", err, c)
-		}
-	}
-
-	// 准备告警事件信息
-	eventInfo, err := c.prepareEventInfo(event)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to prepare event info: %v processor: %v", err, c)
-	}
-
-	// 调用AI模型生成总结
-	summary, err := c.generateAISummary(eventInfo)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to generate AI summary: %v processor: %v", err, c)
-	}
-
-	// 将总结添加到annotations字段
-	if event.AnnotationsJSON == nil {
-		event.AnnotationsJSON = make(map[string]string)
-	}
-	event.AnnotationsJSON["ai_summary"] = summary
-
-	// 更新Annotations字段
-	b, err := json.Marshal(event.AnnotationsJSON)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to marshal annotations: %v processor: %v", err, c)
-	}
-	event.Annotations = string(b)
-
-	return event, "", nil
-}
-
-func (c *AISummaryConfig) initHTTPClient() error {
-	transport := &http.Transport{
-		TLSClientConfig: &tls.Config{InsecureSkipVerify: c.SkipSSLVerify},
-	}
-
-	if c.Proxy != "" {
-		proxyURL, err := url.Parse(c.Proxy)
-		if err != nil {
-			return fmt.Errorf("failed to parse proxy url: %v", err)
-		}
-		transport.Proxy = http.ProxyURL(proxyURL)
-	}
-
-	c.Client = &http.Client{
-		Timeout:   time.Duration(c.Timeout) * time.Millisecond,
-		Transport: transport,
-	}
-	return nil
-}
-
-func (c *AISummaryConfig) prepareEventInfo(event *models.AlertCurEvent) (string, error) {
-	var defs = []string{
-		"{{$event := .}}",
-	}
-
-	text := strings.Join(append(defs, c.PromptTemplate), "")
-	t, err := template.New("prompt").Funcs(template.FuncMap(tplx.TemplateFuncMap)).Parse(text)
-	if err != nil {
-		return "", fmt.Errorf("failed to parse prompt template: %v", err)
-	}
-
-	var body bytes.Buffer
-	err = t.Execute(&body, event)
-	if err != nil {
-		return "", fmt.Errorf("failed to execute prompt template: %v", err)
-	}
-
-	return body.String(), nil
-}
-
-func (c *AISummaryConfig) generateAISummary(eventInfo string) (string, error) {
-	// 构建基础请求参数
-	reqParams := map[string]interface{}{
-		"model": c.ModelName,
-		"messages": []Message{
-			{
-				Role:    "user",
-				Content: eventInfo,
-			},
-		},
-	}
-
-	// 合并自定义参数
-	for k, v := range c.CustomParams {
-		reqParams[k] = v
-	}
-
-	// 序列化请求体
-	jsonData, err := json.Marshal(reqParams)
-	if err != nil {
-		return "", fmt.Errorf("failed to marshal request body: %v", err)
-	}
-
-	// 创建HTTP请求
-	req, err := http.NewRequest("POST", c.URL, bytes.NewBuffer(jsonData))
-	if err != nil {
-		return "", fmt.Errorf("failed to create request: %v", err)
-	}
-
-	// 设置请求头
-	req.Header.Set("Authorization", "Bearer "+c.APIKey)
-	req.Header.Set("Content-Type", "application/json")
-	for k, v := range c.Headers {
-		req.Header.Set(k, v)
-	}
-
-	// 发送请求
-	resp, err := c.Client.Do(req)
-	if err != nil {
-		return "", fmt.Errorf("failed to send request: %v", err)
-	}
-	defer resp.Body.Close()
-
-	// 检查响应状态码
-	if resp.StatusCode > HTTP_STATUS_SUCCESS_MAX {
-		body, _ := io.ReadAll(resp.Body)
-		return "", fmt.Errorf("unexpected status code: %d, body: %s", resp.StatusCode, string(body))
-	}
-
-	// 读取响应
-	body, err := io.ReadAll(resp.Body)
-	if err != nil {
-		return "", fmt.Errorf("failed to read response body: %v", err)
-	}
-
-	// 解析响应
-	var chatResp ChatCompletionResponse
-	if err := json.Unmarshal(body, &chatResp); err != nil {
-		return "", fmt.Errorf("failed to unmarshal response: %v", err)
-	}
-
-	if len(chatResp.Choices) == 0 {
-		return "", fmt.Errorf("no response from AI model")
-	}
-
-	return chatResp.Choices[0].Message.Content, nil
-}

alert/pipeline/processor/aisummary/ai_summary_test.go

@@ -1,69 +0,0 @@
-package aisummary
-
-import (
-	"testing"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/callback"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/stretchr/testify/assert"
-)
-
-func TestAISummaryConfig_Process(t *testing.T) {
-	// 创建测试配置
-	config := &AISummaryConfig{
-		HTTPConfig: callback.HTTPConfig{
-			URL:           "https://generativelanguage.googleapis.com/v1beta/openai/chat/completions",
-			Timeout:       30000,
-			SkipSSLVerify: true,
-			Headers: map[string]string{
-				"Content-Type": "application/json",
-			},
-		},
-		ModelName:      "gemini-2.0-flash",
-		APIKey:         "*",
-		PromptTemplate: "告警规则：{{$event.RuleName}}\n严重程度：{{$event.Severity}}",
-		CustomParams: map[string]interface{}{
-			"temperature": 0.7,
-			"max_tokens":  2000,
-			"top_p":       0.9,
-		},
-	}
-
-	// 创建测试事件
-	event := &models.AlertCurEvent{
-		RuleName: "Test Rule",
-		Severity: 1,
-		TagsMap: map[string]string{
-			"host": "test-host",
-		},
-		AnnotationsJSON: map[string]string{
-			"description": "Test alert",
-		},
-	}
-
-	// 测试模板处理
-	eventInfo, err := config.prepareEventInfo(event)
-	assert.NoError(t, err)
-	assert.Contains(t, eventInfo, "Test Rule")
-	assert.Contains(t, eventInfo, "1")
-
-	// 测试配置初始化
-	processor, err := config.Init(config)
-	assert.NoError(t, err)
-	assert.NotNil(t, processor)
-
-	// 测试处理函数
-	result, _, err := processor.Process(&ctx.Context{}, event)
-	assert.NoError(t, err)
-	assert.NotNil(t, result)
-	assert.NotEmpty(t, result.AnnotationsJSON["ai_summary"])
-
-	// 展示处理结果
-	t.Log("\n=== 处理结果 ===")
-	t.Logf("告警规则: %s", result.RuleName)
-	t.Logf("严重程度: %d", result.Severity)
-	t.Logf("标签: %v", result.TagsMap)
-	t.Logf("原始注释: %v", result.AnnotationsJSON["description"])
-	t.Logf("AI总结: %s", result.AnnotationsJSON["ai_summary"])
-}

alert/pipeline/processor/callback/callback.go

@@ -1,103 +0,0 @@
-package callback
-
-import (
-	"crypto/tls"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/common"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type HTTPConfig struct {
-	URL           string            `json:"url"`
-	Method        string            `json:"method,omitempty"`
-	Body          string            `json:"body,omitempty"`
-	Headers       map[string]string `json:"header"`
-	AuthUsername  string            `json:"auth_username"`
-	AuthPassword  string            `json:"auth_password"`
-	Timeout       int               `json:"timeout"` // 单位:ms
-	SkipSSLVerify bool              `json:"skip_ssl_verify"`
-	Proxy         string            `json:"proxy"`
-	Client        *http.Client      `json:"-"`
-}
-
-// RelabelConfig
-type CallbackConfig struct {
-	HTTPConfig
-}
-
-func init() {
-	models.RegisterProcessor("callback", &CallbackConfig{})
-}
-
-func (c *CallbackConfig) Init(settings interface{}) (models.Processor, error) {
-	result, err := common.InitProcessor[*CallbackConfig](settings)
-	return result, err
-}
-
-func (c *CallbackConfig) Process(ctx *ctx.Context, event *models.AlertCurEvent) (*models.AlertCurEvent, string, error) {
-	if c.Client == nil {
-		transport := &http.Transport{
-			TLSClientConfig: &tls.Config{InsecureSkipVerify: c.SkipSSLVerify},
-		}
-
-		if c.Proxy != "" {
-			proxyURL, err := url.Parse(c.Proxy)
-			if err != nil {
-				return event, "", fmt.Errorf("failed to parse proxy url: %v processor: %v", err, c)
-			} else {
-				transport.Proxy = http.ProxyURL(proxyURL)
-			}
-		}
-
-		c.Client = &http.Client{
-			Timeout:   time.Duration(c.Timeout) * time.Millisecond,
-			Transport: transport,
-		}
-	}
-
-	headers := make(map[string]string)
-	headers["Content-Type"] = "application/json"
-	for k, v := range c.Headers {
-		headers[k] = v
-	}
-
-	body, err := json.Marshal(event)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to marshal event: %v processor: %v", err, c)
-	}
-
-	req, err := http.NewRequest("POST", c.URL, strings.NewReader(string(body)))
-	if err != nil {
-		return event, "", fmt.Errorf("failed to create request: %v processor: %v", err, c)
-	}
-
-	for k, v := range headers {
-		req.Header.Set(k, v)
-	}
-
-	if c.AuthUsername != "" && c.AuthPassword != "" {
-		req.SetBasicAuth(c.AuthUsername, c.AuthPassword)
-	}
-
-	resp, err := c.Client.Do(req)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to send request: %v processor: %v", err, c)
-	}
-
-	b, err := io.ReadAll(resp.Body)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to read response body: %v processor: %v", err, c)
-	}
-
-	logger.Debugf("callback processor response body: %s", string(b))
-	return event, "callback success", nil
-}

alert/pipeline/processor/common/common.go

@@ -1,24 +0,0 @@
-package common
-
-import (
-	"encoding/json"
-)
-
-// InitProcessor 是一个通用的初始化处理器的方法
-// 使用泛型简化处理器初始化逻辑
-// T 必须是 models.Processor 接口的实现
-func InitProcessor[T any](settings interface{}) (T, error) {
-	var zero T
-	b, err := json.Marshal(settings)
-	if err != nil {
-		return zero, err
-	}
-
-	var result T
-	err = json.Unmarshal(b, &result)
-	if err != nil {
-		return zero, err
-	}
-
-	return result, nil
-}

alert/pipeline/processor/eventdrop/event_drop.go

@@ -1,60 +0,0 @@
-package eventdrop
-
-import (
-	"bytes"
-	"fmt"
-	"strings"
-	texttemplate "text/template"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/common"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/tplx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type EventDropConfig struct {
-	Content string `json:"content"`
-}
-
-func init() {
-	models.RegisterProcessor("event_drop", &EventDropConfig{})
-}
-
-func (c *EventDropConfig) Init(settings interface{}) (models.Processor, error) {
-	result, err := common.InitProcessor[*EventDropConfig](settings)
-	return result, err
-}
-
-func (c *EventDropConfig) Process(ctx *ctx.Context, event *models.AlertCurEvent) (*models.AlertCurEvent, string, error) {
-	// 使用背景是可以根据此处理器，实现对事件进行更加灵活的过滤的逻辑
-	// 在标签过滤和属性过滤都不满足需求时可以使用
-	// 如果模板执行结果为 true，则删除该事件
-
-	var defs = []string{
-		"{{ $event := . }}",
-		"{{ $labels := .TagsMap }}",
-		"{{ $value := .TriggerValue }}",
-	}
-
-	text := strings.Join(append(defs, c.Content), "")
-
-	tpl, err := texttemplate.New("eventdrop").Funcs(tplx.TemplateFuncMap).Parse(text)
-	if err != nil {
-		return event, "", fmt.Errorf("processor failed to parse template: %v processor: %v", err, c)
-	}
-
-	var body bytes.Buffer
-	if err = tpl.Execute(&body, event); err != nil {
-		return event, "", fmt.Errorf("processor failed to execute template: %v processor: %v", err, c)
-	}
-
-	result := strings.TrimSpace(body.String())
-	logger.Infof("processor eventdrop result: %v", result)
-	if result == "true" {
-		logger.Infof("processor eventdrop drop event: %v", event)
-		return event, "drop event success", nil
-	}
-
-	return event, "drop event failed", nil
-}

alert/pipeline/processor/eventupdate/event_update.go

@@ -1,96 +0,0 @@
-package eventupdate
-
-import (
-	"crypto/tls"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/callback"
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/common"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/toolkits/pkg/logger"
-)
-
-// RelabelConfig
-type EventUpdateConfig struct {
-	callback.HTTPConfig
-}
-
-func init() {
-	models.RegisterProcessor("event_update", &EventUpdateConfig{})
-}
-
-func (c *EventUpdateConfig) Init(settings interface{}) (models.Processor, error) {
-	result, err := common.InitProcessor[*EventUpdateConfig](settings)
-	return result, err
-}
-
-func (c *EventUpdateConfig) Process(ctx *ctx.Context, event *models.AlertCurEvent) (*models.AlertCurEvent, string, error) {
-	if c.Client == nil {
-		transport := &http.Transport{
-			TLSClientConfig: &tls.Config{InsecureSkipVerify: c.SkipSSLVerify},
-		}
-
-		if c.Proxy != "" {
-			proxyURL, err := url.Parse(c.Proxy)
-			if err != nil {
-				return event, "", fmt.Errorf("failed to parse proxy url: %v processor: %v", err, c)
-			} else {
-				transport.Proxy = http.ProxyURL(proxyURL)
-			}
-		}
-
-		c.Client = &http.Client{
-			Timeout:   time.Duration(c.Timeout) * time.Millisecond,
-			Transport: transport,
-		}
-	}
-
-	headers := make(map[string]string)
-	headers["Content-Type"] = "application/json"
-	for k, v := range c.Headers {
-		headers[k] = v
-	}
-
-	body, err := json.Marshal(event)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to marshal event: %v processor: %v", err, c)
-	}
-
-	req, err := http.NewRequest("POST", c.URL, strings.NewReader(string(body)))
-	if err != nil {
-		return event, "", fmt.Errorf("failed to create request: %v processor: %v", err, c)
-	}
-
-	for k, v := range headers {
-		req.Header.Set(k, v)
-	}
-
-	if c.AuthUsername != "" && c.AuthPassword != "" {
-		req.SetBasicAuth(c.AuthUsername, c.AuthPassword)
-	}
-
-	resp, err := c.Client.Do(req)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to send request: %v processor: %v", err, c)
-	}
-
-	b, err := io.ReadAll(resp.Body)
-	if err != nil {
-		return nil, "", fmt.Errorf("failed to read response body: %v processor: %v", err, c)
-	}
-	logger.Debugf("event update processor response body: %s", string(b))
-
-	err = json.Unmarshal(b, &event)
-	if err != nil {
-		return event, "", fmt.Errorf("failed to unmarshal response body: %v processor: %v", err, c)
-	}
-
-	return event, "", nil
-}

alert/pipeline/processor/relabel/relabel.go

@@ -1,107 +0,0 @@
-package relabel
-
-import (
-	"fmt"
-	"regexp"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/common"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pushgw/pconf"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-
-	"github.com/prometheus/common/model"
-	"github.com/prometheus/prometheus/prompb"
-)
-
-const (
-	REPLACE_DOT = "___"
-)
-
-// RelabelConfig
-type RelabelConfig struct {
-	SourceLabels  []string `json:"source_labels"`
-	Separator     string   `json:"separator"`
-	Regex         string   `json:"regex"`
-	RegexCompiled *regexp.Regexp
-	If            string `json:"if"`
-	IfRegex       *regexp.Regexp
-	Modulus       uint64 `json:"modulus"`
-	TargetLabel   string `json:"target_label"`
-	Replacement   string `json:"replacement"`
-	Action        string `json:"action"`
-}
-
-func init() {
-	models.RegisterProcessor("relabel", &RelabelConfig{})
-}
-
-func (r *RelabelConfig) Init(settings interface{}) (models.Processor, error) {
-	result, err := common.InitProcessor[*RelabelConfig](settings)
-	return result, err
-}
-
-func (r *RelabelConfig) Process(ctx *ctx.Context, event *models.AlertCurEvent) (*models.AlertCurEvent, string, error) {
-	sourceLabels := make([]model.LabelName, len(r.SourceLabels))
-	for i := range r.SourceLabels {
-		sourceLabels[i] = model.LabelName(strings.ReplaceAll(r.SourceLabels[i], ".", REPLACE_DOT))
-	}
-
-	relabelConfigs := []*pconf.RelabelConfig{
-		{
-			SourceLabels:  sourceLabels,
-			Separator:     r.Separator,
-			Regex:         r.Regex,
-			RegexCompiled: r.RegexCompiled,
-			If:            r.If,
-			IfRegex:       r.IfRegex,
-			Modulus:       r.Modulus,
-			TargetLabel:   r.TargetLabel,
-			Replacement:   r.Replacement,
-			Action:        r.Action,
-		},
-	}
-
-	EventRelabel(event, relabelConfigs)
-	return event, "", nil
-}
-
-func EventRelabel(event *models.AlertCurEvent, relabelConfigs []*pconf.RelabelConfig) {
-	labels := make([]prompb.Label, len(event.TagsJSON))
-	event.OriginalTagsJSON = make([]string, len(event.TagsJSON))
-	for i, tag := range event.TagsJSON {
-		label := strings.SplitN(tag, "=", 2)
-		if len(label) != 2 {
-			continue
-		}
-		event.OriginalTagsJSON[i] = tag
-
-		label[0] = strings.ReplaceAll(string(label[0]), ".", REPLACE_DOT)
-		labels[i] = prompb.Label{Name: label[0], Value: label[1]}
-	}
-
-	for i := 0; i < len(relabelConfigs); i++ {
-		if relabelConfigs[i].Replacement == "" {
-			relabelConfigs[i].Replacement = "$1"
-		}
-
-		if relabelConfigs[i].Separator == "" {
-			relabelConfigs[i].Separator = ";"
-		}
-
-		if relabelConfigs[i].Regex == "" {
-			relabelConfigs[i].Regex = "(.*)"
-		}
-	}
-
-	gotLabels := writer.Process(labels, relabelConfigs...)
-	event.TagsJSON = make([]string, len(gotLabels))
-	event.TagsMap = make(map[string]string, len(gotLabels))
-	for i, label := range gotLabels {
-		label.Name = strings.ReplaceAll(string(label.Name), REPLACE_DOT, ".")
-		event.TagsJSON[i] = fmt.Sprintf("%s=%s", label.Name, label.Value)
-		event.TagsMap[label.Name] = label.Value
-	}
-	event.Tags = strings.Join(event.TagsJSON, ",,")
-}

alert/process/alert_cur_event.go

@@ -1,74 +0,0 @@
-package process
-
-import (
-	"sync"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type AlertCurEventMap struct {
-	sync.RWMutex
-	Data map[string]*models.AlertCurEvent
-}
-
-func NewAlertCurEventMap(data map[string]*models.AlertCurEvent) *AlertCurEventMap {
-	if data == nil {
-		return &AlertCurEventMap{
-			Data: make(map[string]*models.AlertCurEvent),
-		}
-	}
-	return &AlertCurEventMap{
-		Data: data,
-	}
-}
-
-func (a *AlertCurEventMap) SetAll(data map[string]*models.AlertCurEvent) {
-	a.Lock()
-	defer a.Unlock()
-	a.Data = data
-}
-
-func (a *AlertCurEventMap) Set(key string, value *models.AlertCurEvent) {
-	a.Lock()
-	defer a.Unlock()
-	a.Data[key] = value
-}
-
-func (a *AlertCurEventMap) Get(key string) (*models.AlertCurEvent, bool) {
-	a.RLock()
-	defer a.RUnlock()
-	event, exists := a.Data[key]
-	return event, exists
-}
-
-func (a *AlertCurEventMap) UpdateLastEvalTime(key string, lastEvalTime int64) {
-	a.Lock()
-	defer a.Unlock()
-	event, exists := a.Data[key]
-	if !exists {
-		return
-	}
-	event.LastEvalTime = lastEvalTime
-}
-
-func (a *AlertCurEventMap) Delete(key string) {
-	a.Lock()
-	defer a.Unlock()
-	delete(a.Data, key)
-}
-
-func (a *AlertCurEventMap) Keys() []string {
-	a.RLock()
-	defer a.RUnlock()
-	keys := make([]string, 0, len(a.Data))
-	for k := range a.Data {
-		keys = append(keys, k)
-	}
-	return keys
-}
-
-func (a *AlertCurEventMap) GetAll() map[string]*models.AlertCurEvent {
-	a.RLock()
-	defer a.RUnlock()
-	return a.Data
-}

alert/process/process.go

@@ -1,664 +0,0 @@
-package process
-
-import (
-	"bytes"
-	"encoding/json"
-	"fmt"
-	"html/template"
-	"sort"
-	"strings"
-	"sync"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/common"
-	"github.com/ccfos/nightingale/v6/alert/dispatch"
-	"github.com/ccfos/nightingale/v6/alert/mute"
-	"github.com/ccfos/nightingale/v6/alert/pipeline/processor/relabel"
-	"github.com/ccfos/nightingale/v6/alert/queue"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/tplx"
-
-	"github.com/robfig/cron/v3"
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/str"
-)
-
-type EventMuteHookFunc func(event *models.AlertCurEvent) bool
-
-type ExternalProcessorsType struct {
-	ExternalLock sync.RWMutex
-	Processors   map[string]*Processor
-}
-
-var ExternalProcessors ExternalProcessorsType
-
-func NewExternalProcessors() *ExternalProcessorsType {
-	return &ExternalProcessorsType{
-		Processors: make(map[string]*Processor),
-	}
-}
-
-func (e *ExternalProcessorsType) GetExternalAlertRule(datasourceId, id int64) (*Processor, bool) {
-	e.ExternalLock.RLock()
-	defer e.ExternalLock.RUnlock()
-	processor, has := e.Processors[common.RuleKey(datasourceId, id)]
-	return processor, has
-}
-
-type HandleEventFunc func(event *models.AlertCurEvent)
-
-type Processor struct {
-	datasourceId int64
-	EngineName   string
-
-	rule                 *models.AlertRule
-	fires                *AlertCurEventMap
-	pendings             *AlertCurEventMap
-	pendingsUseByRecover *AlertCurEventMap
-	inhibit              bool
-
-	tagsMap   map[string]string
-	tagsArr   []string
-	groupName string
-
-	alertRuleCache          *memsto.AlertRuleCacheType
-	TargetCache             *memsto.TargetCacheType
-	TargetsOfAlertRuleCache *memsto.TargetsOfAlertRuleCacheType
-	BusiGroupCache          *memsto.BusiGroupCacheType
-	alertMuteCache          *memsto.AlertMuteCacheType
-	datasourceCache         *memsto.DatasourceCacheType
-
-	ctx   *ctx.Context
-	Stats *astats.Stats
-
-	HandleFireEventHook    HandleEventFunc
-	HandleRecoverEventHook HandleEventFunc
-	EventMuteHook          EventMuteHookFunc
-
-	ScheduleEntry    cron.Entry
-	PromEvalInterval int
-}
-
-func (p *Processor) Key() string {
-	return common.RuleKey(p.datasourceId, p.rule.Id)
-}
-
-func (p *Processor) DatasourceId() int64 {
-	return p.datasourceId
-}
-
-func (p *Processor) Hash() string {
-	return str.MD5(fmt.Sprintf("%d_%s_%s_%d",
-		p.rule.Id,
-		p.rule.CronPattern,
-		p.rule.RuleConfig,
-		p.datasourceId,
-	))
-}
-
-func NewProcessor(engineName string, rule *models.AlertRule, datasourceId int64, alertRuleCache *memsto.AlertRuleCacheType,
-	targetCache *memsto.TargetCacheType, targetsOfAlertRuleCache *memsto.TargetsOfAlertRuleCacheType,
-	busiGroupCache *memsto.BusiGroupCacheType, alertMuteCache *memsto.AlertMuteCacheType, datasourceCache *memsto.DatasourceCacheType, ctx *ctx.Context,
-	stats *astats.Stats) *Processor {
-
-	p := &Processor{
-		EngineName:   engineName,
-		datasourceId: datasourceId,
-		rule:         rule,
-
-		TargetCache:             targetCache,
-		TargetsOfAlertRuleCache: targetsOfAlertRuleCache,
-		BusiGroupCache:          busiGroupCache,
-		alertMuteCache:          alertMuteCache,
-		alertRuleCache:          alertRuleCache,
-		datasourceCache:         datasourceCache,
-
-		ctx:   ctx,
-		Stats: stats,
-
-		HandleFireEventHook:    func(event *models.AlertCurEvent) {},
-		HandleRecoverEventHook: func(event *models.AlertCurEvent) {},
-		EventMuteHook:          func(event *models.AlertCurEvent) bool { return false },
-	}
-
-	p.mayHandleGroup()
-	return p
-}
-
-func (p *Processor) Handle(anomalyPoints []models.AnomalyPoint, from string, inhibit bool) {
-	// 有可能rule的一些配置已经发生变化，比如告警接收人、callbacks等
-	// 这些信息的修改是不会引起worker restart的，但是确实会影响告警处理逻辑
-	// 所以，这里直接从memsto.AlertRuleCache中获取并覆盖
-	p.inhibit = inhibit
-	cachedRule := p.alertRuleCache.Get(p.rule.Id)
-	if cachedRule == nil {
-		logger.Errorf("rule not found %+v", anomalyPoints)
-		p.Stats.CounterRuleEvalErrorTotal.WithLabelValues(fmt.Sprintf("%v", p.DatasourceId()), "handle_event", p.BusiGroupCache.GetNameByBusiGroupId(p.rule.GroupId), fmt.Sprintf("%v", p.rule.Id)).Inc()
-		return
-	}
-
-	// 在 rule 变化之前取到 ruleHash
-	ruleHash := p.rule.Hash()
-
-	p.rule = cachedRule
-	now := time.Now().Unix()
-	alertingKeys := map[string]struct{}{}
-
-	// 根据 event 的 tag 将 events 分组，处理告警抑制的情况
-	eventsMap := make(map[string][]*models.AlertCurEvent)
-	for _, anomalyPoint := range anomalyPoints {
-		event := p.BuildEvent(anomalyPoint, from, now, ruleHash)
-		event.NotifyRuleIds = cachedRule.NotifyRuleIds
-		// 如果 event 被 mute 了,本质也是 fire 的状态,这里无论如何都添加到 alertingKeys 中,防止 fire 的事件自动恢复了
-		hash := event.Hash
-		alertingKeys[hash] = struct{}{}
-		isMuted, detail, muteId := mute.IsMuted(cachedRule, event, p.TargetCache, p.alertMuteCache)
-		if isMuted {
-			logger.Debugf("rule_eval:%s event:%v is muted, detail:%s", p.Key(), event, detail)
-			p.Stats.CounterMuteTotal.WithLabelValues(
-				fmt.Sprintf("%v", event.GroupName),
-				fmt.Sprintf("%v", p.rule.Id),
-				fmt.Sprintf("%v", muteId),
-				fmt.Sprintf("%v", p.datasourceId),
-			).Inc()
-			continue
-		}
-
-		if p.EventMuteHook(event) {
-			logger.Debugf("rule_eval:%s event:%v is muted by hook", p.Key(), event)
-			p.Stats.CounterMuteTotal.WithLabelValues(
-				fmt.Sprintf("%v", event.GroupName),
-				fmt.Sprintf("%v", p.rule.Id),
-				fmt.Sprintf("%v", 0),
-				fmt.Sprintf("%v", p.datasourceId),
-			).Inc()
-			continue
-		}
-
-		tagHash := TagHash(anomalyPoint)
-		eventsMap[tagHash] = append(eventsMap[tagHash], event)
-	}
-
-	for _, events := range eventsMap {
-		p.handleEvent(events)
-	}
-
-	if from == "inner" {
-		p.HandleRecover(alertingKeys, now, inhibit)
-	}
-}
-
-func (p *Processor) BuildEvent(anomalyPoint models.AnomalyPoint, from string, now int64, ruleHash string) *models.AlertCurEvent {
-	p.fillTags(anomalyPoint)
-
-	hash := Hash(p.rule.Id, p.datasourceId, anomalyPoint)
-	ds := p.datasourceCache.GetById(p.datasourceId)
-	var dsName string
-	if ds != nil {
-		dsName = ds.Name
-	}
-
-	event := p.rule.GenerateNewEvent(p.ctx)
-
-	bg := p.BusiGroupCache.GetByBusiGroupId(p.rule.GroupId)
-	if bg != nil {
-		event.GroupName = bg.Name
-	}
-
-	event.TriggerTime = anomalyPoint.Timestamp
-	event.TagsMap = p.tagsMap
-	event.DatasourceId = p.datasourceId
-	event.Cluster = dsName
-	event.Hash = hash
-	event.TriggerValue = anomalyPoint.ReadableValue()
-	event.TriggerValues = anomalyPoint.Values
-	event.TriggerValuesJson = models.EventTriggerValues{ValuesWithUnit: anomalyPoint.ValuesUnit}
-	event.TagsJSON = p.tagsArr
-	event.Tags = strings.Join(p.tagsArr, ",,")
-	event.IsRecovered = false
-	event.Callbacks = p.rule.Callbacks
-	event.CallbacksJSON = p.rule.CallbacksJSON
-	event.Annotations = p.rule.Annotations
-	event.RuleConfig = p.rule.RuleConfig
-	event.RuleConfigJson = p.rule.RuleConfigJson
-	event.Severity = anomalyPoint.Severity
-	event.ExtraConfig = p.rule.ExtraConfigJSON
-	event.PromQl = anomalyPoint.Query
-	event.RecoverConfig = anomalyPoint.RecoverConfig
-	event.RuleHash = ruleHash
-
-	if anomalyPoint.TriggerType == models.TriggerTypeNodata {
-		event.TriggerValue = "nodata"
-		ruleConfig := models.RuleQuery{}
-		json.Unmarshal([]byte(p.rule.RuleConfig), &ruleConfig)
-		ruleConfig.TriggerType = anomalyPoint.TriggerType
-		b, _ := json.Marshal(ruleConfig)
-		event.RuleConfig = string(b)
-	}
-
-	if err := json.Unmarshal([]byte(p.rule.Annotations), &event.AnnotationsJSON); err != nil {
-		event.AnnotationsJSON = make(map[string]string) // 解析失败时使用空 map
-		logger.Warningf("unmarshal annotations json failed: %v, rule: %d", err, p.rule.Id)
-	}
-
-	if event.TriggerValues != "" && strings.Count(event.TriggerValues, "$") > 1 {
-		// TriggerValues 有多个变量，将多个变量都放到 TriggerValue 中
-		event.TriggerValue = event.TriggerValues
-	}
-
-	if from == "inner" {
-		event.LastEvalTime = now
-	} else {
-		event.LastEvalTime = event.TriggerTime
-	}
-
-	// 生成事件之后，立马进程 relabel 处理
-	Relabel(p.rule, event)
-
-	// 放到 Relabel(p.rule, event) 下面，为了处理 relabel 之后，标签里才出现 ident 的情况
-	p.mayHandleIdent(event)
-
-	if event.TargetIdent != "" {
-		if pt, exist := p.TargetCache.Get(event.TargetIdent); exist {
-			pt.GroupNames = p.BusiGroupCache.GetNamesByBusiGroupIds(pt.GroupIds)
-			event.Target = pt
-		} else {
-			logger.Infof("fill event target error, ident: %s doesn't exist in cache.", event.TargetIdent)
-		}
-	}
-
-	return event
-}
-
-func Relabel(rule *models.AlertRule, event *models.AlertCurEvent) {
-	if rule == nil {
-		return
-	}
-
-	// need to keep the original label
-	event.OriginalTags = event.Tags
-	event.OriginalTagsJSON = event.TagsJSON
-
-	if len(rule.EventRelabelConfig) == 0 {
-		return
-	}
-
-	relabel.EventRelabel(event, rule.EventRelabelConfig)
-}
-
-func (p *Processor) HandleRecover(alertingKeys map[string]struct{}, now int64, inhibit bool) {
-	for _, hash := range p.pendings.Keys() {
-		if _, has := alertingKeys[hash]; has {
-			continue
-		}
-		p.pendings.Delete(hash)
-	}
-
-	hashArr := make([]string, 0, len(alertingKeys))
-	for hash, _ := range p.fires.GetAll() {
-		if _, has := alertingKeys[hash]; has {
-			continue
-		}
-
-		hashArr = append(hashArr, hash)
-	}
-	p.HandleRecoverEvent(hashArr, now, inhibit)
-
-}
-
-func (p *Processor) HandleRecoverEvent(hashArr []string, now int64, inhibit bool) {
-	cachedRule := p.rule
-	if cachedRule == nil {
-		return
-	}
-
-	if !inhibit {
-		for _, hash := range hashArr {
-			p.RecoverSingle(false, hash, now, nil)
-		}
-		return
-	}
-
-	eventMap := make(map[string]models.AlertCurEvent)
-	for _, hash := range hashArr {
-		event, has := p.fires.Get(hash)
-		if !has {
-			continue
-		}
-
-		e, exists := eventMap[event.Tags]
-		if !exists {
-			eventMap[event.Tags] = *event
-			continue
-		}
-
-		if e.Severity > event.Severity {
-			// hash 对应的恢复事件的被抑制了，把之前的事件删除
-			p.fires.Delete(e.Hash)
-			p.pendings.Delete(e.Hash)
-			models.AlertCurEventDelByHash(p.ctx, e.Hash)
-			eventMap[event.Tags] = *event
-		}
-	}
-
-	for _, event := range eventMap {
-		p.RecoverSingle(false, event.Hash, now, nil)
-	}
-}
-
-func (p *Processor) RecoverSingle(byRecover bool, hash string, now int64, value *string, values ...string) {
-	cachedRule := p.rule
-	if cachedRule == nil {
-		return
-	}
-
-	event, has := p.fires.Get(hash)
-	if !has {
-		return
-	}
-
-	// 如果配置了留观时长，就不能立马恢复了
-	if cachedRule.RecoverDuration > 0 {
-		lastPendingEvent, has := p.pendingsUseByRecover.Get(hash)
-		if !has {
-			// 说明没有产生过异常点，就不需要恢复了
-			logger.Debugf("rule_eval:%s event:%v do not has pending event, not recover", p.Key(), event)
-			return
-		}
-
-		if now-lastPendingEvent.LastEvalTime < cachedRule.RecoverDuration {
-			logger.Debugf("rule_eval:%s event:%v not recover", p.Key(), event)
-			return
-		}
-	}
-
-	// 如果设置了恢复条件，则不能在此处恢复，必须依靠 recoverPoint 来恢复
-	if event.RecoverConfig.JudgeType != models.Origin && !byRecover {
-		logger.Debugf("rule_eval:%s event:%v not recover", p.Key(), event)
-		return
-	}
-
-	if value != nil {
-		event.TriggerValue = *value
-		if len(values) > 0 {
-			event.TriggerValues = values[0]
-		}
-	}
-
-	// 没查到触发阈值的vector，姑且就认为这个vector的值恢复了
-	// 我确实无法分辨，是prom中有值但是未满足阈值所以没返回，还是prom中确实丢了一些点导致没有数据可以返回，尴尬
-	p.fires.Delete(hash)
-	p.pendings.Delete(hash)
-	p.pendingsUseByRecover.Delete(hash)
-
-	// 可能是因为调整了promql才恢复的，所以事件里边要体现最新的promql，否则用户会比较困惑
-	// 当然，其实rule的各个字段都可能发生变化了，都更新一下吧
-	cachedRule.UpdateEvent(event)
-	event.IsRecovered = true
-	event.LastEvalTime = now
-
-	p.HandleRecoverEventHook(event)
-	p.pushEventToQueue(event)
-}
-
-func (p *Processor) handleEvent(events []*models.AlertCurEvent) {
-	var fireEvents []*models.AlertCurEvent
-	// severity 初始为最低优先级, 一定为遇到比自己优先级高的事件
-	severity := models.SeverityLowest
-	for _, event := range events {
-		if event == nil {
-			continue
-		}
-
-		if _, has := p.pendingsUseByRecover.Get(event.Hash); has {
-			p.pendingsUseByRecover.UpdateLastEvalTime(event.Hash, event.LastEvalTime)
-		} else {
-			p.pendingsUseByRecover.Set(event.Hash, event)
-		}
-
-		event.PromEvalInterval = p.PromEvalInterval
-		if p.rule.PromForDuration == 0 {
-			fireEvents = append(fireEvents, event)
-			if severity > event.Severity {
-				severity = event.Severity
-			}
-			continue
-		}
-
-		var preTriggerTime int64 // 第一个 pending event 的触发时间
-		preEvent, has := p.pendings.Get(event.Hash)
-		if has {
-			p.pendings.UpdateLastEvalTime(event.Hash, event.LastEvalTime)
-			preTriggerTime = preEvent.TriggerTime
-		} else {
-			p.pendings.Set(event.Hash, event)
-			preTriggerTime = event.TriggerTime
-		}
-
-		if event.LastEvalTime-preTriggerTime+int64(event.PromEvalInterval) >= int64(p.rule.PromForDuration) {
-			fireEvents = append(fireEvents, event)
-			if severity > event.Severity {
-				severity = event.Severity
-			}
-			continue
-		}
-	}
-
-	p.inhibitEvent(fireEvents, severity)
-}
-
-func (p *Processor) inhibitEvent(events []*models.AlertCurEvent, highSeverity int) {
-	for _, event := range events {
-		if p.inhibit && event.Severity > highSeverity {
-			logger.Debugf("rule_eval:%s event:%+v inhibit highSeverity:%d", p.Key(), event, highSeverity)
-			continue
-		}
-		p.fireEvent(event)
-	}
-}
-
-func (p *Processor) fireEvent(event *models.AlertCurEvent) {
-	// As p.rule maybe outdated, use rule from cache
-	cachedRule := p.rule
-	if cachedRule == nil {
-		return
-	}
-
-	logger.Debugf("rule_eval:%s event:%+v fire", p.Key(), event)
-	if fired, has := p.fires.Get(event.Hash); has {
-		p.fires.UpdateLastEvalTime(event.Hash, event.LastEvalTime)
-		event.FirstTriggerTime = fired.FirstTriggerTime
-		p.HandleFireEventHook(event)
-
-		if cachedRule.NotifyRepeatStep == 0 {
-			logger.Debugf("rule_eval:%s event:%+v repeat is zero nothing to do", p.Key(), event)
-			// 说明不想重复通知，那就直接返回了，nothing to do
-			// do not need to send alert again
-			return
-		}
-
-		// 之前发送过告警了，这次是否要继续发送，要看是否过了通道静默时间
-		if event.LastEvalTime >= fired.LastSentTime+int64(cachedRule.NotifyRepeatStep)*60 {
-			if cachedRule.NotifyMaxNumber == 0 {
-				// 最大可以发送次数如果是0，表示不想限制最大发送次数，一直发即可
-				event.NotifyCurNumber = fired.NotifyCurNumber + 1
-				p.pushEventToQueue(event)
-			} else {
-				// 有最大发送次数的限制，就要看已经发了几次了，是否达到了最大发送次数
-				if fired.NotifyCurNumber >= cachedRule.NotifyMaxNumber {
-					logger.Debugf("rule_eval:%s event:%+v reach max number", p.Key(), event)
-					return
-				} else {
-					event.NotifyCurNumber = fired.NotifyCurNumber + 1
-					p.pushEventToQueue(event)
-				}
-			}
-		}
-	} else {
-		event.NotifyCurNumber = 1
-		event.FirstTriggerTime = event.TriggerTime
-		p.HandleFireEventHook(event)
-		p.pushEventToQueue(event)
-	}
-}
-
-func (p *Processor) pushEventToQueue(e *models.AlertCurEvent) {
-	if !e.IsRecovered {
-		e.LastSentTime = e.LastEvalTime
-		p.fires.Set(e.Hash, e)
-	}
-
-	dispatch.LogEvent(e, "push_queue")
-	if !queue.EventQueue.PushFront(e) {
-		logger.Warningf("event_push_queue: queue is full, event:%+v", e)
-		p.Stats.CounterRuleEvalErrorTotal.WithLabelValues(fmt.Sprintf("%v", p.DatasourceId()), "push_event_queue", p.BusiGroupCache.GetNameByBusiGroupId(p.rule.GroupId), fmt.Sprintf("%v", p.rule.Id)).Inc()
-	}
-}
-
-func (p *Processor) RecoverAlertCurEventFromDb() {
-	p.pendings = NewAlertCurEventMap(nil)
-	p.pendingsUseByRecover = NewAlertCurEventMap(nil)
-
-	curEvents, err := models.AlertCurEventGetByRuleIdAndDsId(p.ctx, p.rule.Id, p.datasourceId)
-	if err != nil {
-		logger.Errorf("recover event from db for rule:%s failed, err:%s", p.Key(), err)
-		p.Stats.CounterRuleEvalErrorTotal.WithLabelValues(fmt.Sprintf("%v", p.DatasourceId()), "get_recover_event", p.BusiGroupCache.GetNameByBusiGroupId(p.rule.GroupId), fmt.Sprintf("%v", p.rule.Id)).Inc()
-		p.fires = NewAlertCurEventMap(nil)
-		return
-	}
-
-	fireMap := make(map[string]*models.AlertCurEvent)
-	pendingsUseByRecoverMap := make(map[string]*models.AlertCurEvent)
-	for _, event := range curEvents {
-		alertRule := p.alertRuleCache.Get(event.RuleId)
-		if alertRule == nil {
-			continue
-		}
-		event.NotifyRuleIds = alertRule.NotifyRuleIds
-
-		if event.Cate == models.HOST {
-			target, exists := p.TargetCache.Get(event.TargetIdent)
-			if exists && target.EngineName != p.EngineName && !(p.ctx.IsCenter && target.EngineName == "") {
-				// 如果是 host rule，且 target 的 engineName 不是当前的 engineName 或者是中心机房 target EngineName 为空，就跳过
-				continue
-			}
-		}
-
-		event.DB2Mem()
-		target, exists := p.TargetCache.Get(event.TargetIdent)
-		if exists {
-			target.GroupNames = p.BusiGroupCache.GetNamesByBusiGroupIds(target.GroupIds)
-			event.Target = target
-		}
-
-		fireMap[event.Hash] = event
-		e := *event
-		pendingsUseByRecoverMap[event.Hash] = &e
-	}
-
-	p.fires = NewAlertCurEventMap(fireMap)
-
-	// 修改告警规则，或者进程重启之后，需要重新加载 pendingsUseByRecover
-	p.pendingsUseByRecover = NewAlertCurEventMap(pendingsUseByRecoverMap)
-}
-
-func (p *Processor) fillTags(anomalyPoint models.AnomalyPoint) {
-	// handle series tags
-	tagsMap := make(map[string]string)
-	for label, value := range anomalyPoint.Labels {
-		tagsMap[string(label)] = string(value)
-	}
-
-	var e = &models.AlertCurEvent{
-		TagsMap: tagsMap,
-	}
-
-	// handle rule tags
-	for _, tag := range p.rule.AppendTagsJSON {
-		arr := strings.SplitN(tag, "=", 2)
-
-		var defs = []string{
-			"{{$labels := .TagsMap}}",
-			"{{$value := .TriggerValue}}",
-		}
-		tagValue := arr[1]
-		text := strings.Join(append(defs, tagValue), "")
-		t, err := template.New(fmt.Sprint(p.rule.Id)).Funcs(template.FuncMap(tplx.TemplateFuncMap)).Parse(text)
-		if err != nil {
-			tagValue = fmt.Sprintf("parse tag value failed, err:%s", err)
-			tagsMap[arr[0]] = tagValue
-			continue
-		}
-
-		var body bytes.Buffer
-		err = t.Execute(&body, e)
-		if err != nil {
-			tagValue = fmt.Sprintf("parse tag value failed, err:%s", err)
-			tagsMap[arr[0]] = tagValue
-			continue
-		}
-
-		tagsMap[arr[0]] = body.String()
-	}
-
-	tagsMap["rulename"] = p.rule.Name
-	p.tagsMap = tagsMap
-
-	// handle tagsArr
-	p.tagsArr = labelMapToArr(tagsMap)
-}
-
-func (p *Processor) mayHandleIdent(event *models.AlertCurEvent) {
-	// handle ident
-	if ident, has := event.TagsMap["ident"]; has {
-		if target, exists := p.TargetCache.Get(ident); exists {
-			event.TargetIdent = target.Ident
-			event.TargetNote = target.Note
-		} else {
-			event.TargetIdent = ident
-			event.TargetNote = ""
-		}
-	} else {
-		event.TargetIdent = ""
-		event.TargetNote = ""
-	}
-}
-
-func (p *Processor) mayHandleGroup() {
-	// handle bg
-	bg := p.BusiGroupCache.GetByBusiGroupId(p.rule.GroupId)
-	if bg != nil {
-		p.groupName = bg.Name
-	}
-}
-
-func (p *Processor) DeleteProcessEvent(hash string) {
-	p.fires.Delete(hash)
-	p.pendings.Delete(hash)
-	p.pendingsUseByRecover.Delete(hash)
-}
-
-func labelMapToArr(m map[string]string) []string {
-	numLabels := len(m)
-
-	labelStrings := make([]string, 0, numLabels)
-	for label, value := range m {
-		labelStrings = append(labelStrings, fmt.Sprintf("%s=%s", label, value))
-	}
-
-	if numLabels > 1 {
-		sort.Strings(labelStrings)
-	}
-	return labelStrings
-}
-
-func Hash(ruleId, datasourceId int64, vector models.AnomalyPoint) string {
-	return str.MD5(fmt.Sprintf("%d_%s_%d_%d_%s", ruleId, vector.Labels.String(), datasourceId, vector.Severity, vector.Query))
-}
-
-func TagHash(vector models.AnomalyPoint) string {
-	return str.MD5(vector.Labels.String())
-}

alert/queue/queue.go

@@ -1,18 +0,0 @@
-package queue
-
-import (
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/toolkits/pkg/container/list"
-)
-
-var EventQueue = list.NewSafeListLimited(10000000)
-
-func ReportQueueSize(stats *astats.Stats) {
-	for {
-		time.Sleep(time.Second)
-
-		stats.GaugeAlertQueueSize.Set(float64(EventQueue.Len()))
-	}
-}

alert/record/prom_rule.go

@@ -1,119 +0,0 @@
-package record
-
-import (
-	"context"
-	"fmt"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-	"github.com/robfig/cron/v3"
-
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/str"
-)
-
-type RecordRuleContext struct {
-	datasourceId int64
-	quit         chan struct{}
-
-	scheduler   *cron.Cron
-	rule        *models.RecordingRule
-	promClients *prom.PromClientMap
-	stats       *astats.Stats
-}
-
-func NewRecordRuleContext(rule *models.RecordingRule, datasourceId int64, promClients *prom.PromClientMap, writers *writer.WritersType, stats *astats.Stats) *RecordRuleContext {
-	rrc := &RecordRuleContext{
-		datasourceId: datasourceId,
-		quit:         make(chan struct{}),
-		rule:         rule,
-		promClients:  promClients,
-		stats:        stats,
-	}
-
-	if rule.CronPattern == "" && rule.PromEvalInterval != 0 {
-		rule.CronPattern = fmt.Sprintf("@every %ds", rule.PromEvalInterval)
-	}
-
-	rrc.scheduler = cron.New(cron.WithSeconds(), cron.WithChain(cron.SkipIfStillRunning(cron.DefaultLogger)))
-	_, err := rrc.scheduler.AddFunc(rule.CronPattern, func() {
-		rrc.Eval()
-	})
-
-	if err != nil {
-		logger.Errorf("add cron pattern error: %v", err)
-	}
-
-	return rrc
-}
-
-func (rrc *RecordRuleContext) Key() string {
-	return fmt.Sprintf("record-%d-%d", rrc.datasourceId, rrc.rule.Id)
-}
-
-func (rrc *RecordRuleContext) Hash() string {
-	return str.MD5(fmt.Sprintf("%d_%s_%s_%d_%s_%s",
-		rrc.rule.Id,
-		rrc.rule.CronPattern,
-		rrc.rule.PromQl,
-		rrc.datasourceId,
-		rrc.rule.AppendTags,
-		rrc.rule.Name,
-	))
-}
-
-func (rrc *RecordRuleContext) Prepare() {}
-
-func (rrc *RecordRuleContext) Start() {
-	logger.Infof("eval:%s started", rrc.Key())
-	rrc.scheduler.Start()
-}
-
-func (rrc *RecordRuleContext) Eval() {
-	rrc.stats.CounterRecordEval.WithLabelValues(fmt.Sprintf("%d", rrc.datasourceId)).Inc()
-	promql := strings.TrimSpace(rrc.rule.PromQl)
-	if promql == "" {
-		logger.Errorf("eval:%s promql is blank", rrc.Key())
-		return
-	}
-
-	if rrc.promClients.IsNil(rrc.datasourceId) {
-		logger.Errorf("eval:%s reader client is nil", rrc.Key())
-		rrc.stats.CounterRecordEvalErrorTotal.WithLabelValues(fmt.Sprintf("%d", rrc.datasourceId)).Inc()
-		return
-	}
-
-	value, warnings, err := rrc.promClients.GetCli(rrc.datasourceId).Query(context.Background(), promql, time.Now())
-	if err != nil {
-		logger.Errorf("eval:%s promql:%s, error:%v", rrc.Key(), promql, err)
-		rrc.stats.CounterRecordEvalErrorTotal.WithLabelValues(fmt.Sprintf("%d", rrc.datasourceId)).Inc()
-		return
-	}
-
-	if len(warnings) > 0 {
-		logger.Errorf("eval:%s promql:%s, warnings:%v", rrc.Key(), promql, warnings)
-		rrc.stats.CounterRecordEvalErrorTotal.WithLabelValues(fmt.Sprintf("%d", rrc.datasourceId)).Inc()
-		return
-	}
-
-	ts := ConvertToTimeSeries(value, rrc.rule)
-	if len(ts) != 0 {
-		err := rrc.promClients.GetWriterCli(rrc.datasourceId).Write(ts)
-		if err != nil {
-			logger.Errorf("eval:%s promql:%s, error:%v", rrc.Key(), promql, err)
-			rrc.stats.CounterRecordEvalErrorTotal.WithLabelValues(fmt.Sprintf("%d", rrc.datasourceId)).Inc()
-		}
-	}
-}
-
-func (rrc *RecordRuleContext) Stop() {
-	logger.Infof("%s stopped", rrc.Key())
-
-	c := rrc.scheduler.Stop()
-	<-c.Done()
-	close(rrc.quit)
-}

alert/record/sample.go

@@ -1,122 +0,0 @@
-package record
-
-import (
-	"math"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/prometheus/common/model"
-	"github.com/prometheus/prometheus/prompb"
-)
-
-const (
-	LabelName = "__name__"
-)
-
-func ConvertToTimeSeries(value model.Value, rule *models.RecordingRule) (lst []prompb.TimeSeries) {
-	switch value.Type() {
-	case model.ValVector:
-		items, ok := value.(model.Vector)
-		if !ok {
-			return
-		}
-
-		for _, item := range items {
-			if math.IsNaN(float64(item.Value)) {
-				continue
-			}
-			s := prompb.Sample{}
-			s.Timestamp = time.Unix(item.Timestamp.Unix(), 0).UnixNano() / 1e6
-			s.Value = float64(item.Value)
-			l := labelsToLabelsProto(item.Metric, rule)
-			lst = append(lst, prompb.TimeSeries{
-				Labels:  l,
-				Samples: []prompb.Sample{s},
-			})
-		}
-	case model.ValMatrix:
-		items, ok := value.(model.Matrix)
-		if !ok {
-			return
-		}
-
-		for _, item := range items {
-			if len(item.Values) == 0 {
-				return
-			}
-
-			last := item.Values[len(item.Values)-1]
-
-			if math.IsNaN(float64(last.Value)) {
-				continue
-			}
-			l := labelsToLabelsProto(item.Metric, rule)
-			var slst []prompb.Sample
-			for _, v := range item.Values {
-				if math.IsNaN(float64(v.Value)) {
-					continue
-				}
-				slst = append(slst, prompb.Sample{
-					Timestamp: time.Unix(v.Timestamp.Unix(), 0).UnixNano() / 1e6,
-					Value:     float64(v.Value),
-				})
-			}
-			lst = append(lst, prompb.TimeSeries{
-				Labels:  l,
-				Samples: slst,
-			})
-		}
-	case model.ValScalar:
-		item, ok := value.(*model.Scalar)
-		if !ok {
-			return
-		}
-
-		if math.IsNaN(float64(item.Value)) {
-			return
-		}
-
-		lst = append(lst, prompb.TimeSeries{
-			Labels:  nil,
-			Samples: []prompb.Sample{{Value: float64(item.Value), Timestamp: time.Unix(item.Timestamp.Unix(), 0).UnixNano() / 1e6}},
-		})
-	default:
-		return
-	}
-
-	return
-}
-
-func labelsToLabelsProto(labels model.Metric, rule *models.RecordingRule) (result []prompb.Label) {
-	//name
-	nameLs := prompb.Label{
-		Name:  LabelName,
-		Value: rule.Name,
-	}
-	result = append(result, nameLs)
-	for k, v := range labels {
-		if k == LabelName {
-			continue
-		}
-		if model.LabelNameRE.MatchString(string(k)) {
-			result = append(result, prompb.Label{
-				Name:  string(k),
-				Value: string(v),
-			})
-		}
-	}
-	if len(rule.AppendTagsJSON) != 0 {
-		for _, v := range rule.AppendTagsJSON {
-			index := strings.Index(v, "=")
-			if model.LabelNameRE.MatchString(v[:index]) {
-				result = append(result, prompb.Label{
-					Name:  v[:index],
-					Value: v[index+1:],
-				})
-			}
-		}
-	}
-	return result
-}

alert/record/scheduler.go

@@ -1,99 +0,0 @@
-package record
-
-import (
-	"context"
-	"fmt"
-	"strconv"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/naming"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-)
-
-type Scheduler struct {
-	// key: hash
-	recordRules map[string]*RecordRuleContext
-
-	aconf aconf.Alert
-
-	recordingRuleCache *memsto.RecordingRuleCacheType
-
-	promClients *prom.PromClientMap
-	writers     *writer.WritersType
-
-	stats *astats.Stats
-
-	datasourceCache *memsto.DatasourceCacheType
-}
-
-func NewScheduler(aconf aconf.Alert, rrc *memsto.RecordingRuleCacheType, promClients *prom.PromClientMap, writers *writer.WritersType, stats *astats.Stats, datasourceCache *memsto.DatasourceCacheType) *Scheduler {
-	scheduler := &Scheduler{
-		aconf:       aconf,
-		recordRules: make(map[string]*RecordRuleContext),
-
-		recordingRuleCache: rrc,
-
-		promClients: promClients,
-		writers:     writers,
-
-		stats: stats,
-
-		datasourceCache: datasourceCache,
-	}
-
-	go scheduler.LoopSyncRules(context.Background())
-	return scheduler
-}
-
-func (s *Scheduler) LoopSyncRules(ctx context.Context) {
-	time.Sleep(time.Duration(s.aconf.EngineDelay) * time.Second)
-	duration := 9000 * time.Millisecond
-	for {
-		select {
-		case <-ctx.Done():
-			return
-		case <-time.After(duration):
-			s.syncRecordRules()
-		}
-	}
-}
-
-func (s *Scheduler) syncRecordRules() {
-	ids := s.recordingRuleCache.GetRuleIds()
-	recordRules := make(map[string]*RecordRuleContext)
-	for _, id := range ids {
-		rule := s.recordingRuleCache.Get(id)
-		if rule == nil {
-			continue
-		}
-
-		datasourceIds := s.datasourceCache.GetIDsByDsCateAndQueries("prometheus", rule.DatasourceQueries)
-		for _, dsId := range datasourceIds {
-			if !naming.DatasourceHashRing.IsHit(strconv.FormatInt(dsId, 10), fmt.Sprintf("%d", rule.Id), s.aconf.Heartbeat.Endpoint) {
-				continue
-			}
-
-			recordRule := NewRecordRuleContext(rule, dsId, s.promClients, s.writers, s.stats)
-			recordRules[recordRule.Hash()] = recordRule
-		}
-	}
-
-	for hash, rule := range recordRules {
-		if _, has := s.recordRules[hash]; !has {
-			rule.Prepare()
-			rule.Start()
-			s.recordRules[hash] = rule
-		}
-	}
-
-	for hash, rule := range s.recordRules {
-		if _, has := recordRules[hash]; !has {
-			rule.Stop()
-			delete(s.recordRules, hash)
-		}
-	}
-}

alert/router/router.go

@@ -1,79 +0,0 @@
-package router
-
-import (
-	"net/http"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/process"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/httpx"
-
-	"github.com/gin-gonic/gin"
-)
-
-type Router struct {
-	HTTP               httpx.Config
-	Alert              aconf.Alert
-	AlertMuteCache     *memsto.AlertMuteCacheType
-	TargetCache        *memsto.TargetCacheType
-	BusiGroupCache     *memsto.BusiGroupCacheType
-	AlertStats         *astats.Stats
-	Ctx                *ctx.Context
-	ExternalProcessors *process.ExternalProcessorsType
-}
-
-func New(httpConfig httpx.Config, alert aconf.Alert, amc *memsto.AlertMuteCacheType, tc *memsto.TargetCacheType, bgc *memsto.BusiGroupCacheType,
-	astats *astats.Stats, ctx *ctx.Context, externalProcessors *process.ExternalProcessorsType) *Router {
-	return &Router{
-		HTTP:               httpConfig,
-		Alert:              alert,
-		AlertMuteCache:     amc,
-		TargetCache:        tc,
-		BusiGroupCache:     bgc,
-		AlertStats:         astats,
-		Ctx:                ctx,
-		ExternalProcessors: externalProcessors,
-	}
-}
-
-func (rt *Router) Config(r *gin.Engine) {
-	if !rt.HTTP.APIForService.Enable {
-		return
-	}
-
-	service := r.Group("/v1/n9e")
-	if len(rt.HTTP.APIForService.BasicAuth) > 0 {
-		service.Use(gin.BasicAuth(rt.HTTP.APIForService.BasicAuth))
-	}
-	service.POST("/event", rt.pushEventToQueue)
-	service.POST("/event-persist", rt.eventPersist)
-	service.POST("/make-event", rt.makeEvent)
-}
-
-func Render(c *gin.Context, data, msg interface{}) {
-	if msg == nil {
-		if data == nil {
-			data = struct{}{}
-		}
-		c.JSON(http.StatusOK, gin.H{"data": data, "error": ""})
-	} else {
-		c.JSON(http.StatusOK, gin.H{"error": gin.H{"message": msg}})
-	}
-}
-
-func Dangerous(c *gin.Context, v interface{}, code ...int) {
-	if v == nil {
-		return
-	}
-
-	switch t := v.(type) {
-	case string:
-		if t != "" {
-			c.JSON(http.StatusOK, gin.H{"error": gin.H{"message": v}})
-		}
-	case error:
-		c.JSON(http.StatusOK, gin.H{"error": gin.H{"message": t.Error()}})
-	}
-}

alert/router/router_event.go

@@ -1,147 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"strconv"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/dispatch"
-	"github.com/ccfos/nightingale/v6/alert/mute"
-	"github.com/ccfos/nightingale/v6/alert/naming"
-	"github.com/ccfos/nightingale/v6/alert/process"
-	"github.com/ccfos/nightingale/v6/alert/queue"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/poster"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-func (rt *Router) pushEventToQueue(c *gin.Context) {
-	var event *models.AlertCurEvent
-	ginx.BindJSON(c, &event)
-	if event.RuleId == 0 {
-		ginx.Bomb(200, "event is illegal")
-	}
-
-	event.TagsMap = make(map[string]string)
-	for i := 0; i < len(event.TagsJSON); i++ {
-		pair := strings.TrimSpace(event.TagsJSON[i])
-		if pair == "" {
-			continue
-		}
-
-		arr := strings.SplitN(pair, "=", 2)
-		if len(arr) != 2 {
-			continue
-		}
-
-		event.TagsMap[arr[0]] = arr[1]
-	}
-	hit, _ :=  mute.EventMuteStrategy(event, rt.AlertMuteCache)
-	if hit {
-		logger.Infof("event_muted: rule_id=%d %s", event.RuleId, event.Hash)
-		ginx.NewRender(c).Message(nil)
-		return
-	}
-
-	if err := event.ParseRule("rule_name"); err != nil {
-		event.RuleName = fmt.Sprintf("failed to parse rule name: %v", err)
-	}
-
-	if err := event.ParseRule("rule_note"); err != nil {
-		event.RuleNote = fmt.Sprintf("failed to parse rule note: %v", err)
-	}
-
-	if err := event.ParseRule("annotations"); err != nil {
-		event.RuleNote = fmt.Sprintf("failed to parse rule note: %v", err)
-	}
-
-	// 如果 rule_note 中有 ; 前缀，则使用 rule_note 替换 tags 中的内容
-	if strings.HasPrefix(event.RuleNote, ";") {
-		event.RuleNote = strings.TrimPrefix(event.RuleNote, ";")
-		event.Tags = strings.ReplaceAll(event.RuleNote, " ", ",,")
-		event.TagsJSON = strings.Split(event.Tags, ",,")
-	} else {
-		event.Tags = strings.Join(event.TagsJSON, ",,")
-	}
-
-	event.Callbacks = strings.Join(event.CallbacksJSON, " ")
-	event.NotifyChannels = strings.Join(event.NotifyChannelsJSON, " ")
-	event.NotifyGroups = strings.Join(event.NotifyGroupsJSON, " ")
-
-	dispatch.LogEvent(event, "http_push_queue")
-	if !queue.EventQueue.PushFront(event) {
-		msg := fmt.Sprintf("event:%+v push_queue err: queue is full", event)
-		ginx.Bomb(200, msg)
-		logger.Warningf(msg)
-	}
-	ginx.NewRender(c).Message(nil)
-}
-
-func (rt *Router) eventPersist(c *gin.Context) {
-	var event *models.AlertCurEvent
-	ginx.BindJSON(c, &event)
-	event.FE2DB()
-	err := models.EventPersist(rt.Ctx, event)
-	ginx.NewRender(c).Data(event.Id, err)
-}
-
-type eventForm struct {
-	Alert         bool                  `json:"alert"`
-	AnomalyPoints []models.AnomalyPoint `json:"vectors"`
-	RuleId        int64                 `json:"rule_id"`
-	DatasourceId  int64                 `json:"datasource_id"`
-	Inhibit       bool                  `json:"inhibit"`
-}
-
-func (rt *Router) makeEvent(c *gin.Context) {
-	var events []*eventForm
-	ginx.BindJSON(c, &events)
-	//now := time.Now().Unix()
-	for i := 0; i < len(events); i++ {
-		node, err := naming.DatasourceHashRing.GetNode(strconv.FormatInt(events[i].DatasourceId, 10), fmt.Sprintf("%d", events[i].RuleId))
-		if err != nil {
-			logger.Warningf("event:%+v get node err:%v", events[i], err)
-			ginx.Bomb(200, "event node not exists")
-		}
-
-		if node != rt.Alert.Heartbeat.Endpoint {
-			err := forwardEvent(events[i], node)
-			if err != nil {
-				logger.Warningf("event:%+v forward err:%v", events[i], err)
-				ginx.Bomb(200, "event forward error")
-			}
-			continue
-		}
-
-		ruleWorker, exists := rt.ExternalProcessors.GetExternalAlertRule(events[i].DatasourceId, events[i].RuleId)
-		logger.Debugf("handle event:%+v exists:%v", events[i], exists)
-		if !exists {
-			ginx.Bomb(200, "rule not exists")
-		}
-
-		if events[i].Alert {
-			go ruleWorker.Handle(events[i].AnomalyPoints, "http", events[i].Inhibit)
-		} else {
-			for _, vector := range events[i].AnomalyPoints {
-				readableString := vector.ReadableValue()
-				go ruleWorker.RecoverSingle(false, process.Hash(events[i].RuleId, events[i].DatasourceId, vector), vector.Timestamp, &readableString)
-			}
-		}
-	}
-	ginx.NewRender(c).Message(nil)
-}
-
-// event 不归本实例处理，转发给对应的实例
-func forwardEvent(event *eventForm, instance string) error {
-	ur := fmt.Sprintf("http://%s/v1/n9e/make-event", instance)
-	res, code, err := poster.PostJSON(ur, time.Second*5, []*eventForm{event}, 3)
-	if err != nil {
-		return err
-	}
-	logger.Infof("forward event: result=succ url=%s code=%d event:%v response=%s", ur, code, event, string(res))
-	return nil
-}

alert/sender/callback.go

@@ -1,207 +0,0 @@
-package sender
-
-import (
-	"html/template"
-	"net/url"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/poster"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type (
-	// CallBacker 进行回调的接口
-	CallBacker interface {
-		CallBack(ctx CallBackContext)
-	}
-
-	// CallBackContext 回调时所需的上下文
-	CallBackContext struct {
-		Ctx         *ctx.Context
-		CallBackURL string
-		Users       []*models.User
-		Rule        *models.AlertRule
-		Events      []*models.AlertCurEvent
-		Stats       *astats.Stats
-		BatchSend   bool
-	}
-
-	DefaultCallBacker struct{}
-)
-
-func BuildCallBackContext(ctx *ctx.Context, callBackURL string, rule *models.AlertRule, events []*models.AlertCurEvent,
-	uids []int64, userCache *memsto.UserCacheType, batchSend bool, stats *astats.Stats) CallBackContext {
-	users := userCache.GetByUserIds(uids)
-
-	newCallBackUrl, _ := events[0].ParseURL(callBackURL)
-	return CallBackContext{
-		Ctx:         ctx,
-		CallBackURL: newCallBackUrl,
-		Rule:        rule,
-		Events:      events,
-		Users:       users,
-		BatchSend:   batchSend,
-		Stats:       stats,
-	}
-}
-
-func ExtractAtsParams(rawURL string) []string {
-	ans := make([]string, 0, 1)
-	parsedURL, err := url.Parse(rawURL)
-	if err != nil {
-		logger.Errorf("ExtractAtsParams(url=%s), err: %v", rawURL, err)
-		return ans
-	}
-
-	queryParams := parsedURL.Query()
-	atParam := queryParams.Get("ats")
-	if atParam == "" {
-		return ans
-	}
-
-	// Split the atParam by comma and return the result as a slice
-	return strings.Split(atParam, ",")
-}
-
-func NewCallBacker(
-	key string,
-	targetCache *memsto.TargetCacheType,
-	userCache *memsto.UserCacheType,
-	taskTplCache *memsto.TaskTplCache,
-	tpls map[string]*template.Template,
-) CallBacker {
-
-	switch key {
-	case models.IbexDomain: // Distribute to Ibex
-		return &IbexCallBacker{
-			targetCache:  targetCache,
-			userCache:    userCache,
-			taskTplCache: taskTplCache,
-		}
-	case models.DefaultDomain: // default callback
-		return &DefaultCallBacker{}
-	case models.DingtalkDomain:
-		return &DingtalkSender{tpl: tpls[models.Dingtalk]}
-	case models.WecomDomain:
-		return &WecomSender{tpl: tpls[models.Wecom]}
-	case models.FeishuDomain:
-		return &FeishuSender{tpl: tpls[models.Feishu]}
-	case models.FeishuCardDomain:
-		return &FeishuCardSender{tpl: tpls[models.FeishuCard]}
-	//case models.Mm:
-	//	return &MmSender{tpl: tpls[models.Mm]}
-	case models.TelegramDomain:
-		return &TelegramSender{tpl: tpls[models.Telegram]}
-	case models.LarkDomain:
-		return &LarkSender{tpl: tpls[models.Lark]}
-	case models.LarkCardDomain:
-		return &LarkCardSender{tpl: tpls[models.LarkCard]}
-	}
-
-	return nil
-}
-
-func (c *DefaultCallBacker) CallBack(ctx CallBackContext) {
-	if len(ctx.CallBackURL) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-
-	event := ctx.Events[0]
-
-	if ctx.BatchSend {
-		webhookConf := &models.Webhook{
-			Type:          models.RuleCallback,
-			Enable:        true,
-			Url:           ctx.CallBackURL,
-			Timeout:       5,
-			RetryCount:    3,
-			RetryInterval: 10,
-			Batch:         1000,
-		}
-
-		PushCallbackEvent(ctx.Ctx, webhookConf, event, ctx.Stats)
-		return
-	}
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, event, "callback", ctx.Stats, ctx.Events)
-}
-
-func doSendAndRecord(ctx *ctx.Context, url, token string, body interface{}, channel string,
-	stats *astats.Stats, events []*models.AlertCurEvent) {
-	res, err := doSend(url, body, channel, stats)
-	NotifyRecord(ctx, events, 0, channel, token, res, err)
-}
-
-func NotifyRecord(ctx *ctx.Context, evts []*models.AlertCurEvent, notifyRuleID int64, channel, target, res string, err error) {
-	// 一个通知可能对应多个 event，都需要记录
-	notis := make([]*models.NotificaitonRecord, 0, len(evts))
-	for _, evt := range evts {
-		noti := models.NewNotificationRecord(evt, notifyRuleID, channel, target)
-		if err != nil {
-			noti.SetStatus(models.NotiStatusFailure)
-			noti.SetDetails(err.Error())
-		} else if res != "" {
-			noti.SetDetails(string(res))
-		}
-		notis = append(notis, noti)
-	}
-
-	if !ctx.IsCenter {
-		err := poster.PostByUrls(ctx, "/v1/n9e/notify-record", notis)
-		if err != nil {
-			logger.Errorf("add notis:%v failed, err: %v", notis, err)
-		}
-		return
-	}
-
-	PushNotifyRecords(notis)
-}
-
-func doSend(url string, body interface{}, channel string, stats *astats.Stats) (string, error) {
-	stats.AlertNotifyTotal.WithLabelValues(channel).Inc()
-
-	res, code, err := poster.PostJSON(url, time.Second*5, body, 3)
-	if err != nil {
-		logger.Errorf("%s_sender: result=fail url=%s code=%d error=%v req:%v response=%s", channel, url, code, err, body, string(res))
-		stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-		return "", err
-	}
-
-	logger.Infof("%s_sender: result=succ url=%s code=%d req:%v response=%s", channel, url, code, body, string(res))
-	return string(res), nil
-}
-
-type TaskCreateReply struct {
-	Err string `json:"err"`
-	Dat int64  `json:"dat"` // task.id
-}
-
-func PushCallbackEvent(ctx *ctx.Context, webhook *models.Webhook, event *models.AlertCurEvent, stats *astats.Stats) {
-	CallbackEventQueueLock.RLock()
-	queue := CallbackEventQueue[webhook.Url]
-	CallbackEventQueueLock.RUnlock()
-
-	if queue == nil {
-		queue = &WebhookQueue{
-			eventQueue: NewSafeEventQueue(QueueMaxSize),
-			closeCh:    make(chan struct{}),
-		}
-
-		CallbackEventQueueLock.Lock()
-		CallbackEventQueue[webhook.Url] = queue
-		CallbackEventQueueLock.Unlock()
-
-		StartConsumer(ctx, queue, webhook.Batch, webhook, stats)
-	}
-
-	succ := queue.eventQueue.Push(event)
-	if !succ {
-		logger.Warningf("Write channel(%s) full, current channel size: %d event:%v", webhook.Url, queue.eventQueue.Len(), event)
-	}
-}

alert/sender/dingtalk.go

@@ -1,123 +0,0 @@
-package sender
-
-import (
-	"html/template"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type dingtalkMarkdown struct {
-	Title string `json:"title"`
-	Text  string `json:"text"`
-}
-
-type dingtalkAt struct {
-	AtMobiles []string `json:"atMobiles"`
-	IsAtAll   bool     `json:"isAtAll"`
-}
-
-type dingtalk struct {
-	Msgtype  string           `json:"msgtype"`
-	Markdown dingtalkMarkdown `json:"markdown"`
-	At       dingtalkAt       `json:"at"`
-}
-
-var (
-	_ CallBacker = (*DingtalkSender)(nil)
-)
-
-type DingtalkSender struct {
-	tpl *template.Template
-}
-
-func (ds *DingtalkSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-
-	urls, ats, tokens := ds.extract(ctx.Users)
-	if len(urls) == 0 {
-		return
-	}
-	message := BuildTplMessage(models.Dingtalk, ds.tpl, ctx.Events)
-
-	for i, url := range urls {
-		var body dingtalk
-		// NoAt in url
-		if strings.Contains(url, "noat=1") {
-			body = dingtalk{
-				Msgtype: "markdown",
-				Markdown: dingtalkMarkdown{
-					Title: ctx.Events[0].RuleName,
-					Text:  message,
-				},
-			}
-		} else {
-			body = dingtalk{
-				Msgtype: "markdown",
-				Markdown: dingtalkMarkdown{
-					Title: ctx.Events[0].RuleName,
-					Text:  message + "\n" + strings.Join(ats, " "),
-				},
-				At: dingtalkAt{
-					AtMobiles: ats,
-					IsAtAll:   false,
-				},
-			}
-		}
-
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.Dingtalk, ctx.Stats, ctx.Events)
-	}
-}
-
-func (ds *DingtalkSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	body := dingtalk{
-		Msgtype: "markdown",
-		Markdown: dingtalkMarkdown{
-			Title: ctx.Events[0].RuleName,
-		},
-	}
-
-	ats := ExtractAtsParams(ctx.CallBackURL)
-	message := BuildTplMessage(models.Dingtalk, ds.tpl, ctx.Events)
-
-	if len(ats) > 0 {
-		body.Markdown.Text = message + "\n@" + strings.Join(ats, "@")
-		body.At = dingtalkAt{
-			AtMobiles: ats,
-			IsAtAll:   false,
-		}
-	} else {
-		// NoAt in url
-		body.Markdown.Text = message
-	}
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, body, "callback", ctx.Stats, ctx.Events)
-}
-
-// extract urls and ats from Users
-func (ds *DingtalkSender) extract(users []*models.User) ([]string, []string, []string) {
-	urls := make([]string, 0, len(users))
-	ats := make([]string, 0, len(users))
-	tokens := make([]string, 0, len(users))
-
-	for _, user := range users {
-		if user.Phone != "" {
-			ats = append(ats, "@"+user.Phone)
-		}
-		if token, has := user.ExtractToken(models.Dingtalk); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://oapi.dingtalk.com/robot/send?access_token=" + token
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, ats, tokens
-}

alert/sender/email.go

@@ -1,232 +0,0 @@
-package sender
-
-import (
-	"crypto/tls"
-	"errors"
-	"html/template"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/logger"
-
-	"gopkg.in/gomail.v2"
-)
-
-var mailch chan *EmailContext
-
-type EmailSender struct {
-	subjectTpl *template.Template
-	contentTpl *template.Template
-	smtp       aconf.SMTPConfig
-}
-
-type EmailContext struct {
-	events []*models.AlertCurEvent
-	mail   *gomail.Message
-}
-
-func (es *EmailSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	tos := extract(ctx.Users)
-	var subject string
-
-	if es.subjectTpl != nil {
-		subject = BuildTplMessage(models.Email, es.subjectTpl, []*models.AlertCurEvent{ctx.Events[0]})
-	} else {
-		subject = ctx.Events[0].RuleName
-	}
-	content := BuildTplMessage(models.Email, es.contentTpl, ctx.Events)
-	es.WriteEmail(subject, content, tos, ctx.Events)
-
-	ctx.Stats.AlertNotifyTotal.WithLabelValues(models.Email).Add(float64(len(tos)))
-}
-
-func extract(users []*models.User) []string {
-	tos := make([]string, 0, len(users))
-	for _, u := range users {
-		if u.Email != "" {
-			tos = append(tos, u.Email)
-		}
-	}
-	return tos
-}
-
-func SendEmail(subject, content string, tos []string, stmp aconf.SMTPConfig) error {
-	conf := stmp
-
-	d := gomail.NewDialer(conf.Host, conf.Port, conf.User, conf.Pass)
-	if conf.InsecureSkipVerify {
-		d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
-	}
-
-	m := gomail.NewMessage()
-
-	m.SetHeader("From", stmp.From)
-	m.SetHeader("To", tos...)
-	m.SetHeader("Subject", subject)
-	m.SetBody("text/html", content)
-
-	err := d.DialAndSend(m)
-	if err != nil {
-		return errors.New("email_sender: failed to send: " + err.Error())
-	}
-	return nil
-}
-
-func (es *EmailSender) WriteEmail(subject, content string, tos []string, events []*models.AlertCurEvent) {
-	m := gomail.NewMessage()
-
-	m.SetHeader("From", es.smtp.From)
-	m.SetHeader("To", tos...)
-	m.SetHeader("Subject", subject)
-	m.SetBody("text/html", content)
-
-	mailch <- &EmailContext{events, m}
-}
-
-func dialSmtp(d *gomail.Dialer) gomail.SendCloser {
-	for {
-		select {
-		case <-mailQuit:
-			// Note that Sendcloser is not obtained below,
-			// and the outgoing signal (with configuration changes) exits the current dial
-			return nil
-		default:
-			if s, err := d.Dial(); err != nil {
-				logger.Errorf("email_sender: failed to dial smtp: %s", err)
-			} else {
-				return s
-			}
-			time.Sleep(time.Second)
-		}
-	}
-}
-
-var mailQuit = make(chan struct{})
-
-func RestartEmailSender(ctx *ctx.Context, smtp aconf.SMTPConfig) {
-	// Notify internal start exit
-	mailQuit <- struct{}{}
-	startEmailSender(ctx, smtp)
-}
-
-var smtpConfig aconf.SMTPConfig
-
-func InitEmailSender(ctx *ctx.Context, ncc *memsto.NotifyConfigCacheType) {
-	mailch = make(chan *EmailContext, 100000)
-	go updateSmtp(ctx, ncc)
-	smtpConfig = ncc.GetSMTP()
-	go startEmailSender(ctx, smtpConfig)
-}
-
-func updateSmtp(ctx *ctx.Context, ncc *memsto.NotifyConfigCacheType) {
-	for {
-		time.Sleep(1 * time.Minute)
-		smtp := ncc.GetSMTP()
-		if smtpConfig.Host != smtp.Host || smtpConfig.Batch != smtp.Batch || smtpConfig.From != smtp.From ||
-			smtpConfig.Pass != smtp.Pass || smtpConfig.User != smtp.User || smtpConfig.Port != smtp.Port ||
-			smtpConfig.InsecureSkipVerify != smtp.InsecureSkipVerify { //diff
-			smtpConfig = smtp
-			RestartEmailSender(ctx, smtp)
-		}
-	}
-}
-
-func startEmailSender(ctx *ctx.Context, smtp aconf.SMTPConfig) {
-	conf := smtp
-	if conf.Host == "" || conf.Port == 0 {
-		logger.Warning("SMTP configurations invalid")
-		<-mailQuit
-		return
-	}
-	logger.Infof("start email sender... conf.Host:%+v,conf.Port:%+v", conf.Host, conf.Port)
-
-	d := gomail.NewDialer(conf.Host, conf.Port, conf.User, conf.Pass)
-	if conf.InsecureSkipVerify {
-		d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
-	}
-
-	var s gomail.SendCloser
-	var open bool
-	var size int
-	for {
-		select {
-		case <-mailQuit:
-			return
-		case m, ok := <-mailch:
-			if !ok {
-				return
-			}
-
-			if !open {
-				s = dialSmtp(d)
-				if s == nil {
-					// Indicates that the dialing failed and exited the current goroutine directly,
-					// but put the Message back in the mailch
-					mailch <- m
-					return
-				}
-				open = true
-			}
-			var err error
-			if err = gomail.Send(s, m.mail); err != nil {
-				logger.Errorf("email_sender: failed to send: %s", err)
-
-				// close and retry
-				if err := s.Close(); err != nil {
-					logger.Warningf("email_sender: failed to close smtp connection: %s", err)
-				}
-
-				s = dialSmtp(d)
-				if s == nil {
-					// Indicates that the dialing failed and exited the current goroutine directly,
-					// but put the Message back in the mailch
-					mailch <- m
-					return
-				}
-				open = true
-
-				if err = gomail.Send(s, m.mail); err != nil {
-					logger.Errorf("email_sender: failed to retry send: %s", err)
-				}
-			} else {
-				logger.Infof("email_sender: result=succ subject=%v to=%v",
-					m.mail.GetHeader("Subject"), m.mail.GetHeader("To"))
-			}
-
-			for _, to := range m.mail.GetHeader("To") {
-				msg := ""
-				if err == nil {
-					msg = "ok"
-				}
-				NotifyRecord(ctx, m.events, 0, models.Email, to, msg, err)
-			}
-
-			size++
-
-			if size >= conf.Batch {
-				if err := s.Close(); err != nil {
-					logger.Warningf("email_sender: failed to close smtp connection: %s", err)
-				}
-				open = false
-				size = 0
-			}
-
-		// Close the connection to the SMTP server if no email was sent in
-		// the last 30 seconds.
-		case <-time.After(30 * time.Second):
-			if open {
-				if err := s.Close(); err != nil {
-					logger.Warningf("email_sender: failed to close smtp connection: %s", err)
-				}
-				open = false
-			}
-		}
-	}
-}

alert/sender/feishu.go

@@ -1,102 +0,0 @@
-package sender
-
-import (
-	"fmt"
-	"html/template"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type feishuContent struct {
-	Text string `json:"text"`
-}
-
-type feishuAt struct {
-	AtMobiles []string `json:"atMobiles"`
-	IsAtAll   bool     `json:"isAtAll"`
-}
-
-type feishu struct {
-	Msgtype string        `json:"msg_type"`
-	Content feishuContent `json:"content"`
-	At      feishuAt      `json:"at"`
-}
-
-var (
-	_ CallBacker = (*FeishuSender)(nil)
-)
-
-type FeishuSender struct {
-	tpl *template.Template
-}
-
-func (fs *FeishuSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	ats := ExtractAtsParams(ctx.CallBackURL)
-	message := BuildTplMessage(models.Feishu, fs.tpl, ctx.Events)
-
-	if len(ats) > 0 {
-		atTags := ""
-		for _, at := range ats {
-			atTags += fmt.Sprintf("<at user_id=\"%s\"></at> ", at)
-		}
-		message = atTags + message
-	}
-
-	body := feishu{
-		Msgtype: "text",
-		Content: feishuContent{
-			Text: message,
-		},
-	}
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, body, "callback", ctx.Stats, ctx.Events)
-}
-
-func (fs *FeishuSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	urls, ats, tokens := fs.extract(ctx.Users)
-	message := BuildTplMessage(models.Feishu, fs.tpl, ctx.Events)
-	for i, url := range urls {
-		body := feishu{
-			Msgtype: "text",
-			Content: feishuContent{
-				Text: message,
-			},
-		}
-		if !strings.Contains(url, "noat=1") {
-			body.At = feishuAt{
-				AtMobiles: ats,
-				IsAtAll:   false,
-			}
-		}
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.Feishu, ctx.Stats, ctx.Events)
-	}
-}
-
-func (fs *FeishuSender) extract(users []*models.User) ([]string, []string, []string) {
-	urls := make([]string, 0, len(users))
-	ats := make([]string, 0, len(users))
-	tokens := make([]string, 0, len(users))
-
-	for _, user := range users {
-		if user.Phone != "" {
-			ats = append(ats, user.Phone)
-		}
-		if token, has := user.ExtractToken(models.Feishu); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://open.feishu.cn/open-apis/bot/v2/hook/" + token
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, ats, tokens
-}

alert/sender/feishucard.go

@@ -1,180 +0,0 @@
-package sender
-
-import (
-	"fmt"
-	"html/template"
-	"net/url"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type Conf struct {
-	WideScreenMode bool `json:"wide_screen_mode"`
-	EnableForward  bool `json:"enable_forward"`
-}
-
-type Te struct {
-	Content string `json:"content"`
-	Tag     string `json:"tag"`
-}
-
-type Element struct {
-	Tag      string    `json:"tag"`
-	Text     Te        `json:"text"`
-	Content  string    `json:"content"`
-	Elements []Element `json:"elements"`
-}
-
-type Titles struct {
-	Content string `json:"content"`
-	Tag     string `json:"tag"`
-}
-
-type Headers struct {
-	Title    Titles `json:"title"`
-	Template string `json:"template"`
-}
-
-type Cards struct {
-	Config   Conf      `json:"config"`
-	Elements []Element `json:"elements"`
-	Header   Headers   `json:"header"`
-}
-
-type feishuCard struct {
-	feishu
-	Card Cards `json:"card"`
-}
-
-type FeishuCardSender struct {
-	tpl *template.Template
-}
-
-const (
-	Recovered = "recovered"
-	Triggered = "triggered"
-)
-
-func createFeishuCardBody() feishuCard {
-	return feishuCard{
-		feishu: feishu{Msgtype: "interactive"},
-		Card: Cards{
-			Config: Conf{
-				WideScreenMode: true,
-				EnableForward:  true,
-			},
-			Header: Headers{
-				Title: Titles{
-					Tag: "plain_text",
-				},
-			},
-			Elements: []Element{
-				{
-					Tag: "div",
-					Text: Te{
-						Tag: "lark_md",
-					},
-				},
-				{
-					Tag: "hr",
-				},
-				{
-					Tag: "note",
-					Elements: []Element{
-						{
-							Tag: "lark_md",
-						},
-					},
-				},
-			},
-		},
-	}
-}
-
-func (fs *FeishuCardSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	ats := ExtractAtsParams(ctx.CallBackURL)
-	message := BuildTplMessage(models.FeishuCard, fs.tpl, ctx.Events)
-
-	if len(ats) > 0 {
-		atTags := ""
-		for _, at := range ats {
-			if strings.Contains(at, "@") {
-				atTags += fmt.Sprintf("<at email=\"%s\" ></at>", at)
-			} else {
-				atTags += fmt.Sprintf("<at id=\"%s\" ></at>", at)
-			}
-		}
-		message = atTags + message
-	}
-
-	color := "red"
-	lowerUnicode := strings.ToLower(message)
-	if strings.Count(lowerUnicode, Recovered) > 0 && strings.Count(lowerUnicode, Triggered) > 0 {
-		color = "orange"
-	} else if strings.Count(lowerUnicode, Recovered) > 0 {
-		color = "green"
-	}
-
-	SendTitle := fmt.Sprintf("🔔 %s", ctx.Events[0].RuleName)
-	body := createFeishuCardBody()
-	body.Card.Header.Title.Content = SendTitle
-	body.Card.Header.Template = color
-	body.Card.Elements[0].Text.Content = message
-	body.Card.Elements[2].Elements[0].Content = SendTitle
-
-	// This is to be compatible with the feishucard interface, if with query string parameters, the request will fail
-	// Remove query parameters from the URL,
-	parsedURL, err := url.Parse(ctx.CallBackURL)
-	if err != nil {
-		return
-	}
-	parsedURL.RawQuery = ""
-
-	doSendAndRecord(ctx.Ctx, parsedURL.String(), parsedURL.String(), body, "callback", ctx.Stats, ctx.Events)
-}
-
-func (fs *FeishuCardSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	urls, tokens := fs.extract(ctx.Users)
-	message := BuildTplMessage(models.FeishuCard, fs.tpl, ctx.Events)
-	color := "red"
-	lowerUnicode := strings.ToLower(message)
-	if strings.Count(lowerUnicode, Recovered) > 0 && strings.Count(lowerUnicode, Triggered) > 0 {
-		color = "orange"
-	} else if strings.Count(lowerUnicode, Recovered) > 0 {
-		color = "green"
-	}
-
-	SendTitle := fmt.Sprintf("🔔 %s", ctx.Events[0].RuleName)
-	body := createFeishuCardBody()
-	body.Card.Header.Title.Content = SendTitle
-	body.Card.Header.Template = color
-	body.Card.Elements[0].Text.Content = message
-	body.Card.Elements[2].Elements[0].Content = SendTitle
-	for i, url := range urls {
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.FeishuCard, ctx.Stats, ctx.Events)
-	}
-}
-
-func (fs *FeishuCardSender) extract(users []*models.User) ([]string, []string) {
-	urls := make([]string, 0, len(users))
-	tokens := make([]string, 0, len(users))
-	for i := range users {
-		if token, has := users[i].ExtractToken(models.FeishuCard); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://open.feishu.cn/open-apis/bot/v2/hook/" + strings.TrimSpace(token)
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, tokens
-}

alert/sender/ibex.go

@@ -1,287 +0,0 @@
-// @Author: Ciusyan 6/5/24
-
-package sender
-
-import (
-	"encoding/json"
-	"fmt"
-	"strconv"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	imodels "github.com/flashcatcloud/ibex/src/models"
-	"github.com/flashcatcloud/ibex/src/storage"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-var (
-	_ CallBacker = (*IbexCallBacker)(nil)
-)
-
-type IbexCallBacker struct {
-	targetCache  *memsto.TargetCacheType
-	userCache    *memsto.UserCacheType
-	taskTplCache *memsto.TaskTplCache
-}
-
-func (c *IbexCallBacker) CallBack(ctx CallBackContext) {
-	if len(ctx.CallBackURL) == 0 || len(ctx.Events) == 0 {
-		logger.Warningf("event_callback_ibex: url or events is empty, url: %s, events: %+v", ctx.CallBackURL, ctx.Events)
-		return
-	}
-
-	event := ctx.Events[0]
-
-	if event.IsRecovered {
-		logger.Infof("event_callback_ibex: event is recovered, event: %+v", event)
-		return
-	}
-
-	c.handleIbex(ctx.Ctx, ctx.CallBackURL, event)
-}
-
-func (c *IbexCallBacker) handleIbex(ctx *ctx.Context, url string, event *models.AlertCurEvent) {
-	logger.Infof("event_callback_ibex: url: %s, event: %+v", url, event)
-	if imodels.DB() == nil && ctx.IsCenter {
-		logger.Warningf("event_callback_ibex: db is nil, event: %+v", event)
-		return
-	}
-
-	arr := strings.Split(url, "/")
-
-	var idstr string
-	var host string
-
-	if len(arr) > 1 {
-		idstr = arr[1]
-	}
-
-	if len(arr) > 2 {
-		host = arr[2]
-	}
-
-	id, err := strconv.ParseInt(idstr, 10, 64)
-	if err != nil {
-		logger.Errorf("event_callback_ibex: failed to parse url: %s event: %+v", url, event)
-		return
-	}
-
-	if host == "" {
-		// 用户在callback url中没有传入host，就从event中解析
-		host = event.TargetIdent
-
-		if host == "" {
-			if ident, has := event.TagsMap["ident"]; has {
-				host = ident
-			}
-		}
-	}
-
-	if host == "" {
-		logger.Errorf("event_callback_ibex: failed to get host, id: %d, event: %+v", id, event)
-		return
-	}
-
-	CallIbex(ctx, id, host, c.taskTplCache, c.targetCache, c.userCache, event)
-}
-
-func CallIbex(ctx *ctx.Context, id int64, host string,
-	taskTplCache *memsto.TaskTplCache, targetCache *memsto.TargetCacheType,
-	userCache *memsto.UserCacheType, event *models.AlertCurEvent) {
-	logger.Infof("event_callback_ibex: id: %d, host: %s, event: %+v", id, host, event)
-
-	tpl := taskTplCache.Get(id)
-	if tpl == nil {
-		logger.Errorf("event_callback_ibex: no such tpl(%d), event: %+v", id, event)
-		return
-	}
-	// check perm
-	// tpl.GroupId - host - account 三元组校验权限
-	can, err := canDoIbex(tpl.UpdateBy, tpl, host, targetCache, userCache)
-	if err != nil {
-		logger.Errorf("event_callback_ibex: check perm fail: %v, event: %+v", err, event)
-		return
-	}
-
-	if !can {
-		logger.Errorf("event_callback_ibex: user(%s) no permission, event: %+v", tpl.UpdateBy, event)
-		return
-	}
-
-	tagsMap := make(map[string]string)
-	for i := 0; i < len(event.TagsJSON); i++ {
-		pair := strings.TrimSpace(event.TagsJSON[i])
-		if pair == "" {
-			continue
-		}
-
-		arr := strings.SplitN(pair, "=", 2)
-		if len(arr) != 2 {
-			continue
-		}
-
-		tagsMap[arr[0]] = arr[1]
-	}
-	// 附加告警级别  告警触发值标签
-	tagsMap["alert_severity"] = strconv.Itoa(event.Severity)
-	tagsMap["alert_trigger_value"] = event.TriggerValue
-	tagsMap["is_recovered"] = strconv.FormatBool(event.IsRecovered)
-
-	tags, err := json.Marshal(tagsMap)
-	if err != nil {
-		logger.Errorf("event_callback_ibex: failed to marshal tags to json: %v, event: %+v", tagsMap, event)
-		return
-	}
-
-	// call ibex
-	in := models.TaskForm{
-		Title:          tpl.Title + " FH: " + host,
-		Account:        tpl.Account,
-		Batch:          tpl.Batch,
-		Tolerance:      tpl.Tolerance,
-		Timeout:        tpl.Timeout,
-		Pause:          tpl.Pause,
-		Script:         tpl.Script,
-		Args:           tpl.Args,
-		Stdin:          string(tags),
-		Action:         "start",
-		Creator:        tpl.UpdateBy,
-		Hosts:          []string{host},
-		AlertTriggered: true,
-	}
-
-	id, err = TaskAdd(in, tpl.UpdateBy, ctx.IsCenter)
-	if err != nil {
-		logger.Errorf("event_callback_ibex: call ibex fail: %v, event: %+v", err, event)
-		return
-	}
-
-	// write db
-	record := models.TaskRecord{
-		Id:        id,
-		EventId:   event.Id,
-		GroupId:   tpl.GroupId,
-		Title:     in.Title,
-		Account:   in.Account,
-		Batch:     in.Batch,
-		Tolerance: in.Tolerance,
-		Timeout:   in.Timeout,
-		Pause:     in.Pause,
-		Script:    in.Script,
-		Args:      in.Args,
-		CreateAt:  time.Now().Unix(),
-		CreateBy:  in.Creator,
-	}
-
-	if err = record.Add(ctx); err != nil {
-		logger.Errorf("event_callback_ibex: persist task_record fail: %v, event: %+v", err, event)
-	}
-}
-
-func canDoIbex(username string, tpl *models.TaskTpl, host string, targetCache *memsto.TargetCacheType, userCache *memsto.UserCacheType) (bool, error) {
-	user := userCache.GetByUsername(username)
-	if user != nil && user.IsAdmin() {
-		return true, nil
-	}
-
-	target, has := targetCache.Get(host)
-	if !has {
-		return false, nil
-	}
-
-	return target.MatchGroupId(tpl.GroupId), nil
-}
-
-func TaskAdd(f models.TaskForm, authUser string, isCenter bool) (int64, error) {
-	if storage.Cache == nil {
-		logger.Warningf("event_callback_ibex: redis cache is nil, task: %+v", f)
-		return 0, fmt.Errorf("redis cache is nil")
-	}
-
-	hosts := cleanHosts(f.Hosts)
-	if len(hosts) == 0 {
-		return 0, fmt.Errorf("arg(hosts) empty")
-	}
-
-	taskMeta := &imodels.TaskMeta{
-		Title:     f.Title,
-		Account:   f.Account,
-		Batch:     f.Batch,
-		Tolerance: f.Tolerance,
-		Timeout:   f.Timeout,
-		Pause:     f.Pause,
-		Script:    f.Script,
-		Args:      f.Args,
-		Stdin:     f.Stdin,
-		Creator:   f.Creator,
-	}
-
-	err := taskMeta.CleanFields()
-	if err != nil {
-		return 0, err
-	}
-
-	taskMeta.HandleFH(hosts[0])
-
-	// 任务类型分为"告警规则触发"和"n9e center用户下发"两种；
-	// 边缘机房"告警规则触发"的任务不需要规划，并且它可能是失联的，无法使用db资源，所以放入redis缓存中，直接下发给agentd执行
-	if !isCenter && f.AlertTriggered {
-		if err := taskMeta.Create(); err != nil {
-			// 当网络不连通时，生成唯一的id，防止边缘机房中不同任务的id相同；
-			// 方法是，redis自增id去防止同一个机房的不同n9e edge生成的id相同；
-			// 但没法防止不同边缘机房生成同样的id，所以，生成id的数据不会上报存入数据库，只用于闭环执行。
-			taskMeta.Id, err = storage.IdGet()
-			if err != nil {
-				return 0, err
-			}
-		}
-
-		taskHost := imodels.TaskHost{
-			Id:     taskMeta.Id,
-			Host:   hosts[0],
-			Status: "running",
-		}
-		if err = taskHost.Create(); err != nil {
-			logger.Warningf("task_add_fail: authUser=%s title=%s err=%s", authUser, taskMeta.Title, err.Error())
-		}
-
-		// 缓存任务元信息和待下发的任务
-		err = taskMeta.Cache(hosts[0])
-		if err != nil {
-			return 0, err
-		}
-
-	} else {
-		// 如果是中心机房，还是保持之前的逻辑
-		err = taskMeta.Save(hosts, f.Action)
-		if err != nil {
-			return 0, err
-		}
-	}
-
-	logger.Infof("task_add_succ: authUser=%s title=%s", authUser, taskMeta.Title)
-	return taskMeta.Id, nil
-}
-
-func cleanHosts(formHosts []string) []string {
-	cnt := len(formHosts)
-	arr := make([]string, 0, cnt)
-	for i := 0; i < cnt; i++ {
-		item := strings.TrimSpace(formHosts[i])
-		if item == "" {
-			continue
-		}
-
-		if strings.HasPrefix(item, "#") {
-			continue
-		}
-
-		arr = append(arr, item)
-	}
-
-	return arr
-}

alert/sender/lark.go

@@ -1,65 +0,0 @@
-package sender
-
-import (
-	"html/template"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-var (
-	_ CallBacker = (*LarkSender)(nil)
-)
-
-type LarkSender struct {
-	tpl *template.Template
-}
-
-func (lk *LarkSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	body := feishu{
-		Msgtype: "text",
-		Content: feishuContent{
-			Text: BuildTplMessage(models.Lark, lk.tpl, ctx.Events),
-		},
-	}
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, body, "callback", ctx.Stats, ctx.Events)
-}
-
-func (lk *LarkSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	urls, tokens := lk.extract(ctx.Users)
-	message := BuildTplMessage(models.Lark, lk.tpl, ctx.Events)
-	for i, url := range urls {
-		body := feishu{
-			Msgtype: "text",
-			Content: feishuContent{
-				Text: message,
-			},
-		}
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.Lark, ctx.Stats, ctx.Events)
-	}
-}
-
-func (lk *LarkSender) extract(users []*models.User) ([]string, []string) {
-	urls := make([]string, 0, len(users))
-	tokens := make([]string, 0, len(users))
-
-	for _, user := range users {
-		if token, has := user.ExtractToken(models.Lark); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://open.larksuite.com/open-apis/bot/v2/hook/" + token
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, tokens
-}

alert/sender/larkcard.go

@@ -1,101 +0,0 @@
-package sender
-
-import (
-	"fmt"
-	"html/template"
-	"net/url"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type LarkCardSender struct {
-	tpl *template.Template
-}
-
-func (fs *LarkCardSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	ats := ExtractAtsParams(ctx.CallBackURL)
-	message := BuildTplMessage(models.LarkCard, fs.tpl, ctx.Events)
-
-	if len(ats) > 0 {
-		atTags := ""
-		for _, at := range ats {
-			if strings.Contains(at, "@") {
-				atTags += fmt.Sprintf("<at email=\"%s\" ></at>", at)
-			} else {
-				atTags += fmt.Sprintf("<at id=\"%s\" ></at>", at)
-			}
-		}
-		message = atTags + message
-	}
-
-	color := "red"
-	lowerUnicode := strings.ToLower(message)
-	if strings.Count(lowerUnicode, Recovered) > 0 && strings.Count(lowerUnicode, Triggered) > 0 {
-		color = "orange"
-	} else if strings.Count(lowerUnicode, Recovered) > 0 {
-		color = "green"
-	}
-
-	SendTitle := fmt.Sprintf("🔔 %s", ctx.Events[0].RuleName)
-	body := createFeishuCardBody()
-	body.Card.Header.Title.Content = SendTitle
-	body.Card.Header.Template = color
-	body.Card.Elements[0].Text.Content = message
-	body.Card.Elements[2].Elements[0].Content = SendTitle
-
-	// This is to be compatible with the Larkcard interface, if with query string parameters, the request will fail
-	// Remove query parameters from the URL,
-	parsedURL, err := url.Parse(ctx.CallBackURL)
-	if err != nil {
-		return
-	}
-	parsedURL.RawQuery = ""
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, body, "callback", ctx.Stats, ctx.Events)
-}
-
-func (fs *LarkCardSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	urls, tokens := fs.extract(ctx.Users)
-	message := BuildTplMessage(models.LarkCard, fs.tpl, ctx.Events)
-	color := "red"
-	lowerUnicode := strings.ToLower(message)
-	if strings.Count(lowerUnicode, Recovered) > 0 && strings.Count(lowerUnicode, Triggered) > 0 {
-		color = "orange"
-	} else if strings.Count(lowerUnicode, Recovered) > 0 {
-		color = "green"
-	}
-
-	SendTitle := fmt.Sprintf("🔔 %s", ctx.Events[0].RuleName)
-	body := createFeishuCardBody()
-	body.Card.Header.Title.Content = SendTitle
-	body.Card.Header.Template = color
-	body.Card.Elements[0].Text.Content = message
-	body.Card.Elements[2].Elements[0].Content = SendTitle
-	for i, url := range urls {
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.LarkCard, ctx.Stats, ctx.Events)
-	}
-}
-
-func (fs *LarkCardSender) extract(users []*models.User) ([]string, []string) {
-	urls := make([]string, 0, len(users))
-	tokens := make([]string, 0)
-	for i := range users {
-		if token, has := users[i].ExtractToken(models.Lark); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://open.larksuite.com/open-apis/bot/v2/hook/" + strings.TrimSpace(token)
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, tokens
-}

alert/sender/mm.go

@@ -1,117 +0,0 @@
-package sender
-
-import (
-	"html/template"
-	"net/url"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type MatterMostMessage struct {
-	Text   string
-	Tokens []string
-	Stats  *astats.Stats
-}
-
-type mm struct {
-	Channel  string `json:"channel"`
-	Username string `json:"username"`
-	Text     string `json:"text"`
-}
-
-type MmSender struct {
-	tpl *template.Template
-}
-
-func (ms *MmSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-
-	urls := ms.extract(ctx.Users)
-	if len(urls) == 0 {
-		return
-	}
-	message := BuildTplMessage(models.Mm, ms.tpl, ctx.Events)
-
-	SendMM(ctx.Ctx, MatterMostMessage{
-		Text:   message,
-		Tokens: urls,
-		Stats:  ctx.Stats,
-	}, ctx.Events, models.Mm)
-}
-
-func (ms *MmSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-	message := BuildTplMessage(models.Mm, ms.tpl, ctx.Events)
-
-	SendMM(ctx.Ctx, MatterMostMessage{
-		Text:   message,
-		Tokens: []string{ctx.CallBackURL},
-		Stats:  ctx.Stats,
-	}, ctx.Events, "callback")
-}
-
-func (ms *MmSender) extract(users []*models.User) []string {
-	tokens := make([]string, 0, len(users))
-	for _, user := range users {
-		if token, has := user.ExtractToken(models.Mm); has {
-			tokens = append(tokens, token)
-		}
-	}
-	return tokens
-}
-
-func SendMM(ctx *ctx.Context, message MatterMostMessage, events []*models.AlertCurEvent, channel string) {
-	for i := 0; i < len(message.Tokens); i++ {
-		u, err := url.Parse(message.Tokens[i])
-		if err != nil {
-			logger.Errorf("mm_sender: failed to parse error=%v", err)
-			NotifyRecord(ctx, events, 0, channel, message.Tokens[i], "", err)
-			continue
-		}
-
-		v, err := url.ParseQuery(u.RawQuery)
-		if err != nil {
-			logger.Errorf("mm_sender: failed to parse query error=%v", err)
-		}
-
-		channels := v["channel"] // do not get
-		txt := ""
-		atuser := v["atuser"]
-		if len(atuser) != 0 {
-			txt = strings.Join(MapStrToStr(atuser, func(u string) string {
-				return "@" + u
-			}), ",") + "\n"
-		}
-		username := v.Get("username")
-		if err != nil {
-			logger.Errorf("mm_sender: failed to parse error=%v", err)
-		}
-		// simple concatenating
-		ur := u.Scheme + "://" + u.Host + u.Path
-		for _, channel := range channels {
-			body := mm{
-				Channel:  channel,
-				Username: username,
-				Text:     txt + message.Text,
-			}
-			doSendAndRecord(ctx, ur, message.Tokens[i], body, channel, message.Stats, events)
-		}
-	}
-}
-
-func MapStrToStr(arr []string, fn func(s string) string) []string {
-	var newArray = []string{}
-	for _, it := range arr {
-		newArray = append(newArray, fn(it))
-	}
-	return newArray
-}

alert/sender/notify_record_queue.go

@@ -1,75 +0,0 @@
-package sender
-
-import (
-	"errors"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/toolkits/pkg/container/list"
-	"github.com/toolkits/pkg/logger"
-)
-
-// 通知记录队列，最大长度 1000000
-var NotifyRecordQueue = list.NewSafeListLimited(1000000)
-
-// 每秒上报通知记录队列大小
-func ReportNotifyRecordQueueSize(stats *astats.Stats) {
-	for {
-		time.Sleep(time.Second)
-		stats.GaugeNotifyRecordQueueSize.Set(float64(NotifyRecordQueue.Len()))
-	}
-}
-
-// 推送通知记录到队列
-// 若队列满 则返回 error
-func PushNotifyRecords(records []*models.NotificaitonRecord) error {
-	for _, record := range records {
-		if ok := NotifyRecordQueue.PushFront(record); !ok {
-			logger.Warningf("notify record queue is full, record: %+v", record)
-			return errors.New("notify record queue is full")
-		}
-	}
-
-	return nil
-}
-
-type NotifyRecordConsumer struct {
-	ctx *ctx.Context
-}
-
-func NewNotifyRecordConsumer(ctx *ctx.Context) *NotifyRecordConsumer {
-	return &NotifyRecordConsumer{
-		ctx: ctx,
-	}
-}
-
-// 消费通知记录队列 每 100ms 检测一次队列是否为空
-func (c *NotifyRecordConsumer) LoopConsume() {
-	duration := time.Duration(100) * time.Millisecond
-	for {
-		// 无论队列是否为空 都需要等待
-		time.Sleep(duration)
-
-		inotis := NotifyRecordQueue.PopBackBy(100)
-
-		if len(inotis) == 0 {
-			continue
-		}
-
-		// 类型转换，不然 CreateInBatches 会报错
-		notis := make([]*models.NotificaitonRecord, 0, len(inotis))
-		for _, inoti := range inotis {
-			notis = append(notis, inoti.(*models.NotificaitonRecord))
-		}
-
-		c.consume(notis)
-	}
-}
-
-func (c *NotifyRecordConsumer) consume(notis []*models.NotificaitonRecord) {
-	if err := models.DB(c.ctx).CreateInBatches(notis, 100).Error; err != nil {
-		logger.Errorf("add notis:%v failed, err: %v", notis, err)
-	}
-}

alert/sender/plugin.go

@@ -1,136 +0,0 @@
-package sender
-
-import (
-	"bytes"
-	"fmt"
-	"os"
-	"os/exec"
-	"time"
-	"unicode/utf8"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/file"
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/sys"
-)
-
-func MayPluginNotify(ctx *ctx.Context, noticeBytes []byte, notifyScript models.NotifyScript,
-	stats *astats.Stats, event *models.AlertCurEvent) {
-	if len(noticeBytes) == 0 {
-		return
-	}
-	alertingCallScript(ctx, noticeBytes, notifyScript, stats, event)
-}
-
-func alertingCallScript(ctx *ctx.Context, stdinBytes []byte, notifyScript models.NotifyScript,
-	stats *astats.Stats, event *models.AlertCurEvent) {
-	// not enable or no notify.py? do nothing
-	config := notifyScript
-	if !config.Enable || config.Content == "" {
-		return
-	}
-
-	channel := "script"
-	stats.AlertNotifyTotal.WithLabelValues(channel).Inc()
-	fpath := ".notify_scriptt"
-	if config.Type == 1 {
-		fpath = config.Content
-	} else {
-		rewrite := true
-		if file.IsExist(fpath) {
-			oldContent, err := file.ToString(fpath)
-			if err != nil {
-				logger.Errorf("event_script_notify_fail: read script file err: %v", err)
-				stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-				return
-			}
-
-			if oldContent == config.Content {
-				rewrite = false
-			}
-		}
-
-		if rewrite {
-			_, err := file.WriteString(fpath, config.Content)
-			if err != nil {
-				logger.Errorf("event_script_notify_fail: write script file err: %v", err)
-				stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-				return
-			}
-
-			err = os.Chmod(fpath, 0777)
-			if err != nil {
-				logger.Errorf("event_script_notify_fail: chmod script file err: %v", err)
-				stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-				return
-			}
-		}
-		fpath = "./" + fpath
-	}
-
-	cmd := exec.Command(fpath)
-	cmd.Stdin = bytes.NewReader(stdinBytes)
-
-	// combine stdout and stderr
-	var buf bytes.Buffer
-	cmd.Stdout = &buf
-	cmd.Stderr = &buf
-
-	err := startCmd(cmd)
-	if err != nil {
-		logger.Errorf("event_script_notify_fail: run cmd err: %v", err)
-		return
-	}
-
-	err, isTimeout := sys.WrapTimeout(cmd, time.Duration(config.Timeout)*time.Second)
-
-	res := buf.String()
-
-	// 截断超出长度的输出
-	if len(res) > 512 {
-		// 确保在有效的UTF-8字符边界处截断
-		validLen := 0
-		for i := 0; i < 512 && i < len(res); {
-			_, size := utf8.DecodeRuneInString(res[i:])
-			if i+size > 512 {
-				break
-			}
-			i += size
-			validLen = i
-		}
-		res = res[:validLen] + "..."
-	}
-
-	NotifyRecord(ctx, []*models.AlertCurEvent{event}, 0, channel, cmd.String(), res, buildErr(err, isTimeout))
-
-	if isTimeout {
-		if err == nil {
-			logger.Errorf("event_script_notify_fail: timeout and killed process %s", fpath)
-		}
-
-		if err != nil {
-			logger.Errorf("event_script_notify_fail: kill process %s occur error %v", fpath, err)
-			stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-		}
-		return
-	}
-
-	if err != nil {
-		logger.Errorf("event_script_notify_fail: exec script %s occur error: %v, output: %s", fpath, err, res)
-		stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-		return
-	}
-
-	logger.Infof("event_script_notify_ok: exec %s output: %s", fpath, res)
-}
-
-func buildErr(err error, isTimeout bool) error {
-	if err == nil && !isTimeout {
-		return nil
-	} else {
-		return fmt.Errorf("is_timeout: %v, err: %v", isTimeout, err)
-	}
-}

alert/sender/sender.go

@@ -1,85 +0,0 @@
-package sender
-
-import (
-	"bytes"
-	"html/template"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-)
-
-type (
-	// Sender 发送消息通知的接口
-	Sender interface {
-		Send(ctx MessageContext)
-	}
-
-	// MessageContext 一个event所生成的告警通知的上下文
-	MessageContext struct {
-		Users  []*models.User
-		Rule   *models.AlertRule
-		Events []*models.AlertCurEvent
-		Stats  *astats.Stats
-		Ctx    *ctx.Context
-	}
-)
-
-func NewSender(key string, tpls map[string]*template.Template, smtp ...aconf.SMTPConfig) Sender {
-	switch key {
-	case models.Dingtalk:
-		return &DingtalkSender{tpl: tpls[models.Dingtalk]}
-	case models.Wecom:
-		return &WecomSender{tpl: tpls[models.Wecom]}
-	case models.Feishu:
-		return &FeishuSender{tpl: tpls[models.Feishu]}
-	case models.FeishuCard:
-		return &FeishuCardSender{tpl: tpls[models.FeishuCard]}
-	case models.Email:
-		return &EmailSender{subjectTpl: tpls[models.EmailSubject], contentTpl: tpls[models.Email], smtp: smtp[0]}
-	case models.Mm:
-		return &MmSender{tpl: tpls[models.Mm]}
-	case models.Telegram:
-		return &TelegramSender{tpl: tpls[models.Telegram]}
-	case models.Lark:
-		return &LarkSender{tpl: tpls[models.Lark]}
-	case models.LarkCard:
-		return &LarkCardSender{tpl: tpls[models.LarkCard]}
-	}
-	return nil
-}
-
-func BuildMessageContext(ctx *ctx.Context, rule *models.AlertRule, events []*models.AlertCurEvent,
-	uids []int64, userCache *memsto.UserCacheType, stats *astats.Stats) MessageContext {
-	users := userCache.GetByUserIds(uids)
-	return MessageContext{
-		Rule:   rule,
-		Events: events,
-		Users:  users,
-		Stats:  stats,
-		Ctx:    ctx,
-	}
-}
-
-type BuildTplMessageFunc func(channel string, tpl *template.Template, events []*models.AlertCurEvent) string
-
-var BuildTplMessage BuildTplMessageFunc = buildTplMessage
-
-func buildTplMessage(channel string, tpl *template.Template, events []*models.AlertCurEvent) string {
-	if tpl == nil {
-		return "tpl for current sender not found, please check configuration"
-	}
-
-	var content string
-	for _, event := range events {
-		var body bytes.Buffer
-		if err := tpl.Execute(&body, event); err != nil {
-			return err.Error()
-		}
-		content += body.String() + "\n\n"
-	}
-
-	return content
-}

alert/sender/telegram.go

@@ -1,98 +0,0 @@
-package sender
-
-import (
-	"errors"
-	"html/template"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type TelegramMessage struct {
-	Text   string
-	Tokens []string
-	Stats  *astats.Stats
-}
-
-type telegram struct {
-	ParseMode string `json:"parse_mode"`
-	Text      string `json:"text"`
-}
-
-var (
-	_ CallBacker = (*TelegramSender)(nil)
-)
-
-type TelegramSender struct {
-	tpl *template.Template
-}
-
-func (ts *TelegramSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	message := BuildTplMessage(models.Telegram, ts.tpl, ctx.Events)
-	SendTelegram(ctx.Ctx, TelegramMessage{
-		Text:   message,
-		Tokens: []string{ctx.CallBackURL},
-		Stats:  ctx.Stats,
-	}, ctx.Events, "callback")
-}
-
-func (ts *TelegramSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	tokens := ts.extract(ctx.Users)
-	message := BuildTplMessage(models.Telegram, ts.tpl, ctx.Events)
-
-	SendTelegram(ctx.Ctx, TelegramMessage{
-		Text:   message,
-		Tokens: tokens,
-		Stats:  ctx.Stats,
-	}, ctx.Events, models.Telegram)
-}
-
-func (ts *TelegramSender) extract(users []*models.User) []string {
-	tokens := make([]string, 0, len(users))
-	for _, user := range users {
-		if token, has := user.ExtractToken(models.Telegram); has {
-			tokens = append(tokens, token)
-		}
-	}
-	return tokens
-}
-
-func SendTelegram(ctx *ctx.Context, message TelegramMessage, events []*models.AlertCurEvent, channel string) {
-	for i := 0; i < len(message.Tokens); i++ {
-		if !strings.Contains(message.Tokens[i], "/") && !strings.HasPrefix(message.Tokens[i], "https://") {
-			logger.Errorf("telegram_sender: result=fail invalid token=%s", message.Tokens[i])
-			NotifyRecord(ctx, events, 0, channel, message.Tokens[i], "", errors.New("invalid token"))
-			continue
-		}
-		var url string
-		if strings.HasPrefix(message.Tokens[i], "https://") || strings.HasPrefix(message.Tokens[i], "http://") {
-			url = message.Tokens[i]
-		} else {
-			array := strings.Split(message.Tokens[i], "/")
-			if len(array) != 2 {
-				logger.Errorf("telegram_sender: result=fail invalid token=%s", message.Tokens[i])
-				continue
-			}
-			botToken := array[0]
-			chatId := array[1]
-			url = "https://api.telegram.org/bot" + botToken + "/sendMessage?chat_id=" + chatId
-		}
-		body := telegram{
-			ParseMode: "markdown",
-			Text:      message.Text,
-		}
-
-		doSendAndRecord(ctx, url, message.Tokens[i], body, channel, message.Stats, events)
-	}
-}

alert/sender/webhook.go

@@ -1,182 +0,0 @@
-package sender
-
-import (
-	"bytes"
-	"crypto/tls"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net/http"
-	"sync"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-func sendWebhook(webhook *models.Webhook, event interface{}, stats *astats.Stats) (bool, string, error) {
-	channel := "webhook"
-	if webhook.Type == models.RuleCallback {
-		channel = "callback"
-	}
-
-	conf := webhook
-	if conf.Url == "" || !conf.Enable {
-		return false, "", nil
-	}
-	bs, err := json.Marshal(event)
-	if err != nil {
-		logger.Errorf("%s alertingWebhook failed to marshal event:%+v err:%v", channel, event, err)
-		return false, "", err
-	}
-
-	bf := bytes.NewBuffer(bs)
-
-	req, err := http.NewRequest("POST", conf.Url, bf)
-	if err != nil {
-		logger.Warningf("%s alertingWebhook failed to new reques event:%s err:%v", channel, string(bs), err)
-		return true, "", err
-	}
-
-	req.Header.Set("Content-Type", "application/json")
-	if conf.BasicAuthUser != "" && conf.BasicAuthPass != "" {
-		req.SetBasicAuth(conf.BasicAuthUser, conf.BasicAuthPass)
-	}
-
-	if len(conf.Headers) > 0 && len(conf.Headers)%2 == 0 {
-		for i := 0; i < len(conf.Headers); i += 2 {
-			if conf.Headers[i] == "host" || conf.Headers[i] == "Host" {
-				req.Host = conf.Headers[i+1]
-				continue
-			}
-			req.Header.Set(conf.Headers[i], conf.Headers[i+1])
-		}
-	}
-	insecureSkipVerify := false
-	if webhook != nil {
-		insecureSkipVerify = webhook.SkipVerify
-	}
-
-	if conf.Client == nil {
-		logger.Warningf("event_%s, event:%s, url: [%s], error: [%s]", channel, string(bs), conf.Url, "client is nil")
-		conf.Client = &http.Client{
-			Timeout: time.Duration(conf.Timeout) * time.Second,
-			Transport: &http.Transport{
-				TLSClientConfig: &tls.Config{InsecureSkipVerify: insecureSkipVerify},
-			},
-		}
-	}
-
-	stats.AlertNotifyTotal.WithLabelValues(channel).Inc()
-	var resp *http.Response
-	var body []byte
-	resp, err = conf.Client.Do(req)
-
-	if err != nil {
-		stats.AlertNotifyErrorTotal.WithLabelValues(channel).Inc()
-		logger.Errorf("event_%s_fail, event:%s, url: [%s], error: [%s]", channel, string(bs), conf.Url, err)
-		return true, "", err
-	}
-
-	if resp.Body != nil {
-		defer resp.Body.Close()
-		body, _ = io.ReadAll(resp.Body)
-	}
-
-	if resp.StatusCode == 429 {
-		logger.Errorf("event_%s_fail, url: %s, response code: %d, body: %s event:%s", channel, conf.Url, resp.StatusCode, string(body), string(bs))
-		return true, string(body), fmt.Errorf("status code is 429")
-	}
-
-	logger.Debugf("event_%s_succ, url: %s, response code: %d, body: %s event:%s", channel, conf.Url, resp.StatusCode, string(body), string(bs))
-	return false, string(body), nil
-}
-
-func SingleSendWebhooks(ctx *ctx.Context, webhooks map[string]*models.Webhook, event *models.AlertCurEvent, stats *astats.Stats) {
-	for _, conf := range webhooks {
-		retryCount := 0
-		for retryCount < 3 {
-			needRetry, res, err := sendWebhook(conf, event, stats)
-			NotifyRecord(ctx, []*models.AlertCurEvent{event}, 0, "webhook", conf.Url, res, err)
-			if !needRetry {
-				break
-			}
-			retryCount++
-			time.Sleep(time.Minute * 1 * time.Duration(retryCount))
-		}
-	}
-}
-
-func BatchSendWebhooks(ctx *ctx.Context, webhooks map[string]*models.Webhook, event *models.AlertCurEvent, stats *astats.Stats) {
-	for _, conf := range webhooks {
-		logger.Infof("push event:%+v to queue:%v", event, conf)
-		PushEvent(ctx, conf, event, stats)
-	}
-}
-
-var EventQueue = make(map[string]*WebhookQueue)
-var CallbackEventQueue = make(map[string]*WebhookQueue)
-var CallbackEventQueueLock sync.RWMutex
-var EventQueueLock sync.RWMutex
-
-const QueueMaxSize = 100000
-
-type WebhookQueue struct {
-	eventQueue *SafeEventQueue
-	closeCh    chan struct{}
-}
-
-func PushEvent(ctx *ctx.Context, webhook *models.Webhook, event *models.AlertCurEvent, stats *astats.Stats) {
-	EventQueueLock.RLock()
-	queue := EventQueue[webhook.Url]
-	EventQueueLock.RUnlock()
-
-	if queue == nil {
-		queue = &WebhookQueue{
-			eventQueue: NewSafeEventQueue(QueueMaxSize),
-			closeCh:    make(chan struct{}),
-		}
-
-		EventQueueLock.Lock()
-		EventQueue[webhook.Url] = queue
-		EventQueueLock.Unlock()
-
-		StartConsumer(ctx, queue, webhook.Batch, webhook, stats)
-	}
-
-	succ := queue.eventQueue.Push(event)
-	if !succ {
-		stats.AlertNotifyErrorTotal.WithLabelValues("push_event_queue").Inc()
-		logger.Warningf("Write channel(%s) full, current channel size: %d event:%v", webhook.Url, queue.eventQueue.Len(), event)
-	}
-}
-
-func StartConsumer(ctx *ctx.Context, queue *WebhookQueue, popSize int, webhook *models.Webhook, stats *astats.Stats) {
-	for {
-		select {
-		case <-queue.closeCh:
-			logger.Infof("event queue:%v closed", queue)
-			return
-		default:
-			events := queue.eventQueue.PopN(popSize)
-			if len(events) == 0 {
-				time.Sleep(time.Millisecond * 400)
-				continue
-			}
-
-			retryCount := 0
-			for retryCount < webhook.RetryCount {
-				needRetry, res, err := sendWebhook(webhook, events, stats)
-				go NotifyRecord(ctx, events, 0, "webhook", webhook.Url, res, err)
-				if !needRetry {
-					break
-				}
-				retryCount++
-				time.Sleep(time.Second * time.Duration(webhook.RetryInterval) * time.Duration(retryCount))
-			}
-		}
-	}
-}

alert/sender/webhook_event_queue.go

@@ -1,109 +0,0 @@
-package sender
-
-import (
-	"container/list"
-	"sync"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type SafeEventQueue struct {
-	lock        sync.RWMutex
-	maxSize     int
-	queueHigh   *list.List
-	queueMiddle *list.List
-	queueLow    *list.List
-}
-
-const (
-	High   = 1
-	Middle = 2
-	Low    = 3
-)
-
-func NewSafeEventQueue(maxSize int) *SafeEventQueue {
-	return &SafeEventQueue{
-		maxSize:     maxSize,
-		lock:        sync.RWMutex{},
-		queueHigh:   list.New(),
-		queueMiddle: list.New(),
-		queueLow:    list.New(),
-	}
-}
-
-func (spq *SafeEventQueue) Len() int {
-	spq.lock.RLock()
-	defer spq.lock.RUnlock()
-	return spq.queueHigh.Len() + spq.queueMiddle.Len() + spq.queueLow.Len()
-}
-
-// len 无锁读取长度，不要在本文件外调用
-func (spq *SafeEventQueue) len() int {
-	return spq.queueHigh.Len() + spq.queueMiddle.Len() + spq.queueLow.Len()
-}
-
-func (spq *SafeEventQueue) Push(event *models.AlertCurEvent) bool {
-	spq.lock.Lock()
-	defer spq.lock.Unlock()
-
-	for spq.len() > spq.maxSize {
-		return false
-	}
-
-	switch event.Severity {
-	case High:
-		spq.queueHigh.PushBack(event)
-	case Middle:
-		spq.queueMiddle.PushBack(event)
-	case Low:
-		spq.queueLow.PushBack(event)
-	default:
-		return false
-	}
-
-	return true
-}
-
-// pop 无锁弹出事件，不要在本文件外调用
-func (spq *SafeEventQueue) pop() *models.AlertCurEvent {
-	if spq.len() == 0 {
-		return nil
-	}
-
-	var elem interface{}
-
-	if spq.queueHigh.Len() > 0 {
-		elem = spq.queueHigh.Remove(spq.queueHigh.Front())
-	} else if spq.queueMiddle.Len() > 0 {
-		elem = spq.queueMiddle.Remove(spq.queueMiddle.Front())
-	} else {
-		elem = spq.queueLow.Remove(spq.queueLow.Front())
-	}
-	event, ok := elem.(*models.AlertCurEvent)
-	if !ok {
-		return nil
-	}
-	return event
-}
-
-func (spq *SafeEventQueue) Pop() *models.AlertCurEvent {
-	spq.lock.Lock()
-	defer spq.lock.Unlock()
-	return spq.pop()
-}
-
-func (spq *SafeEventQueue) PopN(n int) []*models.AlertCurEvent {
-	spq.lock.Lock()
-	defer spq.lock.Unlock()
-
-	events := make([]*models.AlertCurEvent, 0, n)
-	count := 0
-	for count < n && spq.len() > 0 {
-		event := spq.pop()
-		if event != nil {
-			events = append(events, event)
-		}
-		count++
-	}
-	return events
-}

alert/sender/webhook_event_queue_test.go

@@ -1,157 +0,0 @@
-package sender
-
-import (
-	"sync"
-	"testing"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/stretchr/testify/assert"
-)
-
-func TestSafePriorityQueue_ConcurrentPushPop(t *testing.T) {
-	spq := NewSafeEventQueue(100000)
-
-	var wg sync.WaitGroup
-	numGoroutines := 100
-	numEvents := 1000
-
-	// 并发 Push
-	wg.Add(numGoroutines)
-	for i := 0; i < numGoroutines; i++ {
-		go func(goroutineID int) {
-			defer wg.Done()
-			for j := 0; j < numEvents; j++ {
-				event := &models.AlertCurEvent{
-					Severity:    goroutineID%3 + 1,
-					TriggerTime: time.Now().UnixNano(),
-				}
-				spq.Push(event)
-			}
-		}(i)
-	}
-	wg.Wait()
-
-	// 检查队列长度是否正确
-	expectedLen := numGoroutines * numEvents
-	assert.Equal(t, expectedLen, spq.Len(), "Queue length mismatch after concurrent pushes")
-
-	// 并发 Pop
-	wg.Add(numGoroutines)
-	for i := 0; i < numGoroutines; i++ {
-		go func() {
-			defer wg.Done()
-			for {
-				event := spq.Pop()
-				if event == nil {
-					return
-				}
-			}
-		}()
-	}
-	wg.Wait()
-
-	// 最终队列应该为空
-	assert.Equal(t, 0, spq.Len(), "Queue should be empty after concurrent pops")
-}
-
-func TestSafePriorityQueue_ConcurrentPopMax(t *testing.T) {
-	spq := NewSafeEventQueue(100000)
-
-	// 添加初始数据
-	for i := 0; i < 1000; i++ {
-		spq.Push(&models.AlertCurEvent{
-			Severity:    i%3 + 1,
-			TriggerTime: time.Now().UnixNano(),
-		})
-	}
-
-	var wg sync.WaitGroup
-	numGoroutines := 10
-	popMax := 100
-
-	// 并发 PopN
-	wg.Add(numGoroutines)
-	for i := 0; i < numGoroutines; i++ {
-		go func() {
-			defer wg.Done()
-			events := spq.PopN(popMax)
-			assert.LessOrEqual(t, len(events), popMax, "PopN exceeded maximum")
-		}()
-	}
-	wg.Wait()
-
-	// 检查队列长度是否正确
-	expectedRemaining := 1000 - (numGoroutines * popMax)
-	if expectedRemaining < 0 {
-		expectedRemaining = 0
-	}
-	assert.Equal(t, expectedRemaining, spq.Len(), "Queue length mismatch after concurrent PopN")
-}
-
-func TestSafePriorityQueue_ConcurrentPushPopWithDifferentSeverities(t *testing.T) {
-	spq := NewSafeEventQueue(100000)
-
-	var wg sync.WaitGroup
-	numGoroutines := 50
-	numEvents := 500
-
-	// 并发 Push 不同优先级的事件
-	wg.Add(numGoroutines)
-	for i := 0; i < numGoroutines; i++ {
-		go func(goroutineID int) {
-			defer wg.Done()
-			for j := 0; j < numEvents; j++ {
-				event := &models.AlertCurEvent{
-					Severity:    goroutineID%3 + 1, // 模拟不同的 Severity
-					TriggerTime: time.Now().UnixNano(),
-				}
-				spq.Push(event)
-			}
-		}(i)
-	}
-	wg.Wait()
-
-	// 检查队列长度是否正确
-	expectedLen := numGoroutines * numEvents
-	assert.Equal(t, expectedLen, spq.Len(), "Queue length mismatch after concurrent pushes")
-
-	// 检查事件的顺序是否按照优先级排列
-	var lastEvent *models.AlertCurEvent
-	for spq.Len() > 0 {
-		event := spq.Pop()
-		if lastEvent != nil {
-			assert.LessOrEqual(t, lastEvent.Severity, event.Severity, "Events are not in correct priority order")
-		}
-		lastEvent = event
-	}
-}
-
-func TestSafePriorityQueue_ExceedMaxSize(t *testing.T) {
-	spq := NewSafeEventQueue(5)
-
-	// 插入超过最大容量的事件
-	for i := 0; i < 10; i++ {
-		spq.Push(&models.AlertCurEvent{
-			Severity:    i % 3,
-			TriggerTime: int64(i),
-		})
-	}
-
-	// 验证队列的长度是否不超过 maxSize
-	assert.LessOrEqual(t, spq.Len(), spq.maxSize)
-
-	// 验证队列中剩余事件的内容
-	expectedEvents := 5
-	if spq.Len() < 5 {
-		expectedEvents = spq.Len()
-	}
-
-	// 检查最后存入的事件是否是按优先级排序
-	for i := 0; i < expectedEvents; i++ {
-		event := spq.Pop()
-		if event != nil {
-			assert.LessOrEqual(t, event.Severity, 2)
-		}
-	}
-}

alert/sender/webhook_queue.go

@@ -1,111 +0,0 @@
-package sender
-
-import (
-	"container/list"
-	"sync"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type SafeList struct {
-	sync.RWMutex
-	L *list.List
-}
-
-func NewSafeList() *SafeList {
-	return &SafeList{L: list.New()}
-}
-
-func (sl *SafeList) PushFront(v interface{}) *list.Element {
-	sl.Lock()
-	e := sl.L.PushFront(v)
-	sl.Unlock()
-	return e
-}
-
-func (sl *SafeList) PushFrontBatch(vs []interface{}) {
-	sl.Lock()
-	for _, item := range vs {
-		sl.L.PushFront(item)
-	}
-	sl.Unlock()
-}
-
-func (sl *SafeList) PopBack(max int) []*models.AlertCurEvent {
-	sl.Lock()
-
-	count := sl.L.Len()
-	if count == 0 {
-		sl.Unlock()
-		return []*models.AlertCurEvent{}
-	}
-
-	if count > max {
-		count = max
-	}
-
-	items := make([]*models.AlertCurEvent, 0, count)
-	for i := 0; i < count; i++ {
-		item := sl.L.Remove(sl.L.Back())
-		sample, ok := item.(*models.AlertCurEvent)
-		if ok {
-			items = append(items, sample)
-		}
-	}
-
-	sl.Unlock()
-	return items
-}
-
-func (sl *SafeList) RemoveAll() {
-	sl.Lock()
-	sl.L.Init()
-	sl.Unlock()
-}
-
-func (sl *SafeList) Len() int {
-	sl.RLock()
-	size := sl.L.Len()
-	sl.RUnlock()
-	return size
-}
-
-// SafeList with Limited Size
-type SafeListLimited struct {
-	maxSize int
-	SL      *SafeList
-}
-
-func NewSafeListLimited(maxSize int) *SafeListLimited {
-	return &SafeListLimited{SL: NewSafeList(), maxSize: maxSize}
-}
-
-func (sll *SafeListLimited) PopBack(max int) []*models.AlertCurEvent {
-	return sll.SL.PopBack(max)
-}
-
-func (sll *SafeListLimited) PushFront(v interface{}) bool {
-	if sll.SL.Len() >= sll.maxSize {
-		return false
-	}
-
-	sll.SL.PushFront(v)
-	return true
-}
-
-func (sll *SafeListLimited) PushFrontBatch(vs []interface{}) bool {
-	if sll.SL.Len() >= sll.maxSize {
-		return false
-	}
-
-	sll.SL.PushFrontBatch(vs)
-	return true
-}
-
-func (sll *SafeListLimited) RemoveAll() {
-	sll.SL.RemoveAll()
-}
-
-func (sll *SafeListLimited) Len() int {
-	return sll.SL.Len()
-}

alert/sender/wecom.go

@@ -1,74 +0,0 @@
-package sender
-
-import (
-	"html/template"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-)
-
-type wecomMarkdown struct {
-	Content string `json:"content"`
-}
-
-type wecom struct {
-	Msgtype  string        `json:"msgtype"`
-	Markdown wecomMarkdown `json:"markdown"`
-}
-
-var (
-	_ CallBacker = (*WecomSender)(nil)
-)
-
-type WecomSender struct {
-	tpl *template.Template
-}
-
-func (ws *WecomSender) CallBack(ctx CallBackContext) {
-	if len(ctx.Events) == 0 || len(ctx.CallBackURL) == 0 {
-		return
-	}
-
-	message := BuildTplMessage(models.Wecom, ws.tpl, ctx.Events)
-	body := wecom{
-		Msgtype: "markdown",
-		Markdown: wecomMarkdown{
-			Content: message,
-		},
-	}
-
-	doSendAndRecord(ctx.Ctx, ctx.CallBackURL, ctx.CallBackURL, body, "callback", ctx.Stats, ctx.Events)
-}
-
-func (ws *WecomSender) Send(ctx MessageContext) {
-	if len(ctx.Users) == 0 || len(ctx.Events) == 0 {
-		return
-	}
-	urls, tokens := ws.extract(ctx.Users)
-	message := BuildTplMessage(models.Wecom, ws.tpl, ctx.Events)
-	for i, url := range urls {
-		body := wecom{
-			Msgtype: "markdown",
-			Markdown: wecomMarkdown{
-				Content: message,
-			},
-		}
-		doSendAndRecord(ctx.Ctx, url, tokens[i], body, models.Wecom, ctx.Stats, ctx.Events)
-	}
-}
-
-func (ws *WecomSender) extract(users []*models.User) ([]string, []string) {
-	urls := make([]string, 0, len(users))
-	tokens := make([]string, 0, len(users))
-	for _, user := range users {
-		if token, has := user.ExtractToken(models.Wecom); has {
-			url := token
-			if !strings.HasPrefix(token, "https://") && !strings.HasPrefix(token, "http://") {
-				url = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=" + token
-			}
-			urls = append(urls, url)
-			tokens = append(tokens, token)
-		}
-	}
-	return urls, tokens
-}

center/cconf/conf.go

@@ -1,42 +0,0 @@
-package cconf
-
-import "time"
-
-type Center struct {
-	Plugins                []Plugin
-	MetricsYamlFile        string
-	OpsYamlFile            string
-	BuiltinIntegrationsDir string
-	I18NHeaderKey          string
-	MetricDesc             MetricDescType
-	AnonymousAccess        AnonymousAccess
-	UseFileAssets          bool
-	FlashDuty              FlashDuty
-	EventHistoryGroupView  bool
-	CleanNotifyRecordDay   int
-	MigrateBusiGroupLabel  bool
-}
-
-type Plugin struct {
-	Id       int64  `json:"id"`
-	Category string `json:"category"`
-	Type     string `json:"plugin_type"`
-	TypeName string `json:"plugin_type_name"`
-}
-
-type FlashDuty struct {
-	Api     string
-	Headers map[string]string
-	Timeout time.Duration
-}
-
-type AnonymousAccess struct {
-	PromQuerier bool
-	AlertDetail bool
-}
-
-func (c *Center) PreCheck() {
-	if len(c.Plugins) == 0 {
-		c.Plugins = Plugins
-	}
-}

center/cconf/event_example.go

@@ -1,60 +0,0 @@
-package cconf
-
-const EVENT_EXAMPLE = `
-{
-    "id": 1000000,
-    "cate": "prometheus",
-    "datasource_id": 1,
-    "group_id": 1,
-    "group_name": "Default Busi Group",
-    "hash": "2cb966f9ba1cdc7af94c3796e855955a",
-    "rule_id": 23,
-    "rule_name": "测试告警",
-    "rule_note": "测试告警",
-    "rule_prod": "metric",
-    "rule_config": {
-        "queries": [
-            {
-                "key": "all_hosts",
-                "op": "==",
-                "values": []
-            }
-        ],
-        "triggers": [
-            {
-                "duration": 3,
-                "percent": 10,
-                "severity": 3,
-                "type": "pct_target_miss"
-            }
-        ]
-    },
-    "prom_for_duration": 60,
-    "prom_eval_interval": 30,
-    "callbacks": ["https://n9e.github.io"],
-    "notify_recovered": 1,
-    "notify_channels": ["dingtalk"],
-    "notify_groups": [],
-    "notify_groups_obj": null,
-    "target_ident": "host01",
-    "target_note": "机器备注",
-    "trigger_time": 1677229517,
-    "trigger_value": "2273533952",
-    "tags": [
-        "__name__=disk_free",
-        "dc=qcloud-dev",
-        "device=vda1",
-        "fstype=ext4",
-        "ident=tt-fc-dev00.nj"
-    ],
-    "is_recovered": false,
-    "notify_users_obj": null,
-    "last_eval_time": 1677229517,
-    "last_sent_time": 1677229517,
-    "notify_cur_number": 1,
-    "first_trigger_time": 1677229517,
-    "annotations": {
-        "summary": "测试告警"
-    }
-}
-`

center/cconf/metric.go

@@ -1,52 +0,0 @@
-package cconf
-
-import (
-	"path"
-
-	"github.com/toolkits/pkg/file"
-)
-
-// metricDesc , As load map happens before read map, there is no necessary to use concurrent map for metric desc store
-type MetricDescType struct {
-	CommonDesc map[string]string `yaml:",inline" json:"common"`
-	Zh         map[string]string `yaml:"zh" json:"zh"`
-	En         map[string]string `yaml:"en" json:"en"`
-}
-
-var MetricDesc MetricDescType
-
-// GetMetricDesc , if metric is not registered, empty string will be returned
-func GetMetricDesc(lang, metric string) string {
-	var m map[string]string
-
-	switch lang {
-	case "en":
-		m = MetricDesc.En
-	default:
-		m = MetricDesc.Zh
-	}
-
-	if m != nil {
-		if desc, ok := m[metric]; ok {
-			return desc
-		}
-	}
-
-	if MetricDesc.CommonDesc != nil {
-		if desc, ok := MetricDesc.CommonDesc[metric]; ok {
-			return desc
-		}
-	}
-
-	return ""
-}
-func LoadMetricsYaml(configDir, metricsYamlFile string) error {
-	fp := metricsYamlFile
-	if fp == "" {
-		fp = path.Join(configDir, "metrics.yaml")
-	}
-	if !file.IsExist(fp) {
-		return nil
-	}
-	return file.ReadYaml(fp, &MetricDesc)
-}

center/cconf/ops.go

@@ -1,305 +0,0 @@
-package cconf
-
-import (
-	"fmt"
-	"path"
-
-	"github.com/toolkits/pkg/file"
-	"gopkg.in/yaml.v2"
-)
-
-var Operations = Operation{}
-
-type Operation struct {
-	Ops []Ops `yaml:"ops"`
-}
-
-type Ops struct {
-	Name  string     `yaml:"name" json:"name"`
-	Cname string     `yaml:"cname" json:"cname"`
-	Ops   []SingleOp `yaml:"ops" json:"ops"`
-}
-
-// SingleOp Name 为 op 名称；Cname 为展示名称，默认英文
-type SingleOp struct {
-	Name  string `yaml:"name" json:"name"`
-	Cname string `yaml:"cname" json:"cname"`
-}
-
-func TransformNames(name []string, nameToName map[string]string) []string {
-	var ret []string
-	for _, n := range name {
-		if v, has := nameToName[n]; has {
-			ret = append(ret, v)
-		}
-	}
-	return ret
-}
-
-func LoadOpsYaml(configDir string, opsYamlFile string) error {
-	fp := opsYamlFile
-	if fp == "" {
-		fp = path.Join(configDir, "ops.yaml")
-	}
-	if !file.IsExist(fp) {
-		return nil
-	}
-
-	hash, _ := file.MD5(fp)
-	if hash == "2f91a9ed265cf2024e266dc1d538ee77" {
-		// ops.yaml 是老的默认文件，删除
-		file.Remove(fp)
-		return nil
-	}
-
-	return file.ReadYaml(fp, &Operations)
-}
-
-func GetAllOps(ops []Ops) []SingleOp {
-	var ret []SingleOp
-	for _, op := range ops {
-		ret = append(ret, op.Ops...)
-	}
-	return ret
-}
-
-func MergeOperationConf() error {
-	var opsBuiltIn Operation
-	err := yaml.Unmarshal([]byte(builtInOps), &opsBuiltIn)
-	if err != nil {
-		return fmt.Errorf("cannot parse builtInOps: %s", err.Error())
-	}
-	configOpsMap := make(map[string]struct{})
-	for _, op := range Operations.Ops {
-		configOpsMap[op.Name] = struct{}{}
-	}
-	//If the opBu.Name is not a constant in the target (Operations.Ops), add Ops from the built-in options
-	for _, opBu := range opsBuiltIn.Ops {
-		if _, has := configOpsMap[opBu.Name]; !has {
-			Operations.Ops = append(Operations.Ops, opBu)
-		}
-	}
-	return nil
-}
-
-const (
-	builtInOps = `
-ops:
-- name: Infrastructure
-  cname: Infrastructure
-  ops:
-    - name: /targets
-      cname: Host - View
-    - name: /targets/put
-      cname: Host - Modify
-    - name: /targets/del
-      cname: Host - Delete
-    - name: /targets/bind
-      cname: Host - Bind Uncategorized
-
-- name: Explorer
-  cname: Explorer
-  ops:
-    - name: /metric/explorer
-      cname: Metrics Explorer
-    - name: /object/explorer
-      cname: Quick View
-    - name: /metrics-built-in
-      cname: Built-in Metric - View
-    - name: /builtin-metrics/add
-      cname: Built-in Metric - Add
-    - name: /builtin-metrics/put
-      cname: Built-in Metric - Modify
-    - name: /builtin-metrics/del
-      cname: Built-in Metric - Delete
-    - name: /recording-rules
-      cname: Recording Rule - View
-    - name: /recording-rules/add
-      cname: Recording Rule - Add
-    - name: /recording-rules/put
-      cname: Recording Rule - Modify
-    - name: /recording-rules/del
-      cname: Recording Rule - Delete
-    - name: /log/explorer
-      cname: Logs Explorer
-    - name: /log/index-patterns # 前端有个管理索引模式的页面，所以需要一个权限点来控制，后面应该改成侧拉板
-      cname: Index Pattern - View
-    - name: /log/index-patterns/add
-      cname: Index Pattern - Add
-    - name: /log/index-patterns/put
-      cname: Index Pattern - Modify
-    - name: /log/index-patterns/del
-      cname: Index Pattern - Delete
-    - name: /dashboards
-      cname: Dashboard - View
-    - name: /dashboards/add
-      cname: Dashboard - Add
-    - name: /dashboards/put
-      cname: Dashboard - Modify
-    - name: /dashboards/del
-      cname: Dashboard - Delete
-    - name: /public-dashboards
-      cname: Dashboard - View Public
-
-- name: alerting
-  cname: Alerting
-  ops:
-    - name: /alert-rules
-      cname: Alerting Rule - View
-    - name: /alert-rules/add
-      cname: Alerting Rule - Add
-    - name: /alert-rules/put
-      cname: Alerting Rule - Modify
-    - name: /alert-rules/del
-      cname: Alerting Rule - Delete
-    - name: /alert-mutes
-      cname: Mutting Rule - View
-    - name: /alert-mutes/add
-      cname: Mutting Rule - Add
-    - name: /alert-mutes/put
-      cname: Mutting Rule - Modify
-    - name: /alert-mutes/del
-      cname: Mutting Rule - Delete
-    - name: /alert-subscribes
-      cname: Subscribing Rule - View
-    - name: /alert-subscribes/add
-      cname: Subscribing Rule - Add
-    - name: /alert-subscribes/put
-      cname: Subscribing Rule - Modify
-    - name: /alert-subscribes/del
-      cname: Subscribing Rule - Delete
-    - name: /job-tpls
-      cname: Self-healing-Script - View
-    - name: /job-tpls/add
-      cname: Self-healing-Script - Add
-    - name: /job-tpls/put
-      cname: Self-healing-Script - Modify
-    - name: /job-tpls/del
-      cname: Self-healing-Script - Delete
-    - name: /job-tasks
-      cname: Self-healing-Job - View
-    - name: /job-tasks/add
-      cname: Self-healing-Job - Add
-    - name: /job-tasks/put
-      cname: Self-healing-Job - Modify
-    - name: /alert-cur-events
-      cname: Active Event - View
-    - name: /alert-cur-events/del
-      cname: Active Event - Delete
-    - name: /alert-his-events
-      cname: Historical Event - View
-
-- name: Notification
-  cname: Notification
-  ops:
-    - name: /notification-rules
-      cname: Notification Rule - View
-    - name: /notification-rules/add
-      cname: Notification Rule - Add
-    - name: /notification-rules/put
-      cname: Notification Rule - Modify
-    - name: /notification-rules/del
-      cname: Notification Rule - Delete
-    - name: /notification-channels
-      cname: Media Type - View
-    - name: /notification-channels/add
-      cname: Media Type - Add
-    - name: /notification-channels/put
-      cname: Media Type - Modify
-    - name: /notification-channels/del
-      cname: Media Type - Delete
-    - name: /notification-templates
-      cname: Message Template - View
-    - name: /notification-templates/add
-      cname: Message Template - Add
-    - name: /notification-templates/put
-      cname: Message Template - Modify
-    - name: /notification-templates/del
-      cname: Message Template - Delete
-    - name: /event-pipelines
-      cname: Event Pipeline - View
-    - name: /event-pipelines/add
-      cname: Event Pipeline - Add
-    - name: /event-pipelines/put
-      cname: Event Pipeline - Modify
-    - name: /event-pipelines/del
-      cname: Event Pipeline - Delete
-    - name: /help/notification-settings # 用于控制老版本的通知设置菜单是否展示
-      cname: Notification Settings - View
-    - name: /help/notification-tpls # 用于控制老版本的通知模板菜单是否展示
-      cname: Notification Templates - View
-
-- name: Integrations
-  cname: Integrations
-  ops:
-    - name: /datasources # 用于控制能否看到数据源列表页面的菜单。只有 Admin 才能修改、删除数据源
-      cname: Data Source - View
-    - name: /components
-      cname: Component - View
-    - name: /components/add
-      cname: Component - Add
-    - name: /components/put
-      cname: Component - Modify
-    - name: /components/del
-      cname: Component - Delete
-    - name: /embedded-products
-      cname: Embedded Product - View
-    - name: /embedded-product/add
-      cname: Embedded Product - Add
-    - name: /embedded-product/put
-      cname: Embedded Product - Modify
-    - name: /embedded-product/delete
-      cname: Embedded Product - Delete
-
-- name: Organization
-  cname: Organization
-  ops:
-    - name: /users
-      cname: User - View
-    - name: /users/add
-      cname: User - Add
-    - name: /users/put
-      cname: User - Modify
-    - name: /users/del
-      cname: User - Delete
-    - name: /user-groups
-      cname: Team - View
-    - name: /user-groups/add
-      cname: Team - Add
-    - name: /user-groups/put
-      cname: Team - Modify
-    - name: /user-groups/del
-      cname: Team - Delete
-    - name: /busi-groups
-      cname: Business Group - View
-    - name: /busi-groups/add
-      cname: Business Group - Add
-    - name: /busi-groups/put
-      cname: Business Group - Modify
-    - name: /busi-groups/del
-      cname: Business Group - Delete
-    - name: /roles
-      cname: Role - View
-    - name: /roles/add
-      cname: Role - Add
-    - name: /roles/put
-      cname: Role - Modify
-    - name: /roles/del
-      cname: Role - Delete
-
-- name: System Settings
-  cname: System Settings
-  ops:
-    - name: /system/site-settings # 仅用于控制能否展示菜单，只有 Admin 才能修改、删除
-      cname: View Site Settings
-    - name: /system/variable-settings
-      cname: View Variable Settings
-    - name: /system/sso-settings
-      cname: View SSO Settings
-    - name: /system/alerting-engines
-      cname: View Alerting Engines
-    - name: /system/version
-      cname: View Product Version
-
-`
-)

center/cconf/plugin.go

@@ -1,46 +0,0 @@
-package cconf
-
-var Plugins = []Plugin{
-	{
-		Id:       1,
-		Category: "timeseries",
-		Type:     "prometheus",
-		TypeName: "Prometheus Like",
-	},
-	{
-		Id:       2,
-		Category: "logging",
-		Type:     "elasticsearch",
-		TypeName: "Elasticsearch",
-	},
-	{
-		Id:       3,
-		Category: "loki",
-		Type:     "loki",
-		TypeName: "Loki",
-	},
-	{
-		Id:       4,
-		Category: "timeseries",
-		Type:     "tdengine",
-		TypeName: "TDengine",
-	},
-	{
-		Id:       5,
-		Category: "logging",
-		Type:     "ck",
-		TypeName: "ClickHouse",
-	},
-	{
-		Id:       6,
-		Category: "timeseries",
-		Type:     "mysql",
-		TypeName: "MySQL",
-	},
-	{
-		Id:       7,
-		Category: "timeseries",
-		Type:     "pgsql",
-		TypeName: "PostgreSQL",
-	},
-}

center/cconf/rsa/rsa_conf.go

@@ -1,105 +0,0 @@
-package rsa
-
-import (
-	"os"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/httpx"
-	"github.com/ccfos/nightingale/v6/pkg/secu"
-
-	"github.com/pkg/errors"
-	"github.com/toolkits/pkg/file"
-	"github.com/toolkits/pkg/logger"
-)
-
-func InitRSAConfig(ctx *ctx.Context, rsaConfig *httpx.RSAConfig) error {
-
-	// 1.Load RSA keys from Database
-	rsaPassWord, err := models.ConfigsGet(ctx, models.RSA_PASSWORD)
-	if err != nil {
-		return errors.WithMessagef(err, "cannot query config(%s)", models.RSA_PASSWORD)
-	}
-	privateKeyVal, err := models.ConfigsGet(ctx, models.RSA_PRIVATE_KEY)
-	if err != nil {
-		return errors.WithMessagef(err, "cannot query config(%s)", models.RSA_PRIVATE_KEY)
-	}
-	publicKeyVal, err := models.ConfigsGet(ctx, models.RSA_PUBLIC_KEY)
-	if err != nil {
-		return errors.WithMessagef(err, "cannot query config(%s)", models.RSA_PUBLIC_KEY)
-	}
-	if rsaPassWord != "" && privateKeyVal != "" && publicKeyVal != "" {
-		rsaConfig.RSAPassWord = rsaPassWord
-		rsaConfig.RSAPrivateKey = []byte(privateKeyVal)
-		rsaConfig.RSAPublicKey = []byte(publicKeyVal)
-		return nil
-	}
-
-	// 2.Read RSA configuration from file if exists
-	if file.IsExist(rsaConfig.RSAPrivateKeyPath) && file.IsExist(rsaConfig.RSAPublicKeyPath) {
-		//password already read from config
-		rsaConfig.RSAPrivateKey, rsaConfig.RSAPublicKey, err = readConfigFile(rsaConfig)
-		if err != nil {
-			return errors.WithMessage(err, "failed to read rsa config from file")
-		}
-		return nil
-	}
-	// 3.Generate RSA keys if not exist
-	rsaConfig.RSAPassWord, rsaConfig.RSAPrivateKey, rsaConfig.RSAPublicKey, err = initRSAKeyPairs(ctx, rsaConfig.RSAPassWord)
-	if err != nil {
-		return errors.WithMessage(err, "failed to generate rsa key pair")
-	}
-	return nil
-}
-
-func initRSAKeyPairs(ctx *ctx.Context, rsaPassWord string) (password string, privateByte, publicByte []byte, err error) {
-
-	// Generate RSA keys
-
-	// Generate RSA password
-	if rsaPassWord != "" {
-		logger.Debug("Using existing RSA password")
-		password = rsaPassWord
-		err = models.ConfigsSet(ctx, models.RSA_PASSWORD, password)
-		if err != nil {
-			err = errors.WithMessagef(err, "failed to set config(%s)", models.RSA_PASSWORD)
-			return
-		}
-	} else {
-		password, err = models.InitRSAPassWord(ctx)
-		if err != nil {
-			err = errors.WithMessage(err, "failed to generate rsa password")
-			return
-		}
-	}
-	privateByte, publicByte, err = secu.GenerateRsaKeyPair(password)
-	if err != nil {
-		err = errors.WithMessage(err, "failed to generate rsa key pair")
-		return
-	}
-	// Save generated RSA keys
-	err = models.ConfigsSet(ctx, models.RSA_PRIVATE_KEY, string(privateByte))
-	if err != nil {
-		err = errors.WithMessagef(err, "failed to set config(%s)", models.RSA_PRIVATE_KEY)
-		return
-	}
-	err = models.ConfigsSet(ctx, models.RSA_PUBLIC_KEY, string(publicByte))
-	if err != nil {
-		err = errors.WithMessagef(err, "failed to set config(%s)", models.RSA_PUBLIC_KEY)
-		return
-	}
-	return
-}
-
-func readConfigFile(rsaConfig *httpx.RSAConfig) (privateBuf, publicBuf []byte, err error) {
-	publicBuf, err = os.ReadFile(rsaConfig.RSAPublicKeyPath)
-	if err != nil {
-		err = errors.WithMessagef(err, "could not read RSAPublicKeyPath %q", rsaConfig.RSAPublicKeyPath)
-		return
-	}
-	privateBuf, err = os.ReadFile(rsaConfig.RSAPrivateKeyPath)
-	if err != nil {
-		err = errors.WithMessagef(err, "could not read RSAPrivateKeyPath %q", rsaConfig.RSAPrivateKeyPath)
-	}
-	return
-}

center/cconf/sql_tpl.go

@@ -1,15 +0,0 @@
-package cconf
-
-var TDengineSQLTpl = map[string]string{
-	"load5":                "SELECT _wstart as ts, last(load5) FROM $database.system WHERE host = '$server' and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"process_total":        "SELECT _wstart as ts, last(total) FROM $database.processes WHERE host = '$server' and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"thread_total":         "SELECT _wstart as ts, last(total) FROM $database.threads WHERE host = '$server' and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"cpu_idle":             "SELECT _wstart as ts, last(usage_idle) * -1 + 100 FROM $database.cpu WHERE (host = '$server' and cpu = 'cpu-total') and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"mem_used_percent":     "SELECT _wstart as ts, last(used_percent) FROM $database.mem WHERE (host = '$server') and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"disk_used_percent":    "SELECT _wstart as ts, last(used_percent) FROM $database.disk WHERE (host = '$server' and path = '/') and _ts >= $from and _ts <= $to interval($interval) fill(null)",
-	"cpu_context_switches": "select ts, derivative(context_switches, 1s, 0) as context FROM (SELECT _wstart as ts, avg(context_switches) as context_switches FROM $database.kernel WHERE host = '$server' and _ts >= $from and _ts <= $to interval($interval) )",
-	"tcp":                  "SELECT _wstart as ts, avg(tcp_close) as CLOSED, avg(tcp_close_wait) as CLOSE_WAIT, avg(tcp_closing) as CLOSING, avg(tcp_established) as ESTABLISHED, avg(tcp_fin_wait1) as FIN_WAIT1, avg(tcp_fin_wait2) as FIN_WAIT2, avg(tcp_last_ack) as LAST_ACK, avg(tcp_syn_recv) as SYN_RECV, avg(tcp_syn_sent) as SYN_SENT, avg(tcp_time_wait) as TIME_WAIT FROM $database.netstat WHERE host = '$server' and _ts >= $from and _ts <= $to interval($interval)",
-	"net_bytes_recv":       "SELECT _wstart as ts, derivative(bytes_recv,1s, 1) as bytes_in FROM $database.net WHERE host = '$server' and interface = '$netif' and _ts >= $from and _ts <= $to group by tbname",
-	"net_bytes_sent":       "SELECT _wstart as ts, derivative(bytes_sent,1s, 1) as bytes_out FROM $database.net WHERE host = '$server' and interface = '$netif' and _ts >= $from and _ts <= $to group by tbname",
-	"disk_total":           "SELECT _wstart as ts, avg(total) AS total, avg(used) as used FROM $database.disk WHERE   path = '$mountpoint' and _ts >= $from and _ts <= $to  interval($interval) group by host",
-}

center/center.go

@@ -1,161 +0,0 @@
-package center
-
-import (
-	"context"
-	"fmt"
-
-	"github.com/ccfos/nightingale/v6/dscache"
-
-	"github.com/ccfos/nightingale/v6/alert"
-	"github.com/ccfos/nightingale/v6/alert/astats"
-	"github.com/ccfos/nightingale/v6/alert/dispatch"
-	"github.com/ccfos/nightingale/v6/alert/process"
-	alertrt "github.com/ccfos/nightingale/v6/alert/router"
-	"github.com/ccfos/nightingale/v6/center/cconf"
-	"github.com/ccfos/nightingale/v6/center/cconf/rsa"
-	"github.com/ccfos/nightingale/v6/center/integration"
-	"github.com/ccfos/nightingale/v6/center/metas"
-	centerrt "github.com/ccfos/nightingale/v6/center/router"
-	"github.com/ccfos/nightingale/v6/center/sso"
-	"github.com/ccfos/nightingale/v6/conf"
-	"github.com/ccfos/nightingale/v6/cron"
-	"github.com/ccfos/nightingale/v6/dumper"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/models/migrate"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/flashduty"
-	"github.com/ccfos/nightingale/v6/pkg/httpx"
-	"github.com/ccfos/nightingale/v6/pkg/i18nx"
-	"github.com/ccfos/nightingale/v6/pkg/logx"
-	"github.com/ccfos/nightingale/v6/pkg/macros"
-	"github.com/ccfos/nightingale/v6/pkg/version"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/ccfos/nightingale/v6/pushgw/idents"
-	pushgwrt "github.com/ccfos/nightingale/v6/pushgw/router"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-	"github.com/ccfos/nightingale/v6/storage"
-	"github.com/flashcatcloud/ibex/src/cmd/ibex"
-)
-
-func Initialize(configDir string, cryptoKey string) (func(), error) {
-	config, err := conf.InitConfig(configDir, cryptoKey)
-	if err != nil {
-		return nil, fmt.Errorf("failed to init config: %v", err)
-	}
-
-	cconf.LoadMetricsYaml(configDir, config.Center.MetricsYamlFile)
-	cconf.LoadOpsYaml(configDir, config.Center.OpsYamlFile)
-
-	cconf.MergeOperationConf()
-
-	if config.Alert.Heartbeat.EngineName == "" {
-		config.Alert.Heartbeat.EngineName = "default"
-	}
-
-	logxClean, err := logx.Init(config.Log)
-	if err != nil {
-		return nil, err
-	}
-
-	i18nx.Init(configDir)
-	flashduty.Init(config.Center.FlashDuty)
-
-	db, err := storage.New(config.DB)
-	if err != nil {
-		return nil, err
-	}
-	ctx := ctx.NewContext(context.Background(), db, true)
-	migrate.Migrate(db)
-	isRootInit := models.InitRoot(ctx)
-
-	config.HTTP.JWTAuth.SigningKey = models.InitJWTSigningKey(ctx)
-
-	err = rsa.InitRSAConfig(ctx, &config.HTTP.RSA)
-	if err != nil {
-		return nil, err
-	}
-
-	go integration.Init(ctx, config.Center.BuiltinIntegrationsDir)
-	var redis storage.Redis
-	redis, err = storage.NewRedis(config.Redis)
-	if err != nil {
-		return nil, err
-	}
-
-	metas := metas.New(redis)
-	idents := idents.New(ctx, redis, config.Pushgw)
-
-	syncStats := memsto.NewSyncStats()
-	alertStats := astats.NewSyncStats()
-
-	if config.Center.MigrateBusiGroupLabel || models.CanMigrateBg(ctx) {
-		models.MigrateBg(ctx, config.Pushgw.BusiGroupLabelKey)
-	}
-	if models.CanMigrateEP(ctx) {
-		models.MigrateEP(ctx)
-	}
-
-	configCache := memsto.NewConfigCache(ctx, syncStats, config.HTTP.RSA.RSAPrivateKey, config.HTTP.RSA.RSAPassWord)
-	busiGroupCache := memsto.NewBusiGroupCache(ctx, syncStats)
-	targetCache := memsto.NewTargetCache(ctx, syncStats, redis)
-	dsCache := memsto.NewDatasourceCache(ctx, syncStats)
-	alertMuteCache := memsto.NewAlertMuteCache(ctx, syncStats)
-	alertRuleCache := memsto.NewAlertRuleCache(ctx, syncStats)
-	notifyConfigCache := memsto.NewNotifyConfigCache(ctx, configCache)
-	userCache := memsto.NewUserCache(ctx, syncStats)
-	userGroupCache := memsto.NewUserGroupCache(ctx, syncStats)
-	taskTplCache := memsto.NewTaskTplCache(ctx)
-	configCvalCache := memsto.NewCvalCache(ctx, syncStats)
-	notifyRuleCache := memsto.NewNotifyRuleCache(ctx, syncStats)
-	notifyChannelCache := memsto.NewNotifyChannelCache(ctx, syncStats)
-	messageTemplateCache := memsto.NewMessageTemplateCache(ctx, syncStats)
-	userTokenCache := memsto.NewUserTokenCache(ctx, syncStats)
-
-	sso := sso.Init(config.Center, ctx, configCache)
-	promClients := prom.NewPromClient(ctx)
-
-	dispatch.InitRegisterQueryFunc(promClients)
-
-	externalProcessors := process.NewExternalProcessors()
-
-	macros.RegisterMacro(macros.MacroInVain)
-	dscache.Init(ctx, false)
-	alert.Start(config.Alert, config.Pushgw, syncStats, alertStats, externalProcessors, targetCache, busiGroupCache, alertMuteCache, alertRuleCache, notifyConfigCache, taskTplCache, dsCache, ctx, promClients, userCache, userGroupCache, notifyRuleCache, notifyChannelCache, messageTemplateCache)
-
-	writers := writer.NewWriters(config.Pushgw)
-
-	go version.GetGithubVersion()
-
-	go cron.CleanNotifyRecord(ctx, config.Center.CleanNotifyRecordDay)
-
-	alertrtRouter := alertrt.New(config.HTTP, config.Alert, alertMuteCache, targetCache, busiGroupCache, alertStats, ctx, externalProcessors)
-	centerRouter := centerrt.New(config.HTTP, config.Center, config.Alert, config.Ibex,
-		cconf.Operations, dsCache, notifyConfigCache, promClients,
-		redis, sso, ctx, metas, idents, targetCache, userCache, userGroupCache, userTokenCache)
-	pushgwRouter := pushgwrt.New(config.HTTP, config.Pushgw, config.Alert, targetCache, busiGroupCache, idents, metas, writers, ctx)
-
-	r := httpx.GinEngine(config.Global.RunMode, config.HTTP, configCvalCache.PrintBodyPaths, configCvalCache.PrintAccessLog)
-
-	centerRouter.Config(r)
-	alertrtRouter.Config(r)
-	pushgwRouter.Config(r)
-	dumper.ConfigRouter(r)
-
-	if config.Ibex.Enable {
-		migrate.MigrateIbexTables(db)
-		ibex.ServerStart(true, db, redis, config.HTTP.APIForService.BasicAuth, config.Alert.Heartbeat, &config.CenterApi, r, centerRouter, config.Ibex, config.HTTP.Port)
-	}
-
-	httpClean := httpx.Init(config.HTTP, r)
-
-	fmt.Printf("please view n9e at  http://%v:%v\n", config.Alert.Heartbeat.IP, config.HTTP.Port)
-	if isRootInit {
-		fmt.Println("username/password: root/root.2020")
-	}
-
-	return func() {
-		logxClean()
-		httpClean()
-	}, nil
-}

center/cstats/stats.go

@@ -1,62 +0,0 @@
-package cstats
-
-import (
-	"time"
-
-	"github.com/prometheus/client_golang/prometheus"
-)
-
-const (
-	namespace = "n9e"
-	subsystem = "center"
-)
-
-var (
-	uptime = prometheus.NewCounter(
-		prometheus.CounterOpts{
-			Namespace: namespace,
-			Subsystem: subsystem,
-			Name:      "uptime",
-			Help:      "HTTP service uptime.",
-		},
-	)
-
-	RequestDuration = prometheus.NewHistogramVec(
-		prometheus.HistogramOpts{
-			Namespace: namespace,
-			Subsystem: subsystem,
-			Buckets:   prometheus.DefBuckets,
-			Name:      "http_request_duration_seconds",
-			Help:      "HTTP request latencies in seconds.",
-		}, []string{"code", "path", "method"},
-	)
-
-	RedisOperationLatency = prometheus.NewHistogramVec(
-		prometheus.HistogramOpts{
-			Namespace: namespace,
-			Subsystem: subsystem,
-			Name:      "redis_operation_latency_seconds",
-			Help:      "Histogram of latencies for Redis operations",
-			Buckets:   []float64{.005, .01, .025, .05, .1, .25, .5, 1, 2.5, 5},
-		},
-		[]string{"operation", "status"},
-	)
-)
-
-func init() {
-	// Register the summary and the histogram with Prometheus's default registry.
-	prometheus.MustRegister(
-		uptime,
-		RequestDuration,
-		RedisOperationLatency,
-	)
-
-	go recordUptime()
-}
-
-// recordUptime increases service uptime per second.
-func recordUptime() {
-	for range time.Tick(time.Second) {
-		uptime.Inc()
-	}
-}

center/integration/init.go

@@ -1,389 +0,0 @@
-package integration
-
-import (
-	"encoding/json"
-	"path"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/toolkits/pkg/file"
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/runner"
-)
-
-const SYSTEM = "system"
-
-func Init(ctx *ctx.Context, builtinIntegrationsDir string) {
-	err := models.InitBuiltinPayloads(ctx)
-	if err != nil {
-		logger.Warning("init old builtinPayloads fail ", err)
-		return
-	}
-
-	if res, err := models.ConfigsSelectByCkey(ctx, "disable_integration_init"); err != nil {
-		logger.Error("fail to get value 'disable_integration_init' from configs", err)
-		return
-	} else if len(res) != 0 {
-		logger.Info("disable_integration_init is set, skip integration init")
-		return
-	}
-
-	fp := builtinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	// var fileList []string
-	dirList, err := file.DirsUnder(fp)
-	if err != nil {
-		logger.Warning("read builtin component dir fail ", err)
-		return
-	}
-
-	for _, dir := range dirList {
-		// components icon
-		componentDir := fp + "/" + dir
-		component := models.BuiltinComponent{
-			Ident: dir,
-		}
-
-		// get logo name
-		// /api/n9e/integrations/icon/AliYun/aliyun.png
-		files, err := file.FilesUnder(componentDir + "/icon")
-		if err == nil && len(files) > 0 {
-			component.Logo = "/api/n9e/integrations/icon/" + component.Ident + "/" + files[0]
-		} else if err != nil {
-			logger.Warningf("read builtin component icon dir fail %s %v", component.Ident, err)
-		}
-
-		// get description
-		files, err = file.FilesUnder(componentDir + "/markdown")
-		if err == nil && len(files) > 0 {
-			var readmeFile string
-			for _, file := range files {
-				if strings.HasSuffix(strings.ToLower(file), "md") {
-					readmeFile = componentDir + "/markdown/" + file
-					break
-				}
-			}
-			if readmeFile != "" {
-				component.Readme, _ = file.ReadString(readmeFile)
-			}
-		} else if err != nil {
-			logger.Warningf("read builtin component markdown dir fail %s %v", component.Ident, err)
-		}
-
-		exists, _ := models.BuiltinComponentExists(ctx, &component)
-		if !exists {
-			err = component.Add(ctx, SYSTEM)
-			if err != nil {
-				logger.Warning("add builtin component fail ", component, err)
-				continue
-			}
-		} else {
-			old, err := models.BuiltinComponentGet(ctx, "ident = ?", component.Ident)
-			if err != nil {
-				logger.Warning("get builtin component fail ", component, err)
-				continue
-			}
-
-			if old == nil {
-				logger.Warning("get builtin component nil ", component)
-				continue
-			}
-
-			if old.UpdatedBy == SYSTEM {
-				now := time.Now().Unix()
-				old.CreatedAt = now
-				old.UpdatedAt = now
-				old.Readme = component.Readme
-				old.UpdatedBy = SYSTEM
-
-				err = models.DB(ctx).Model(old).Select("*").Updates(old).Error
-				if err != nil {
-					logger.Warning("update builtin component fail ", old, err)
-				}
-			}
-			component.ID = old.ID
-		}
-
-		// delete uuid is emtpy
-		err = models.DB(ctx).Exec("delete from builtin_payloads where uuid = 0 and type != 'collect' and (updated_by = 'system' or updated_by = '')").Error
-		if err != nil {
-			logger.Warning("delete builtin payloads fail ", err)
-		}
-
-		// delete builtin metrics uuid is emtpy
-		err = models.DB(ctx).Exec("delete from builtin_metrics where uuid = 0 and (updated_by = 'system' or updated_by = '')").Error
-		if err != nil {
-			logger.Warning("delete builtin metrics fail ", err)
-		}
-
-		// 删除 uuid%1000 不为 0 uuid > 1000000000000000000 且 type 为 dashboard 的记录
-		err = models.DB(ctx).Exec("delete from builtin_payloads where uuid%1000 != 0 and uuid > 1000000000000000000 and type = 'dashboard' and updated_by = 'system'").Error
-		if err != nil {
-			logger.Warning("delete builtin payloads fail ", err)
-		}
-
-		// alerts
-		files, err = file.FilesUnder(componentDir + "/alerts")
-		if err == nil && len(files) > 0 {
-			for _, f := range files {
-				fp := componentDir + "/alerts/" + f
-				bs, err := file.ReadBytes(fp)
-				if err != nil {
-					logger.Warning("read builtin component alerts file fail ", f, err)
-					continue
-				}
-
-				alerts := []models.AlertRule{}
-				err = json.Unmarshal(bs, &alerts)
-				if err != nil {
-					logger.Warning("parse builtin component alerts file fail ", f, err)
-					continue
-				}
-
-				newAlerts := []models.AlertRule{}
-				writeAlertFileFlag := false
-				for _, alert := range alerts {
-					if alert.UUID == 0 {
-						writeAlertFileFlag = true
-						alert.UUID = time.Now().UnixNano()
-					}
-
-					newAlerts = append(newAlerts, alert)
-					content, err := json.Marshal(alert)
-					if err != nil {
-						logger.Warning("marshal builtin alert fail ", alert, err)
-						continue
-					}
-
-					cate := strings.Replace(f, ".json", "", -1)
-					builtinAlert := models.BuiltinPayload{
-						ComponentID: component.ID,
-						Type:        "alert",
-						Cate:        cate,
-						Name:        alert.Name,
-						Tags:        alert.AppendTags,
-						Content:     string(content),
-						UUID:        alert.UUID,
-					}
-
-					old, err := models.BuiltinPayloadGet(ctx, "uuid = ?", alert.UUID)
-					if err != nil {
-						logger.Warning("get builtin alert fail ", builtinAlert, err)
-						continue
-					}
-
-					if old == nil {
-						err := builtinAlert.Add(ctx, SYSTEM)
-						if err != nil {
-							logger.Warning("add builtin alert fail ", builtinAlert, err)
-						}
-						continue
-					}
-
-					if old.UpdatedBy == SYSTEM {
-						old.ComponentID = component.ID
-						old.Content = string(content)
-						old.Name = alert.Name
-						old.Tags = alert.AppendTags
-						err = models.DB(ctx).Model(old).Select("*").Updates(old).Error
-						if err != nil {
-							logger.Warningf("update builtin alert:%+v fail %v", builtinAlert, err)
-						}
-					}
-				}
-
-				if writeAlertFileFlag {
-					bs, err = json.MarshalIndent(newAlerts, "", "    ")
-					if err != nil {
-						logger.Warning("marshal builtin alerts fail ", newAlerts, err)
-						continue
-					}
-
-					_, err = file.WriteBytes(fp, bs)
-					if err != nil {
-						logger.Warning("write builtin alerts file fail ", f, err)
-					}
-				}
-
-			}
-		}
-
-		// dashboards
-		files, err = file.FilesUnder(componentDir + "/dashboards")
-		if err == nil && len(files) > 0 {
-			for _, f := range files {
-				fp := componentDir + "/dashboards/" + f
-				bs, err := file.ReadBytes(fp)
-				if err != nil {
-					logger.Warning("read builtin component dashboards file fail ", f, err)
-					continue
-				}
-
-				dashboard := BuiltinBoard{}
-				err = json.Unmarshal(bs, &dashboard)
-				if err != nil {
-					logger.Warning("parse builtin component dashboards file fail ", f, err)
-					continue
-				}
-
-				if dashboard.UUID == 0 {
-					time.Sleep(time.Microsecond)
-					dashboard.UUID = time.Now().UnixMicro()
-					// 补全文件中的 uuid
-					bs, err = json.MarshalIndent(dashboard, "", "    ")
-					if err != nil {
-						logger.Warning("marshal builtin dashboard fail ", dashboard, err)
-						continue
-					}
-
-					_, err = file.WriteBytes(fp, bs)
-					if err != nil {
-						logger.Warning("write builtin dashboard file fail ", f, err)
-					}
-				}
-
-				content, err := json.Marshal(dashboard)
-				if err != nil {
-					logger.Warning("marshal builtin dashboard fail ", dashboard, err)
-					continue
-				}
-
-				builtinDashboard := models.BuiltinPayload{
-					ComponentID: component.ID,
-					Type:        "dashboard",
-					Cate:        "",
-					Name:        dashboard.Name,
-					Tags:        dashboard.Tags,
-					Content:     string(content),
-					UUID:        dashboard.UUID,
-				}
-
-				old, err := models.BuiltinPayloadGet(ctx, "uuid = ?", dashboard.UUID)
-				if err != nil {
-					logger.Warning("get builtin alert fail ", builtinDashboard, err)
-					continue
-				}
-
-				if old == nil {
-					err := builtinDashboard.Add(ctx, SYSTEM)
-					if err != nil {
-						logger.Warning("add builtin alert fail ", builtinDashboard, err)
-					}
-					continue
-				}
-
-				if old.UpdatedBy == SYSTEM {
-					old.ComponentID = component.ID
-					old.Content = string(content)
-					old.Name = dashboard.Name
-					old.Tags = dashboard.Tags
-					err = models.DB(ctx).Model(old).Select("*").Updates(old).Error
-					if err != nil {
-						logger.Warningf("update builtin alert:%+v fail %v", builtinDashboard, err)
-					}
-				}
-			}
-		} else if err != nil {
-			logger.Warningf("read builtin component dash dir fail %s %v", component.Ident, err)
-		}
-
-		// metrics
-		files, err = file.FilesUnder(componentDir + "/metrics")
-		if err == nil && len(files) > 0 {
-			for _, f := range files {
-				fp := componentDir + "/metrics/" + f
-				bs, err := file.ReadBytes(fp)
-				if err != nil {
-					logger.Warning("read builtin component metrics file fail", f, err)
-					continue
-				}
-
-				metrics := []models.BuiltinMetric{}
-				newMetrics := []models.BuiltinMetric{}
-				err = json.Unmarshal(bs, &metrics)
-				if err != nil {
-					logger.Warning("parse builtin component metrics file fail", f, err)
-					continue
-				}
-
-				writeMetricFileFlag := false
-				for _, metric := range metrics {
-					if metric.UUID == 0 {
-						writeMetricFileFlag = true
-						metric.UUID = time.Now().UnixNano()
-					}
-					newMetrics = append(newMetrics, metric)
-
-					old, err := models.BuiltinMetricGet(ctx, "uuid = ?", metric.UUID)
-					if err != nil {
-						logger.Warning("get builtin metrics fail ", metric, err)
-						continue
-					}
-
-					if old == nil {
-						err := metric.Add(ctx, SYSTEM)
-						if err != nil {
-							logger.Warning("add builtin metrics fail ", metric, err)
-						}
-						continue
-					}
-
-					if old.UpdatedBy == SYSTEM {
-						old.Collector = metric.Collector
-						old.Typ = metric.Typ
-						old.Name = metric.Name
-						old.Unit = metric.Unit
-						old.Note = metric.Note
-						old.Lang = metric.Lang
-						old.Expression = metric.Expression
-
-						err = models.DB(ctx).Model(old).Select("*").Updates(old).Error
-						if err != nil {
-							logger.Warningf("update builtin metric:%+v fail %v", metric, err)
-						}
-					}
-				}
-
-				if writeMetricFileFlag {
-					bs, err = json.MarshalIndent(newMetrics, "", "    ")
-					if err != nil {
-						logger.Warning("marshal builtin metrics fail ", newMetrics, err)
-						continue
-					}
-
-					_, err = file.WriteBytes(fp, bs)
-					if err != nil {
-						logger.Warning("write builtin metrics file fail ", f, err)
-					}
-				}
-
-			}
-		} else if err != nil {
-			logger.Warningf("read builtin component metrics dir fail %s %v", component.Ident, err)
-		}
-	}
-}
-
-type BuiltinBoard struct {
-	Id         int64       `json:"id" gorm:"primaryKey"`
-	GroupId    int64       `json:"group_id"`
-	Name       string      `json:"name"`
-	Ident      string      `json:"ident"`
-	Tags       string      `json:"tags"`
-	CreateAt   int64       `json:"create_at"`
-	CreateBy   string      `json:"create_by"`
-	UpdateAt   int64       `json:"update_at"`
-	UpdateBy   string      `json:"update_by"`
-	Configs    interface{} `json:"configs" gorm:"-"`
-	Public     int         `json:"public"`      // 0: false, 1: true
-	PublicCate int         `json:"public_cate"` // 0: anonymous, 1: login, 2: busi
-	Bgids      []int64     `json:"bgids" gorm:"-"`
-	BuiltIn    int         `json:"built_in"` // 0: false, 1: true
-	Hide       int         `json:"hide"`     // 0: false, 1: true
-	UUID       int64       `json:"uuid"`
-}

center/metas/metas.go

@@ -1,140 +0,0 @@
-package metas
-
-import (
-	"context"
-	"encoding/json"
-	"sync"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/center/cstats"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/storage"
-
-	"github.com/toolkits/pkg/logger"
-)
-
-type Set struct {
-	sync.RWMutex
-	items map[string]models.HostMeta
-	redis storage.Redis
-}
-
-func New(redis storage.Redis) *Set {
-	set := &Set{
-		items: make(map[string]models.HostMeta),
-		redis: redis,
-	}
-
-	set.Init()
-	return set
-}
-
-func (s *Set) Init() {
-	go s.LoopPersist()
-}
-
-func (s *Set) MSet(items map[string]models.HostMeta) {
-	s.Lock()
-	defer s.Unlock()
-	for ident, meta := range items {
-		s.items[ident] = meta
-	}
-}
-
-func (s *Set) Set(ident string, meta models.HostMeta) {
-	s.Lock()
-	defer s.Unlock()
-	s.items[ident] = meta
-}
-
-func (s *Set) LoopPersist() {
-	for {
-		time.Sleep(time.Second)
-		s.persist()
-	}
-}
-
-func (s *Set) persist() {
-	var items map[string]models.HostMeta
-
-	s.Lock()
-	if len(s.items) == 0 {
-		s.Unlock()
-		return
-	}
-
-	items = s.items
-	s.items = make(map[string]models.HostMeta)
-	s.Unlock()
-
-	s.updateMeta(items)
-}
-
-func (s *Set) updateMeta(items map[string]models.HostMeta) {
-	m := make(map[string]models.HostMeta, 100)
-	num := 0
-
-	for _, meta := range items {
-		m[meta.Hostname] = meta
-		num++
-		if num == 100 {
-			if err := s.updateTargets(m); err != nil {
-				logger.Errorf("failed to update targets: %v", err)
-			}
-			m = make(map[string]models.HostMeta, 100)
-			num = 0
-		}
-	}
-
-	if err := s.updateTargets(m); err != nil {
-		logger.Errorf("failed to update targets: %v", err)
-	}
-}
-
-func (s *Set) updateTargets(m map[string]models.HostMeta) error {
-	if s.redis == nil {
-		logger.Warningf("redis is nil")
-		return nil
-	}
-
-	count := int64(len(m))
-	if count == 0 {
-		return nil
-	}
-
-	newMap := make(map[string]interface{}, count)
-	extendMap := make(map[string]interface{})
-	for ident, meta := range m {
-		if meta.ExtendInfo != nil {
-			extendMeta := meta.ExtendInfo
-			meta.ExtendInfo = make(map[string]interface{})
-			extendMetaStr, err := json.Marshal(extendMeta)
-			if err != nil {
-				return err
-			}
-			extendMap[models.WrapExtendIdent(ident)] = extendMetaStr
-		}
-		newMap[models.WrapIdent(ident)] = meta
-	}
-
-	start := time.Now()
-	err := storage.MSet(context.Background(), s.redis, newMap)
-	if err != nil {
-		cstats.RedisOperationLatency.WithLabelValues("mset_target_meta", "fail").Observe(time.Since(start).Seconds())
-		return err
-	} else {
-		cstats.RedisOperationLatency.WithLabelValues("mset_target_meta", "success").Observe(time.Since(start).Seconds())
-	}
-
-	if len(extendMap) > 0 {
-		err = storage.MSet(context.Background(), s.redis, extendMap)
-		if err != nil {
-			cstats.RedisOperationLatency.WithLabelValues("mset_target_extend", "fail").Observe(time.Since(start).Seconds())
-			return err
-		} else {
-			cstats.RedisOperationLatency.WithLabelValues("mset_target_extend", "success").Observe(time.Since(start).Seconds())
-		}
-	}
-
-	return err
-}

center/router/router.go

@@ -1,707 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"net/http"
-	"path"
-	"runtime"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/alert/aconf"
-	"github.com/ccfos/nightingale/v6/center/cconf"
-	"github.com/ccfos/nightingale/v6/center/cstats"
-	"github.com/ccfos/nightingale/v6/center/metas"
-	"github.com/ccfos/nightingale/v6/center/sso"
-	"github.com/ccfos/nightingale/v6/conf"
-	_ "github.com/ccfos/nightingale/v6/front/statik"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/aop"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/httpx"
-	"github.com/ccfos/nightingale/v6/pkg/version"
-	"github.com/ccfos/nightingale/v6/prom"
-	"github.com/ccfos/nightingale/v6/pushgw/idents"
-	"github.com/ccfos/nightingale/v6/storage"
-	"gorm.io/gorm"
-
-	"github.com/gin-gonic/gin"
-	"github.com/rakyll/statik/fs"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/runner"
-)
-
-type Router struct {
-	HTTP              httpx.Config
-	Center            cconf.Center
-	Ibex              conf.Ibex
-	Alert             aconf.Alert
-	Operations        cconf.Operation
-	DatasourceCache   *memsto.DatasourceCacheType
-	NotifyConfigCache *memsto.NotifyConfigCacheType
-	PromClients       *prom.PromClientMap
-	Redis             storage.Redis
-	MetaSet           *metas.Set
-	IdentSet          *idents.Set
-	TargetCache       *memsto.TargetCacheType
-	Sso               *sso.SsoClient
-	UserCache         *memsto.UserCacheType
-	UserGroupCache    *memsto.UserGroupCacheType
-	UserTokenCache    *memsto.UserTokenCacheType
-	Ctx               *ctx.Context
-
-	HeartbeatHook       HeartbeatHookFunc
-	TargetDeleteHook    models.TargetDeleteHookFunc
-	AlertRuleModifyHook AlertRuleModifyHookFunc
-}
-
-func New(httpConfig httpx.Config, center cconf.Center, alert aconf.Alert, ibex conf.Ibex,
-	operations cconf.Operation, ds *memsto.DatasourceCacheType, ncc *memsto.NotifyConfigCacheType,
-	pc *prom.PromClientMap, redis storage.Redis,
-	sso *sso.SsoClient, ctx *ctx.Context, metaSet *metas.Set, idents *idents.Set,
-	tc *memsto.TargetCacheType, uc *memsto.UserCacheType, ugc *memsto.UserGroupCacheType, utc *memsto.UserTokenCacheType) *Router {
-	return &Router{
-		HTTP:                httpConfig,
-		Center:              center,
-		Alert:               alert,
-		Ibex:                ibex,
-		Operations:          operations,
-		DatasourceCache:     ds,
-		NotifyConfigCache:   ncc,
-		PromClients:         pc,
-		Redis:               redis,
-		MetaSet:             metaSet,
-		IdentSet:            idents,
-		TargetCache:         tc,
-		Sso:                 sso,
-		UserCache:           uc,
-		UserGroupCache:      ugc,
-		UserTokenCache:      utc,
-		Ctx:                 ctx,
-		HeartbeatHook:       func(ident string) map[string]interface{} { return nil },
-		TargetDeleteHook:    func(tx *gorm.DB, idents []string) error { return nil },
-		AlertRuleModifyHook: func(ar *models.AlertRule) {},
-	}
-}
-
-func stat() gin.HandlerFunc {
-	return func(c *gin.Context) {
-		start := time.Now()
-		c.Next()
-
-		code := fmt.Sprintf("%d", c.Writer.Status())
-		method := c.Request.Method
-		labels := []string{code, c.FullPath(), method}
-
-		cstats.RequestDuration.WithLabelValues(labels...).Observe(time.Since(start).Seconds())
-	}
-}
-
-func languageDetector(i18NHeaderKey string) gin.HandlerFunc {
-	headerKey := i18NHeaderKey
-	return func(c *gin.Context) {
-		if headerKey != "" {
-			lang := c.GetHeader(headerKey)
-			if lang != "" {
-				if strings.HasPrefix(lang, "zh_HK") {
-					c.Request.Header.Set("X-Language", "zh_HK")
-				} else if strings.HasPrefix(lang, "zh") {
-					c.Request.Header.Set("X-Language", "zh_CN")
-				} else if strings.HasPrefix(lang, "en") {
-					c.Request.Header.Set("X-Language", "en")
-				} else {
-					c.Request.Header.Set("X-Language", lang)
-				}
-			} else {
-				c.Request.Header.Set("X-Language", "zh_CN")
-			}
-		}
-		c.Next()
-	}
-}
-
-func (rt *Router) configNoRoute(r *gin.Engine, fs *http.FileSystem) {
-	r.NoRoute(func(c *gin.Context) {
-		arr := strings.Split(c.Request.URL.Path, ".")
-		suffix := arr[len(arr)-1]
-
-		switch suffix {
-		case "png", "jpeg", "jpg", "svg", "ico", "gif", "css", "js", "html", "htm", "gz", "zip", "map", "ttf", "md":
-			if !rt.Center.UseFileAssets {
-				c.FileFromFS(c.Request.URL.Path, *fs)
-			} else {
-				cwdarr := []string{"/"}
-				if runtime.GOOS == "windows" {
-					cwdarr[0] = ""
-				}
-				cwdarr = append(cwdarr, strings.Split(runner.Cwd, "/")...)
-				cwdarr = append(cwdarr, "pub")
-				cwdarr = append(cwdarr, strings.Split(c.Request.URL.Path, "/")...)
-				c.File(path.Join(cwdarr...))
-			}
-		default:
-			if !rt.Center.UseFileAssets {
-				c.FileFromFS("/", *fs)
-			} else {
-				cwdarr := []string{"/"}
-				if runtime.GOOS == "windows" {
-					cwdarr[0] = ""
-				}
-				cwdarr = append(cwdarr, strings.Split(runner.Cwd, "/")...)
-				cwdarr = append(cwdarr, "pub")
-				cwdarr = append(cwdarr, "index.html")
-				c.File(path.Join(cwdarr...))
-			}
-		}
-	})
-}
-
-func (rt *Router) Config(r *gin.Engine) {
-
-	r.Use(stat())
-	r.Use(languageDetector(rt.Center.I18NHeaderKey))
-	r.Use(aop.Recovery())
-
-	statikFS, err := fs.New()
-	if err != nil {
-		logger.Errorf("cannot create statik fs: %v", err)
-	}
-
-	if !rt.Center.UseFileAssets {
-		r.StaticFS("/pub", statikFS)
-	}
-
-	pagesPrefix := "/api/n9e"
-	pages := r.Group(pagesPrefix)
-	{
-
-		if rt.Center.AnonymousAccess.PromQuerier {
-			pages.Any("/proxy/:id/*url", rt.dsProxy)
-			pages.POST("/query-range-batch", rt.promBatchQueryRange)
-			pages.POST("/query-instant-batch", rt.promBatchQueryInstant)
-			pages.GET("/datasource/brief", rt.datasourceBriefs)
-			pages.POST("/datasource/query", rt.datasourceQuery)
-
-			pages.POST("/ds-query", rt.QueryData)
-			pages.POST("/logs-query", rt.QueryLogV2)
-
-			pages.POST("/tdengine-databases", rt.tdengineDatabases)
-			pages.POST("/tdengine-tables", rt.tdengineTables)
-			pages.POST("/tdengine-columns", rt.tdengineColumns)
-
-			pages.POST("/log-query-batch", rt.QueryLogBatch)
-
-			// 数据库元数据接口
-			pages.POST("/db-databases", rt.ShowDatabases)
-			pages.POST("/db-tables", rt.ShowTables)
-			pages.POST("/db-desc-table", rt.DescribeTable)
-
-			// es 专用接口
-			pages.POST("/indices", rt.auth(), rt.user(), rt.QueryIndices)
-			pages.POST("/es-variable", rt.auth(), rt.user(), rt.QueryESVariable)
-			pages.POST("/fields", rt.auth(), rt.user(), rt.QueryFields)
-			pages.POST("/log-query", rt.auth(), rt.user(), rt.QueryLog)
-		} else {
-			pages.Any("/proxy/:id/*url", rt.auth(), rt.dsProxy)
-			pages.POST("/query-range-batch", rt.auth(), rt.promBatchQueryRange)
-			pages.POST("/query-instant-batch", rt.auth(), rt.promBatchQueryInstant)
-			pages.GET("/datasource/brief", rt.auth(), rt.user(), rt.datasourceBriefs)
-			pages.POST("/datasource/query", rt.auth(), rt.user(), rt.datasourceQuery)
-
-			pages.POST("/ds-query", rt.auth(), rt.QueryData)
-			pages.POST("/logs-query", rt.auth(), rt.QueryLogV2)
-
-			pages.POST("/tdengine-databases", rt.auth(), rt.tdengineDatabases)
-			pages.POST("/tdengine-tables", rt.auth(), rt.tdengineTables)
-			pages.POST("/tdengine-columns", rt.auth(), rt.tdengineColumns)
-
-			pages.POST("/log-query-batch", rt.auth(), rt.user(), rt.QueryLogBatch)
-
-			// 数据库元数据接口
-			pages.POST("/db-databases", rt.auth(), rt.user(), rt.ShowDatabases)
-			pages.POST("/db-tables", rt.auth(), rt.user(), rt.ShowTables)
-			pages.POST("/db-desc-table", rt.auth(), rt.user(), rt.DescribeTable)
-
-			// es 专用接口
-			pages.POST("/indices", rt.auth(), rt.user(), rt.QueryIndices)
-			pages.POST("/es-variable", rt.QueryESVariable)
-			pages.POST("/fields", rt.QueryFields)
-			pages.POST("/log-query", rt.QueryLog)
-		}
-
-		pages.GET("/sql-template", rt.QuerySqlTemplate)
-		pages.POST("/auth/login", rt.jwtMock(), rt.loginPost)
-		pages.POST("/auth/logout", rt.jwtMock(), rt.auth(), rt.user(), rt.logoutPost)
-		pages.POST("/auth/refresh", rt.jwtMock(), rt.refreshPost)
-		pages.POST("/auth/captcha", rt.jwtMock(), rt.generateCaptcha)
-		pages.POST("/auth/captcha-verify", rt.jwtMock(), rt.captchaVerify)
-		pages.GET("/auth/ifshowcaptcha", rt.ifShowCaptcha)
-
-		pages.GET("/auth/sso-config", rt.ssoConfigNameGet)
-		pages.GET("/auth/rsa-config", rt.rsaConfigGet)
-		pages.GET("/auth/redirect", rt.loginRedirect)
-		pages.GET("/auth/redirect/cas", rt.loginRedirectCas)
-		pages.GET("/auth/redirect/oauth", rt.loginRedirectOAuth)
-		pages.GET("/auth/callback", rt.loginCallback)
-		pages.GET("/auth/callback/cas", rt.loginCallbackCas)
-		pages.GET("/auth/callback/oauth", rt.loginCallbackOAuth)
-		pages.GET("/auth/perms", rt.allPerms)
-
-		pages.GET("/metrics/desc", rt.metricsDescGetFile)
-		pages.POST("/metrics/desc", rt.metricsDescGetMap)
-
-		pages.GET("/notify-channels", rt.notifyChannelsGets)
-		pages.GET("/contact-keys", rt.contactKeysGets)
-
-		pages.GET("/self/perms", rt.auth(), rt.user(), rt.permsGets)
-		pages.GET("/self/profile", rt.auth(), rt.user(), rt.selfProfileGet)
-		pages.PUT("/self/profile", rt.auth(), rt.user(), rt.selfProfilePut)
-		pages.PUT("/self/password", rt.auth(), rt.user(), rt.selfPasswordPut)
-		pages.GET("/self/token", rt.auth(), rt.user(), rt.getToken)
-		pages.POST("/self/token", rt.auth(), rt.user(), rt.addToken)
-		pages.DELETE("/self/token/:id", rt.auth(), rt.user(), rt.deleteToken)
-
-		pages.GET("/users", rt.auth(), rt.user(), rt.perm("/users"), rt.userGets)
-		pages.POST("/users", rt.auth(), rt.user(), rt.perm("/users/add"), rt.userAddPost)
-		pages.GET("/user/:id/profile", rt.auth(), rt.userProfileGet)
-		pages.PUT("/user/:id/profile", rt.auth(), rt.user(), rt.perm("/users/put"), rt.userProfilePut)
-		pages.PUT("/user/:id/password", rt.auth(), rt.user(), rt.perm("/users/put"), rt.userPasswordPut)
-		pages.DELETE("/user/:id", rt.auth(), rt.user(), rt.perm("/users/del"), rt.userDel)
-
-		pages.GET("/metric-views", rt.auth(), rt.metricViewGets)
-		pages.DELETE("/metric-views", rt.auth(), rt.user(), rt.metricViewDel)
-		pages.POST("/metric-views", rt.auth(), rt.user(), rt.metricViewAdd)
-		pages.PUT("/metric-views", rt.auth(), rt.user(), rt.metricViewPut)
-
-		pages.GET("/builtin-metric-filters", rt.auth(), rt.user(), rt.metricFilterGets)
-		pages.DELETE("/builtin-metric-filters", rt.auth(), rt.user(), rt.metricFilterDel)
-		pages.POST("/builtin-metric-filters", rt.auth(), rt.user(), rt.metricFilterAdd)
-		pages.PUT("/builtin-metric-filters", rt.auth(), rt.user(), rt.metricFilterPut)
-		pages.POST("/builtin-metric-promql", rt.auth(), rt.user(), rt.getMetricPromql)
-
-		pages.POST("/builtin-metrics", rt.auth(), rt.user(), rt.perm("/builtin-metrics/add"), rt.builtinMetricsAdd)
-		pages.PUT("/builtin-metrics", rt.auth(), rt.user(), rt.perm("/builtin-metrics/put"), rt.builtinMetricsPut)
-		pages.DELETE("/builtin-metrics", rt.auth(), rt.user(), rt.perm("/builtin-metrics/del"), rt.builtinMetricsDel)
-		pages.GET("/builtin-metrics", rt.auth(), rt.user(), rt.builtinMetricsGets)
-		pages.GET("/builtin-metrics/types", rt.auth(), rt.user(), rt.builtinMetricsTypes)
-		pages.GET("/builtin-metrics/types/default", rt.auth(), rt.user(), rt.builtinMetricsDefaultTypes)
-		pages.GET("/builtin-metrics/collectors", rt.auth(), rt.user(), rt.builtinMetricsCollectors)
-
-		pages.GET("/user-groups", rt.auth(), rt.user(), rt.userGroupGets)
-		pages.POST("/user-groups", rt.auth(), rt.user(), rt.perm("/user-groups/add"), rt.userGroupAdd)
-		pages.GET("/user-group/:id", rt.auth(), rt.user(), rt.userGroupGet)
-		pages.PUT("/user-group/:id", rt.auth(), rt.user(), rt.perm("/user-groups/put"), rt.userGroupWrite(), rt.userGroupPut)
-		pages.DELETE("/user-group/:id", rt.auth(), rt.user(), rt.perm("/user-groups/del"), rt.userGroupWrite(), rt.userGroupDel)
-		pages.POST("/user-group/:id/members", rt.auth(), rt.user(), rt.perm("/user-groups/put"), rt.userGroupWrite(), rt.userGroupMemberAdd)
-		pages.DELETE("/user-group/:id/members", rt.auth(), rt.user(), rt.perm("/user-groups/put"), rt.userGroupWrite(), rt.userGroupMemberDel)
-
-		pages.GET("/busi-groups", rt.auth(), rt.user(), rt.busiGroupGets)
-		pages.POST("/busi-groups", rt.auth(), rt.user(), rt.perm("/busi-groups/add"), rt.busiGroupAdd)
-		pages.GET("/busi-groups/alertings", rt.auth(), rt.busiGroupAlertingsGets)
-		pages.GET("/busi-group/:id", rt.auth(), rt.user(), rt.bgro(), rt.busiGroupGet)
-		pages.PUT("/busi-group/:id", rt.auth(), rt.user(), rt.perm("/busi-groups/put"), rt.bgrw(), rt.busiGroupPut)
-		pages.POST("/busi-group/:id/members", rt.auth(), rt.user(), rt.perm("/busi-groups/put"), rt.bgrw(), rt.busiGroupMemberAdd)
-		pages.DELETE("/busi-group/:id/members", rt.auth(), rt.user(), rt.perm("/busi-groups/put"), rt.bgrw(), rt.busiGroupMemberDel)
-		pages.DELETE("/busi-group/:id", rt.auth(), rt.user(), rt.perm("/busi-groups/del"), rt.bgrw(), rt.busiGroupDel)
-		pages.GET("/busi-group/:id/perm/:perm", rt.auth(), rt.user(), rt.checkBusiGroupPerm)
-		pages.GET("/busi-groups/tags", rt.auth(), rt.user(), rt.busiGroupsGetTags)
-
-		pages.GET("/targets", rt.auth(), rt.user(), rt.targetGets)
-		pages.GET("/target/extra-meta", rt.auth(), rt.user(), rt.targetExtendInfoByIdent)
-		pages.POST("/target/list", rt.auth(), rt.user(), rt.targetGetsByHostFilter)
-		pages.DELETE("/targets", rt.auth(), rt.user(), rt.perm("/targets/del"), rt.targetDel)
-		pages.GET("/targets/tags", rt.auth(), rt.user(), rt.targetGetTags)
-		pages.POST("/targets/tags", rt.auth(), rt.user(), rt.perm("/targets/put"), rt.targetBindTagsByFE)
-		pages.DELETE("/targets/tags", rt.auth(), rt.user(), rt.perm("/targets/put"), rt.targetUnbindTagsByFE)
-		pages.PUT("/targets/note", rt.auth(), rt.user(), rt.perm("/targets/put"), rt.targetUpdateNote)
-		pages.PUT("/targets/bgids", rt.auth(), rt.user(), rt.perm("/targets/put"), rt.targetBindBgids)
-
-		pages.POST("/builtin-cate-favorite", rt.auth(), rt.user(), rt.builtinCateFavoriteAdd)
-		pages.DELETE("/builtin-cate-favorite/:name", rt.auth(), rt.user(), rt.builtinCateFavoriteDel)
-
-		pages.GET("/integrations/icon/:cate/:name", rt.builtinIcon)
-
-		// pages.GET("/builtin-boards", rt.builtinBoardGets)
-		// pages.GET("/builtin-board/:name", rt.builtinBoardGet)
-		// pages.GET("/dashboards/builtin/list", rt.builtinBoardGets)
-		// pages.GET("/builtin-boards-cates", rt.auth(), rt.user(), rt.builtinBoardCateGets)
-		// pages.POST("/builtin-boards-detail", rt.auth(), rt.user(), rt.builtinBoardDetailGets)
-		// pages.GET("/integrations/makedown/:cate", rt.builtinMarkdown)
-
-		pages.GET("/busi-groups/public-boards", rt.auth(), rt.user(), rt.perm("/dashboards"), rt.publicBoardGets)
-		pages.GET("/busi-groups/boards", rt.auth(), rt.user(), rt.perm("/dashboards"), rt.boardGetsByGids)
-		pages.GET("/busi-group/:id/boards", rt.auth(), rt.user(), rt.perm("/dashboards"), rt.bgro(), rt.boardGets)
-		pages.POST("/busi-group/:id/boards", rt.auth(), rt.user(), rt.perm("/dashboards/add"), rt.bgrw(), rt.boardAdd)
-		pages.POST("/busi-group/:id/board/:bid/clone", rt.auth(), rt.user(), rt.perm("/dashboards/add"), rt.bgrw(), rt.boardClone)
-		pages.POST("/busi-groups/boards/clones", rt.auth(), rt.user(), rt.perm("/dashboards/add"), rt.boardBatchClone)
-
-		pages.GET("/boards", rt.auth(), rt.user(), rt.boardGetsByBids)
-		pages.GET("/board/:bid", rt.boardGet)
-		pages.GET("/board/:bid/pure", rt.boardPureGet)
-		pages.PUT("/board/:bid", rt.auth(), rt.user(), rt.perm("/dashboards/put"), rt.boardPut)
-		pages.PUT("/board/:bid/configs", rt.auth(), rt.user(), rt.perm("/dashboards/put"), rt.boardPutConfigs)
-		pages.PUT("/board/:bid/public", rt.auth(), rt.user(), rt.perm("/dashboards/put"), rt.boardPutPublic)
-		pages.DELETE("/boards", rt.auth(), rt.user(), rt.perm("/dashboards/del"), rt.boardDel)
-
-		pages.GET("/share-charts", rt.chartShareGets)
-		pages.POST("/share-charts", rt.auth(), rt.chartShareAdd)
-
-		pages.POST("/dashboard-annotations", rt.auth(), rt.user(), rt.perm("/dashboards/put"), rt.dashAnnotationAdd)
-		pages.GET("/dashboard-annotations", rt.dashAnnotationGets)
-		pages.PUT("/dashboard-annotation/:id", rt.auth(), rt.user(), rt.perm("/dashboards/put"), rt.dashAnnotationPut)
-		pages.DELETE("/dashboard-annotation/:id", rt.auth(), rt.user(), rt.perm("/dashboards/del"), rt.dashAnnotationDel)
-
-		// pages.GET("/alert-rules/builtin/alerts-cates", rt.auth(), rt.user(), rt.builtinAlertCateGets)
-		// pages.GET("/alert-rules/builtin/list", rt.auth(), rt.user(), rt.builtinAlertRules)
-		pages.GET("/alert-rules/callbacks", rt.auth(), rt.user(), rt.alertRuleCallbacks)
-
-		pages.GET("/busi-groups/alert-rules", rt.auth(), rt.user(), rt.perm("/alert-rules"), rt.alertRuleGetsByGids)
-		pages.GET("/busi-group/:id/alert-rules", rt.auth(), rt.user(), rt.perm("/alert-rules"), rt.alertRuleGets)
-		pages.POST("/busi-group/:id/alert-rules", rt.auth(), rt.user(), rt.perm("/alert-rules/add"), rt.bgrw(), rt.alertRuleAddByFE)
-		pages.POST("/busi-group/:id/alert-rules/import", rt.auth(), rt.user(), rt.perm("/alert-rules/add"), rt.bgrw(), rt.alertRuleAddByImport)
-		pages.POST("/busi-group/:id/alert-rules/import-prom-rule", rt.auth(),
-			rt.user(), rt.perm("/alert-rules/add"), rt.bgrw(), rt.alertRuleAddByImportPromRule)
-		pages.DELETE("/busi-group/:id/alert-rules", rt.auth(), rt.user(), rt.perm("/alert-rules/del"), rt.bgrw(), rt.alertRuleDel)
-		pages.PUT("/busi-group/:id/alert-rules/fields", rt.auth(), rt.user(), rt.perm("/alert-rules/put"), rt.bgrw(), rt.alertRulePutFields)
-		pages.PUT("/busi-group/:id/alert-rule/:arid", rt.auth(), rt.user(), rt.perm("/alert-rules/put"), rt.alertRulePutByFE)
-		pages.GET("/alert-rule/:arid", rt.auth(), rt.user(), rt.perm("/alert-rules"), rt.alertRuleGet)
-		pages.GET("/alert-rule/:arid/pure", rt.auth(), rt.user(), rt.perm("/alert-rules"), rt.alertRulePureGet)
-		pages.PUT("/busi-group/alert-rule/validate", rt.auth(), rt.user(), rt.perm("/alert-rules/put"), rt.alertRuleValidation)
-		pages.POST("/relabel-test", rt.auth(), rt.user(), rt.relabelTest)
-		pages.POST("/busi-group/:id/alert-rules/clone", rt.auth(), rt.user(), rt.perm("/alert-rules/add"), rt.bgrw(), rt.cloneToMachine)
-		pages.POST("/busi-groups/alert-rules/clones", rt.auth(), rt.user(), rt.perm("/alert-rules/add"), rt.batchAlertRuleClone)
-
-		pages.GET("/busi-groups/recording-rules", rt.auth(), rt.user(), rt.perm("/recording-rules"), rt.recordingRuleGetsByGids)
-		pages.GET("/busi-group/:id/recording-rules", rt.auth(), rt.user(), rt.perm("/recording-rules"), rt.recordingRuleGets)
-		pages.POST("/busi-group/:id/recording-rules", rt.auth(), rt.user(), rt.perm("/recording-rules/add"), rt.bgrw(), rt.recordingRuleAddByFE)
-		pages.DELETE("/busi-group/:id/recording-rules", rt.auth(), rt.user(), rt.perm("/recording-rules/del"), rt.bgrw(), rt.recordingRuleDel)
-		pages.PUT("/busi-group/:id/recording-rule/:rrid", rt.auth(), rt.user(), rt.perm("/recording-rules/put"), rt.bgrw(), rt.recordingRulePutByFE)
-		pages.GET("/recording-rule/:rrid", rt.auth(), rt.user(), rt.perm("/recording-rules"), rt.recordingRuleGet)
-		pages.PUT("/busi-group/:id/recording-rules/fields", rt.auth(), rt.user(), rt.perm("/recording-rules/put"), rt.recordingRulePutFields)
-
-		pages.GET("/busi-groups/alert-mutes", rt.auth(), rt.user(), rt.perm("/alert-mutes"), rt.alertMuteGetsByGids)
-		pages.GET("/busi-group/:id/alert-mutes", rt.auth(), rt.user(), rt.perm("/alert-mutes"), rt.bgro(), rt.alertMuteGetsByBG)
-		pages.POST("/busi-group/:id/alert-mutes/preview", rt.auth(), rt.user(), rt.perm("/alert-mutes/add"), rt.bgrw(), rt.alertMutePreview)
-		pages.POST("/busi-group/:id/alert-mutes", rt.auth(), rt.user(), rt.perm("/alert-mutes/add"), rt.bgrw(), rt.alertMuteAdd)
-		pages.DELETE("/busi-group/:id/alert-mutes", rt.auth(), rt.user(), rt.perm("/alert-mutes/del"), rt.bgrw(), rt.alertMuteDel)
-		pages.PUT("/busi-group/:id/alert-mute/:amid", rt.auth(), rt.user(), rt.perm("/alert-mutes/put"), rt.alertMutePutByFE)
-		pages.GET("/busi-group/:id/alert-mute/:amid", rt.auth(), rt.user(), rt.perm("/alert-mutes"), rt.alertMuteGet)
-		pages.PUT("/busi-group/:id/alert-mutes/fields", rt.auth(), rt.user(), rt.perm("/alert-mutes/put"), rt.bgrw(), rt.alertMutePutFields)
-		pages.POST("/alert-mute-tryrun", rt.auth(), rt.user(), rt.perm("/alert-mutes/add"), rt.alertMuteTryRun)
-
-		pages.GET("/busi-groups/alert-subscribes", rt.auth(), rt.user(), rt.perm("/alert-subscribes"), rt.alertSubscribeGetsByGids)
-		pages.GET("/busi-group/:id/alert-subscribes", rt.auth(), rt.user(), rt.perm("/alert-subscribes"), rt.bgro(), rt.alertSubscribeGets)
-		pages.GET("/alert-subscribe/:sid", rt.auth(), rt.user(), rt.perm("/alert-subscribes"), rt.alertSubscribeGet)
-		pages.POST("/busi-group/:id/alert-subscribes", rt.auth(), rt.user(), rt.perm("/alert-subscribes/add"), rt.bgrw(), rt.alertSubscribeAdd)
-		pages.PUT("/busi-group/:id/alert-subscribes", rt.auth(), rt.user(), rt.perm("/alert-subscribes/put"), rt.bgrw(), rt.alertSubscribePut)
-		pages.DELETE("/busi-group/:id/alert-subscribes", rt.auth(), rt.user(), rt.perm("/alert-subscribes/del"), rt.bgrw(), rt.alertSubscribeDel)
-
-		pages.GET("/alert-cur-event/:eid", rt.alertCurEventGet)
-		pages.GET("/alert-his-event/:eid", rt.alertHisEventGet)
-		pages.GET("/event-notify-records/:eid", rt.notificationRecordList)
-
-		// card logic
-		pages.GET("/alert-cur-events/list", rt.auth(), rt.user(), rt.alertCurEventsList)
-		pages.GET("/alert-cur-events/card", rt.auth(), rt.user(), rt.alertCurEventsCard)
-		pages.POST("/alert-cur-events/card/details", rt.auth(), rt.alertCurEventsCardDetails)
-		pages.GET("/alert-his-events/list", rt.auth(), rt.user(), rt.alertHisEventsList)
-		pages.DELETE("/alert-his-events", rt.auth(), rt.admin(), rt.alertHisEventsDelete)
-		pages.DELETE("/alert-cur-events", rt.auth(), rt.user(), rt.perm("/alert-cur-events/del"), rt.alertCurEventDel)
-		pages.GET("/alert-cur-events/stats", rt.auth(), rt.alertCurEventsStatistics)
-
-		pages.GET("/alert-aggr-views", rt.auth(), rt.alertAggrViewGets)
-		pages.DELETE("/alert-aggr-views", rt.auth(), rt.user(), rt.alertAggrViewDel)
-		pages.POST("/alert-aggr-views", rt.auth(), rt.user(), rt.alertAggrViewAdd)
-		pages.PUT("/alert-aggr-views", rt.auth(), rt.user(), rt.alertAggrViewPut)
-
-		pages.GET("/busi-groups/task-tpls", rt.auth(), rt.user(), rt.perm("/job-tpls"), rt.taskTplGetsByGids)
-		pages.GET("/busi-group/:id/task-tpls", rt.auth(), rt.user(), rt.perm("/job-tpls"), rt.bgro(), rt.taskTplGets)
-		pages.POST("/busi-group/:id/task-tpls", rt.auth(), rt.user(), rt.perm("/job-tpls/add"), rt.bgrw(), rt.taskTplAdd)
-		pages.DELETE("/busi-group/:id/task-tpl/:tid", rt.auth(), rt.user(), rt.perm("/job-tpls/del"), rt.bgrw(), rt.taskTplDel)
-		pages.POST("/busi-group/:id/task-tpls/tags", rt.auth(), rt.user(), rt.perm("/job-tpls/put"), rt.bgrw(), rt.taskTplBindTags)
-		pages.DELETE("/busi-group/:id/task-tpls/tags", rt.auth(), rt.user(), rt.perm("/job-tpls/put"), rt.bgrw(), rt.taskTplUnbindTags)
-		pages.GET("/busi-group/:id/task-tpl/:tid", rt.auth(), rt.user(), rt.perm("/job-tpls"), rt.bgro(), rt.taskTplGet)
-		pages.PUT("/busi-group/:id/task-tpl/:tid", rt.auth(), rt.user(), rt.perm("/job-tpls/put"), rt.bgrw(), rt.taskTplPut)
-
-		pages.GET("/busi-groups/tasks", rt.auth(), rt.user(), rt.perm("/job-tasks"), rt.taskGetsByGids)
-		pages.GET("/busi-group/:id/tasks", rt.auth(), rt.user(), rt.perm("/job-tasks"), rt.bgro(), rt.taskGets)
-		pages.POST("/busi-group/:id/tasks", rt.auth(), rt.user(), rt.perm("/job-tasks/add"), rt.bgrw(), rt.taskAdd)
-
-		pages.GET("/servers", rt.auth(), rt.user(), rt.serversGet)
-		pages.GET("/server-clusters", rt.auth(), rt.user(), rt.serverClustersGet)
-
-		pages.POST("/datasource/list", rt.auth(), rt.user(), rt.datasourceList)
-		pages.POST("/datasource/plugin/list", rt.auth(), rt.pluginList)
-		pages.POST("/datasource/upsert", rt.auth(), rt.admin(), rt.datasourceUpsert)
-		pages.POST("/datasource/desc", rt.auth(), rt.admin(), rt.datasourceGet)
-		pages.POST("/datasource/status/update", rt.auth(), rt.admin(), rt.datasourceUpdataStatus)
-		pages.DELETE("/datasource/", rt.auth(), rt.admin(), rt.datasourceDel)
-
-		pages.GET("/roles", rt.auth(), rt.user(), rt.perm("/roles"), rt.roleGets)
-		pages.POST("/roles", rt.auth(), rt.user(), rt.perm("/roles/add"), rt.roleAdd)
-		pages.PUT("/roles", rt.auth(), rt.user(), rt.perm("/roles/put"), rt.rolePut)
-		pages.DELETE("/role/:id", rt.auth(), rt.user(), rt.perm("/roles/del"), rt.roleDel)
-
-		pages.GET("/role/:id/ops", rt.auth(), rt.user(), rt.perm("/roles"), rt.operationOfRole)
-		pages.PUT("/role/:id/ops", rt.auth(), rt.user(), rt.perm("/roles/put"), rt.roleBindOperation)
-		pages.GET("/operation", rt.operations)
-
-		pages.GET("/notify-tpls", rt.auth(), rt.user(), rt.notifyTplGets)
-		pages.PUT("/notify-tpl/content", rt.auth(), rt.user(), rt.notifyTplUpdateContent)
-		pages.PUT("/notify-tpl", rt.auth(), rt.user(), rt.notifyTplUpdate)
-		pages.POST("/notify-tpl", rt.auth(), rt.user(), rt.notifyTplAdd)
-		pages.DELETE("/notify-tpl/:id", rt.auth(), rt.user(), rt.notifyTplDel)
-		pages.POST("/notify-tpl/preview", rt.auth(), rt.user(), rt.notifyTplPreview)
-
-		pages.GET("/sso-configs", rt.auth(), rt.admin(), rt.ssoConfigGets)
-		pages.PUT("/sso-config", rt.auth(), rt.admin(), rt.ssoConfigUpdate)
-
-		pages.GET("/webhooks", rt.auth(), rt.user(), rt.webhookGets)
-		pages.PUT("/webhooks", rt.auth(), rt.admin(), rt.webhookPuts)
-
-		pages.GET("/notify-script", rt.auth(), rt.user(), rt.perm("/help/notification-settings"), rt.notifyScriptGet)
-		pages.PUT("/notify-script", rt.auth(), rt.admin(), rt.notifyScriptPut)
-
-		pages.GET("/notify-channel", rt.auth(), rt.user(), rt.perm("/help/notification-settings"), rt.notifyChannelGets)
-		pages.PUT("/notify-channel", rt.auth(), rt.admin(), rt.notifyChannelPuts)
-
-		pages.GET("/notify-contact", rt.auth(), rt.user(), rt.notifyContactGets)
-		pages.PUT("/notify-contact", rt.auth(), rt.admin(), rt.notifyContactPuts)
-
-		pages.GET("/notify-config", rt.auth(), rt.user(), rt.perm("/help/notification-settings"), rt.notifyConfigGet)
-		pages.PUT("/notify-config", rt.auth(), rt.admin(), rt.notifyConfigPut)
-		pages.PUT("/smtp-config-test", rt.auth(), rt.admin(), rt.attemptSendEmail)
-
-		pages.GET("/es-index-pattern", rt.auth(), rt.esIndexPatternGet)
-		pages.GET("/es-index-pattern-list", rt.auth(), rt.esIndexPatternGetList)
-		pages.POST("/es-index-pattern", rt.auth(), rt.user(), rt.perm("/log/index-patterns/add"), rt.esIndexPatternAdd)
-		pages.PUT("/es-index-pattern", rt.auth(), rt.user(), rt.perm("/log/index-patterns/put"), rt.esIndexPatternPut)
-		pages.DELETE("/es-index-pattern", rt.auth(), rt.user(), rt.perm("/log/index-patterns/del"), rt.esIndexPatternDel)
-
-		pages.GET("/embedded-dashboards", rt.auth(), rt.user(), rt.perm("/embedded-dashboards"), rt.embeddedDashboardsGet)
-		pages.PUT("/embedded-dashboards", rt.auth(), rt.user(), rt.perm("/embedded-dashboards/put"), rt.embeddedDashboardsPut)
-
-		// 获取 embedded-product 列表
-		pages.GET("/embedded-product", rt.auth(), rt.user(), rt.embeddedProductGets)
-		pages.GET("/embedded-product/:id", rt.auth(), rt.user(), rt.embeddedProductGet)
-		pages.POST("/embedded-product", rt.auth(), rt.user(), rt.perm("/embedded-product/add"), rt.embeddedProductAdd)
-		pages.PUT("/embedded-product/:id", rt.auth(), rt.user(), rt.perm("/embedded-product/put"), rt.embeddedProductPut)
-		pages.DELETE("/embedded-product/:id", rt.auth(), rt.user(), rt.perm("/embedded-product/delete"), rt.embeddedProductDelete)
-
-		pages.GET("/user-variable-configs", rt.auth(), rt.user(), rt.perm("/help/variable-configs"), rt.userVariableConfigGets)
-		pages.POST("/user-variable-config", rt.auth(), rt.user(), rt.perm("/help/variable-configs"), rt.userVariableConfigAdd)
-		pages.PUT("/user-variable-config/:id", rt.auth(), rt.user(), rt.perm("/help/variable-configs"), rt.userVariableConfigPut)
-		pages.DELETE("/user-variable-config/:id", rt.auth(), rt.user(), rt.perm("/help/variable-configs"), rt.userVariableConfigDel)
-
-		pages.GET("/config", rt.auth(), rt.admin(), rt.configGetByKey)
-		pages.PUT("/config", rt.auth(), rt.admin(), rt.configPutByKey)
-		pages.GET("/site-info", rt.siteInfo)
-
-		// source token 相关路由
-		pages.POST("/source-token", rt.auth(), rt.user(), rt.sourceTokenAdd)
-
-		// for admin api
-		pages.GET("/user/busi-groups", rt.auth(), rt.admin(), rt.userBusiGroupsGets)
-
-		pages.GET("/builtin-components", rt.auth(), rt.user(), rt.builtinComponentsGets)
-		pages.POST("/builtin-components", rt.auth(), rt.user(), rt.perm("/components/add"), rt.builtinComponentsAdd)
-		pages.PUT("/builtin-components", rt.auth(), rt.user(), rt.perm("/components/put"), rt.builtinComponentsPut)
-		pages.DELETE("/builtin-components", rt.auth(), rt.user(), rt.perm("/components/del"), rt.builtinComponentsDel)
-
-		pages.GET("/builtin-payloads", rt.auth(), rt.user(), rt.builtinPayloadsGets)
-		pages.GET("/builtin-payloads/cates", rt.auth(), rt.user(), rt.builtinPayloadcatesGet)
-		pages.POST("/builtin-payloads", rt.auth(), rt.user(), rt.perm("/components/add"), rt.builtinPayloadsAdd)
-		pages.GET("/builtin-payload/:id", rt.auth(), rt.user(), rt.perm("/components"), rt.builtinPayloadGet)
-		pages.PUT("/builtin-payloads", rt.auth(), rt.user(), rt.perm("/components/put"), rt.builtinPayloadsPut)
-		pages.DELETE("/builtin-payloads", rt.auth(), rt.user(), rt.perm("/components/del"), rt.builtinPayloadsDel)
-		pages.GET("/builtin-payload", rt.auth(), rt.user(), rt.builtinPayloadsGetByUUIDOrID)
-
-		pages.POST("/message-templates", rt.auth(), rt.user(), rt.perm("/notification-templates/add"), rt.messageTemplatesAdd)
-		pages.DELETE("/message-templates", rt.auth(), rt.user(), rt.perm("/notification-templates/del"), rt.messageTemplatesDel)
-		pages.PUT("/message-template/:id", rt.auth(), rt.user(), rt.perm("/notification-templates/put"), rt.messageTemplatePut)
-		pages.GET("/message-template/:id", rt.auth(), rt.user(), rt.perm("/notification-templates"), rt.messageTemplateGet)
-		pages.GET("/message-templates", rt.auth(), rt.user(), rt.messageTemplatesGet)
-		pages.POST("/events-message", rt.auth(), rt.user(), rt.eventsMessage)
-
-		pages.POST("/notify-rules", rt.auth(), rt.user(), rt.perm("/notification-rules/add"), rt.notifyRulesAdd)
-		pages.DELETE("/notify-rules", rt.auth(), rt.user(), rt.perm("/notification-rules/del"), rt.notifyRulesDel)
-		pages.PUT("/notify-rule/:id", rt.auth(), rt.user(), rt.perm("/notification-rules/put"), rt.notifyRulePut)
-		pages.GET("/notify-rule/:id", rt.auth(), rt.user(), rt.perm("/notification-rules"), rt.notifyRuleGet)
-		pages.GET("/notify-rules", rt.auth(), rt.user(), rt.perm("/notification-rules"), rt.notifyRulesGet)
-		pages.POST("/notify-rule/test", rt.auth(), rt.user(), rt.perm("/notification-rules"), rt.notifyTest)
-		pages.GET("/notify-rule/custom-params", rt.auth(), rt.user(), rt.perm("/notification-rules"), rt.notifyRuleCustomParamsGet)
-		pages.POST("/notify-rule/event-pipelines-tryrun", rt.auth(), rt.user(), rt.perm("/notification-rules/add"), rt.tryRunEventProcessorByNotifyRule)
-
-		// 事件Pipeline相关路由
-		pages.GET("/event-pipelines", rt.auth(), rt.user(), rt.perm("/event-pipelines"), rt.eventPipelinesList)
-		pages.POST("/event-pipeline", rt.auth(), rt.user(), rt.perm("/event-pipelines/add"), rt.addEventPipeline)
-		pages.PUT("/event-pipeline", rt.auth(), rt.user(), rt.perm("/event-pipelines/put"), rt.updateEventPipeline)
-		pages.GET("/event-pipeline/:id", rt.auth(), rt.user(), rt.perm("/event-pipelines"), rt.getEventPipeline)
-		pages.DELETE("/event-pipelines", rt.auth(), rt.user(), rt.perm("/event-pipelines/del"), rt.deleteEventPipelines)
-		pages.POST("/event-pipeline-tryrun", rt.auth(), rt.user(), rt.perm("/event-pipelines"), rt.tryRunEventPipeline)
-		pages.POST("/event-processor-tryrun", rt.auth(), rt.user(), rt.perm("/event-pipelines"), rt.tryRunEventProcessor)
-
-		pages.POST("/notify-channel-configs", rt.auth(), rt.user(), rt.perm("/notification-channels/add"), rt.notifyChannelsAdd)
-		pages.DELETE("/notify-channel-configs", rt.auth(), rt.user(), rt.perm("/notification-channels/del"), rt.notifyChannelsDel)
-		pages.PUT("/notify-channel-config/:id", rt.auth(), rt.user(), rt.perm("/notification-channels/put"), rt.notifyChannelPut)
-		pages.GET("/notify-channel-config/:id", rt.auth(), rt.user(), rt.perm("/notification-channels"), rt.notifyChannelGet)
-		pages.GET("/notify-channel-configs", rt.auth(), rt.user(), rt.perm("/notification-channels"), rt.notifyChannelsGet)
-		pages.GET("/simplified-notify-channel-configs", rt.notifyChannelsGetForNormalUser)
-		pages.GET("/flashduty-channel-list/:id", rt.auth(), rt.user(), rt.flashDutyNotifyChannelsGet)
-		pages.GET("/notify-channel-config", rt.auth(), rt.user(), rt.notifyChannelGetBy)
-		pages.GET("/notify-channel-config/idents", rt.notifyChannelIdentsGet)
-	}
-
-	r.GET("/api/n9e/versions", func(c *gin.Context) {
-		v := version.Version
-		lastIndex := strings.LastIndex(version.Version, "-")
-		if lastIndex != -1 {
-			v = version.Version[:lastIndex]
-		}
-
-		gv := version.GithubVersion.Load()
-		if gv != nil {
-			ginx.NewRender(c).Data(gin.H{"version": v, "github_verison": gv.(string)}, nil)
-		} else {
-			ginx.NewRender(c).Data(gin.H{"version": v, "github_verison": ""}, nil)
-		}
-	})
-
-	if rt.HTTP.APIForService.Enable {
-		service := r.Group("/v1/n9e")
-		if len(rt.HTTP.APIForService.BasicAuth) > 0 {
-			service.Use(gin.BasicAuth(rt.HTTP.APIForService.BasicAuth))
-		}
-		{
-			service.Any("/prometheus/*url", rt.dsProxy)
-			service.POST("/users", rt.userAddPost)
-			service.PUT("/user/:id", rt.userProfilePutByService)
-			service.DELETE("/user/:id", rt.userDel)
-			service.GET("/users", rt.userFindAll)
-
-			service.GET("/user-groups", rt.userGroupGetsByService)
-			service.GET("/user-group-members", rt.userGroupMemberGetsByService)
-
-			service.GET("/targets", rt.targetGetsByService)
-			service.GET("/target/extra-meta", rt.targetExtendInfoByIdent)
-			service.POST("/target/list", rt.targetGetsByHostFilter)
-			service.DELETE("/targets", rt.targetDelByService)
-			service.GET("/targets/tags", rt.targetGetTags)
-			service.POST("/targets/tags", rt.targetBindTagsByService)
-			service.DELETE("/targets/tags", rt.targetUnbindTagsByService)
-			service.PUT("/targets/note", rt.targetUpdateNoteByService)
-			service.PUT("/targets/bgid", rt.targetUpdateBgidByService)
-
-			service.POST("/targets-of-host-query", rt.targetsOfHostQuery)
-
-			service.POST("/alert-rules", rt.alertRuleAddByService)
-			service.POST("/alert-rule-add", rt.alertRuleAddOneByService)
-			service.DELETE("/alert-rules", rt.alertRuleDelByService)
-			service.PUT("/alert-rule/:arid", rt.alertRulePutByService)
-			service.GET("/alert-rule/:arid", rt.alertRuleGet)
-			service.GET("/alert-rules", rt.alertRulesGetByService)
-
-			service.GET("/alert-subscribes", rt.alertSubscribeGetsByService)
-
-			service.GET("/busi-groups", rt.busiGroupGetsByService)
-
-			service.GET("/datasources", rt.datasourceGetsByService)
-			service.GET("/datasource-ids", rt.getDatasourceIds)
-			service.POST("/server-heartbeat", rt.serverHeartbeat)
-			service.GET("/servers-active", rt.serversActive)
-
-			service.GET("/recording-rules", rt.recordingRuleGetsByService)
-
-			service.GET("/alert-mutes", rt.alertMuteGets)
-			service.POST("/alert-mutes", rt.alertMuteAddByService)
-			service.DELETE("/alert-mutes", rt.alertMuteDel)
-
-			service.GET("/alert-cur-events", rt.alertCurEventsList)
-			service.GET("/alert-cur-events-get-by-rid", rt.alertCurEventsGetByRid)
-			service.GET("/alert-his-events", rt.alertHisEventsList)
-			service.GET("/alert-his-event/:eid", rt.alertHisEventGet)
-
-			service.GET("/task-tpl/:tid", rt.taskTplGetByService)
-			service.GET("/task-tpls", rt.taskTplGetsByService)
-			service.GET("/task-tpl/statistics", rt.taskTplStatistics)
-
-			service.GET("/config/:id", rt.configGet)
-			service.GET("/configs", rt.configsGet)
-			service.GET("/config", rt.configGetByKey)
-			service.GET("/all-configs", rt.configGetAll)
-			service.PUT("/configs", rt.configsPut)
-			service.POST("/configs", rt.configsPost)
-			service.DELETE("/configs", rt.configsDel)
-
-			service.POST("/conf-prop/encrypt", rt.confPropEncrypt)
-			service.POST("/conf-prop/decrypt", rt.confPropDecrypt)
-
-			service.GET("/statistic", rt.statistic)
-
-			service.GET("/notify-tpls", rt.notifyTplGets)
-
-			service.POST("/task-record-add", rt.taskRecordAdd)
-
-			service.GET("/user-variable/decrypt", rt.userVariableGetDecryptByService)
-
-			service.GET("/targets-of-alert-rule", rt.targetsOfAlertRule)
-
-			service.POST("/notify-record", rt.notificationRecordAdd)
-
-			service.GET("/alert-cur-events-del-by-hash", rt.alertCurEventDelByHash)
-
-			service.POST("/center/heartbeat", rt.heartbeat)
-
-			service.GET("/es-index-pattern-list", rt.esIndexPatternGetList)
-
-			service.GET("/notify-rules", rt.notifyRulesGetByService)
-
-			service.GET("/notify-channels", rt.notifyChannelConfigGets)
-
-			service.GET("/message-templates", rt.messageTemplateGets)
-
-			service.GET("/event-pipelines", rt.eventPipelinesListByService)
-		}
-	}
-
-	if rt.HTTP.APIForAgent.Enable {
-		heartbeat := r.Group("/v1/n9e")
-		{
-			if len(rt.HTTP.APIForAgent.BasicAuth) > 0 {
-				heartbeat.Use(gin.BasicAuth(rt.HTTP.APIForAgent.BasicAuth))
-			}
-			heartbeat.POST("/heartbeat", rt.heartbeat)
-		}
-	}
-
-	rt.configNoRoute(r, &statikFS)
-
-}
-
-func Render(c *gin.Context, data, msg interface{}) {
-	if msg == nil {
-		if data == nil {
-			data = struct{}{}
-		}
-		c.JSON(http.StatusOK, gin.H{"data": data, "error": ""})
-	} else {
-		c.JSON(http.StatusOK, gin.H{"error": gin.H{"message": msg}})
-	}
-}
-
-func Dangerous(c *gin.Context, v interface{}, code ...int) {
-	if v == nil {
-		return
-	}
-
-	switch t := v.(type) {
-	case string:
-		if t != "" {
-			c.JSON(http.StatusOK, gin.H{"error": v})
-		}
-	case error:
-		c.JSON(http.StatusOK, gin.H{"error": t.Error()})
-	}
-}

center/router/router_alert_aggr_view.go

@@ -1,79 +0,0 @@
-package router
-
-import (
-	"net/http"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-// no param
-func (rt *Router) alertAggrViewGets(c *gin.Context) {
-	lst, err := models.AlertAggrViewGets(rt.Ctx, c.MustGet("userid"))
-	ginx.NewRender(c).Data(lst, err)
-}
-
-// body: name, rule, cate
-func (rt *Router) alertAggrViewAdd(c *gin.Context) {
-	var f models.AlertAggrView
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	if !me.IsAdmin() {
-		// 管理员可以选择当前这个视图是公开呢，还是私有，普通用户的话就只能是私有的
-		f.Cate = 1
-	}
-
-	f.Id = 0
-	f.CreateBy = me.Id
-	ginx.Dangerous(f.Add(rt.Ctx))
-
-	ginx.NewRender(c).Data(f, nil)
-}
-
-// body: ids
-func (rt *Router) alertAggrViewDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	me := c.MustGet("user").(*models.User)
-	if me.IsAdmin() {
-		ginx.NewRender(c).Message(models.AlertAggrViewDel(rt.Ctx, f.Ids))
-	} else {
-		ginx.NewRender(c).Message(models.AlertAggrViewDel(rt.Ctx, f.Ids, me.Id))
-	}
-}
-
-// body: id, name, rule, cate
-func (rt *Router) alertAggrViewPut(c *gin.Context) {
-	var f models.AlertAggrView
-	ginx.BindJSON(c, &f)
-
-	view, err := models.AlertAggrViewGet(rt.Ctx, "id = ?", f.Id)
-	ginx.Dangerous(err)
-
-	if view == nil {
-		ginx.NewRender(c).Message("no such item(id: %d)", f.Id)
-		return
-	}
-
-	me := c.MustGet("user").(*models.User)
-	if !me.IsAdmin() {
-		f.Cate = 1
-
-		if view.CreateBy != me.Id {
-			ginx.NewRender(c, http.StatusForbidden).Message("forbidden")
-			return
-		}
-	}
-	view.Name = f.Name
-	view.Rule = f.Rule
-	view.Cate = f.Cate
-	if view.CreateBy == 0 {
-		view.CreateBy = me.Id
-	}
-	ginx.NewRender(c).Message(view.Update(rt.Ctx))
-}

center/router/router_alert_cur_event.go

@@ -1,307 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"net/http"
-	"sort"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func getUserGroupIds(ctx *gin.Context, rt *Router, myGroups bool) ([]int64, error) {
-	if !myGroups {
-		return nil, nil
-	}
-	me := ctx.MustGet("user").(*models.User)
-	return models.MyGroupIds(rt.Ctx, me.Id)
-}
-
-func (rt *Router) alertCurEventsCard(c *gin.Context) {
-	stime, etime := getTimeRange(c)
-	severity := strx.IdsInt64ForAPI(ginx.QueryStr(c, "severity", ""), ",")
-	query := ginx.QueryStr(c, "query", "")
-	myGroups := ginx.QueryBool(c, "my_groups", false) // 是否只看自己组，默认false
-
-	var gids []int64
-	var err error
-	if myGroups {
-		gids, err = getUserGroupIds(c, rt, myGroups)
-		ginx.Dangerous(err)
-		if len(gids) == 0 {
-			gids = append(gids, -1)
-		}
-	}
-
-	viewId := ginx.QueryInt64(c, "view_id")
-
-	alertView, err := models.GetAlertAggrViewByViewID(rt.Ctx, viewId)
-	ginx.Dangerous(err)
-
-	if alertView == nil {
-		ginx.Bomb(http.StatusNotFound, "alert aggr view not found")
-	}
-
-	dsIds := queryDatasourceIds(c)
-
-	prod := ginx.QueryStr(c, "prods", "")
-	if prod == "" {
-		prod = ginx.QueryStr(c, "rule_prods", "")
-	}
-	prods := []string{}
-	if prod != "" {
-		prods = strings.Split(prod, ",")
-	}
-
-	cate := ginx.QueryStr(c, "cate", "$all")
-	cates := []string{}
-	if cate != "$all" {
-		cates = strings.Split(cate, ",")
-	}
-
-	bgids, err := GetBusinessGroupIds(c, rt.Ctx, rt.Center.EventHistoryGroupView, myGroups)
-	ginx.Dangerous(err)
-
-	// 最多获取50000个，获取太多也没啥意义
-	list, err := models.AlertCurEventsGet(rt.Ctx, prods, bgids, stime, etime, severity, dsIds,
-		cates, 0, query, 50000, 0, []int64{})
-	ginx.Dangerous(err)
-
-	cardmap := make(map[string]*AlertCard)
-	for _, event := range list {
-		title, err := event.GenCardTitle(alertView.Rule)
-		ginx.Dangerous(err)
-		if _, has := cardmap[title]; has {
-			cardmap[title].Total++
-			cardmap[title].EventIds = append(cardmap[title].EventIds, event.Id)
-			if event.Severity < cardmap[title].Severity {
-				cardmap[title].Severity = event.Severity
-			}
-		} else {
-			cardmap[title] = &AlertCard{
-				Total:    1,
-				EventIds: []int64{event.Id},
-				Title:    title,
-				Severity: event.Severity,
-			}
-		}
-
-		if cardmap[title].Severity < 1 {
-			cardmap[title].Severity = 3
-		}
-	}
-
-	titles := make([]string, 0, len(cardmap))
-	for title := range cardmap {
-		titles = append(titles, title)
-	}
-
-	sort.Strings(titles)
-
-	cards := make([]*AlertCard, len(titles))
-	for i := 0; i < len(titles); i++ {
-		cards[i] = cardmap[titles[i]]
-	}
-
-	sort.SliceStable(cards, func(i, j int) bool {
-		if cards[i].Severity != cards[j].Severity {
-			return cards[i].Severity < cards[j].Severity
-		}
-		return cards[i].Total > cards[j].Total
-	})
-
-	ginx.NewRender(c).Data(cards, nil)
-}
-
-type AlertCard struct {
-	Title    string  `json:"title"`
-	Total    int     `json:"total"`
-	EventIds []int64 `json:"event_ids"`
-	Severity int     `json:"severity"`
-}
-
-func (rt *Router) alertCurEventsCardDetails(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-
-	list, err := models.AlertCurEventGetByIds(rt.Ctx, f.Ids)
-	if err == nil {
-		cache := make(map[int64]*models.UserGroup)
-		for i := 0; i < len(list); i++ {
-			list[i].FillNotifyGroups(rt.Ctx, cache)
-		}
-	}
-
-	ginx.NewRender(c).Data(list, err)
-}
-
-// alertCurEventsGetByRid
-func (rt *Router) alertCurEventsGetByRid(c *gin.Context) {
-	rid := ginx.QueryInt64(c, "rid")
-	dsId := ginx.QueryInt64(c, "dsid")
-	ginx.NewRender(c).Data(models.AlertCurEventGetByRuleIdAndDsId(rt.Ctx, rid, dsId))
-}
-
-// 列表方式，拉取活跃告警
-func (rt *Router) alertCurEventsList(c *gin.Context) {
-	stime, etime := getTimeRange(c)
-	severity := strx.IdsInt64ForAPI(ginx.QueryStr(c, "severity", ""), ",")
-	query := ginx.QueryStr(c, "query", "")
-	limit := ginx.QueryInt(c, "limit", 20)
-	myGroups := ginx.QueryBool(c, "my_groups", false) // 是否只看自己组，默认false
-
-	dsIds := queryDatasourceIds(c)
-
-	eventIds := strx.IdsInt64ForAPI(ginx.QueryStr(c, "event_ids", ""), ",")
-
-	prod := ginx.QueryStr(c, "prods", "")
-	if prod == "" {
-		prod = ginx.QueryStr(c, "rule_prods", "")
-	}
-
-	prods := []string{}
-	if prod != "" {
-		prods = strings.Split(prod, ",")
-	}
-
-	cate := ginx.QueryStr(c, "cate", "$all")
-	cates := []string{}
-	if cate != "$all" {
-		cates = strings.Split(cate, ",")
-	}
-
-	ruleId := ginx.QueryInt64(c, "rid", 0)
-
-	bgids, err := GetBusinessGroupIds(c, rt.Ctx, rt.Center.EventHistoryGroupView, myGroups)
-	ginx.Dangerous(err)
-
-	total, err := models.AlertCurEventTotal(rt.Ctx, prods, bgids, stime, etime, severity, dsIds,
-		cates, ruleId, query, eventIds)
-	ginx.Dangerous(err)
-
-	list, err := models.AlertCurEventsGet(rt.Ctx, prods, bgids, stime, etime, severity, dsIds,
-		cates, ruleId, query, limit, ginx.Offset(c, limit), eventIds)
-	ginx.Dangerous(err)
-
-	cache := make(map[int64]*models.UserGroup)
-
-	for i := 0; i < len(list); i++ {
-		list[i].FillNotifyGroups(rt.Ctx, cache)
-	}
-
-	ginx.NewRender(c).Data(gin.H{
-		"list":  list,
-		"total": total,
-	}, nil)
-}
-
-func (rt *Router) alertCurEventDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	rt.checkCurEventBusiGroupRWPermission(c, f.Ids)
-
-	ginx.NewRender(c).Message(models.AlertCurEventDel(rt.Ctx, f.Ids))
-}
-
-func (rt *Router) checkCurEventBusiGroupRWPermission(c *gin.Context, ids []int64) {
-	set := make(map[int64]struct{})
-
-	// event group id is 0, ignore perm check
-	set[0] = struct{}{}
-
-	for i := 0; i < len(ids); i++ {
-		event, err := models.AlertCurEventGetById(rt.Ctx, ids[i])
-		ginx.Dangerous(err)
-		if event == nil {
-			continue
-		}
-		if _, has := set[event.GroupId]; !has {
-			rt.bgrwCheck(c, event.GroupId)
-			set[event.GroupId] = struct{}{}
-		}
-	}
-}
-
-func (rt *Router) alertCurEventGet(c *gin.Context) {
-	eid := ginx.UrlParamInt64(c, "eid")
-	event, err := GetCurEventDetail(rt.Ctx, eid)
-
-	hasPermission := HasPermission(rt.Ctx, c, "event", fmt.Sprintf("%d", eid), rt.Center.AnonymousAccess.AlertDetail)
-	if !hasPermission {
-		rt.auth()(c)
-		rt.user()(c)
-		rt.bgroCheck(c, event.GroupId)
-	}
-
-	ginx.NewRender(c).Data(event, err)
-}
-
-func GetCurEventDetail(ctx *ctx.Context, eid int64) (*models.AlertCurEvent, error) {
-	event, err := models.AlertCurEventGetById(ctx, eid)
-	if err != nil {
-		return nil, err
-	}
-
-	if event == nil {
-		return nil, fmt.Errorf("no such active event")
-	}
-
-	ruleConfig, needReset := models.FillRuleConfigTplName(ctx, event.RuleConfig)
-	if needReset {
-		event.RuleConfigJson = ruleConfig
-	}
-
-	event.LastEvalTime = event.TriggerTime
-	event.NotifyVersion, err = GetEventNotifyVersion(ctx, event.RuleId, event.NotifyRuleIds)
-	ginx.Dangerous(err)
-
-	event.NotifyRules, err = GetEventNorifyRuleNames(ctx, event.NotifyRuleIds)
-	return event, err
-}
-
-func GetEventNorifyRuleNames(ctx *ctx.Context, notifyRuleIds []int64) ([]*models.EventNotifyRule, error) {
-	notifyRuleNames := make([]*models.EventNotifyRule, 0)
-	notifyRules, err := models.NotifyRulesGet(ctx, "id in ?", notifyRuleIds)
-	if err != nil {
-		return nil, err
-	}
-
-	for _, notifyRule := range notifyRules {
-		notifyRuleNames = append(notifyRuleNames, &models.EventNotifyRule{
-			Id:   notifyRule.ID,
-			Name: notifyRule.Name,
-		})
-	}
-	return notifyRuleNames, nil
-}
-
-func GetEventNotifyVersion(ctx *ctx.Context, ruleId int64, notifyRuleIds []int64) (int, error) {
-	if len(notifyRuleIds) != 0 {
-		// 如果存在 notify_rule_ids，则认为使用新的告警通知方式
-		return 1, nil
-	}
-
-	rule, err := models.AlertRuleGetById(ctx, ruleId)
-	if err != nil {
-		return 0, err
-	}
-	return rule.NotifyVersion, nil
-}
-
-func (rt *Router) alertCurEventsStatistics(c *gin.Context) {
-
-	ginx.NewRender(c).Data(models.AlertCurEventStatistics(rt.Ctx, time.Now()), nil)
-}
-
-func (rt *Router) alertCurEventDelByHash(c *gin.Context) {
-	hash := ginx.QueryStr(c, "hash")
-	ginx.NewRender(c).Message(models.AlertCurEventDelByHash(rt.Ctx, hash))
-}

center/router/router_alert_his_event.go

@@ -1,191 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"net/http"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-	"golang.org/x/exp/slices"
-)
-
-func getTimeRange(c *gin.Context) (stime, etime int64) {
-	stime = ginx.QueryInt64(c, "stime", 0)
-	etime = ginx.QueryInt64(c, "etime", 0)
-	hours := ginx.QueryInt64(c, "hours", 0)
-	now := time.Now().Unix()
-	if hours != 0 {
-		stime = now - 3600*hours
-		etime = now + 3600*24
-	}
-
-	if stime != 0 && etime == 0 {
-		etime = now + 3600*24
-	}
-	return
-}
-
-func (rt *Router) alertHisEventsList(c *gin.Context) {
-	stime, etime := getTimeRange(c)
-
-	severity := ginx.QueryInt(c, "severity", -1)
-	recovered := ginx.QueryInt(c, "is_recovered", -1)
-	query := ginx.QueryStr(c, "query", "")
-	limit := ginx.QueryInt(c, "limit", 20)
-	dsIds := queryDatasourceIds(c)
-
-	prod := ginx.QueryStr(c, "prods", "")
-	if prod == "" {
-		prod = ginx.QueryStr(c, "rule_prods", "")
-	}
-
-	prods := []string{}
-	if prod != "" {
-		prods = strings.Split(prod, ",")
-	}
-
-	cate := ginx.QueryStr(c, "cate", "$all")
-	cates := []string{}
-	if cate != "$all" {
-		cates = strings.Split(cate, ",")
-	}
-
-	ruleId := ginx.QueryInt64(c, "rid", 0)
-
-	bgids, err := GetBusinessGroupIds(c, rt.Ctx, rt.Center.EventHistoryGroupView, false)
-	ginx.Dangerous(err)
-
-	total, err := models.AlertHisEventTotal(rt.Ctx, prods, bgids, stime, etime, severity,
-		recovered, dsIds, cates, ruleId, query)
-	ginx.Dangerous(err)
-
-	list, err := models.AlertHisEventGets(rt.Ctx, prods, bgids, stime, etime, severity, recovered,
-		dsIds, cates, ruleId, query, limit, ginx.Offset(c, limit))
-	ginx.Dangerous(err)
-
-	cache := make(map[int64]*models.UserGroup)
-	for i := 0; i < len(list); i++ {
-		list[i].FillNotifyGroups(rt.Ctx, cache)
-	}
-
-	ginx.NewRender(c).Data(gin.H{
-		"list":  list,
-		"total": total,
-	}, nil)
-}
-
-type alertHisEventsDeleteForm struct {
-	Severities []int `json:"severities"`
-	Timestamp  int64 `json:"timestamp" binding:"required"`
-}
-
-func (rt *Router) alertHisEventsDelete(c *gin.Context) {
-	var f alertHisEventsDeleteForm
-	ginx.BindJSON(c, &f)
-	// 校验
-	if f.Timestamp == 0 {
-		ginx.Bomb(http.StatusBadRequest, "timestamp parameter is required")
-		return
-	}
-
-	user := c.MustGet("user").(*models.User)
-
-	// 启动后台清理任务
-	go func() {
-		limit := 100
-		for {
-			n, err := models.AlertHisEventBatchDelete(rt.Ctx, f.Timestamp, f.Severities, limit)
-			if err != nil {
-				logger.Errorf("Failed to delete alert history events: operator=%s, timestamp=%d, severities=%v, error=%v",
-					user.Username, f.Timestamp, f.Severities, err)
-				break
-			}
-			logger.Debugf("Successfully deleted alert history events: operator=%s, timestamp=%d, severities=%v, deleted=%d",
-				user.Username, f.Timestamp, f.Severities, n)
-			if n < int64(limit) {
-				break // 已经删完
-			}
-
-			time.Sleep(100 * time.Millisecond) // 防止锁表
-		}
-	}()
-	ginx.NewRender(c).Message("Alert history events deletion started")
-}
-
-func (rt *Router) alertHisEventGet(c *gin.Context) {
-	eid := ginx.UrlParamInt64(c, "eid")
-	event, err := models.AlertHisEventGetById(rt.Ctx, eid)
-	ginx.Dangerous(err)
-	if event == nil {
-		ginx.Bomb(404, "No such alert event")
-	}
-
-	hasPermission := HasPermission(rt.Ctx, c, "event", fmt.Sprintf("%d", eid), rt.Center.AnonymousAccess.AlertDetail)
-	if !hasPermission {
-		rt.auth()(c)
-		rt.user()(c)
-		rt.bgroCheck(c, event.GroupId)
-	}
-
-	ruleConfig, needReset := models.FillRuleConfigTplName(rt.Ctx, event.RuleConfig)
-	if needReset {
-		event.RuleConfigJson = ruleConfig
-	}
-
-	event.NotifyVersion, err = GetEventNotifyVersion(rt.Ctx, event.RuleId, event.NotifyRuleIds)
-	ginx.Dangerous(err)
-
-	event.NotifyRules, err = GetEventNorifyRuleNames(rt.Ctx, event.NotifyRuleIds)
-	ginx.NewRender(c).Data(event, err)
-}
-
-func GetBusinessGroupIds(c *gin.Context, ctx *ctx.Context, onlySelfGroupView bool, myGroups bool) ([]int64, error) {
-	bgid := ginx.QueryInt64(c, "bgid", 0)
-	var bgids []int64
-
-	if strings.HasPrefix(c.Request.URL.Path, "/v1") {
-		// 如果请求路径以 /v1 开头，不查询用户信息
-		if bgid > 0 {
-			return []int64{bgid}, nil
-		}
-
-		return bgids, nil
-	}
-
-	user := c.MustGet("user").(*models.User)
-	if myGroups || (onlySelfGroupView && !user.IsAdmin()) {
-		// 1. 页面上勾选了我的业务组，需要查询用户所属的业务组
-		// 2. 如果 onlySelfGroupView 为 true，表示只允许查询用户所属的业务组
-		bussGroupIds, err := models.MyBusiGroupIds(ctx, user.Id)
-		if err != nil {
-			return nil, err
-		}
-
-		if len(bussGroupIds) == 0 {
-			// 如果没查到用户属于任何业务组，需要返回一个0，否则会导致查询到全部告警历史
-			return []int64{0}, nil
-		}
-
-		if bgid > 0 {
-			if !slices.Contains(bussGroupIds, bgid) && !user.IsAdmin() {
-				return nil, fmt.Errorf("business group ID not allowed")
-			}
-
-			return []int64{bgid}, nil
-		}
-
-		return bussGroupIds, nil
-	}
-
-	if bgid > 0 {
-		return []int64{bgid}, nil
-	}
-
-	return bgids, nil
-}

center/router/router_alert_rule.go

@@ -1,735 +0,0 @@
-package router
-
-import (
-	"encoding/json"
-	"fmt"
-	"net/http"
-	"regexp"
-	"strconv"
-	"strings"
-	"time"
-
-	"gopkg.in/yaml.v2"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-	"github.com/ccfos/nightingale/v6/pushgw/pconf"
-	"github.com/ccfos/nightingale/v6/pushgw/writer"
-
-	"github.com/gin-gonic/gin"
-	"github.com/jinzhu/copier"
-	"github.com/prometheus/prometheus/prompb"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/i18n"
-)
-
-type AlertRuleModifyHookFunc func(ar *models.AlertRule)
-
-// Return all, front-end search and paging
-func (rt *Router) alertRuleGets(c *gin.Context) {
-	busiGroupId := ginx.UrlParamInt64(c, "id")
-	ars, err := models.AlertRuleGets(rt.Ctx, busiGroupId)
-	if err == nil {
-		cache := make(map[int64]*models.UserGroup)
-		for i := 0; i < len(ars); i++ {
-			ars[i].FillNotifyGroups(rt.Ctx, cache)
-			ars[i].FillSeverities()
-		}
-	}
-	ginx.NewRender(c).Data(ars, err)
-}
-
-func getAlertCueEventTimeRange(c *gin.Context) (stime, etime int64) {
-	stime = ginx.QueryInt64(c, "stime", 0)
-	etime = ginx.QueryInt64(c, "etime", 0)
-	if etime == 0 {
-		etime = time.Now().Unix()
-	}
-	if stime == 0 || stime >= etime {
-		stime = etime - 30*24*int64(time.Hour.Seconds())
-	}
-	return
-}
-
-func (rt *Router) alertRuleGetsByGids(c *gin.Context) {
-	gids := strx.IdsInt64ForAPI(ginx.QueryStr(c, "gids", ""), ",")
-	if len(gids) > 0 {
-		for _, gid := range gids {
-			rt.bgroCheck(c, gid)
-		}
-	} else {
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			var err error
-			gids, err = models.MyBusiGroupIds(rt.Ctx, me.Id)
-			ginx.Dangerous(err)
-
-			if len(gids) == 0 {
-				ginx.NewRender(c).Data([]int{}, nil)
-				return
-			}
-		}
-	}
-
-	ars, err := models.AlertRuleGetsByBGIds(rt.Ctx, gids)
-	if err == nil {
-		cache := make(map[int64]*models.UserGroup)
-		rids := make([]int64, 0, len(ars))
-		names := make([]string, 0, len(ars))
-		for i := 0; i < len(ars); i++ {
-			ars[i].FillNotifyGroups(rt.Ctx, cache)
-			ars[i].FillSeverities()
-
-			if len(ars[i].DatasourceQueries) != 0 {
-				ars[i].DatasourceIdsJson = rt.DatasourceCache.GetIDsByDsCateAndQueries(ars[i].Cate, ars[i].DatasourceQueries)
-			}
-
-			rids = append(rids, ars[i].Id)
-			names = append(names, ars[i].UpdateBy)
-		}
-
-		stime, etime := getAlertCueEventTimeRange(c)
-		cnt := models.AlertCurEventCountByRuleId(rt.Ctx, rids, stime, etime)
-		if cnt != nil {
-			for i := 0; i < len(ars); i++ {
-				ars[i].CurEventCount = cnt[ars[i].Id]
-			}
-		}
-
-		users := models.UserMapGet(rt.Ctx, "username in (?)", names)
-		if users != nil {
-			for i := 0; i < len(ars); i++ {
-				if user, exist := users[ars[i].UpdateBy]; exist {
-					ars[i].UpdateByNickname = user.Nickname
-				}
-			}
-		}
-	}
-	ginx.NewRender(c).Data(ars, err)
-}
-
-func (rt *Router) alertRulesGetByService(c *gin.Context) {
-	prods := []string{}
-	prodStr := ginx.QueryStr(c, "prods", "")
-	if prodStr != "" {
-		prods = strings.Split(ginx.QueryStr(c, "prods", ""), ",")
-	}
-
-	query := ginx.QueryStr(c, "query", "")
-	algorithm := ginx.QueryStr(c, "algorithm", "")
-	cluster := ginx.QueryStr(c, "cluster", "")
-	cate := ginx.QueryStr(c, "cate", "$all")
-	cates := []string{}
-	if cate != "$all" {
-		cates = strings.Split(cate, ",")
-	}
-
-	disabled := ginx.QueryInt(c, "disabled", -1)
-	ars, err := models.AlertRulesGetsBy(rt.Ctx, prods, query, algorithm, cluster, cates, disabled)
-	if err == nil {
-		cache := make(map[int64]*models.UserGroup)
-		for i := 0; i < len(ars); i++ {
-			ars[i].FillNotifyGroups(rt.Ctx, cache)
-
-			if len(ars[i].DatasourceQueries) != 0 {
-				ars[i].DatasourceIdsJson = rt.DatasourceCache.GetIDsByDsCateAndQueries(ars[i].Cate, ars[i].DatasourceQueries)
-			}
-		}
-	}
-	ginx.NewRender(c).Data(ars, err)
-}
-
-// single or import
-func (rt *Router) alertRuleAddByFE(c *gin.Context) {
-	username := c.MustGet("username").(string)
-
-	var lst []models.AlertRule
-	ginx.BindJSON(c, &lst)
-
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	bgid := ginx.UrlParamInt64(c, "id")
-	reterr := rt.alertRuleAdd(lst, username, bgid, c.GetHeader("X-Language"))
-
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-func (rt *Router) alertRuleAddByImport(c *gin.Context) {
-	username := c.MustGet("username").(string)
-
-	var lst []models.AlertRule
-	ginx.BindJSON(c, &lst)
-
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	for i := range lst {
-		if len(lst[i].DatasourceQueries) == 0 {
-			lst[i].DatasourceQueries = []models.DatasourceQuery{
-				models.DataSourceQueryAll,
-			}
-		}
-	}
-
-	bgid := ginx.UrlParamInt64(c, "id")
-	reterr := rt.alertRuleAdd(lst, username, bgid, c.GetHeader("X-Language"))
-
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-type promRuleForm struct {
-	Payload           string                   `json:"payload" binding:"required"`
-	DatasourceQueries []models.DatasourceQuery `json:"datasource_queries" binding:"required"`
-	Disabled          int                      `json:"disabled" binding:"gte=0,lte=1"`
-}
-
-func (rt *Router) alertRuleAddByImportPromRule(c *gin.Context) {
-	var f promRuleForm
-	ginx.Dangerous(c.BindJSON(&f))
-
-	var pr struct {
-		Groups []models.PromRuleGroup `yaml:"groups"`
-	}
-	err := yaml.Unmarshal([]byte(f.Payload), &pr)
-	if err != nil {
-		ginx.Bomb(http.StatusBadRequest, "invalid yaml format, please use the example format. err: %v", err)
-	}
-
-	if len(pr.Groups) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input yaml is empty")
-	}
-
-	lst := models.DealPromGroup(pr.Groups, f.DatasourceQueries, f.Disabled)
-	username := c.MustGet("username").(string)
-	bgid := ginx.UrlParamInt64(c, "id")
-	ginx.NewRender(c).Data(rt.alertRuleAdd(lst, username, bgid, c.GetHeader("X-Language")), nil)
-}
-
-func (rt *Router) alertRuleAddByService(c *gin.Context) {
-	var lst []models.AlertRule
-	ginx.BindJSON(c, &lst)
-
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-	reterr := rt.alertRuleAddForService(lst, "")
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-func (rt *Router) alertRuleAddOneByService(c *gin.Context) {
-	var f models.AlertRule
-	ginx.BindJSON(c, &f)
-
-	err := f.FE2DB()
-	ginx.Dangerous(err)
-
-	err = f.Add(rt.Ctx)
-	ginx.NewRender(c).Data(f.Id, err)
-}
-
-func (rt *Router) alertRuleAddForService(lst []models.AlertRule, username string) map[string]string {
-	count := len(lst)
-	// alert rule name -> error string
-	reterr := make(map[string]string)
-	for i := 0; i < count; i++ {
-		lst[i].Id = 0
-		if username != "" {
-			lst[i].CreateBy = username
-			lst[i].UpdateBy = username
-		}
-
-		if err := lst[i].FE2DB(); err != nil {
-			reterr[lst[i].Name] = err.Error()
-			continue
-		}
-
-		if err := lst[i].Add(rt.Ctx); err != nil {
-			reterr[lst[i].Name] = err.Error()
-		} else {
-			reterr[lst[i].Name] = ""
-		}
-	}
-	return reterr
-}
-
-func (rt *Router) alertRuleAdd(lst []models.AlertRule, username string, bgid int64, lang string) map[string]string {
-	count := len(lst)
-	// alert rule name -> error string
-	reterr := make(map[string]string)
-	for i := 0; i < count; i++ {
-		lst[i].Id = 0
-		lst[i].GroupId = bgid
-		if username != "" {
-			lst[i].CreateBy = username
-			lst[i].UpdateBy = username
-		}
-
-		if err := lst[i].FE2DB(); err != nil {
-			reterr[lst[i].Name] = i18n.Sprintf(lang, err.Error())
-			continue
-		}
-
-		if err := lst[i].Add(rt.Ctx); err != nil {
-			reterr[lst[i].Name] = i18n.Sprintf(lang, err.Error())
-		} else {
-			reterr[lst[i].Name] = ""
-		}
-	}
-	return reterr
-}
-
-func (rt *Router) alertRuleDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	// param(busiGroupId) for protect
-	ginx.NewRender(c).Message(models.AlertRuleDels(rt.Ctx, f.Ids, ginx.UrlParamInt64(c, "id")))
-}
-
-func (rt *Router) alertRuleDelByService(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-	ginx.NewRender(c).Message(models.AlertRuleDels(rt.Ctx, f.Ids))
-}
-
-func (rt *Router) alertRulePutByFE(c *gin.Context) {
-	var f models.AlertRule
-	ginx.BindJSON(c, &f)
-
-	arid := ginx.UrlParamInt64(c, "arid")
-	ar, err := models.AlertRuleGetById(rt.Ctx, arid)
-	ginx.Dangerous(err)
-
-	if ar == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such AlertRule")
-		return
-	}
-
-	rt.bgrwCheck(c, ar.GroupId)
-
-	f.UpdateBy = c.MustGet("username").(string)
-	ginx.NewRender(c).Message(ar.Update(rt.Ctx, f))
-}
-
-func (rt *Router) alertRulePutByService(c *gin.Context) {
-	var f models.AlertRule
-	ginx.BindJSON(c, &f)
-
-	arid := ginx.UrlParamInt64(c, "arid")
-	ar, err := models.AlertRuleGetById(rt.Ctx, arid)
-	ginx.Dangerous(err)
-
-	if ar == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such AlertRule")
-		return
-	}
-	ginx.NewRender(c).Message(ar.Update(rt.Ctx, f))
-}
-
-type alertRuleFieldForm struct {
-	Ids    []int64                `json:"ids"`
-	Fields map[string]interface{} `json:"fields"`
-	Action string                 `json:"action"`
-}
-
-// update one field: cluster note severity disabled prom_eval_interval prom_for_duration notify_channels notify_groups notify_recovered notify_repeat_step callbacks runbook_url append_tags
-func (rt *Router) alertRulePutFields(c *gin.Context) {
-	var f alertRuleFieldForm
-	ginx.BindJSON(c, &f)
-
-	if len(f.Fields) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "fields empty")
-	}
-
-	f.Fields["update_by"] = c.MustGet("username").(string)
-	f.Fields["update_at"] = time.Now().Unix()
-
-	for i := 0; i < len(f.Ids); i++ {
-		ar, err := models.AlertRuleGetById(rt.Ctx, f.Ids[i])
-		ginx.Dangerous(err)
-
-		if ar == nil {
-			continue
-		}
-
-		if f.Action == "update_triggers" {
-			if triggers, has := f.Fields["triggers"]; has {
-				originRule := ar.RuleConfigJson.(map[string]interface{})
-				originRule["triggers"] = triggers
-				b, err := json.Marshal(originRule)
-				ginx.Dangerous(err)
-				ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"rule_config": string(b)}))
-				continue
-			}
-		}
-
-		if f.Action == "annotations_add" {
-			if annotations, has := f.Fields["annotations"]; has {
-				annotationsMap := annotations.(map[string]interface{})
-				for k, v := range annotationsMap {
-					ar.AnnotationsJSON[k] = v.(string)
-				}
-				b, err := json.Marshal(ar.AnnotationsJSON)
-				ginx.Dangerous(err)
-				ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"annotations": string(b)}))
-				continue
-			}
-		}
-
-		if f.Action == "annotations_del" {
-			if annotations, has := f.Fields["annotations"]; has {
-				annotationsKeys := annotations.(map[string]interface{})
-				for key := range annotationsKeys {
-					delete(ar.AnnotationsJSON, key)
-				}
-				b, err := json.Marshal(ar.AnnotationsJSON)
-				ginx.Dangerous(err)
-				ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"annotations": string(b)}))
-				continue
-			}
-		}
-
-		if f.Action == "callback_add" {
-			// 增加一个 callback 地址
-			if callbacks, has := f.Fields["callbacks"]; has {
-				callback := callbacks.(string)
-				if !strings.Contains(ar.Callbacks, callback) {
-					ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"callbacks": ar.Callbacks + " " + callback}))
-					continue
-				}
-			}
-		}
-
-		if f.Action == "callback_del" {
-			// 删除一个 callback 地址
-			if callbacks, has := f.Fields["callbacks"]; has {
-				callback := callbacks.(string)
-				ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"callbacks": strings.ReplaceAll(ar.Callbacks, callback, "")}))
-				continue
-			}
-		}
-
-		if f.Action == "datasource_change" {
-			// 修改数据源
-			if datasourceQueries, has := f.Fields["datasource_queries"]; has {
-				bytes, err := json.Marshal(datasourceQueries)
-				ginx.Dangerous(err)
-				ginx.Dangerous(ar.UpdateFieldsMap(rt.Ctx, map[string]interface{}{"datasource_queries": bytes}))
-				continue
-			}
-		}
-
-		for k, v := range f.Fields {
-			// 检查 v 是否为各种切片类型
-			switch v.(type) {
-			case []interface{}, []int64, []int, []string:
-				// 将切片转换为 JSON 字符串
-				bytes, err := json.Marshal(v)
-				ginx.Dangerous(err)
-				ginx.Dangerous(ar.UpdateColumn(rt.Ctx, k, string(bytes)))
-			default:
-				ginx.Dangerous(ar.UpdateColumn(rt.Ctx, k, v))
-			}
-		}
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-func (rt *Router) alertRuleGet(c *gin.Context) {
-	arid := ginx.UrlParamInt64(c, "arid")
-
-	ar, err := models.AlertRuleGetById(rt.Ctx, arid)
-	ginx.Dangerous(err)
-
-	if ar == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such AlertRule")
-		return
-	}
-
-	if len(ar.DatasourceQueries) != 0 {
-		ar.DatasourceIdsJson = rt.DatasourceCache.GetIDsByDsCateAndQueries(ar.Cate, ar.DatasourceQueries)
-	}
-
-	err = ar.FillNotifyGroups(rt.Ctx, make(map[int64]*models.UserGroup))
-	ginx.Dangerous(err)
-
-	rt.AlertRuleModifyHook(ar)
-	ginx.NewRender(c).Data(ar, err)
-}
-
-func (rt *Router) alertRulePureGet(c *gin.Context) {
-	arid := ginx.UrlParamInt64(c, "arid")
-
-	ar, err := models.AlertRuleGetById(rt.Ctx, arid)
-	ginx.Dangerous(err)
-
-	if ar == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such AlertRule")
-		return
-	}
-
-	ginx.NewRender(c).Data(ar, err)
-}
-
-// pre validation before save rule
-func (rt *Router) alertRuleValidation(c *gin.Context) {
-	var f models.AlertRule //new
-	ginx.BindJSON(c, &f)
-
-	if len(f.NotifyChannelsJSON) > 0 && len(f.NotifyGroupsJSON) > 0 { //Validation NotifyChannels
-		ngids := make([]int64, 0, len(f.NotifyChannelsJSON))
-		for i := range f.NotifyGroupsJSON {
-			id, _ := strconv.ParseInt(f.NotifyGroupsJSON[i], 10, 64)
-			ngids = append(ngids, id)
-		}
-		userGroups := rt.UserGroupCache.GetByUserGroupIds(ngids)
-		uids := make([]int64, 0)
-		for i := range userGroups {
-			uids = append(uids, userGroups[i].UserIds...)
-		}
-		users := rt.UserCache.GetByUserIds(uids)
-		//If any users have a certain notify channel's token, it will be okay. Otherwise, this notify channel is absent of tokens.
-		ancs := make([]string, 0, len(f.NotifyChannelsJSON)) //absent Notify Channels
-		for i := range f.NotifyChannelsJSON {
-			flag := true
-			//ignore non-default channels
-			switch f.NotifyChannelsJSON[i] {
-			case models.Dingtalk, models.Wecom, models.Feishu, models.Mm,
-				models.Telegram, models.Email, models.FeishuCard:
-				// do nothing
-			default:
-				continue
-			}
-			//default channels
-			for ui := range users {
-				if _, b := users[ui].ExtractToken(f.NotifyChannelsJSON[i]); b {
-					flag = false
-					break
-				}
-			}
-			if flag {
-				ancs = append(ancs, f.NotifyChannelsJSON[i])
-			}
-		}
-
-		if len(ancs) > 0 {
-			ginx.NewRender(c).Message("All users are missing notify channel configurations. Please check for missing tokens (each channel should be configured with at least one user). %s", ancs)
-			return
-		}
-
-	}
-
-	ginx.NewRender(c).Message("")
-}
-
-func (rt *Router) alertRuleCallbacks(c *gin.Context) {
-	user := c.MustGet("user").(*models.User)
-	bussGroupIds, err := models.MyBusiGroupIds(rt.Ctx, user.Id)
-	ginx.Dangerous(err)
-
-	ars, err := models.AlertRuleGetsByBGIds(rt.Ctx, bussGroupIds)
-	ginx.Dangerous(err)
-
-	var callbacks []string
-	callbackFilter := make(map[string]struct{})
-	for i := range ars {
-		for _, callback := range ars[i].CallbacksJSON {
-			if _, ok := callbackFilter[callback]; !ok {
-				callbackFilter[callback] = struct{}{}
-				callbacks = append(callbacks, callback)
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(callbacks, nil)
-}
-
-type alertRuleTestForm struct {
-	Configs []*pconf.RelabelConfig `json:"configs"`
-	Tags    []string               `json:"tags"`
-}
-
-func (rt *Router) relabelTest(c *gin.Context) {
-	var f alertRuleTestForm
-	ginx.BindJSON(c, &f)
-
-	if len(f.Tags) == 0 || len(f.Configs) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "relabel config is empty")
-	}
-
-	labels := make([]prompb.Label, len(f.Tags))
-	for i, tag := range f.Tags {
-		label := strings.SplitN(tag, "=", 2)
-		if len(label) != 2 {
-			ginx.Bomb(http.StatusBadRequest, "tag:%s format error", tag)
-		}
-
-		labels[i] = prompb.Label{Name: label[0], Value: label[1]}
-	}
-
-	for i := 0; i < len(f.Configs); i++ {
-		if f.Configs[i].Replacement == "" {
-			f.Configs[i].Replacement = "$1"
-		}
-
-		if f.Configs[i].Separator == "" {
-			f.Configs[i].Separator = ";"
-		}
-
-		if f.Configs[i].Regex == "" {
-			f.Configs[i].Regex = "(.*)"
-		}
-	}
-
-	relabels := writer.Process(labels, f.Configs...)
-
-	var tags []string
-	for _, label := range relabels {
-		tags = append(tags, fmt.Sprintf("%s=%s", label.Name, label.Value))
-	}
-
-	ginx.NewRender(c).Data(tags, nil)
-}
-
-type identListForm struct {
-	Ids       []int64  `json:"ids"`
-	IdentList []string `json:"ident_list"`
-}
-
-func containsIdentOperator(s string) bool {
-	pattern := `ident\s*(!=|!~|=~)`
-	matched, err := regexp.MatchString(pattern, s)
-	if err != nil {
-		return false
-	}
-	return matched
-}
-
-func (rt *Router) cloneToMachine(c *gin.Context) {
-	var f identListForm
-	ginx.BindJSON(c, &f)
-
-	if len(f.IdentList) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "ident_list is empty")
-	}
-
-	alertRules, err := models.AlertRuleGetsByIds(rt.Ctx, f.Ids)
-	ginx.Dangerous(err)
-
-	re := regexp.MustCompile(`ident\s*=\s*\\".*?\\"`)
-
-	user := c.MustGet("username").(string)
-	now := time.Now().Unix()
-
-	newRules := make([]*models.AlertRule, 0)
-
-	reterr := make(map[string]map[string]string)
-
-	for i := range alertRules {
-		errMsg := make(map[string]string)
-
-		if alertRules[i].Cate != "prometheus" {
-			errMsg["all"] = "Only Prometheus rule can be cloned to machines"
-			reterr[alertRules[i].Name] = errMsg
-			continue
-		}
-
-		if containsIdentOperator(alertRules[i].RuleConfig) {
-			errMsg["all"] = "promql is missing ident"
-			reterr[alertRules[i].Name] = errMsg
-			continue
-		}
-
-		for j := range f.IdentList {
-			alertRules[i].RuleConfig = re.ReplaceAllString(alertRules[i].RuleConfig, fmt.Sprintf(`ident=\"%s\"`, f.IdentList[j]))
-
-			newRule := &models.AlertRule{}
-			if err := copier.Copy(newRule, alertRules[i]); err != nil {
-				errMsg[f.IdentList[j]] = fmt.Sprintf("fail to clone rule, err: %s", err)
-				continue
-			}
-
-			newRule.Id = 0
-			newRule.Name = alertRules[i].Name + "_" + f.IdentList[j]
-			newRule.CreateBy = user
-			newRule.UpdateBy = user
-			newRule.UpdateAt = now
-			newRule.CreateAt = now
-			newRule.RuleConfig = alertRules[i].RuleConfig
-
-			exist, err := models.AlertRuleExists(rt.Ctx, 0, newRule.GroupId, newRule.Name)
-			if err != nil {
-				errMsg[f.IdentList[j]] = err.Error()
-				continue
-			}
-
-			if exist {
-				errMsg[f.IdentList[j]] = fmt.Sprintf("rule already exists, ruleName: %s", newRule.Name)
-				continue
-			}
-
-			newRules = append(newRules, newRule)
-		}
-
-		if len(errMsg) > 0 {
-			reterr[alertRules[i].Name] = errMsg
-		}
-	}
-
-	ginx.NewRender(c).Data(reterr, models.InsertAlertRule(rt.Ctx, newRules))
-}
-
-type alertBatchCloneForm struct {
-	RuleIds []int64 `json:"rule_ids"`
-	Bgids   []int64 `json:"bgids"`
-}
-
-// 批量克隆告警规则
-func (rt *Router) batchAlertRuleClone(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-
-	var f alertBatchCloneForm
-	ginx.BindJSON(c, &f)
-
-	// 校验 bgids 操作权限
-	for _, bgid := range f.Bgids {
-		rt.bgrwCheck(c, bgid)
-	}
-
-	reterr := make(map[string]string, len(f.RuleIds))
-	lang := c.GetHeader("X-Language")
-
-	for _, arid := range f.RuleIds {
-		ar, err := models.AlertRuleGetById(rt.Ctx, arid)
-		for _, bgid := range f.Bgids {
-			// 为了让 bgid 和 arid 对应，将上面的 err 放到这里处理
-			if err != nil {
-				reterr[fmt.Sprintf("%d-%d", arid, bgid)] = i18n.Sprintf(lang, err.Error())
-				continue
-			}
-
-			if ar == nil {
-				reterr[fmt.Sprintf("%d-%d", arid, bgid)] = i18n.Sprintf(lang, "alert rule not found")
-				continue
-			}
-
-			newAr := ar.Clone(me.Username, bgid)
-			err = newAr.Add(rt.Ctx)
-			if err != nil {
-				reterr[fmt.Sprintf("%d-%d", arid, bgid)] = i18n.Sprintf(lang, err.Error())
-				continue
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(reterr, nil)
-}

center/router/router_alert_subscribe.go

@@ -1,162 +0,0 @@
-package router
-
-import (
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-// Return all, front-end search and paging
-func (rt *Router) alertSubscribeGets(c *gin.Context) {
-	bgid := ginx.UrlParamInt64(c, "id")
-	lst, err := models.AlertSubscribeGets(rt.Ctx, bgid)
-	ginx.Dangerous(err)
-
-	ugcache := make(map[int64]*models.UserGroup)
-	rulecache := make(map[int64]string)
-
-	for i := 0; i < len(lst); i++ {
-		ginx.Dangerous(lst[i].FillUserGroups(rt.Ctx, ugcache))
-		ginx.Dangerous(lst[i].FillRuleNames(rt.Ctx, rulecache))
-		ginx.Dangerous(lst[i].FillDatasourceIds(rt.Ctx))
-		ginx.Dangerous(lst[i].DB2FE())
-	}
-
-	ginx.NewRender(c).Data(lst, err)
-}
-
-func (rt *Router) alertSubscribeGetsByGids(c *gin.Context) {
-	gids := strx.IdsInt64ForAPI(ginx.QueryStr(c, "gids", ""), ",")
-	if len(gids) > 0 {
-		for _, gid := range gids {
-			rt.bgroCheck(c, gid)
-		}
-	} else {
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			var err error
-			gids, err = models.MyBusiGroupIds(rt.Ctx, me.Id)
-			ginx.Dangerous(err)
-
-			if len(gids) == 0 {
-				ginx.NewRender(c).Data([]int{}, nil)
-				return
-			}
-		}
-	}
-
-	lst, err := models.AlertSubscribeGetsByBGIds(rt.Ctx, gids)
-	ginx.Dangerous(err)
-
-	ugcache := make(map[int64]*models.UserGroup)
-	rulecache := make(map[int64]string)
-
-	for i := 0; i < len(lst); i++ {
-		ginx.Dangerous(lst[i].FillUserGroups(rt.Ctx, ugcache))
-		ginx.Dangerous(lst[i].FillRuleNames(rt.Ctx, rulecache))
-		ginx.Dangerous(lst[i].FillDatasourceIds(rt.Ctx))
-		ginx.Dangerous(lst[i].DB2FE())
-	}
-
-	ginx.NewRender(c).Data(lst, err)
-}
-
-func (rt *Router) alertSubscribeGet(c *gin.Context) {
-	subid := ginx.UrlParamInt64(c, "sid")
-
-	sub, err := models.AlertSubscribeGet(rt.Ctx, "id=?", subid)
-	ginx.Dangerous(err)
-
-	if sub == nil {
-		ginx.NewRender(c, 404).Message("No such alert subscribe")
-		return
-	}
-
-	ugcache := make(map[int64]*models.UserGroup)
-	ginx.Dangerous(sub.FillUserGroups(rt.Ctx, ugcache))
-
-	rulecache := make(map[int64]string)
-	ginx.Dangerous(sub.FillRuleNames(rt.Ctx, rulecache))
-	ginx.Dangerous(sub.FillDatasourceIds(rt.Ctx))
-	ginx.Dangerous(sub.DB2FE())
-
-	ginx.NewRender(c).Data(sub, nil)
-}
-
-func (rt *Router) alertSubscribeAdd(c *gin.Context) {
-	var f models.AlertSubscribe
-	ginx.BindJSON(c, &f)
-
-	username := c.MustGet("username").(string)
-	f.CreateBy = username
-	f.UpdateBy = username
-	f.GroupId = ginx.UrlParamInt64(c, "id")
-
-	if f.GroupId <= 0 {
-		ginx.Bomb(http.StatusBadRequest, "group_id invalid")
-	}
-
-	ginx.NewRender(c).Message(f.Add(rt.Ctx))
-}
-
-func (rt *Router) alertSubscribePut(c *gin.Context) {
-	var fs []models.AlertSubscribe
-	ginx.BindJSON(c, &fs)
-
-	timestamp := time.Now().Unix()
-	username := c.MustGet("username").(string)
-	for i := 0; i < len(fs); i++ {
-		fs[i].UpdateBy = username
-		fs[i].UpdateAt = timestamp
-		//After adding the function of batch subscription alert rules, rule_ids is used instead of rule_id.
-		//When the subscription rules are updated, set rule_id=0 to prevent the wrong subscription caused by the old rule_id.
-		fs[i].RuleId = 0
-		ginx.Dangerous(fs[i].Update(
-			rt.Ctx,
-			"name",
-			"disabled",
-			"prod",
-			"cate",
-			"datasource_ids",
-			"cluster",
-			"rule_id",
-			"rule_ids",
-			"tags",
-			"redefine_severity",
-			"new_severity",
-			"redefine_channels",
-			"new_channels",
-			"user_group_ids",
-			"update_at",
-			"update_by",
-			"webhooks",
-			"for_duration",
-			"redefine_webhooks",
-			"severities",
-			"extra_config",
-			"busi_groups",
-			"note",
-			"notify_rule_ids",
-		))
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-func (rt *Router) alertSubscribeDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	ginx.NewRender(c).Message(models.AlertSubscribeDel(rt.Ctx, f.Ids))
-}
-
-func (rt *Router) alertSubscribeGetsByService(c *gin.Context) {
-	lst, err := models.AlertSubscribeGetsByService(rt.Ctx)
-	ginx.NewRender(c).Data(lst, err)
-}

center/router/router_board.go

@@ -1,365 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/i18n"
-)
-
-type boardForm struct {
-	Name       string  `json:"name"`
-	Ident      string  `json:"ident"`
-	Tags       string  `json:"tags"`
-	Configs    string  `json:"configs"`
-	Public     int     `json:"public"`
-	PublicCate int     `json:"public_cate"`
-	Bgids      []int64 `json:"bgids"`
-}
-
-func (rt *Router) boardAdd(c *gin.Context) {
-	var f boardForm
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-
-	board := &models.Board{
-		GroupId:  ginx.UrlParamInt64(c, "id"),
-		Name:     f.Name,
-		Ident:    f.Ident,
-		Tags:     f.Tags,
-		Configs:  f.Configs,
-		CreateBy: me.Username,
-		UpdateBy: me.Username,
-	}
-
-	err := board.Add(rt.Ctx)
-	ginx.Dangerous(err)
-
-	if f.Configs != "" {
-		ginx.Dangerous(models.BoardPayloadSave(rt.Ctx, board.Id, f.Configs))
-	}
-
-	ginx.NewRender(c).Data(board, nil)
-}
-
-func (rt *Router) boardGet(c *gin.Context) {
-	bid := ginx.UrlParamStr(c, "bid")
-	board, err := models.BoardGet(rt.Ctx, "ident = ?", bid)
-	ginx.Dangerous(err)
-
-	if board == nil {
-		board, err = models.BoardGet(rt.Ctx, "id = ?", bid)
-		ginx.Dangerous(err)
-	}
-
-	if board == nil {
-		ginx.Bomb(http.StatusNotFound, "No such dashboard")
-	}
-
-	if board.Public == 0 {
-		rt.auth()(c)
-		rt.user()(c)
-
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			// check permission
-			rt.bgroCheck(c, board.GroupId)
-		}
-	}
-
-	if board.PublicCate == models.PublicLogin {
-		rt.auth()(c)
-	} else if board.PublicCate == models.PublicBusi {
-		rt.auth()(c)
-		rt.user()(c)
-
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			bgids, err := models.MyBusiGroupIds(rt.Ctx, me.Id)
-			ginx.Dangerous(err)
-			if len(bgids) == 0 {
-				ginx.Bomb(http.StatusForbidden, "forbidden")
-			}
-
-			ok, err := models.BoardBusigroupCheck(rt.Ctx, board.Id, bgids)
-			ginx.Dangerous(err)
-			if !ok {
-				ginx.Bomb(http.StatusForbidden, "forbidden")
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(board, nil)
-}
-
-// 根据 bids 参数，获取多个 board
-func (rt *Router) boardGetsByBids(c *gin.Context) {
-	bids := strx.IdsInt64ForAPI(ginx.QueryStr(c, "bids", ""), ",")
-	boards, err := models.BoardGetsByBids(rt.Ctx, bids)
-	ginx.Dangerous(err)
-	ginx.NewRender(c).Data(boards, err)
-}
-
-func (rt *Router) boardPureGet(c *gin.Context) {
-	board, err := models.BoardGetByID(rt.Ctx, ginx.UrlParamInt64(c, "bid"))
-	ginx.Dangerous(err)
-
-	if board == nil {
-		ginx.Bomb(http.StatusNotFound, "No such dashboard")
-	}
-
-	ginx.NewRender(c).Data(board, nil)
-}
-
-// bgrwCheck
-func (rt *Router) boardDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	for i := 0; i < len(f.Ids); i++ {
-		bid := f.Ids[i]
-
-		board, err := models.BoardGet(rt.Ctx, "id = ?", bid)
-		ginx.Dangerous(err)
-
-		if board == nil {
-			continue
-		}
-
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			// check permission
-			rt.bgrwCheck(c, board.GroupId)
-		}
-
-		ginx.Dangerous(board.Del(rt.Ctx))
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-func (rt *Router) Board(id int64) *models.Board {
-	obj, err := models.BoardGet(rt.Ctx, "id=?", id)
-	ginx.Dangerous(err)
-
-	if obj == nil {
-		ginx.Bomb(http.StatusNotFound, "No such dashboard")
-	}
-
-	return obj
-}
-
-// bgrwCheck
-func (rt *Router) boardPut(c *gin.Context) {
-	var f boardForm
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	bo := rt.Board(ginx.UrlParamInt64(c, "bid"))
-
-	if !me.IsAdmin() {
-		// check permission
-		rt.bgrwCheck(c, bo.GroupId)
-	}
-
-	can, err := bo.CanRenameIdent(rt.Ctx, f.Ident)
-	ginx.Dangerous(err)
-
-	if !can {
-		ginx.Bomb(http.StatusOK, "Ident duplicate")
-	}
-
-	bo.Name = f.Name
-	bo.Ident = f.Ident
-	bo.Tags = f.Tags
-	bo.UpdateBy = me.Username
-	bo.UpdateAt = time.Now().Unix()
-
-	err = bo.Update(rt.Ctx, "name", "ident", "tags", "update_by", "update_at")
-	ginx.NewRender(c).Data(bo, err)
-}
-
-// bgrwCheck
-func (rt *Router) boardPutConfigs(c *gin.Context) {
-	var f boardForm
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-
-	bid := ginx.UrlParamStr(c, "bid")
-	bo, err := models.BoardGet(rt.Ctx, "id = ? or ident = ?", bid, bid)
-	ginx.Dangerous(err)
-
-	if bo == nil {
-		ginx.Bomb(http.StatusNotFound, "No such dashboard")
-	}
-
-	// check permission
-	if !me.IsAdmin() {
-		rt.bgrwCheck(c, bo.GroupId)
-	}
-
-	bo.UpdateBy = me.Username
-	bo.UpdateAt = time.Now().Unix()
-	ginx.Dangerous(bo.Update(rt.Ctx, "update_by", "update_at"))
-
-	bo.Configs = f.Configs
-	ginx.Dangerous(models.BoardPayloadSave(rt.Ctx, bo.Id, f.Configs))
-
-	ginx.NewRender(c).Data(bo, nil)
-}
-
-// bgrwCheck
-func (rt *Router) boardPutPublic(c *gin.Context) {
-	var f boardForm
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	bo := rt.Board(ginx.UrlParamInt64(c, "bid"))
-
-	// check permission
-	if !me.IsAdmin() {
-		rt.bgrwCheck(c, bo.GroupId)
-	}
-
-	bo.Public = f.Public
-	bo.PublicCate = f.PublicCate
-
-	if bo.PublicCate == models.PublicBusi {
-		err := models.BoardBusigroupUpdate(rt.Ctx, bo.Id, f.Bgids)
-		ginx.Dangerous(err)
-	} else {
-		err := models.BoardBusigroupDelByBoardId(rt.Ctx, bo.Id)
-		ginx.Dangerous(err)
-	}
-
-	bo.UpdateBy = me.Username
-	bo.UpdateAt = time.Now().Unix()
-
-	err := bo.Update(rt.Ctx, "public", "public_cate", "update_by", "update_at")
-	ginx.NewRender(c).Data(bo, err)
-}
-
-func (rt *Router) boardGets(c *gin.Context) {
-	bgid := ginx.UrlParamInt64(c, "id")
-	query := ginx.QueryStr(c, "query", "")
-
-	boards, err := models.BoardGetsByGroupId(rt.Ctx, bgid, query)
-	ginx.NewRender(c).Data(boards, err)
-}
-
-func (rt *Router) publicBoardGets(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-
-	bgids, err := models.MyBusiGroupIds(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-
-	boardIds, err := models.BoardIdsByBusiGroupIds(rt.Ctx, bgids)
-	ginx.Dangerous(err)
-
-	boards, err := models.BoardGets(rt.Ctx, "", "public=1 and (public_cate in (?) or id in (?))", []int64{0, 1}, boardIds)
-	ginx.NewRender(c).Data(boards, err)
-}
-
-func (rt *Router) boardGetsByGids(c *gin.Context) {
-	gids := strx.IdsInt64ForAPI(ginx.QueryStr(c, "gids", ""), ",")
-	query := ginx.QueryStr(c, "query", "")
-
-	if len(gids) > 0 {
-		for _, gid := range gids {
-			rt.bgroCheck(c, gid)
-		}
-	} else {
-		me := c.MustGet("user").(*models.User)
-		if !me.IsAdmin() {
-			var err error
-			gids, err = models.MyBusiGroupIds(rt.Ctx, me.Id)
-			ginx.Dangerous(err)
-
-			if len(gids) == 0 {
-				ginx.NewRender(c).Data([]int{}, nil)
-				return
-			}
-		}
-	}
-
-	boardBusigroups, err := models.BoardBusigroupGets(rt.Ctx)
-	ginx.Dangerous(err)
-	m := make(map[int64][]int64)
-	for _, boardBusigroup := range boardBusigroups {
-		m[boardBusigroup.BoardId] = append(m[boardBusigroup.BoardId], boardBusigroup.BusiGroupId)
-	}
-
-	boards, err := models.BoardGetsByBGIds(rt.Ctx, gids, query)
-	ginx.Dangerous(err)
-	for i := 0; i < len(boards); i++ {
-		if ids, ok := m[boards[i].Id]; ok {
-			boards[i].Bgids = ids
-		}
-	}
-
-	ginx.NewRender(c).Data(boards, err)
-}
-
-func (rt *Router) boardClone(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-	bo := rt.Board(ginx.UrlParamInt64(c, "bid"))
-
-	newBoard := bo.Clone(me.Username, bo.GroupId, " Cloned")
-
-	ginx.Dangerous(newBoard.Add(rt.Ctx))
-
-	// clone payload
-	payload, err := models.BoardPayloadGet(rt.Ctx, bo.Id)
-	ginx.Dangerous(err)
-
-	if payload != "" {
-		ginx.Dangerous(models.BoardPayloadSave(rt.Ctx, newBoard.Id, payload))
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-type boardsForm struct {
-	BoardIds []int64 `json:"board_ids"`
-	Bgids    []int64 `json:"bgids"`
-}
-
-func (rt *Router) boardBatchClone(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-	var f boardsForm
-	ginx.BindJSON(c, &f)
-
-	for _, bgid := range f.Bgids {
-		rt.bgrwCheck(c, bgid)
-	}
-
-	reterr := make(map[string]string, len(f.BoardIds))
-	lang := c.GetHeader("X-Language")
-
-	for _, bgid := range f.Bgids {
-		for _, bid := range f.BoardIds {
-			bo := rt.Board(bid)
-			newBoard := bo.Clone(me.Username, bgid, "")
-			payload, err := models.BoardPayloadGet(rt.Ctx, bo.Id)
-			if err != nil {
-				reterr[fmt.Sprintf("%s-%d", newBoard.Name, bgid)] = i18n.Sprintf(lang, err.Error())
-				continue
-			}
-
-			if err = newBoard.AtomicAdd(rt.Ctx, payload); err != nil {
-				reterr[fmt.Sprintf("%s-%d", newBoard.Name, bgid)] = i18n.Sprintf(lang, err.Error())
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(reterr, nil)
-}

center/router/router_builtin.go

@@ -1,340 +0,0 @@
-package router
-
-import (
-	"encoding/json"
-	"fmt"
-	"net/http"
-	"path"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/file"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-	"github.com/toolkits/pkg/runner"
-)
-
-// 创建 builtin_cate
-func (rt *Router) builtinCateFavoriteAdd(c *gin.Context) {
-	var f models.BuiltinCate
-	ginx.BindJSON(c, &f)
-
-	if f.Name == "" {
-		ginx.Bomb(http.StatusBadRequest, "name is empty")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	f.UserId = me.Id
-
-	ginx.NewRender(c).Message(f.Create(rt.Ctx))
-}
-
-// 删除 builtin_cate
-func (rt *Router) builtinCateFavoriteDel(c *gin.Context) {
-	name := ginx.UrlParamStr(c, "name")
-	me := c.MustGet("user").(*models.User)
-
-	ginx.NewRender(c).Message(models.BuiltinCateDelete(rt.Ctx, name, me.Id))
-}
-
-type Payload struct {
-	Cate    string      `json:"cate"`
-	Fname   string      `json:"fname"`
-	Name    string      `json:"name"`
-	Configs interface{} `json:"configs"`
-	Tags    string      `json:"tags"`
-}
-
-type BoardCate struct {
-	Name     string    `json:"name"`
-	IconUrl  string    `json:"icon_url"`
-	Boards   []Payload `json:"boards"`
-	Favorite bool      `json:"favorite"`
-}
-
-func (rt *Router) builtinBoardDetailGets(c *gin.Context) {
-	var payload Payload
-	ginx.BindJSON(c, &payload)
-
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	fn := fp + "/" + payload.Cate + "/dashboards/" + payload.Fname
-	content, err := file.ReadBytes(fn)
-	ginx.Dangerous(err)
-
-	err = json.Unmarshal(content, &payload)
-	ginx.NewRender(c).Data(payload, err)
-}
-
-func (rt *Router) builtinBoardCateGets(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	builtinFavoritesMap, err := models.BuiltinCateGetByUserId(rt.Ctx, me.Id)
-	if err != nil {
-		logger.Warningf("get builtin favorites fail: %v", err)
-	}
-
-	var boardCates []BoardCate
-	dirList, err := file.DirsUnder(fp)
-	ginx.Dangerous(err)
-	for _, dir := range dirList {
-		var boardCate BoardCate
-		boardCate.Name = dir
-		files, err := file.FilesUnder(fp + "/" + dir + "/dashboards")
-		ginx.Dangerous(err)
-		if len(files) == 0 {
-			continue
-		}
-
-		var boards []Payload
-		for _, f := range files {
-			fn := fp + "/" + dir + "/dashboards/" + f
-			content, err := file.ReadBytes(fn)
-			if err != nil {
-				logger.Warningf("add board fail: %v", err)
-				continue
-			}
-
-			var payload Payload
-			err = json.Unmarshal(content, &payload)
-			if err != nil {
-				logger.Warningf("add board:%s fail: %v", fn, err)
-				continue
-			}
-			payload.Cate = dir
-			payload.Fname = f
-			payload.Configs = ""
-			boards = append(boards, payload)
-		}
-		boardCate.Boards = boards
-
-		if _, ok := builtinFavoritesMap[dir]; ok {
-			boardCate.Favorite = true
-		}
-
-		iconFiles, _ := file.FilesUnder(fp + "/" + dir + "/icon")
-		if len(iconFiles) > 0 {
-			boardCate.IconUrl = fmt.Sprintf("/api/n9e/integrations/icon/%s/%s", dir, iconFiles[0])
-		}
-
-		boardCates = append(boardCates, boardCate)
-	}
-	ginx.NewRender(c).Data(boardCates, nil)
-}
-
-func (rt *Router) builtinBoardGets(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	var fileList []string
-	dirList, err := file.DirsUnder(fp)
-	ginx.Dangerous(err)
-	for _, dir := range dirList {
-		files, err := file.FilesUnder(fp + "/" + dir + "/dashboards")
-		ginx.Dangerous(err)
-		fileList = append(fileList, files...)
-	}
-
-	names := make([]string, 0, len(fileList))
-	for _, f := range fileList {
-		if !strings.HasSuffix(f, ".json") {
-			continue
-		}
-
-		name := strings.TrimSuffix(f, ".json")
-		names = append(names, name)
-	}
-
-	ginx.NewRender(c).Data(names, nil)
-}
-
-type AlertCate struct {
-	Name       string             `json:"name"`
-	IconUrl    string             `json:"icon_url"`
-	AlertRules []models.AlertRule `json:"alert_rules"`
-	Favorite   bool               `json:"favorite"`
-}
-
-func (rt *Router) builtinAlertCateGets(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	builtinFavoritesMap, err := models.BuiltinCateGetByUserId(rt.Ctx, me.Id)
-	if err != nil {
-		logger.Warningf("get builtin favorites fail: %v", err)
-	}
-
-	var alertCates []AlertCate
-	dirList, err := file.DirsUnder(fp)
-	ginx.Dangerous(err)
-	for _, dir := range dirList {
-		var alertCate AlertCate
-		alertCate.Name = dir
-		files, err := file.FilesUnder(fp + "/" + dir + "/alerts")
-		ginx.Dangerous(err)
-
-		var alertRules []models.AlertRule
-		for _, f := range files {
-			fn := fp + "/" + dir + "/alerts/" + f
-			content, err := file.ReadBytes(fn)
-			if err != nil {
-				logger.Warningf("add board fail: %v", err)
-				continue
-			}
-
-			var ars []models.AlertRule
-			err = json.Unmarshal(content, &ars)
-			if err != nil {
-				logger.Warningf("add board:%s fail: %v", fn, err)
-				continue
-			}
-			alertRules = append(alertRules, ars...)
-		}
-		alertCate.AlertRules = alertRules
-		iconFiles, _ := file.FilesUnder(fp + "/" + dir + "/icon")
-		if len(iconFiles) > 0 {
-			alertCate.IconUrl = fmt.Sprintf("/api/n9e/integrations/icon/%s/%s", dir, iconFiles[0])
-		}
-
-		if _, ok := builtinFavoritesMap[dir]; ok {
-			alertCate.Favorite = true
-		}
-
-		alertCates = append(alertCates, alertCate)
-	}
-	ginx.NewRender(c).Data(alertCates, nil)
-}
-
-type builtinAlertRulesList struct {
-	Name       string                        `json:"name"`
-	IconUrl    string                        `json:"icon_url"`
-	AlertRules map[string][]models.AlertRule `json:"alert_rules"`
-	Favorite   bool                          `json:"favorite"`
-}
-
-func (rt *Router) builtinAlertRules(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	builtinFavoritesMap, err := models.BuiltinCateGetByUserId(rt.Ctx, me.Id)
-	if err != nil {
-		logger.Warningf("get builtin favorites fail: %v", err)
-	}
-
-	var alertCates []builtinAlertRulesList
-	dirList, err := file.DirsUnder(fp)
-	ginx.Dangerous(err)
-	for _, dir := range dirList {
-		var alertCate builtinAlertRulesList
-		alertCate.Name = dir
-		files, err := file.FilesUnder(fp + "/" + dir + "/alerts")
-		ginx.Dangerous(err)
-		if len(files) == 0 {
-			continue
-		}
-
-		alertRules := make(map[string][]models.AlertRule)
-		for _, f := range files {
-			fn := fp + "/" + dir + "/alerts/" + f
-			content, err := file.ReadBytes(fn)
-			if err != nil {
-				logger.Warningf("add board fail: %v", err)
-				continue
-			}
-
-			var ars []models.AlertRule
-			err = json.Unmarshal(content, &ars)
-			if err != nil {
-				logger.Warningf("add board:%s fail: %v", fn, err)
-				continue
-			}
-			alertRules[strings.TrimSuffix(f, ".json")] = ars
-		}
-
-		alertCate.AlertRules = alertRules
-		iconFiles, _ := file.FilesUnder(fp + "/" + dir + "/icon")
-		if len(iconFiles) > 0 {
-			alertCate.IconUrl = fmt.Sprintf("/api/n9e/integrations/icon/%s/%s", dir, iconFiles[0])
-		}
-
-		if _, ok := builtinFavoritesMap[dir]; ok {
-			alertCate.Favorite = true
-		}
-
-		alertCates = append(alertCates, alertCate)
-	}
-	ginx.NewRender(c).Data(alertCates, nil)
-}
-
-// read the json file content
-func (rt *Router) builtinBoardGet(c *gin.Context) {
-	name := ginx.UrlParamStr(c, "name")
-	dirpath := rt.Center.BuiltinIntegrationsDir
-	if dirpath == "" {
-		dirpath = path.Join(runner.Cwd, "integrations")
-	}
-
-	dirList, err := file.DirsUnder(dirpath)
-	ginx.Dangerous(err)
-	for _, dir := range dirList {
-		jsonFile := dirpath + "/" + dir + "/dashboards/" + name + ".json"
-		if file.IsExist(jsonFile) {
-			body, err := file.ReadString(jsonFile)
-			ginx.NewRender(c).Data(body, err)
-			return
-		}
-	}
-
-	ginx.Bomb(http.StatusBadRequest, "%s not found", name)
-}
-
-func (rt *Router) builtinIcon(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-
-	cate := ginx.UrlParamStr(c, "cate")
-	iconPath := fp + "/" + cate + "/icon/" + ginx.UrlParamStr(c, "name")
-	c.File(path.Join(iconPath))
-}
-
-func (rt *Router) builtinMarkdown(c *gin.Context) {
-	fp := rt.Center.BuiltinIntegrationsDir
-	if fp == "" {
-		fp = path.Join(runner.Cwd, "integrations")
-	}
-	cate := ginx.UrlParamStr(c, "cate")
-
-	var markdown []byte
-	markdownDir := fp + "/" + cate + "/markdown"
-	markdownFiles, err := file.FilesUnder(markdownDir)
-	if err != nil {
-		logger.Warningf("get markdown fail: %v", err)
-	} else if len(markdownFiles) > 0 {
-		f := markdownFiles[0]
-		fn := markdownDir + "/" + f
-		markdown, err = file.ReadBytes(fn)
-		if err != nil {
-			logger.Warningf("get collect fail: %v", err)
-		}
-	}
-	ginx.NewRender(c).Data(string(markdown), nil)
-}

center/router/router_builtin_componet.go

@@ -1,93 +0,0 @@
-package router
-
-import (
-	"net/http"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"gorm.io/gorm"
-)
-
-const SYSTEM = "system"
-
-func (rt *Router) builtinComponentsAdd(c *gin.Context) {
-	var lst []models.BuiltinComponent
-	ginx.BindJSON(c, &lst)
-
-	username := Username(c)
-
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	reterr := make(map[string]string)
-	for i := 0; i < count; i++ {
-		if err := lst[i].Add(rt.Ctx, username); err != nil {
-			reterr[lst[i].Ident] = err.Error()
-		}
-	}
-
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-func (rt *Router) builtinComponentsGets(c *gin.Context) {
-	query := ginx.QueryStr(c, "query", "")
-	disabled := ginx.QueryInt(c, "disabled", -1)
-
-	bc, err := models.BuiltinComponentGets(rt.Ctx, query, disabled)
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(bc, nil)
-}
-
-func (rt *Router) builtinComponentsPut(c *gin.Context) {
-	var req models.BuiltinComponent
-	ginx.BindJSON(c, &req)
-
-	bc, err := models.BuiltinComponentGet(rt.Ctx, "id = ?", req.ID)
-	ginx.Dangerous(err)
-
-	if bc == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such builtin component")
-		return
-	}
-
-	if bc.CreatedBy == SYSTEM {
-		req.Ident = bc.Ident
-	}
-
-	username := Username(c)
-	req.UpdatedBy = username
-
-	err = models.DB(rt.Ctx).Transaction(func(tx *gorm.DB) error {
-		tCtx := &ctx.Context{
-			DB: tx,
-		}
-
-		txErr := models.BuiltinMetricBatchUpdateColumn(tCtx, "typ", bc.Ident, req.Ident, req.UpdatedBy)
-		if txErr != nil {
-			return txErr
-		}
-
-		txErr = bc.Update(tCtx, req)
-		if txErr != nil {
-			return txErr
-		}
-		return nil
-	})
-
-	ginx.NewRender(c).Message(err)
-}
-
-func (rt *Router) builtinComponentsDel(c *gin.Context) {
-	var req idsForm
-	ginx.BindJSON(c, &req)
-
-	req.Verify()
-
-	ginx.NewRender(c).Message(models.BuiltinComponentDels(rt.Ctx, req.Ids))
-}

center/router/router_builtin_metric_filter.go

@@ -1,120 +0,0 @@
-package router
-
-import (
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/prom"
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func (rt *Router) metricFilterGets(c *gin.Context) {
-	lst, err := models.MetricFilterGets(rt.Ctx, "")
-	ginx.Dangerous(err)
-	me := c.MustGet("user").(*models.User)
-
-	gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-	arr := make([]models.MetricFilter, 0)
-
-	for _, f := range lst {
-		if me.Username == f.CreateBy {
-			arr = append(arr, f)
-			continue
-		}
-
-		if HasPerm(gids, f.GroupsPerm, false) {
-			arr = append(arr, f)
-		}
-	}
-
-	ginx.NewRender(c).Data(arr, err)
-}
-
-func (rt *Router) metricFilterAdd(c *gin.Context) {
-	var f models.MetricFilter
-	ginx.BindJSON(c, &f)
-	me := c.MustGet("user").(*models.User)
-
-	f.CreateBy = me.Username
-	f.UpdateBy = me.Username
-	ginx.Dangerous(f.Add(rt.Ctx))
-	ginx.NewRender(c).Data(f, nil)
-}
-
-func (rt *Router) metricFilterDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	me := c.MustGet("user").(*models.User)
-
-	for _, id := range f.Ids {
-		old, err := models.MetricFilterGet(rt.Ctx, id)
-		ginx.Dangerous(err)
-
-		if me.Username != old.CreateBy {
-			gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-			ginx.Dangerous(err)
-
-			if !HasPerm(gids, old.GroupsPerm, true) {
-				ginx.NewRender(c).Message("forbidden")
-				return
-			}
-		}
-	}
-
-	ginx.NewRender(c).Message(models.MetricFilterDel(rt.Ctx, f.Ids))
-}
-
-func (rt *Router) metricFilterPut(c *gin.Context) {
-	var f models.MetricFilter
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	old, err := models.MetricFilterGet(rt.Ctx, f.ID)
-	ginx.Dangerous(err)
-
-	if me.Username != old.CreateBy {
-		gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-		ginx.Dangerous(err)
-
-		if !HasPerm(gids, old.GroupsPerm, true) {
-			ginx.NewRender(c).Message("forbidden")
-			return
-		}
-	}
-
-	f.UpdateBy = me.Username
-	ginx.NewRender(c).Message(f.Update(rt.Ctx))
-}
-
-type metricPromqlReq struct {
-	LabelFilter string `json:"label_filter"`
-	Promql      string `json:"promql"`
-}
-
-func (rt *Router) getMetricPromql(c *gin.Context) {
-	var req metricPromqlReq
-	ginx.BindJSON(c, &req)
-
-	promql := prom.AddLabelToPromQL(req.LabelFilter, req.Promql)
-	ginx.NewRender(c).Data(promql, nil)
-}
-
-func HasPerm(gids []int64, gps []models.GroupPerm, checkWrite bool) bool {
-	gmap := make(map[int64]struct{})
-	for _, gp := range gps {
-		if checkWrite && !gp.Write {
-			continue
-		}
-		gmap[gp.Gid] = struct{}{}
-	}
-
-	for _, gid := range gids {
-		if _, ok := gmap[gid]; ok {
-			return true
-		}
-	}
-
-	return false
-}

center/router/router_builtin_metrics.go

@@ -1,113 +0,0 @@
-package router
-
-import (
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/i18n"
-)
-
-// single or import
-func (rt *Router) builtinMetricsAdd(c *gin.Context) {
-	var lst []models.BuiltinMetric
-	ginx.BindJSON(c, &lst)
-	username := Username(c)
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	lang := c.GetHeader("X-Language")
-	if lang == "" {
-		lang = "zh_CN"
-	}
-
-	reterr := make(map[string]string)
-	for i := 0; i < count; i++ {
-		lst[i].Lang = lang
-		lst[i].UUID = time.Now().UnixNano()
-		if err := lst[i].Add(rt.Ctx, username); err != nil {
-			reterr[lst[i].Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-		}
-	}
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-func (rt *Router) builtinMetricsGets(c *gin.Context) {
-	collector := ginx.QueryStr(c, "collector", "")
-	typ := ginx.QueryStr(c, "typ", "")
-	query := ginx.QueryStr(c, "query", "")
-	limit := ginx.QueryInt(c, "limit", 20)
-	lang := c.GetHeader("X-Language")
-	unit := ginx.QueryStr(c, "unit", "")
-	if lang == "" {
-		lang = "zh_CN"
-	}
-
-	bm, err := models.BuiltinMetricGets(rt.Ctx, lang, collector, typ, query, unit, limit, ginx.Offset(c, limit))
-	ginx.Dangerous(err)
-
-	total, err := models.BuiltinMetricCount(rt.Ctx, lang, collector, typ, query, unit)
-	ginx.Dangerous(err)
-	ginx.NewRender(c).Data(gin.H{
-		"list":  bm,
-		"total": total,
-	}, nil)
-}
-
-func (rt *Router) builtinMetricsPut(c *gin.Context) {
-	var req models.BuiltinMetric
-	ginx.BindJSON(c, &req)
-
-	bm, err := models.BuiltinMetricGet(rt.Ctx, "id = ?", req.ID)
-	ginx.Dangerous(err)
-	if bm == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such builtin metric")
-		return
-	}
-	username := Username(c)
-
-	req.UpdatedBy = username
-	ginx.NewRender(c).Message(bm.Update(rt.Ctx, req))
-}
-
-func (rt *Router) builtinMetricsDel(c *gin.Context) {
-	var req idsForm
-	ginx.BindJSON(c, &req)
-	req.Verify()
-
-	ginx.NewRender(c).Message(models.BuiltinMetricDels(rt.Ctx, req.Ids))
-}
-
-func (rt *Router) builtinMetricsDefaultTypes(c *gin.Context) {
-	lst := []string{
-		"Linux",
-		"Procstat",
-		"cAdvisor",
-		"Ping",
-		"MySQL",
-		"ClickHouse",
-	}
-	ginx.NewRender(c).Data(lst, nil)
-}
-
-func (rt *Router) builtinMetricsTypes(c *gin.Context) {
-	collector := ginx.QueryStr(c, "collector", "")
-	query := ginx.QueryStr(c, "query", "")
-	lang := c.GetHeader("X-Language")
-
-	metricTypeList, err := models.BuiltinMetricTypes(rt.Ctx, lang, collector, query)
-	ginx.NewRender(c).Data(metricTypeList, err)
-}
-
-func (rt *Router) builtinMetricsCollectors(c *gin.Context) {
-	typ := ginx.QueryStr(c, "typ", "")
-	query := ginx.QueryStr(c, "query", "")
-	lang := c.GetHeader("X-Language")
-
-	ginx.NewRender(c).Data(models.BuiltinMetricCollectors(rt.Ctx, lang, typ, query))
-}

center/router/router_builtin_payload.go

@@ -1,286 +0,0 @@
-package router
-
-import (
-	"encoding/json"
-	"net/http"
-	"strings"
-	"time"
-
-	"github.com/BurntSushi/toml"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/i18n"
-)
-
-type Board struct {
-	Name    string      `json:"name"`
-	Tags    string      `json:"tags"`
-	Configs interface{} `json:"configs"`
-	UUID    int64       `json:"uuid"`
-}
-
-func (rt *Router) builtinPayloadsAdd(c *gin.Context) {
-	var lst []models.BuiltinPayload
-	ginx.BindJSON(c, &lst)
-
-	username := Username(c)
-
-	count := len(lst)
-	if count == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	reterr := make(map[string]string)
-	for i := 0; i < count; i++ {
-		if lst[i].Type == "alert" {
-			if strings.HasPrefix(strings.TrimSpace(lst[i].Content), "[") {
-				// 处理多个告警规则模板的情况
-				alertRules := []models.AlertRule{}
-				if err := json.Unmarshal([]byte(lst[i].Content), &alertRules); err != nil {
-					reterr[lst[i].Name] = err.Error()
-				}
-
-				for _, rule := range alertRules {
-					if rule.UUID == 0 {
-						rule.UUID = time.Now().UnixMicro()
-					}
-
-					contentBytes, err := json.Marshal(rule)
-					if err != nil {
-						reterr[rule.Name] = err.Error()
-						continue
-					}
-
-					bp := models.BuiltinPayload{
-						Type:        lst[i].Type,
-						ComponentID: lst[i].ComponentID,
-						Cate:        lst[i].Cate,
-						Name:        rule.Name,
-						Tags:        rule.AppendTags,
-						UUID:        rule.UUID,
-						Content:     string(contentBytes),
-						CreatedBy:   username,
-						UpdatedBy:   username,
-					}
-
-					if err := bp.Add(rt.Ctx, username); err != nil {
-						reterr[bp.Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-					}
-				}
-				continue
-			}
-
-			alertRule := models.AlertRule{}
-			if err := json.Unmarshal([]byte(lst[i].Content), &alertRule); err != nil {
-				reterr[lst[i].Name] = err.Error()
-				continue
-			}
-
-			if alertRule.UUID == 0 {
-				alertRule.UUID = time.Now().UnixMicro()
-			}
-
-			contentBytes, err := json.Marshal(alertRule)
-			if err != nil {
-				reterr[alertRule.Name] = err.Error()
-				continue
-			}
-
-			bp := models.BuiltinPayload{
-				Type:        lst[i].Type,
-				ComponentID: lst[i].ComponentID,
-				Cate:        lst[i].Cate,
-				Name:        alertRule.Name,
-				Tags:        alertRule.AppendTags,
-				UUID:        alertRule.UUID,
-				Content:     string(contentBytes),
-				CreatedBy:   username,
-				UpdatedBy:   username,
-			}
-
-			if err := bp.Add(rt.Ctx, username); err != nil {
-				reterr[bp.Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-			}
-		} else if lst[i].Type == "dashboard" {
-			if strings.HasPrefix(strings.TrimSpace(lst[i].Content), "[") {
-				// 处理多个告警规则模板的情况
-				dashboards := []Board{}
-				if err := json.Unmarshal([]byte(lst[i].Content), &dashboards); err != nil {
-					reterr[lst[i].Name] = err.Error()
-				}
-
-				for _, dashboard := range dashboards {
-					if dashboard.UUID == 0 {
-						dashboard.UUID = time.Now().UnixMicro()
-					}
-
-					contentBytes, err := json.Marshal(dashboard)
-					if err != nil {
-						reterr[dashboard.Name] = err.Error()
-						continue
-					}
-
-					bp := models.BuiltinPayload{
-						Type:        lst[i].Type,
-						ComponentID: lst[i].ComponentID,
-						Cate:        lst[i].Cate,
-						Name:        dashboard.Name,
-						Tags:        dashboard.Tags,
-						UUID:        dashboard.UUID,
-						Content:     string(contentBytes),
-						CreatedBy:   username,
-						UpdatedBy:   username,
-					}
-
-					if err := bp.Add(rt.Ctx, username); err != nil {
-						reterr[bp.Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-					}
-				}
-				continue
-			}
-
-			dashboard := Board{}
-			if err := json.Unmarshal([]byte(lst[i].Content), &dashboard); err != nil {
-				reterr[lst[i].Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-				continue
-			}
-
-			if dashboard.UUID == 0 {
-				dashboard.UUID = time.Now().UnixMicro()
-			}
-
-			contentBytes, err := json.Marshal(dashboard)
-			if err != nil {
-				reterr[dashboard.Name] = err.Error()
-				continue
-			}
-
-			bp := models.BuiltinPayload{
-				Type:        lst[i].Type,
-				ComponentID: lst[i].ComponentID,
-				Cate:        lst[i].Cate,
-				Name:        dashboard.Name,
-				Tags:        dashboard.Tags,
-				UUID:        dashboard.UUID,
-				Content:     string(contentBytes),
-				CreatedBy:   username,
-				UpdatedBy:   username,
-			}
-
-			if err := bp.Add(rt.Ctx, username); err != nil {
-				reterr[bp.Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-			}
-		} else {
-			if lst[i].Type == "collect" {
-				c := make(map[string]interface{})
-				if _, err := toml.Decode(lst[i].Content, &c); err != nil {
-					reterr[lst[i].Name] = err.Error()
-					continue
-				}
-			}
-
-			if err := lst[i].Add(rt.Ctx, username); err != nil {
-				reterr[lst[i].Name] = i18n.Sprintf(c.GetHeader("X-Language"), err.Error())
-			}
-		}
-
-	}
-
-	ginx.NewRender(c).Data(reterr, nil)
-}
-
-func (rt *Router) builtinPayloadsGets(c *gin.Context) {
-	typ := ginx.QueryStr(c, "type", "")
-	ComponentID := ginx.QueryInt64(c, "component_id", 0)
-
-	cate := ginx.QueryStr(c, "cate", "")
-	query := ginx.QueryStr(c, "query", "")
-
-	lst, err := models.BuiltinPayloadGets(rt.Ctx, uint64(ComponentID), typ, cate, query)
-	ginx.NewRender(c).Data(lst, err)
-}
-
-func (rt *Router) builtinPayloadcatesGet(c *gin.Context) {
-	typ := ginx.QueryStr(c, "type", "")
-	ComponentID := ginx.QueryInt64(c, "component_id", 0)
-
-	cates, err := models.BuiltinPayloadCates(rt.Ctx, typ, uint64(ComponentID))
-	ginx.NewRender(c).Data(cates, err)
-}
-
-func (rt *Router) builtinPayloadGet(c *gin.Context) {
-	id := ginx.UrlParamInt64(c, "id")
-
-	bp, err := models.BuiltinPayloadGet(rt.Ctx, "id = ?", id)
-	if err != nil {
-		ginx.Bomb(http.StatusInternalServerError, err.Error())
-	}
-	if bp == nil {
-		ginx.Bomb(http.StatusNotFound, "builtin payload not found")
-	}
-
-	ginx.NewRender(c).Data(bp, nil)
-}
-
-func (rt *Router) builtinPayloadsPut(c *gin.Context) {
-	var req models.BuiltinPayload
-	ginx.BindJSON(c, &req)
-
-	bp, err := models.BuiltinPayloadGet(rt.Ctx, "id = ?", req.ID)
-	ginx.Dangerous(err)
-
-	if bp == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such builtin payload")
-		return
-	}
-
-	if req.Type == "alert" {
-		alertRule := models.AlertRule{}
-		if err := json.Unmarshal([]byte(req.Content), &alertRule); err != nil {
-			ginx.Bomb(http.StatusBadRequest, err.Error())
-		}
-
-		req.Name = alertRule.Name
-		req.Tags = alertRule.AppendTags
-	} else if req.Type == "dashboard" {
-		dashboard := Board{}
-		if err := json.Unmarshal([]byte(req.Content), &dashboard); err != nil {
-			ginx.Bomb(http.StatusBadRequest, err.Error())
-		}
-
-		req.Name = dashboard.Name
-		req.Tags = dashboard.Tags
-	} else if req.Type == "collect" {
-		c := make(map[string]interface{})
-		if _, err := toml.Decode(req.Content, &c); err != nil {
-			ginx.Bomb(http.StatusBadRequest, err.Error())
-		}
-	}
-
-	username := Username(c)
-	req.UpdatedBy = username
-
-	ginx.NewRender(c).Message(bp.Update(rt.Ctx, req))
-}
-
-func (rt *Router) builtinPayloadsDel(c *gin.Context) {
-	var req idsForm
-	ginx.BindJSON(c, &req)
-
-	req.Verify()
-
-	ginx.NewRender(c).Message(models.BuiltinPayloadDels(rt.Ctx, req.Ids))
-}
-
-func (rt *Router) builtinPayloadsGetByUUIDOrID(c *gin.Context) {
-	uuid := ginx.QueryInt64(c, "uuid", 0)
-	// 优先以 uuid 为准
-	if uuid != 0 {
-		ginx.NewRender(c).Data(models.BuiltinPayloadGet(rt.Ctx, "uuid = ?", uuid))
-		return
-	}
-
-	id := ginx.QueryInt64(c, "id", 0)
-	ginx.NewRender(c).Data(models.BuiltinPayloadGet(rt.Ctx, "id = ?", id))
-}

center/router/router_captcha.go

@@ -1,114 +0,0 @@
-package router
-
-import (
-	"context"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/storage"
-	"github.com/gin-gonic/gin"
-	captcha "github.com/mojocn/base64Captcha"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type CaptchaRedisStore struct {
-	redis storage.Redis
-}
-
-func (s *CaptchaRedisStore) Set(id string, value string) error {
-	ctx := context.Background()
-	err := s.redis.Set(ctx, id, value, time.Duration(300*time.Second)).Err()
-	if err != nil {
-		logger.Errorf("captcha id set to redis error : %s", err.Error())
-		return err
-	}
-	return nil
-}
-
-func (s *CaptchaRedisStore) Get(id string, clear bool) string {
-	ctx := context.Background()
-	val, err := s.redis.Get(ctx, id).Result()
-	if err != nil {
-		logger.Errorf("captcha id get from redis error : %s", err.Error())
-		return ""
-	}
-
-	if clear {
-		s.redis.Del(ctx, id)
-	}
-
-	return val
-}
-
-func (s *CaptchaRedisStore) Verify(id, answer string, clear bool) bool {
-
-	old := s.Get(id, clear)
-	return old == answer
-}
-
-func (rt *Router) newCaptchaRedisStore() *CaptchaRedisStore {
-	if captchaStore == nil {
-		captchaStore = &CaptchaRedisStore{redis: rt.Redis}
-	}
-	return captchaStore
-}
-
-var captchaStore *CaptchaRedisStore
-
-type CaptchaReqBody struct {
-	Id          string
-	VerifyValue string
-}
-
-// 生成图形验证码
-func (rt *Router) generateCaptcha(c *gin.Context) {
-	var driver = captcha.NewDriverMath(60, 200, 0, captcha.OptionShowHollowLine, nil, nil, []string{"wqy-microhei.ttc"})
-	cc := captcha.NewCaptcha(driver, rt.newCaptchaRedisStore())
-	//data:image/png;base64
-	id, b64s, _, err := cc.Generate()
-
-	if err != nil {
-		ginx.NewRender(c).Message(err)
-		return
-	}
-
-	ginx.NewRender(c).Data(gin.H{
-		"imgdata":   b64s,
-		"captchaid": id,
-	}, nil)
-}
-
-// 验证
-func (rt *Router) captchaVerify(c *gin.Context) {
-
-	var param CaptchaReqBody
-	ginx.BindJSON(c, &param)
-
-	//verify the captcha
-	if captchaStore.Verify(param.Id, param.VerifyValue, true) {
-		ginx.NewRender(c).Message("")
-		return
-	}
-	ginx.NewRender(c).Message("incorrect verification code")
-}
-
-// 验证码开关
-func (rt *Router) ifShowCaptcha(c *gin.Context) {
-
-	if rt.HTTP.ShowCaptcha.Enable {
-		ginx.NewRender(c).Data(gin.H{
-			"show": true,
-		}, nil)
-		return
-	}
-
-	ginx.NewRender(c).Data(gin.H{
-		"show": false,
-	}, nil)
-}
-
-// 验证
-func CaptchaVerify(id string, value string) bool {
-	//verify the captcha
-	return captchaStore.Verify(id, value, true)
-}

center/router/router_chart_share.go

@@ -1,45 +0,0 @@
-package router
-
-import (
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func (rt *Router) chartShareGets(c *gin.Context) {
-	ids := ginx.QueryStr(c, "ids", "")
-	lst, err := models.ChartShareGetsByIds(rt.Ctx, strx.IdsInt64ForAPI(ids, ","))
-	ginx.NewRender(c).Data(lst, err)
-}
-
-type chartShareForm struct {
-	DatasourceId int64  `json:"datasource_id"`
-	Configs      string `json:"configs"`
-}
-
-func (rt *Router) chartShareAdd(c *gin.Context) {
-	username := c.MustGet("username").(string)
-
-	var forms []chartShareForm
-	ginx.BindJSON(c, &forms)
-
-	ids := []int64{}
-	now := time.Now().Unix()
-
-	for _, f := range forms {
-		chart := models.ChartShare{
-			DatasourceId: f.DatasourceId,
-			Configs:      f.Configs,
-			CreateBy:     username,
-			CreateAt:     now,
-		}
-		ginx.Dangerous(chart.Add(rt.Ctx))
-		ids = append(ids, chart.Id)
-	}
-
-	ginx.NewRender(c).Data(ids, nil)
-}

center/router/router_config.go

@@ -1,69 +0,0 @@
-package router
-
-import (
-	"encoding/json"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func (rt *Router) notifyChannelsGets(c *gin.Context) {
-	var labelAndKeys []models.LabelAndKey
-	cval, err := models.ConfigsGet(rt.Ctx, models.NOTIFYCHANNEL)
-	ginx.Dangerous(err)
-
-	if cval == "" {
-		ginx.NewRender(c).Data(labelAndKeys, nil)
-		return
-	}
-
-	var notifyChannels []models.NotifyChannel
-	err = json.Unmarshal([]byte(cval), &notifyChannels)
-	ginx.Dangerous(err)
-
-	for _, v := range notifyChannels {
-		if v.Hide {
-			continue
-		}
-		var labelAndKey models.LabelAndKey
-		labelAndKey.Label = v.Name
-		labelAndKey.Key = v.Ident
-		labelAndKeys = append(labelAndKeys, labelAndKey)
-	}
-
-	ginx.NewRender(c).Data(labelAndKeys, nil)
-}
-
-func (rt *Router) contactKeysGets(c *gin.Context) {
-	var labelAndKeys []models.LabelAndKey
-	cval, err := models.ConfigsGet(rt.Ctx, models.NOTIFYCONTACT)
-	ginx.Dangerous(err)
-
-	if cval == "" {
-		ginx.NewRender(c).Data(labelAndKeys, nil)
-		return
-	}
-
-	var notifyContacts []models.NotifyContact
-	err = json.Unmarshal([]byte(cval), &notifyContacts)
-	ginx.Dangerous(err)
-
-	for _, v := range notifyContacts {
-		if v.Hide {
-			continue
-		}
-		var labelAndKey models.LabelAndKey
-		labelAndKey.Label = v.Name
-		labelAndKey.Key = v.Ident
-		labelAndKeys = append(labelAndKeys, labelAndKey)
-	}
-
-	ginx.NewRender(c).Data(labelAndKeys, nil)
-}
-
-func (rt *Router) siteInfo(c *gin.Context) {
-	config, err := models.ConfigsGet(rt.Ctx, "site_info")
-	ginx.NewRender(c).Data(config, err)
-}

center/router/router_configs.go

@@ -1,96 +0,0 @@
-package router
-
-import (
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-const EMBEDDEDDASHBOARD = "embedded-dashboards"
-
-func (rt *Router) configsGet(c *gin.Context) {
-	prefix := ginx.QueryStr(c, "prefix", "")
-	limit := ginx.QueryInt(c, "limit", 10)
-	configs, err := models.ConfigsGets(rt.Ctx, prefix, limit, ginx.Offset(c, limit))
-	ginx.NewRender(c).Data(configs, err)
-}
-
-func (rt *Router) configGet(c *gin.Context) {
-	id := ginx.UrlParamInt64(c, "id")
-	configs, err := models.ConfigGet(rt.Ctx, id)
-	ginx.NewRender(c).Data(configs, err)
-}
-
-func (rt *Router) configGetAll(c *gin.Context) {
-	config, err := models.ConfigsGetAll(rt.Ctx)
-	ginx.NewRender(c).Data(config, err)
-}
-
-func (rt *Router) configGetByKey(c *gin.Context) {
-	config, err := models.ConfigsGet(rt.Ctx, ginx.QueryStr(c, "key"))
-	ginx.NewRender(c).Data(config, err)
-}
-
-func (rt *Router) configPutByKey(c *gin.Context) {
-	var f models.Configs
-	ginx.BindJSON(c, &f)
-	username := c.MustGet("username").(string)
-	ginx.NewRender(c).Message(models.ConfigsSetWithUname(rt.Ctx, f.Ckey, f.Cval, username))
-}
-
-func (rt *Router) embeddedDashboardsGet(c *gin.Context) {
-	config, err := models.ConfigsGet(rt.Ctx, EMBEDDEDDASHBOARD)
-	ginx.NewRender(c).Data(config, err)
-}
-
-func (rt *Router) embeddedDashboardsPut(c *gin.Context) {
-	var f models.Configs
-	ginx.BindJSON(c, &f)
-	username := c.MustGet("username").(string)
-	ginx.NewRender(c).Message(models.ConfigsSetWithUname(rt.Ctx, EMBEDDEDDASHBOARD, f.Cval, username))
-}
-
-func (rt *Router) configsDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	ginx.NewRender(c).Message(models.ConfigsDel(rt.Ctx, f.Ids))
-}
-
-func (rt *Router) configsPut(c *gin.Context) { //for APIForService
-	var arr []models.Configs
-	ginx.BindJSON(c, &arr)
-	username := c.GetString("user")
-	if username == "" {
-		username = "default"
-	}
-	now := time.Now().Unix()
-	for i := 0; i < len(arr); i++ {
-		arr[i].UpdateBy = username
-		arr[i].UpdateAt = now
-		ginx.Dangerous(arr[i].Update(rt.Ctx))
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-func (rt *Router) configsPost(c *gin.Context) { //for APIForService
-	var arr []models.Configs
-	ginx.BindJSON(c, &arr)
-	username := c.GetString("user")
-	if username == "" {
-		username = "default"
-	}
-	now := time.Now().Unix()
-	for i := 0; i < len(arr); i++ {
-		arr[i].CreateBy = username
-		arr[i].UpdateBy = username
-		arr[i].CreateAt = now
-		arr[i].UpdateAt = now
-		ginx.Dangerous(arr[i].Add(rt.Ctx))
-	}
-
-	ginx.NewRender(c).Message(nil)
-}

center/router/router_crypto.go

@@ -1,63 +0,0 @@
-package router
-
-import (
-	"github.com/ccfos/nightingale/v6/pkg/secu"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-type confPropCrypto struct {
-	Data string `json:"data" binding:"required"`
-	Key  string `json:"key" binding:"required"`
-}
-
-func (rt *Router) confPropEncrypt(c *gin.Context) {
-	var f confPropCrypto
-	ginx.BindJSON(c, &f)
-
-	k := len(f.Key)
-	switch k {
-	default:
-		c.String(400, "The key length should be 16, 24 or 32")
-		return
-	case 16, 24, 32:
-		break
-	}
-
-	s, err := secu.DealWithEncrypt(f.Data, f.Key)
-	if err != nil {
-		c.String(500, err.Error())
-	}
-
-	c.JSON(200, gin.H{
-		"src":     f.Data,
-		"key":     f.Key,
-		"encrypt": s,
-	})
-}
-
-func (rt *Router) confPropDecrypt(c *gin.Context) {
-	var f confPropCrypto
-	ginx.BindJSON(c, &f)
-
-	k := len(f.Key)
-	switch k {
-	default:
-		c.String(400, "The key length should be 16, 24 or 32")
-		return
-	case 16, 24, 32:
-		break
-	}
-
-	s, err := secu.DealWithDecrypt(f.Data, f.Key)
-	if err != nil {
-		c.String(500, err.Error())
-	}
-
-	c.JSON(200, gin.H{
-		"src":     f.Data,
-		"key":     f.Key,
-		"decrypt": s,
-	})
-}

center/router/router_dash_annotation.go

@@ -1,99 +0,0 @@
-package router
-
-import (
-	"fmt"
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func checkAnnotationPermission(c *gin.Context, ctx *ctx.Context, dashboardId int64) {
-	dashboard, err := models.BoardGetByID(ctx, dashboardId)
-	if err != nil {
-		ginx.Bomb(http.StatusInternalServerError, "failed to get dashboard: %v", err)
-	}
-
-	if dashboard == nil {
-		ginx.Bomb(http.StatusNotFound, "dashboard not found")
-	}
-
-	bg := BusiGroup(ctx, dashboard.GroupId)
-	me := c.MustGet("user").(*models.User)
-	can, err := me.CanDoBusiGroup(ctx, bg, "rw")
-	ginx.Dangerous(err)
-
-	if !can {
-		ginx.Bomb(http.StatusForbidden, "forbidden")
-	}
-}
-
-func (rt *Router) dashAnnotationAdd(c *gin.Context) {
-	var f models.DashAnnotation
-	ginx.BindJSON(c, &f)
-
-	username := c.MustGet("username").(string)
-	now := time.Now().Unix()
-
-	checkAnnotationPermission(c, rt.Ctx, f.DashboardId)
-
-	f.CreateBy = username
-	f.CreateAt = now
-	f.UpdateBy = username
-	f.UpdateAt = now
-
-	ginx.NewRender(c).Data(f.Id, f.Add(rt.Ctx))
-}
-
-func (rt *Router) dashAnnotationGets(c *gin.Context) {
-	dashboardId := ginx.QueryInt64(c, "dashboard_id")
-	from := ginx.QueryInt64(c, "from")
-	to := ginx.QueryInt64(c, "to")
-	limit := ginx.QueryInt(c, "limit", 100)
-
-	lst, err := models.DashAnnotationGets(rt.Ctx, dashboardId, from, to, limit)
-	ginx.NewRender(c).Data(lst, err)
-}
-
-func (rt *Router) dashAnnotationPut(c *gin.Context) {
-	var f models.DashAnnotation
-	ginx.BindJSON(c, &f)
-
-	id := ginx.UrlParamInt64(c, "id")
-	annotation, err := getAnnotationById(rt.Ctx, id)
-	ginx.Dangerous(err)
-
-	checkAnnotationPermission(c, rt.Ctx, annotation.DashboardId)
-
-	f.Id = id
-	f.UpdateAt = time.Now().Unix()
-	f.UpdateBy = c.MustGet("username").(string)
-
-	ginx.NewRender(c).Message(f.Update(rt.Ctx))
-}
-
-func (rt *Router) dashAnnotationDel(c *gin.Context) {
-	id := ginx.UrlParamInt64(c, "id")
-
-	annotation, err := getAnnotationById(rt.Ctx, id)
-	ginx.Dangerous(err)
-	checkAnnotationPermission(c, rt.Ctx, annotation.DashboardId)
-
-	ginx.NewRender(c).Message(models.DashAnnotationDel(rt.Ctx, id))
-}
-
-// 可以提取获取注释的通用方法
-func getAnnotationById(ctx *ctx.Context, id int64) (*models.DashAnnotation, error) {
-	annotation, err := models.DashAnnotationGet(ctx, "id=?", id)
-	if err != nil {
-		return nil, err
-	}
-	if annotation == nil {
-		return nil, fmt.Errorf("annotation not found")
-	}
-	return annotation, nil
-}

center/router/router_dashboard.go

@@ -1,19 +0,0 @@
-package router
-
-type ChartPure struct {
-	Configs string `json:"configs"`
-	Weight  int    `json:"weight"`
-}
-
-type ChartGroupPure struct {
-	Name   string      `json:"name"`
-	Weight int         `json:"weight"`
-	Charts []ChartPure `json:"charts"`
-}
-
-type DashboardPure struct {
-	Name        string           `json:"name"`
-	Tags        string           `json:"tags"`
-	Configs     string           `json:"configs"`
-	ChartGroups []ChartGroupPure `json:"chart_groups"`
-}

center/router/router_datasource.go

@@ -1,296 +0,0 @@
-package router
-
-import (
-	"crypto/tls"
-	"fmt"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-func (rt *Router) pluginList(c *gin.Context) {
-	Render(c, rt.Center.Plugins, nil)
-}
-
-type listReq struct {
-	Name     string `json:"name"`
-	Type     string `json:"plugin_type"`
-	Category string `json:"category"`
-}
-
-func (rt *Router) datasourceList(c *gin.Context) {
-	if rt.DatasourceCache.DatasourceCheckHook(c) {
-		Render(c, []int{}, nil)
-		return
-	}
-
-	var req listReq
-	ginx.BindJSON(c, &req)
-
-	typ := req.Type
-	category := req.Category
-	name := req.Name
-
-	user := c.MustGet("user").(*models.User)
-
-	list, err := models.GetDatasourcesGetsBy(rt.Ctx, typ, category, name, "")
-	Render(c, rt.DatasourceCache.DatasourceFilter(list, user), err)
-}
-
-func (rt *Router) datasourceGetsByService(c *gin.Context) {
-	typ := ginx.QueryStr(c, "typ", "")
-	lst, err := models.GetDatasourcesGetsBy(rt.Ctx, typ, "", "", "")
-	ginx.NewRender(c).Data(lst, err)
-}
-
-func (rt *Router) datasourceBriefs(c *gin.Context) {
-	var dss []*models.Datasource
-	list, err := models.GetDatasourcesGetsBy(rt.Ctx, "", "", "", "")
-	ginx.Dangerous(err)
-
-	for _, item := range list {
-		item.AuthJson.BasicAuthPassword = ""
-		if item.PluginType == models.PROMETHEUS {
-			for k, v := range item.SettingsJson {
-				if strings.HasPrefix(k, "prometheus.") {
-					item.SettingsJson[strings.TrimPrefix(k, "prometheus.")] = v
-					delete(item.SettingsJson, k)
-				}
-			}
-		} else if item.PluginType == "cloudwatch" {
-			for k := range item.SettingsJson {
-				if !strings.Contains(k, "region") {
-					delete(item.SettingsJson, k)
-				}
-			}
-		} else {
-			item.SettingsJson = nil
-		}
-		dss = append(dss, item)
-	}
-
-	if !rt.Center.AnonymousAccess.PromQuerier {
-		user := c.MustGet("user").(*models.User)
-		dss = rt.DatasourceCache.DatasourceFilter(dss, user)
-	}
-
-	ginx.NewRender(c).Data(dss, err)
-}
-
-func (rt *Router) datasourceUpsert(c *gin.Context) {
-	if rt.DatasourceCache.DatasourceCheckHook(c) {
-		Render(c, []int{}, nil)
-		return
-	}
-
-	var req models.Datasource
-	ginx.BindJSON(c, &req)
-	username := Username(c)
-	req.UpdatedBy = username
-
-	var err error
-	var count int64
-
-	if !req.ForceSave {
-		if req.PluginType == models.PROMETHEUS || req.PluginType == models.LOKI || req.PluginType == models.TDENGINE {
-			err = DatasourceCheck(req)
-			if err != nil {
-				Dangerous(c, err)
-				return
-			}
-		}
-	}
-
-	if req.Id == 0 {
-		req.CreatedBy = username
-		req.Status = "enabled"
-		count, err = models.GetDatasourcesCountBy(rt.Ctx, "", "", req.Name)
-		if err != nil {
-			Render(c, nil, err)
-			return
-		}
-
-		if count > 0 {
-			Render(c, nil, "name already exists")
-			return
-		}
-		err = req.Add(rt.Ctx)
-	} else {
-		err = req.Update(rt.Ctx, "name", "identifier", "description", "cluster_name", "settings", "http", "auth", "updated_by", "updated_at", "is_default")
-	}
-
-	Render(c, nil, err)
-}
-
-func DatasourceCheck(ds models.Datasource) error {
-	if ds.PluginType == models.PROMETHEUS || ds.PluginType == models.LOKI || ds.PluginType == models.TDENGINE {
-		if ds.HTTPJson.Url == "" {
-			return fmt.Errorf("url is empty")
-		}
-
-		if !strings.HasPrefix(ds.HTTPJson.Url, "http") {
-			return fmt.Errorf("url must start with http or https")
-		}
-	}
-
-	client := &http.Client{
-		Transport: &http.Transport{
-			TLSClientConfig: &tls.Config{
-				InsecureSkipVerify: ds.HTTPJson.TLS.SkipTlsVerify,
-			},
-		},
-	}
-
-	ds.HTTPJson.Url = strings.TrimRight(ds.HTTPJson.Url, "/")
-	var fullURL string
-	req, err := ds.HTTPJson.NewReq(&fullURL)
-	if err != nil {
-		logger.Errorf("Error creating request: %v", err)
-		return fmt.Errorf("request urls:%v failed: %v", ds.HTTPJson.GetUrls(), err)
-	}
-
-	if ds.PluginType == models.PROMETHEUS {
-		subPath := "/api/v1/query"
-		query := url.Values{}
-		if ds.HTTPJson.IsLoki() {
-			subPath = "/api/v1/labels"
-		} else {
-			query.Add("query", "1+1")
-		}
-		fullURL = fmt.Sprintf("%s%s?%s", ds.HTTPJson.Url, subPath, query.Encode())
-
-		req, err = http.NewRequest("GET", fullURL, nil)
-		if err != nil {
-			logger.Errorf("Error creating request: %v", err)
-			return fmt.Errorf("request url:%s failed: %v", fullURL, err)
-		}
-	} else if ds.PluginType == models.TDENGINE {
-		fullURL = fmt.Sprintf("%s/rest/sql", ds.HTTPJson.Url)
-		req, err = http.NewRequest("POST", fullURL, strings.NewReader("show databases"))
-		if err != nil {
-			logger.Errorf("Error creating request: %v", err)
-			return fmt.Errorf("request url:%s failed: %v", fullURL, err)
-		}
-	}
-
-	if ds.PluginType == models.LOKI {
-		subPath := "/api/v1/labels"
-
-		fullURL = fmt.Sprintf("%s%s", ds.HTTPJson.Url, subPath)
-
-		req, err = http.NewRequest("GET", fullURL, nil)
-		if err != nil {
-			logger.Errorf("Error creating request: %v", err)
-			return fmt.Errorf("request url:%s failed: %v", fullURL, err)
-		}
-	}
-
-	if ds.AuthJson.BasicAuthUser != "" {
-		req.SetBasicAuth(ds.AuthJson.BasicAuthUser, ds.AuthJson.BasicAuthPassword)
-	}
-
-	for k, v := range ds.HTTPJson.Headers {
-		req.Header.Set(k, v)
-	}
-
-	resp, err := client.Do(req)
-	if err != nil {
-		logger.Errorf("Error making request: %v\n", err)
-		return fmt.Errorf("request url:%s failed: %v", fullURL, err)
-	}
-	defer resp.Body.Close()
-
-	if resp.StatusCode != 200 {
-		logger.Errorf("Error making request: %v\n", resp.StatusCode)
-		body, _ := io.ReadAll(resp.Body)
-		return fmt.Errorf("request url:%s failed code:%d body:%s", fullURL, resp.StatusCode, string(body))
-	}
-
-	return nil
-}
-
-func (rt *Router) datasourceGet(c *gin.Context) {
-	if rt.DatasourceCache.DatasourceCheckHook(c) {
-		Render(c, []int{}, nil)
-		return
-	}
-
-	var req models.Datasource
-	ginx.BindJSON(c, &req)
-	err := req.Get(rt.Ctx)
-	Render(c, req, err)
-}
-
-func (rt *Router) datasourceUpdataStatus(c *gin.Context) {
-	if rt.DatasourceCache.DatasourceCheckHook(c) {
-		Render(c, []int{}, nil)
-		return
-	}
-
-	var req models.Datasource
-	ginx.BindJSON(c, &req)
-	username := Username(c)
-	req.UpdatedBy = username
-	err := req.Update(rt.Ctx, "status", "updated_by", "updated_at")
-	Render(c, req, err)
-}
-
-func (rt *Router) datasourceDel(c *gin.Context) {
-	if rt.DatasourceCache.DatasourceCheckHook(c) {
-		Render(c, []int{}, nil)
-		return
-	}
-
-	var ids []int64
-	ginx.BindJSON(c, &ids)
-	err := models.DatasourceDel(rt.Ctx, ids)
-	Render(c, nil, err)
-}
-
-func (rt *Router) getDatasourceIds(c *gin.Context) {
-	name := ginx.QueryStr(c, "name")
-	datasourceIds, err := models.GetDatasourceIdsByEngineName(rt.Ctx, name)
-
-	ginx.NewRender(c).Data(datasourceIds, err)
-}
-
-type datasourceQueryForm struct {
-	Cate              string                   `json:"datasource_cate"`
-	DatasourceQueries []models.DatasourceQuery `json:"datasource_queries"`
-}
-
-type datasourceQueryResp struct {
-	ID   int64  `json:"id"`
-	Name string `json:"name"`
-}
-
-func (rt *Router) datasourceQuery(c *gin.Context) {
-	var dsf datasourceQueryForm
-	ginx.BindJSON(c, &dsf)
-	datasources, err := models.GetDatasourcesGetsByTypes(rt.Ctx, []string{dsf.Cate})
-	ginx.Dangerous(err)
-
-	nameToID := make(map[string]int64)
-	IDToName := make(map[int64]string)
-	for _, ds := range datasources {
-		nameToID[ds.Name] = ds.Id
-		IDToName[ds.Id] = ds.Name
-	}
-
-	ids := models.GetDatasourceIDsByDatasourceQueries(dsf.DatasourceQueries, IDToName, nameToID)
-	var req []datasourceQueryResp
-	for _, id := range ids {
-		req = append(req, datasourceQueryResp{
-			ID:   id,
-			Name: IDToName[id],
-		})
-	}
-	ginx.NewRender(c).Data(req, err)
-}

center/router/router_datasource_db.go

@@ -1,101 +0,0 @@
-package router
-
-import (
-	"context"
-
-	"github.com/ccfos/nightingale/v6/dscache"
-	"github.com/ccfos/nightingale/v6/dskit/types"
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-func (rt *Router) ShowDatabases(c *gin.Context) {
-	var f models.QueryParam
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-
-	var databases []string
-	var err error
-	type DatabaseShower interface {
-		ShowDatabases(context.Context) ([]string, error)
-	}
-	switch plug.(type) {
-	case DatabaseShower:
-		databases, err = plug.(DatabaseShower).ShowDatabases(c.Request.Context())
-		ginx.Dangerous(err)
-	default:
-		ginx.Bomb(200, "datasource not exists")
-	}
-
-	if len(databases) == 0 {
-		databases = make([]string, 0)
-	}
-
-	ginx.NewRender(c).Data(databases, nil)
-}
-
-func (rt *Router) ShowTables(c *gin.Context) {
-	var f models.QueryParam
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-
-	// 只接受一个入参
-	tables := make([]string, 0)
-	var err error
-	type TableShower interface {
-		ShowTables(ctx context.Context, database string) ([]string, error)
-	}
-	switch plug.(type) {
-	case TableShower:
-		if len(f.Querys) > 0 {
-			database, ok := f.Querys[0].(string)
-			if ok {
-				tables, err = plug.(TableShower).ShowTables(c.Request.Context(), database)
-			}
-		}
-	default:
-		ginx.Bomb(200, "datasource not exists")
-	}
-	ginx.NewRender(c).Data(tables, err)
-}
-
-func (rt *Router) DescribeTable(c *gin.Context) {
-	var f models.QueryParam
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-	// 只接受一个入参
-	columns := make([]*types.ColumnProperty, 0)
-	var err error
-	type TableDescriber interface {
-		DescribeTable(context.Context, interface{}) ([]*types.ColumnProperty, error)
-	}
-	switch plug.(type) {
-	case TableDescriber:
-		client := plug.(TableDescriber)
-		if len(f.Querys) > 0 {
-			columns, err = client.DescribeTable(c.Request.Context(), f.Querys[0])
-		}
-	default:
-		ginx.Bomb(200, "datasource not exists")
-	}
-
-	ginx.NewRender(c).Data(columns, err)
-}

center/router/router_embedded.go

@@ -1,141 +0,0 @@
-package router
-
-import (
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func (rt *Router) embeddedProductGets(c *gin.Context) {
-	products, err := models.EmbeddedProductGets(rt.Ctx)
-	ginx.Dangerous(err)
-	// 获取当前用户可访问的Group ID 列表
-	me := c.MustGet("user").(*models.User)
-
-	if me.IsAdmin() {
-		ginx.NewRender(c).Data(products, err)
-		return
-	}
-
-	gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-	bgSet := make(map[int64]struct{}, len(gids))
-	for _, id := range gids {
-		bgSet[id] = struct{}{}
-	}
-
-	// 过滤出公开或有权限访问的私有 product link
-	var result []*models.EmbeddedProduct
-	for _, product := range products {
-		if !product.IsPrivate {
-			result = append(result, product)
-			continue
-		}
-
-		for _, tid := range product.TeamIDs {
-			if _, ok := bgSet[tid]; ok {
-				result = append(result, product)
-				break
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(result, err)
-}
-
-func (rt *Router) embeddedProductGet(c *gin.Context) {
-	id := ginx.UrlParamInt64(c, "id")
-	if id <= 0 {
-		ginx.Bomb(400, "invalid id")
-	}
-
-	data, err := models.GetEmbeddedProductByID(rt.Ctx, id)
-	ginx.Dangerous(err)
-
-	me := c.MustGet("user").(*models.User)
-	hashPermission, err := hasEmbeddedProductAccess(rt.Ctx, me, data)
-	ginx.Dangerous(err)
-
-	if !hashPermission {
-		ginx.Bomb(403, "forbidden")
-	}
-
-	ginx.NewRender(c).Data(data, nil)
-}
-
-func (rt *Router) embeddedProductAdd(c *gin.Context) {
-	var eps []models.EmbeddedProduct
-	ginx.BindJSON(c, &eps)
-
-	me := c.MustGet("user").(*models.User)
-
-	for i := range eps {
-		eps[i].CreateBy = me.Nickname
-		eps[i].UpdateBy = me.Nickname
-	}
-
-	err := models.AddEmbeddedProduct(rt.Ctx, eps)
-	ginx.NewRender(c).Message(err)
-}
-
-func (rt *Router) embeddedProductPut(c *gin.Context) {
-	var ep models.EmbeddedProduct
-	id := ginx.UrlParamInt64(c, "id")
-	ginx.BindJSON(c, &ep)
-
-	if id <= 0 {
-		ginx.Bomb(400, "invalid id")
-	}
-
-	oldProduct, err := models.GetEmbeddedProductByID(rt.Ctx, id)
-	ginx.Dangerous(err)
-	me := c.MustGet("user").(*models.User)
-
-	now := time.Now().Unix()
-	oldProduct.Name = ep.Name
-	oldProduct.URL = ep.URL
-	oldProduct.IsPrivate = ep.IsPrivate
-	oldProduct.TeamIDs = ep.TeamIDs
-	oldProduct.UpdateBy = me.Username
-	oldProduct.UpdateAt = now
-
-	err = models.UpdateEmbeddedProduct(rt.Ctx, oldProduct)
-	ginx.NewRender(c).Message(err)
-}
-
-func (rt *Router) embeddedProductDelete(c *gin.Context) {
-	id := ginx.UrlParamInt64(c, "id")
-	if id <= 0 {
-		ginx.Bomb(400, "invalid id")
-	}
-
-	err := models.DeleteEmbeddedProduct(rt.Ctx, id)
-	ginx.NewRender(c).Message(err)
-}
-
-func hasEmbeddedProductAccess(ctx *ctx.Context, user *models.User, ep *models.EmbeddedProduct) (bool, error) {
-	if user.IsAdmin() || !ep.IsPrivate {
-		return true, nil
-	}
-
-	gids, err := models.MyGroupIds(ctx, user.Id)
-	if err != nil {
-		return false, err
-	}
-
-	groupSet := make(map[int64]struct{}, len(gids))
-	for _, gid := range gids {
-		groupSet[gid] = struct{}{}
-	}
-
-	for _, tid := range ep.TeamIDs {
-		if _, ok := groupSet[tid]; ok {
-			return true, nil
-		}
-	}
-
-	return false, nil
-}

center/router/router_es.go

@@ -1,77 +0,0 @@
-package router
-
-import (
-	"github.com/ccfos/nightingale/v6/datasource/es"
-	"github.com/ccfos/nightingale/v6/dscache"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type IndexReq struct {
-	Cate         string `json:"cate"`
-	DatasourceId int64  `json:"datasource_id"`
-	Index        string `json:"index"`
-}
-
-type FieldValueReq struct {
-	Cate         string   `json:"cate"`
-	DatasourceId int64    `json:"datasource_id"`
-	Index        string   `json:"index"`
-	Query        FieldObj `json:"query"`
-}
-
-type FieldObj struct {
-	Find  string `json:"find"`
-	Field string `json:"field"`
-	Query string `json:"query"`
-}
-
-func (rt *Router) QueryIndices(c *gin.Context) {
-	var f IndexReq
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-
-	indices, err := plug.(*es.Elasticsearch).QueryIndices()
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(indices, nil)
-}
-
-func (rt *Router) QueryFields(c *gin.Context) {
-	var f IndexReq
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-
-	fields, err := plug.(*es.Elasticsearch).QueryFields([]string{f.Index})
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(fields, nil)
-}
-
-func (rt *Router) QueryESVariable(c *gin.Context) {
-	var f FieldValueReq
-	ginx.BindJSON(c, &f)
-
-	plug, exists := dscache.DsCache.Get(f.Cate, f.DatasourceId)
-	if !exists {
-		logger.Warningf("cluster:%d not exists", f.DatasourceId)
-		ginx.Bomb(200, "cluster not exists")
-	}
-
-	fields, err := plug.(*es.Elasticsearch).QueryFieldValue([]string{f.Index}, f.Query.Field, f.Query.Query)
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(fields, nil)
-}

center/router/router_es_index_pattern.go

@@ -1,81 +0,0 @@
-package router
-
-import (
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-// 创建 ES Index Pattern
-func (rt *Router) esIndexPatternAdd(c *gin.Context) {
-	var f models.EsIndexPattern
-	ginx.BindJSON(c, &f)
-
-	username := c.MustGet("username").(string)
-	now := time.Now().Unix()
-	f.CreateAt = now
-	f.CreateBy = username
-	f.UpdateAt = now
-	f.UpdateBy = username
-
-	err := f.Add(rt.Ctx)
-	ginx.NewRender(c).Message(err)
-}
-
-// 更新 ES Index Pattern
-func (rt *Router) esIndexPatternPut(c *gin.Context) {
-	var f models.EsIndexPattern
-	ginx.BindJSON(c, &f)
-
-	id := ginx.QueryInt64(c, "id")
-
-	esIndexPattern, err := models.EsIndexPatternGetById(rt.Ctx, id)
-	ginx.Dangerous(err)
-
-	if esIndexPattern == nil {
-		ginx.NewRender(c, http.StatusNotFound).Message("No such EsIndexPattern")
-		return
-	}
-
-	f.UpdateBy = c.MustGet("username").(string)
-
-	ginx.NewRender(c).Message(esIndexPattern.Update(rt.Ctx, f))
-}
-
-// 删除 ES Index Pattern
-func (rt *Router) esIndexPatternDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-
-	if len(f.Ids) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "ids empty")
-	}
-
-	ginx.NewRender(c).Message(models.EsIndexPatternDel(rt.Ctx, f.Ids))
-}
-
-// ES Index Pattern列表
-func (rt *Router) esIndexPatternGetList(c *gin.Context) {
-	datasourceId := ginx.QueryInt64(c, "datasource_id", 0)
-
-	var lst []*models.EsIndexPattern
-	var err error
-	if datasourceId != 0 {
-		lst, err = models.EsIndexPatternGets(rt.Ctx, "datasource_id = ?", datasourceId)
-	} else {
-		lst, err = models.EsIndexPatternGets(rt.Ctx, "")
-	}
-
-	ginx.NewRender(c).Data(lst, err)
-}
-
-// ES Index Pattern 单个数据
-func (rt *Router) esIndexPatternGet(c *gin.Context) {
-	id := ginx.QueryInt64(c, "id")
-
-	item, err := models.EsIndexPatternGet(rt.Ctx, "id=?", id)
-	ginx.NewRender(c).Data(item, err)
-}

center/router/router_event_pipeline.go

@@ -1,245 +0,0 @@
-package router
-
-import (
-	"net/http"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-// 获取事件Pipeline列表
-func (rt *Router) eventPipelinesList(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-	pipelines, err := models.ListEventPipelines(rt.Ctx)
-	ginx.Dangerous(err)
-
-	allTids := make([]int64, 0)
-	for _, pipeline := range pipelines {
-		allTids = append(allTids, pipeline.TeamIds...)
-	}
-	ugMap, err := models.UserGroupIdAndNameMap(rt.Ctx, allTids)
-	ginx.Dangerous(err)
-	for _, pipeline := range pipelines {
-		for _, tid := range pipeline.TeamIds {
-			pipeline.TeamNames = append(pipeline.TeamNames, ugMap[tid])
-		}
-	}
-
-	gids, err := models.MyGroupIdsMap(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-
-	if me.IsAdmin() {
-		ginx.NewRender(c).Data(pipelines, nil)
-		return
-	}
-
-	res := make([]*models.EventPipeline, 0)
-	for _, pipeline := range pipelines {
-		for _, tid := range pipeline.TeamIds {
-			if _, ok := gids[tid]; ok {
-				res = append(res, pipeline)
-				break
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(res, nil)
-}
-
-// 获取单个事件Pipeline详情
-func (rt *Router) getEventPipeline(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-	id := ginx.UrlParamInt64(c, "id")
-	pipeline, err := models.GetEventPipeline(rt.Ctx, id)
-	ginx.Dangerous(err)
-	ginx.Dangerous(me.CheckGroupPermission(rt.Ctx, pipeline.TeamIds))
-
-	err = pipeline.FillTeamNames(rt.Ctx)
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(pipeline, nil)
-}
-
-// 创建事件Pipeline
-func (rt *Router) addEventPipeline(c *gin.Context) {
-	var pipeline models.EventPipeline
-	ginx.BindJSON(c, &pipeline)
-
-	user := c.MustGet("user").(*models.User)
-	now := time.Now().Unix()
-	pipeline.CreateBy = user.Username
-	pipeline.CreateAt = now
-	pipeline.UpdateAt = now
-	pipeline.UpdateBy = user.Username
-
-	err := pipeline.Verify()
-	if err != nil {
-		ginx.Bomb(http.StatusBadRequest, err.Error())
-	}
-
-	ginx.Dangerous(user.CheckGroupPermission(rt.Ctx, pipeline.TeamIds))
-	err = models.CreateEventPipeline(rt.Ctx, &pipeline)
-	ginx.NewRender(c).Message(err)
-}
-
-// 更新事件Pipeline
-func (rt *Router) updateEventPipeline(c *gin.Context) {
-	var f models.EventPipeline
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	f.UpdateBy = me.Username
-	f.UpdateAt = time.Now().Unix()
-
-	pipeline, err := models.GetEventPipeline(rt.Ctx, f.ID)
-	if err != nil {
-		ginx.Bomb(http.StatusNotFound, "No such event pipeline")
-	}
-	ginx.Dangerous(me.CheckGroupPermission(rt.Ctx, pipeline.TeamIds))
-
-	ginx.NewRender(c).Message(pipeline.Update(rt.Ctx, &f))
-}
-
-// 删除事件Pipeline
-func (rt *Router) deleteEventPipelines(c *gin.Context) {
-	var f struct {
-		Ids []int64 `json:"ids"`
-	}
-	ginx.BindJSON(c, &f)
-
-	if len(f.Ids) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "ids required")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	for _, id := range f.Ids {
-		pipeline, err := models.GetEventPipeline(rt.Ctx, id)
-		ginx.Dangerous(err)
-		ginx.Dangerous(me.CheckGroupPermission(rt.Ctx, pipeline.TeamIds))
-	}
-
-	err := models.DeleteEventPipelines(rt.Ctx, f.Ids)
-	ginx.NewRender(c).Message(err)
-}
-
-// 测试事件Pipeline
-func (rt *Router) tryRunEventPipeline(c *gin.Context) {
-	var f struct {
-		EventId        int64                `json:"event_id"`
-		PipelineConfig models.EventPipeline `json:"pipeline_config"`
-	}
-	ginx.BindJSON(c, &f)
-
-	hisEvent, err := models.AlertHisEventGetById(rt.Ctx, f.EventId)
-	if err != nil || hisEvent == nil {
-		ginx.Bomb(http.StatusBadRequest, "event not found")
-	}
-	event := hisEvent.ToCur()
-
-	for _, p := range f.PipelineConfig.ProcessorConfigs {
-		processor, err := models.GetProcessorByType(p.Typ, p.Config)
-		if err != nil {
-			ginx.Bomb(http.StatusBadRequest, "get processor: %+v err: %+v", p, err)
-		}
-		event, _, err = processor.Process(rt.Ctx, event)
-		if err != nil {
-			ginx.Bomb(http.StatusBadRequest, "processor: %+v err: %+v", p, err)
-		}
-
-		if event == nil {
-			ginx.NewRender(c).Data(map[string]interface{}{
-				"event":  event,
-				"result": "event is dropped",
-			}, nil)
-			return
-		}
-	}
-
-	ginx.NewRender(c).Data(event, nil)
-}
-
-// 测试事件处理器
-func (rt *Router) tryRunEventProcessor(c *gin.Context) {
-	var f struct {
-		EventId         int64                  `json:"event_id"`
-		ProcessorConfig models.ProcessorConfig `json:"processor_config"`
-	}
-	ginx.BindJSON(c, &f)
-
-	hisEvent, err := models.AlertHisEventGetById(rt.Ctx, f.EventId)
-	if err != nil || hisEvent == nil {
-		ginx.Bomb(http.StatusBadRequest, "event not found")
-	}
-	event := hisEvent.ToCur()
-
-	processor, err := models.GetProcessorByType(f.ProcessorConfig.Typ, f.ProcessorConfig.Config)
-	if err != nil {
-		ginx.Bomb(http.StatusBadRequest, "get processor err: %+v", err)
-	}
-	event, res, err := processor.Process(rt.Ctx, event)
-	if err != nil {
-		ginx.Bomb(http.StatusBadRequest, "processor err: %+v", err)
-	}
-
-	ginx.NewRender(c).Data(map[string]interface{}{
-		"event":  event,
-		"result": res,
-	}, nil)
-}
-
-func (rt *Router) tryRunEventProcessorByNotifyRule(c *gin.Context) {
-	var f struct {
-		EventId         int64                   `json:"event_id"`
-		PipelineConfigs []models.PipelineConfig `json:"pipeline_configs"`
-	}
-	ginx.BindJSON(c, &f)
-
-	hisEvent, err := models.AlertHisEventGetById(rt.Ctx, f.EventId)
-	if err != nil || hisEvent == nil {
-		ginx.Bomb(http.StatusBadRequest, "event not found")
-	}
-	event := hisEvent.ToCur()
-
-	pids := make([]int64, 0)
-	for _, pc := range f.PipelineConfigs {
-		if pc.Enable {
-			pids = append(pids, pc.PipelineId)
-		}
-	}
-
-	pipelines, err := models.GetEventPipelinesByIds(rt.Ctx, pids)
-	if err != nil {
-		ginx.Bomb(http.StatusBadRequest, "processors not found")
-	}
-
-	for _, pl := range pipelines {
-		for _, p := range pl.ProcessorConfigs {
-			processor, err := models.GetProcessorByType(p.Typ, p.Config)
-			if err != nil {
-				ginx.Bomb(http.StatusBadRequest, "get processor: %+v err: %+v", p, err)
-			}
-
-			event, _, err := processor.Process(rt.Ctx, event)
-			if err != nil {
-				ginx.Bomb(http.StatusBadRequest, "processor: %+v err: %+v", p, err)
-			}
-			if event == nil {
-				ginx.NewRender(c).Data(map[string]interface{}{
-					"event":  event,
-					"result": "event is dropped",
-				}, nil)
-				return
-			}
-		}
-	}
-
-	ginx.NewRender(c).Data(event, nil)
-}
-
-func (rt *Router) eventPipelinesListByService(c *gin.Context) {
-	pipelines, err := models.ListEventPipelines(rt.Ctx)
-	ginx.NewRender(c).Data(pipelines, err)
-}

center/router/router_funcs.go

@@ -1,210 +0,0 @@
-package router
-
-import (
-	"net/http"
-	"strconv"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-const defaultLimit = 300
-
-func (rt *Router) statistic(c *gin.Context) {
-	name := ginx.QueryStr(c, "name")
-	var model interface{}
-	var err error
-	var statistics *models.Statistics
-	switch name {
-	case "alert_mute":
-		model = models.AlertMute{}
-	case "alert_rule":
-		model = models.AlertRule{}
-	case "alert_subscribe":
-		model = models.AlertSubscribe{}
-	case "busi_group":
-		model = models.BusiGroup{}
-	case "recording_rule":
-		model = models.RecordingRule{}
-	case "target":
-		model = models.Target{}
-	case "user":
-		model = models.User{}
-	case "user_group":
-		model = models.UserGroup{}
-	case "notify_rule":
-		model = models.NotifyRule{}
-	case "notify_channel":
-		model = models.NotifyChannel{}
-	case "event_pipeline":
-		statistics, err = models.EventPipelineStatistics(rt.Ctx)
-		ginx.NewRender(c).Data(statistics, err)
-		return
-	case "datasource":
-		// datasource update_at is different from others
-		statistics, err = models.DatasourceStatistics(rt.Ctx)
-		ginx.NewRender(c).Data(statistics, err)
-		return
-	case "user_variable":
-		statistics, err = models.ConfigsUserVariableStatistics(rt.Ctx)
-		ginx.NewRender(c).Data(statistics, err)
-		return
-	case "cval":
-		statistics, err = models.ConfigCvalStatistics(rt.Ctx)
-		ginx.NewRender(c).Data(statistics, err)
-		return
-	case "message_template":
-		statistics, err = models.MessageTemplateStatistics(rt.Ctx)
-		ginx.NewRender(c).Data(statistics, err)
-		return
-	default:
-		ginx.Bomb(http.StatusBadRequest, "invalid name")
-	}
-
-	statistics, err = models.StatisticsGet(rt.Ctx, model)
-	ginx.NewRender(c).Data(statistics, err)
-}
-
-func queryDatasourceIds(c *gin.Context) []int64 {
-	datasourceIds := ginx.QueryStr(c, "datasource_ids", "")
-	datasourceIds = strings.ReplaceAll(datasourceIds, ",", " ")
-	idsStr := strings.Fields(datasourceIds)
-	ids := make([]int64, len(idsStr))
-	for i, idStr := range idsStr {
-		id, _ := strconv.ParseInt(idStr, 10, 64)
-		ids[i] = id
-	}
-	return ids
-}
-
-func queryStrListField(c *gin.Context, fieldName string, sep ...string) []string {
-	str := ginx.QueryStr(c, fieldName, "")
-	if str == "" {
-		return nil
-	}
-
-	lst := []string{str}
-	for _, s := range sep {
-		var newLst []string
-		for _, str := range lst {
-			newLst = append(newLst, strings.Split(str, s)...)
-		}
-		lst = newLst
-	}
-	return lst
-}
-
-type idsForm struct {
-	Ids               []int64 `json:"ids"`
-	IsSyncToFlashDuty bool    `json:"is_sync_to_flashduty"`
-}
-
-func (f idsForm) Verify() {
-	if len(f.Ids) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "ids empty")
-	}
-}
-
-func User(ctx *ctx.Context, id int64) *models.User {
-	obj, err := models.UserGetById(ctx, id)
-	ginx.Dangerous(err)
-
-	if obj == nil {
-		ginx.Bomb(http.StatusNotFound, "No such user")
-	}
-
-	return obj
-}
-
-func UserGroup(ctx *ctx.Context, id int64) *models.UserGroup {
-	obj, err := models.UserGroupGetById(ctx, id)
-	ginx.Dangerous(err)
-
-	if obj == nil {
-		ginx.Bomb(http.StatusNotFound, "No such UserGroup")
-	}
-
-	return obj
-}
-
-func BusiGroup(ctx *ctx.Context, id int64) *models.BusiGroup {
-	obj, err := models.BusiGroupGetById(ctx, id)
-	ginx.Dangerous(err)
-
-	if obj == nil {
-		ginx.Bomb(http.StatusNotFound, "No such BusiGroup")
-	}
-
-	return obj
-}
-
-func Dashboard(ctx *ctx.Context, id int64) *models.Dashboard {
-	obj, err := models.DashboardGet(ctx, "id=?", id)
-	ginx.Dangerous(err)
-
-	if obj == nil {
-		ginx.Bomb(http.StatusNotFound, "No such dashboard")
-	}
-
-	return obj
-}
-
-type DoneIdsReply struct {
-	Err string `json:"err"`
-	Dat struct {
-		List []int64 `json:"list"`
-	} `json:"dat"`
-}
-
-type TaskCreateReply struct {
-	Err string `json:"err"`
-	Dat int64  `json:"dat"` // task.id
-}
-
-func Username(c *gin.Context) string {
-	username := c.GetString(gin.AuthUserKey)
-	if username == "" {
-		user := c.MustGet("user").(*models.User)
-		username = user.Username
-	}
-	return username
-}
-
-func HasPermission(ctx *ctx.Context, c *gin.Context, sourceType, sourceId string, isAnonymousAccess bool) bool {
-	if sourceType == "event" && isAnonymousAccess {
-		return true
-	}
-
-	// 尝试从请求中获取 __token 参数
-	token := ginx.QueryStr(c, "__token", "")
-
-	// 如果有 __token 参数，验证其合法性
-	if token != "" {
-		return ValidateSourceToken(ctx, sourceType, sourceId, token)
-	}
-
-	return false
-}
-
-func ValidateSourceToken(ctx *ctx.Context, sourceType, sourceId, token string) bool {
-	if token == "" {
-		return false
-	}
-
-	// 根据源类型、源ID和令牌获取源令牌记录
-	sourceToken, err := models.GetSourceTokenBySource(ctx, sourceType, sourceId, token)
-	if err != nil {
-		return false
-	}
-
-	// 检查令牌是否过期
-	if sourceToken.IsExpired() {
-		return false
-	}
-
-	return true
-}

center/router/router_heartbeat.go

@@ -1,203 +0,0 @@
-package router
-
-import (
-	"compress/gzip"
-	"encoding/json"
-	"errors"
-	"io/ioutil"
-	"sort"
-	"strconv"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/center/metas"
-	"github.com/ccfos/nightingale/v6/memsto"
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/ctx"
-	"github.com/ccfos/nightingale/v6/pushgw/idents"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type HeartbeatHookFunc func(ident string) map[string]interface{}
-
-func (rt *Router) heartbeat(c *gin.Context) {
-	req, err := HandleHeartbeat(c, rt.Ctx, rt.Alert.Heartbeat.EngineName, rt.MetaSet, rt.IdentSet, rt.TargetCache)
-	ginx.Dangerous(err)
-
-	m := rt.HeartbeatHook(req.Hostname)
-	ginx.NewRender(c).Data(m, err)
-}
-
-func HandleHeartbeat(c *gin.Context, ctx *ctx.Context, engineName string, metaSet *metas.Set, identSet *idents.Set, targetCache *memsto.TargetCacheType) (models.HostMeta, error) {
-	var bs []byte
-	var err error
-	var r *gzip.Reader
-	var req models.HostMeta
-	if c.GetHeader("Content-Encoding") == "gzip" {
-		r, err = gzip.NewReader(c.Request.Body)
-		if err != nil {
-			c.String(400, err.Error())
-			return req, err
-		}
-		defer r.Close()
-		bs, err = ioutil.ReadAll(r)
-		ginx.Dangerous(err)
-	} else {
-		defer c.Request.Body.Close()
-		bs, err = ioutil.ReadAll(c.Request.Body)
-		if err != nil {
-			return req, err
-		}
-	}
-
-	err = json.Unmarshal(bs, &req)
-	if err != nil {
-		return req, err
-	}
-
-	if req.Hostname == "" {
-		return req, errors.New("hostname is required")
-	}
-
-	// maybe from pushgw
-	if req.Offset == 0 {
-		req.Offset = (time.Now().UnixMilli() - req.UnixTime)
-	}
-
-	if req.RemoteAddr == "" {
-		req.RemoteAddr = c.ClientIP()
-	}
-
-	if req.EngineName == "" {
-		req.EngineName = engineName
-	}
-
-	metaSet.Set(req.Hostname, req)
-	var items = make(map[string]struct{})
-	items[req.Hostname] = struct{}{}
-	identSet.MSet(items)
-
-	if target, has := targetCache.Get(req.Hostname); has && target != nil {
-		gidsStr := ginx.QueryStr(c, "gid", "")
-		overwriteGids := ginx.QueryBool(c, "overwrite_gids", false)
-		hostIp := strings.TrimSpace(req.HostIp)
-		gids := strings.Split(gidsStr, ",")
-
-		if overwriteGids {
-			groupIds := make([]int64, 0)
-			for i := range gids {
-				if gids[i] == "" {
-					continue
-				}
-				groupId, err := strconv.ParseInt(gids[i], 10, 64)
-				if err != nil {
-					logger.Warningf("update target:%s group ids failed, err: %v", req.Hostname, err)
-					continue
-				}
-				groupIds = append(groupIds, groupId)
-			}
-
-			err := models.TargetOverrideBgids(ctx, []string{target.Ident}, groupIds, nil)
-			if err != nil {
-				logger.Warningf("update target:%s group ids failed, err: %v", target.Ident, err)
-			}
-		} else if gidsStr != "" {
-			for i := range gids {
-				groupId, err := strconv.ParseInt(gids[i], 10, 64)
-				if err != nil {
-					logger.Warningf("update target:%s group ids failed, err: %v", req.Hostname, err)
-					continue
-				}
-
-				if !target.MatchGroupId(groupId) {
-					err := models.TargetBindBgids(ctx, []string{target.Ident}, []int64{groupId}, nil)
-					if err != nil {
-						logger.Warningf("update target:%s group ids failed, err: %v", target.Ident, err)
-					}
-				}
-			}
-		}
-
-		newTarget := models.Target{}
-		targetNeedUpdate := false
-		if hostIp != "" && hostIp != target.HostIp {
-			newTarget.HostIp = hostIp
-			targetNeedUpdate = true
-		}
-
-		hostTagsMap := target.GetHostTagsMap()
-		hostTagNeedUpdate := false
-		if len(hostTagsMap) != len(req.GlobalLabels) {
-			hostTagNeedUpdate = true
-		} else {
-			for k, v := range req.GlobalLabels {
-				if v == "" {
-					continue
-				}
-
-				if tagv, ok := hostTagsMap[k]; !ok || tagv != v {
-					hostTagNeedUpdate = true
-					break
-				}
-			}
-		}
-
-		if hostTagNeedUpdate {
-			lst := []string{}
-			for k, v := range req.GlobalLabels {
-				lst = append(lst, k+"="+v)
-			}
-			sort.Strings(lst)
-			newTarget.HostTags = lst
-			targetNeedUpdate = true
-		}
-
-		userTagsMap := target.GetTagsMap()
-		userTagNeedUpdate := false
-		userTags := []string{}
-		for k, v := range userTagsMap {
-			if v == "" {
-				continue
-			}
-
-			if _, ok := req.GlobalLabels[k]; !ok {
-				userTags = append(userTags, k+"="+v)
-			} else { // 该key在hostTags中已经存在
-				userTagNeedUpdate = true
-			}
-		}
-
-		if userTagNeedUpdate {
-			newTarget.Tags = strings.Join(userTags, " ") + " "
-			targetNeedUpdate = true
-		}
-
-		if req.EngineName != "" && req.EngineName != target.EngineName {
-			newTarget.EngineName = req.EngineName
-			targetNeedUpdate = true
-		}
-
-		if req.AgentVersion != "" && req.AgentVersion != target.AgentVersion {
-			newTarget.AgentVersion = req.AgentVersion
-			targetNeedUpdate = true
-		}
-
-		if req.OS != "" && req.OS != target.OS {
-			newTarget.OS = req.OS
-			targetNeedUpdate = true
-		}
-
-		if targetNeedUpdate {
-			err := models.DB(ctx).Model(&target).Updates(newTarget).Error
-			if err != nil {
-				logger.Errorf("update target fields failed, err: %v", err)
-			}
-		}
-		logger.Debugf("heartbeat field:%+v target: %v", newTarget, *target)
-	}
-
-	return req, nil
-}

center/router/router_login.go

@@ -1,540 +0,0 @@
-package router
-
-import (
-	"encoding/base64"
-	"fmt"
-	"net/http"
-	"strconv"
-	"strings"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/cas"
-	"github.com/ccfos/nightingale/v6/pkg/ldapx"
-	"github.com/ccfos/nightingale/v6/pkg/oauth2x"
-	"github.com/ccfos/nightingale/v6/pkg/oidcx"
-	"github.com/ccfos/nightingale/v6/pkg/secu"
-
-	"github.com/dgrijalva/jwt-go"
-	"github.com/gin-gonic/gin"
-	"github.com/pelletier/go-toml/v2"
-	"github.com/toolkits/pkg/ginx"
-	"github.com/toolkits/pkg/logger"
-)
-
-type loginForm struct {
-	Username    string `json:"username" binding:"required"`
-	Password    string `json:"password" binding:"required"`
-	Captchaid   string `json:"captchaid"`
-	Verifyvalue string `json:"verifyvalue"`
-}
-
-func (rt *Router) loginPost(c *gin.Context) {
-	var f loginForm
-	ginx.BindJSON(c, &f)
-	logger.Infof("username:%s login from:%s", f.Username, c.ClientIP())
-
-	if rt.HTTP.ShowCaptcha.Enable {
-		if !CaptchaVerify(f.Captchaid, f.Verifyvalue) {
-			ginx.NewRender(c).Message("incorrect verification code")
-			return
-		}
-	}
-	authPassWord := f.Password
-	// need decode
-	if rt.HTTP.RSA.OpenRSA {
-		decPassWord, err := secu.Decrypt(f.Password, rt.HTTP.RSA.RSAPrivateKey, rt.HTTP.RSA.RSAPassWord)
-		if err != nil {
-			logger.Errorf("RSA Decrypt failed: %v username: %s", err, f.Username)
-			ginx.NewRender(c).Message(err)
-			return
-		}
-		authPassWord = decPassWord
-	}
-
-	var user *models.User
-	var err error
-	lc := rt.Sso.LDAP.Copy()
-	if lc.Enable {
-		user, err = ldapx.LdapLogin(rt.Ctx, f.Username, authPassWord, lc.DefaultRoles, lc.DefaultTeams, lc)
-		if err != nil {
-			logger.Debugf("ldap login failed: %v username: %s", err, f.Username)
-			var errLoginInN9e error
-			// to use n9e as the minimum guarantee for login
-			if user, errLoginInN9e = models.PassLogin(rt.Ctx, rt.Redis, f.Username, authPassWord); errLoginInN9e != nil {
-				ginx.NewRender(c).Message("ldap login failed: %v; n9e login failed: %v", err, errLoginInN9e)
-				return
-			}
-		} else {
-			user.RolesLst = strings.Fields(user.Roles)
-		}
-	} else {
-		user, err = models.PassLogin(rt.Ctx, rt.Redis, f.Username, authPassWord)
-		ginx.Dangerous(err)
-	}
-
-	if user == nil {
-		// Theoretically impossible
-		ginx.NewRender(c).Message("Username or password invalid")
-		return
-	}
-
-	userIdentity := fmt.Sprintf("%d-%s", user.Id, user.Username)
-
-	ts, err := rt.createTokens(rt.HTTP.JWTAuth.SigningKey, userIdentity)
-	ginx.Dangerous(err)
-	ginx.Dangerous(rt.createAuth(c.Request.Context(), userIdentity, ts))
-
-	ginx.NewRender(c).Data(gin.H{
-		"user":          user,
-		"access_token":  ts.AccessToken,
-		"refresh_token": ts.RefreshToken,
-	}, nil)
-}
-
-func (rt *Router) logoutPost(c *gin.Context) {
-	logger.Infof("username:%s logout from:%s", c.GetString("username"), c.ClientIP())
-	metadata, err := rt.extractTokenMetadata(c.Request)
-	if err != nil {
-		ginx.NewRender(c, http.StatusBadRequest).Message("failed to parse jwt token")
-		return
-	}
-
-	delErr := rt.deleteTokens(c.Request.Context(), metadata)
-	if delErr != nil {
-		ginx.NewRender(c).Message(http.StatusText(http.StatusInternalServerError))
-		return
-	}
-
-	var logoutAddr string
-	user := c.MustGet("user").(*models.User)
-	switch user.Belong {
-	case "oidc":
-		logoutAddr = rt.Sso.OIDC.GetSsoLogoutAddr()
-	case "cas":
-		logoutAddr = rt.Sso.CAS.GetSsoLogoutAddr()
-	case "oauth2":
-		logoutAddr = rt.Sso.OAuth2.GetSsoLogoutAddr()
-	}
-
-	ginx.NewRender(c).Data(logoutAddr, nil)
-}
-
-type refreshForm struct {
-	RefreshToken string `json:"refresh_token" binding:"required"`
-}
-
-func (rt *Router) refreshPost(c *gin.Context) {
-	var f refreshForm
-	ginx.BindJSON(c, &f)
-
-	// verify the token
-	token, err := jwt.Parse(f.RefreshToken, func(token *jwt.Token) (interface{}, error) {
-		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
-			return nil, fmt.Errorf("unexpected jwt signing method: %v", token.Header["alg"])
-		}
-		return []byte(rt.HTTP.JWTAuth.SigningKey), nil
-	})
-
-	// if there is an error, the token must have expired
-	if err != nil {
-		// redirect to login page
-		ginx.NewRender(c, http.StatusUnauthorized).Message("refresh token expired")
-		return
-	}
-
-	// Since token is valid, get the uuid:
-	claims, ok := token.Claims.(jwt.MapClaims) //the token claims should conform to MapClaims
-	if ok && token.Valid {
-		refreshUuid, ok := claims["refresh_uuid"].(string) //convert the interface to string
-		if !ok {
-			// Theoretically impossible
-			ginx.NewRender(c, http.StatusUnauthorized).Message("failed to parse refresh_uuid from jwt")
-			return
-		}
-
-		// 看这个 token 是否还存在 redis 中
-		val, err := rt.fetchAuth(c.Request.Context(), refreshUuid)
-		if err != nil || val == "" {
-			ginx.NewRender(c, http.StatusUnauthorized).Message("refresh token expired")
-			return
-		}
-
-		userIdentity, ok := claims["user_identity"].(string)
-		if !ok {
-			// Theoretically impossible
-			ginx.NewRender(c, http.StatusUnauthorized).Message("failed to parse user_identity from jwt")
-			return
-		}
-
-		userid, err := strconv.ParseInt(strings.Split(userIdentity, "-")[0], 10, 64)
-		if err != nil {
-			ginx.NewRender(c, http.StatusUnauthorized).Message("failed to parse user_identity from jwt")
-			return
-		}
-
-		u, err := models.UserGetById(rt.Ctx, userid)
-		if err != nil {
-			ginx.NewRender(c, http.StatusInternalServerError).Message("failed to query user by id")
-			return
-		}
-
-		if u == nil {
-			// user already deleted
-			ginx.NewRender(c, http.StatusUnauthorized).Message("user already deleted")
-			return
-		}
-
-		// Delete the previous Refresh Token
-		err = rt.deleteAuth(c.Request.Context(), refreshUuid)
-		if err != nil {
-			ginx.NewRender(c, http.StatusUnauthorized).Message(http.StatusText(http.StatusInternalServerError))
-			return
-		}
-
-		// Delete previous Access Token
-		rt.deleteAuth(c.Request.Context(), strings.Split(refreshUuid, "++")[0])
-
-		// Create new pairs of refresh and access tokens
-		ts, err := rt.createTokens(rt.HTTP.JWTAuth.SigningKey, userIdentity)
-		ginx.Dangerous(err)
-		ginx.Dangerous(rt.createAuth(c.Request.Context(), userIdentity, ts))
-
-		ginx.NewRender(c).Data(gin.H{
-			"access_token":  ts.AccessToken,
-			"refresh_token": ts.RefreshToken,
-		}, nil)
-	} else {
-		// redirect to login page
-		ginx.NewRender(c, http.StatusUnauthorized).Message("refresh token expired")
-	}
-}
-
-func (rt *Router) loginRedirect(c *gin.Context) {
-	redirect := ginx.QueryStr(c, "redirect", "/")
-
-	v, exists := c.Get("userid")
-	if exists {
-		userid := v.(int64)
-		user, err := models.UserGetById(rt.Ctx, userid)
-		ginx.Dangerous(err)
-		if user == nil {
-			ginx.Bomb(200, "user not found")
-		}
-
-		if user.Username != "" { // already login
-			ginx.NewRender(c).Data(redirect, nil)
-			return
-		}
-	}
-
-	if !rt.Sso.OIDC.Enable {
-		ginx.NewRender(c).Data("", nil)
-		return
-	}
-
-	redirect, err := rt.Sso.OIDC.Authorize(rt.Redis, redirect)
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(redirect, err)
-}
-
-type CallbackOutput struct {
-	Redirect     string       `json:"redirect"`
-	User         *models.User `json:"user"`
-	AccessToken  string       `json:"access_token"`
-	RefreshToken string       `json:"refresh_token"`
-}
-
-func (rt *Router) loginCallback(c *gin.Context) {
-	code := ginx.QueryStr(c, "code", "")
-	state := ginx.QueryStr(c, "state", "")
-
-	ret, err := rt.Sso.OIDC.Callback(rt.Redis, c.Request.Context(), code, state)
-	if err != nil {
-		logger.Errorf("sso_callback fail. code:%s, state:%s, get ret: %+v. error: %v", code, state, ret, err)
-		ginx.NewRender(c).Data(CallbackOutput{}, err)
-		return
-	}
-
-	user, err := models.UserGet(rt.Ctx, "username=?", ret.Username)
-	ginx.Dangerous(err)
-
-	if user != nil {
-		if rt.Sso.OIDC.CoverAttributes {
-			updatedFields := user.UpdateSsoFields("oidc", ret.Nickname, ret.Phone, ret.Email)
-			ginx.Dangerous(user.Update(rt.Ctx, "update_at", updatedFields...))
-		}
-	} else {
-		user = new(models.User)
-		user.FullSsoFields("oidc", ret.Username, ret.Nickname, ret.Phone, ret.Email, rt.Sso.OIDC.DefaultRoles)
-		// create user from oidc
-		ginx.Dangerous(user.Add(rt.Ctx))
-
-		if len(rt.Sso.OIDC.DefaultTeams) > 0 {
-			for _, gid := range rt.Sso.OIDC.DefaultTeams {
-				err = models.UserGroupMemberAdd(rt.Ctx, gid, user.Id)
-				if err != nil {
-					logger.Errorf("user:%v UserGroupMemberAdd: %s", user, err)
-				}
-			}
-		}
-	}
-
-	// set user login state
-	userIdentity := fmt.Sprintf("%d-%s", user.Id, user.Username)
-	ts, err := rt.createTokens(rt.HTTP.JWTAuth.SigningKey, userIdentity)
-	ginx.Dangerous(err)
-	ginx.Dangerous(rt.createAuth(c.Request.Context(), userIdentity, ts))
-
-	redirect := "/"
-	if ret.Redirect != "/login" {
-		redirect = ret.Redirect
-	}
-
-	ginx.NewRender(c).Data(CallbackOutput{
-		Redirect:     redirect,
-		User:         user,
-		AccessToken:  ts.AccessToken,
-		RefreshToken: ts.RefreshToken,
-	}, nil)
-}
-
-type RedirectOutput struct {
-	Redirect string `json:"redirect"`
-	State    string `json:"state"`
-}
-
-func (rt *Router) loginRedirectCas(c *gin.Context) {
-	redirect := ginx.QueryStr(c, "redirect", "/")
-
-	v, exists := c.Get("userid")
-	if exists {
-		userid := v.(int64)
-		user, err := models.UserGetById(rt.Ctx, userid)
-		ginx.Dangerous(err)
-		if user == nil {
-			ginx.Bomb(200, "user not found")
-		}
-
-		if user.Username != "" { // already login
-			ginx.NewRender(c).Data(redirect, nil)
-			return
-		}
-	}
-
-	if !rt.Sso.CAS.Enable {
-		logger.Error("cas is not enable")
-		ginx.NewRender(c).Data("", nil)
-		return
-	}
-
-	redirect, state, err := rt.Sso.CAS.Authorize(rt.Redis, redirect)
-
-	ginx.Dangerous(err)
-	ginx.NewRender(c).Data(RedirectOutput{
-		Redirect: redirect,
-		State:    state,
-	}, err)
-}
-
-func (rt *Router) loginCallbackCas(c *gin.Context) {
-	ticket := ginx.QueryStr(c, "ticket", "")
-	state := ginx.QueryStr(c, "state", "")
-	ret, err := rt.Sso.CAS.ValidateServiceTicket(c.Request.Context(), ticket, state, rt.Redis)
-	if err != nil {
-		logger.Errorf("ValidateServiceTicket: %s", err)
-		ginx.NewRender(c).Data("", err)
-		return
-	}
-	user, err := models.UserGet(rt.Ctx, "username=?", ret.Username)
-	if err != nil {
-		logger.Errorf("UserGet: %s", err)
-	}
-	ginx.Dangerous(err)
-	if user != nil {
-		if rt.Sso.CAS.CoverAttributes {
-			updatedFields := user.UpdateSsoFields("cas", ret.Nickname, ret.Phone, ret.Email)
-			ginx.Dangerous(user.Update(rt.Ctx, "update_at", updatedFields...))
-		}
-	} else {
-		user = new(models.User)
-		user.FullSsoFields("cas", ret.Username, ret.Nickname, ret.Phone, ret.Email, rt.Sso.CAS.DefaultRoles)
-		// create user from cas
-		ginx.Dangerous(user.Add(rt.Ctx))
-	}
-
-	// set user login state
-	userIdentity := fmt.Sprintf("%d-%s", user.Id, user.Username)
-	ts, err := rt.createTokens(rt.HTTP.JWTAuth.SigningKey, userIdentity)
-	if err != nil {
-		logger.Errorf("createTokens: %s", err)
-	}
-	ginx.Dangerous(err)
-	ginx.Dangerous(rt.createAuth(c.Request.Context(), userIdentity, ts))
-
-	redirect := "/"
-	if ret.Redirect != "/login" {
-		redirect = ret.Redirect
-	}
-	ginx.NewRender(c).Data(CallbackOutput{
-		Redirect:     redirect,
-		User:         user,
-		AccessToken:  ts.AccessToken,
-		RefreshToken: ts.RefreshToken,
-	}, nil)
-}
-
-func (rt *Router) loginRedirectOAuth(c *gin.Context) {
-	redirect := ginx.QueryStr(c, "redirect", "/")
-
-	v, exists := c.Get("userid")
-	if exists {
-		userid := v.(int64)
-		user, err := models.UserGetById(rt.Ctx, userid)
-		ginx.Dangerous(err)
-		if user == nil {
-			ginx.Bomb(200, "user not found")
-		}
-
-		if user.Username != "" { // already login
-			ginx.NewRender(c).Data(redirect, nil)
-			return
-		}
-	}
-
-	if !rt.Sso.OAuth2.Enable {
-		ginx.NewRender(c).Data("", nil)
-		return
-	}
-
-	redirect, err := rt.Sso.OAuth2.Authorize(rt.Redis, redirect)
-	ginx.Dangerous(err)
-
-	ginx.NewRender(c).Data(redirect, err)
-}
-
-func (rt *Router) loginCallbackOAuth(c *gin.Context) {
-	code := ginx.QueryStr(c, "code", "")
-	state := ginx.QueryStr(c, "state", "")
-
-	ret, err := rt.Sso.OAuth2.Callback(rt.Redis, c.Request.Context(), code, state)
-	if err != nil {
-		logger.Debugf("sso.callback() get ret %+v error %v", ret, err)
-		ginx.NewRender(c).Data(CallbackOutput{}, err)
-		return
-	}
-
-	user, err := models.UserGet(rt.Ctx, "username=?", ret.Username)
-	ginx.Dangerous(err)
-
-	if user != nil {
-		if rt.Sso.OAuth2.CoverAttributes {
-			updatedFields := user.UpdateSsoFields("oauth2", ret.Nickname, ret.Phone, ret.Email)
-			ginx.Dangerous(user.Update(rt.Ctx, "update_at", updatedFields...))
-		}
-	} else {
-		user = new(models.User)
-		user.FullSsoFields("oauth2", ret.Username, ret.Nickname, ret.Phone, ret.Email, rt.Sso.OAuth2.DefaultRoles)
-		// create user from oidc
-		ginx.Dangerous(user.Add(rt.Ctx))
-	}
-
-	// set user login state
-	userIdentity := fmt.Sprintf("%d-%s", user.Id, user.Username)
-	ts, err := rt.createTokens(rt.HTTP.JWTAuth.SigningKey, userIdentity)
-	ginx.Dangerous(err)
-	ginx.Dangerous(rt.createAuth(c.Request.Context(), userIdentity, ts))
-
-	redirect := "/"
-	if ret.Redirect != "/login" {
-		redirect = ret.Redirect
-	}
-
-	ginx.NewRender(c).Data(CallbackOutput{
-		Redirect:     redirect,
-		User:         user,
-		AccessToken:  ts.AccessToken,
-		RefreshToken: ts.RefreshToken,
-	}, nil)
-}
-
-type SsoConfigOutput struct {
-	OidcDisplayName  string `json:"oidcDisplayName"`
-	CasDisplayName   string `json:"casDisplayName"`
-	OauthDisplayName string `json:"oauthDisplayName"`
-}
-
-func (rt *Router) ssoConfigNameGet(c *gin.Context) {
-	var oidcDisplayName, casDisplayName, oauthDisplayName string
-	if rt.Sso.OIDC != nil {
-		oidcDisplayName = rt.Sso.OIDC.GetDisplayName()
-	}
-
-	if rt.Sso.CAS != nil {
-		casDisplayName = rt.Sso.CAS.GetDisplayName()
-	}
-
-	if rt.Sso.OAuth2 != nil {
-		oauthDisplayName = rt.Sso.OAuth2.GetDisplayName()
-	}
-
-	ginx.NewRender(c).Data(SsoConfigOutput{
-		OidcDisplayName:  oidcDisplayName,
-		CasDisplayName:   casDisplayName,
-		OauthDisplayName: oauthDisplayName,
-	}, nil)
-}
-
-func (rt *Router) ssoConfigGets(c *gin.Context) {
-	ginx.NewRender(c).Data(models.SsoConfigGets(rt.Ctx))
-}
-
-func (rt *Router) ssoConfigUpdate(c *gin.Context) {
-	var f models.SsoConfig
-	ginx.BindJSON(c, &f)
-
-	err := f.Update(rt.Ctx)
-	ginx.Dangerous(err)
-
-	switch f.Name {
-	case "LDAP":
-		var config ldapx.Config
-		err := toml.Unmarshal([]byte(f.Content), &config)
-		ginx.Dangerous(err)
-		rt.Sso.LDAP.Reload(config)
-	case "OIDC":
-		var config oidcx.Config
-		err := toml.Unmarshal([]byte(f.Content), &config)
-		ginx.Dangerous(err)
-		rt.Sso.OIDC, err = oidcx.New(config)
-		ginx.Dangerous(err)
-	case "CAS":
-		var config cas.Config
-		err := toml.Unmarshal([]byte(f.Content), &config)
-		ginx.Dangerous(err)
-		rt.Sso.CAS.Reload(config)
-	case "OAuth2":
-		var config oauth2x.Config
-		err := toml.Unmarshal([]byte(f.Content), &config)
-		ginx.Dangerous(err)
-		rt.Sso.OAuth2.Reload(config)
-	}
-
-	ginx.NewRender(c).Message(nil)
-}
-
-type RSAConfigOutput struct {
-	OpenRSA      bool
-	RSAPublicKey string
-}
-
-func (rt *Router) rsaConfigGet(c *gin.Context) {
-	publicKey := ""
-	if len(rt.HTTP.RSA.RSAPublicKey) > 0 {
-		publicKey = base64.StdEncoding.EncodeToString(rt.HTTP.RSA.RSAPublicKey)
-	}
-	ginx.NewRender(c).Data(RSAConfigOutput{
-		OpenRSA:      rt.HTTP.RSA.OpenRSA,
-		RSAPublicKey: publicKey,
-	}, nil)
-}

center/router/router_message_template.go

@@ -1,219 +0,0 @@
-package router
-
-import (
-	"bytes"
-	"fmt"
-	"html/template"
-	"net/http"
-	"strings"
-	"time"
-
-	"github.com/ccfos/nightingale/v6/models"
-	"github.com/ccfos/nightingale/v6/pkg/slice"
-	"github.com/ccfos/nightingale/v6/pkg/strx"
-	"github.com/ccfos/nightingale/v6/pkg/tplx"
-
-	"github.com/gin-gonic/gin"
-	"github.com/google/uuid"
-	"github.com/toolkits/pkg/ginx"
-)
-
-func (rt *Router) messageTemplatesAdd(c *gin.Context) {
-	var lst []*models.MessageTemplate
-	ginx.BindJSON(c, &lst)
-	if len(lst) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "input json is empty")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	isAdmin := me.IsAdmin()
-	idents := make([]string, 0, len(lst))
-	gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-	now := time.Now().Unix()
-	for _, tpl := range lst {
-		// 生成一个唯一的标识符，以后也不允许修改，前端不需要传这个参数
-		tpl.Ident = uuid.New().String()
-
-		ginx.Dangerous(tpl.Verify())
-		if !isAdmin && !slice.HaveIntersection(gids, tpl.UserGroupIds) {
-			ginx.Bomb(http.StatusForbidden, "forbidden")
-		}
-		idents = append(idents, tpl.Ident)
-
-		tpl.CreateBy = me.Username
-		tpl.CreateAt = now
-		tpl.UpdateBy = me.Username
-		tpl.UpdateAt = now
-	}
-
-	lstWithSameId, err := models.MessageTemplatesGet(rt.Ctx, "ident IN ?", idents)
-	ginx.Dangerous(err)
-	if len(lstWithSameId) > 0 {
-		ginx.Bomb(http.StatusBadRequest, "ident already exists")
-	}
-
-	ids := make([]int64, 0, len(lst))
-	for _, tpl := range lst {
-		err := models.Insert(rt.Ctx, tpl)
-		ginx.Dangerous(err)
-
-		ids = append(ids, tpl.ID)
-	}
-	ginx.NewRender(c).Data(ids, nil)
-}
-
-func (rt *Router) messageTemplatesDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	lst, err := models.MessageTemplatesGet(rt.Ctx, "id in (?)", f.Ids)
-	ginx.Dangerous(err)
-	notifyRuleIds, err := models.UsedByNotifyRule(rt.Ctx, models.MsgTplList(lst))
-	ginx.Dangerous(err)
-	if len(notifyRuleIds) > 0 {
-		ginx.NewRender(c).Message(fmt.Errorf("used by notify rule: %v", notifyRuleIds))
-		return
-	}
-	if me := c.MustGet("user").(*models.User); !me.IsAdmin() {
-		gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-		ginx.Dangerous(err)
-		for _, t := range lst {
-			if !slice.HaveIntersection(gids, t.UserGroupIds) {
-				ginx.Bomb(http.StatusForbidden, "forbidden")
-			}
-		}
-	}
-
-	ginx.NewRender(c).Message(models.DB(rt.Ctx).Delete(
-		&models.MessageTemplate{}, "id in (?)", f.Ids).Error)
-}
-
-func (rt *Router) messageTemplatePut(c *gin.Context) {
-	var f models.MessageTemplate
-	ginx.BindJSON(c, &f)
-
-	mt, err := models.MessageTemplateGet(rt.Ctx, "id <> ? and ident = ?", ginx.UrlParamInt64(c, "id"), f.Ident)
-	ginx.Dangerous(err)
-	if mt != nil {
-		ginx.Bomb(http.StatusBadRequest, "message template ident already exists")
-	}
-
-	mt, err = models.MessageTemplateGet(rt.Ctx, "id = ?", ginx.UrlParamInt64(c, "id"))
-	ginx.Dangerous(err)
-	if mt == nil {
-		ginx.Bomb(http.StatusNotFound, "message template not found")
-	}
-
-	me := c.MustGet("user").(*models.User)
-	if !me.IsAdmin() {
-		gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-		ginx.Dangerous(err)
-		if !slice.HaveIntersection(gids, mt.UserGroupIds) {
-			ginx.Bomb(http.StatusForbidden, "forbidden")
-		}
-	}
-
-	f.UpdateBy = me.Username
-	ginx.NewRender(c).Message(mt.Update(rt.Ctx, f))
-}
-
-func (rt *Router) messageTemplateGet(c *gin.Context) {
-	me := c.MustGet("user").(*models.User)
-	gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-
-	tid := ginx.UrlParamInt64(c, "id")
-	mt, err := models.MessageTemplateGet(rt.Ctx, "id = ?", tid)
-	ginx.Dangerous(err)
-	if mt == nil {
-		ginx.Bomb(http.StatusNotFound, "message template not found")
-	}
-	if mt.Private == 1 && !slice.HaveIntersection(gids, mt.UserGroupIds) {
-		ginx.Bomb(http.StatusForbidden, "forbidden")
-	}
-
-	ginx.NewRender(c).Data(mt, nil)
-}
-
-func (rt *Router) messageTemplatesGet(c *gin.Context) {
-	var notifyChannelIdents []string
-	if tmp := ginx.QueryStr(c, "notify_channel_idents", ""); tmp != "" {
-		notifyChannelIdents = strings.Split(tmp, ",")
-	}
-	notifyChannelIds := strx.IdsInt64ForAPI(ginx.QueryStr(c, "notify_channel_ids", ""))
-	if len(notifyChannelIds) > 0 {
-		ginx.Dangerous(models.DB(rt.Ctx).Model(models.NotifyChannelConfig{}).
-			Where("id in (?)", notifyChannelIds).Pluck("ident", &notifyChannelIdents).Error)
-	}
-
-	me := c.MustGet("user").(*models.User)
-	gids, err := models.MyGroupIds(rt.Ctx, me.Id)
-	ginx.Dangerous(err)
-
-	lst, err := models.MessageTemplatesGetBy(rt.Ctx, notifyChannelIdents)
-	ginx.Dangerous(err)
-
-	if me.IsAdmin() {
-		ginx.NewRender(c).Data(lst, nil)
-		return
-	}
-
-	res := make([]*models.MessageTemplate, 0)
-	for _, t := range lst {
-		if slice.HaveIntersection[int64](gids, t.UserGroupIds) || t.Private == 0 {
-			res = append(res, t)
-		}
-	}
-	ginx.NewRender(c).Data(res, nil)
-}
-
-type evtMsgReq struct {
-	EventIds []int64 `json:"event_ids"`
-	Tpl      struct {
-		Content map[string]string `json:"content"`
-	} `json:"tpl"`
-}
-
-func (rt *Router) eventsMessage(c *gin.Context) {
-	var req evtMsgReq
-	ginx.BindJSON(c, &req)
-
-	hisEvents, err := models.AlertHisEventGetByIds(rt.Ctx, req.EventIds)
-	ginx.Dangerous(err)
-
-	if len(hisEvents) == 0 {
-		ginx.Bomb(http.StatusBadRequest, "event not found")
-	}
-
-	ginx.Dangerous(err)
-	events := make([]*models.AlertCurEvent, len(hisEvents))
-	for i, he := range hisEvents {
-		events[i] = he.ToCur()
-	}
-
-	var defs = []string{
-		"{{$events := .}}",
-		"{{$event := index . 0}}",
-	}
-	ret := make(map[string]string, len(req.Tpl.Content))
-	for k, v := range req.Tpl.Content {
-		text := strings.Join(append(defs, v), "")
-		tpl, err := template.New(k).Funcs(tplx.TemplateFuncMap).Parse(text)
-		if err != nil {
-			ret[k] = err.Error()
-			continue
-		}
-
-		var buf bytes.Buffer
-		err = tpl.Execute(&buf, events)
-		if err != nil {
-			ret[k] = err.Error()
-			continue
-		}
-
-		ret[k] = buf.String()
-	}
-	ginx.NewRender(c).Data(ret, nil)
-}

center/router/router_metric_view.go

@@ -1,76 +0,0 @@
-package router
-
-import (
-	"net/http"
-
-	"github.com/ccfos/nightingale/v6/models"
-
-	"github.com/gin-gonic/gin"
-	"github.com/toolkits/pkg/ginx"
-)
-
-// no param
-func (rt *Router) metricViewGets(c *gin.Context) {
-	lst, err := models.MetricViewGets(rt.Ctx, c.MustGet("userid"))
-	ginx.NewRender(c).Data(lst, err)
-}
-
-// body: name, configs, cate
-func (rt *Router) metricViewAdd(c *gin.Context) {
-	var f models.MetricView
-	ginx.BindJSON(c, &f)
-
-	me := c.MustGet("user").(*models.User)
-	if !me.IsAdmin() {
-		// 管理员可以选择当前这个视图是公开呢，还是私有，普通用户的话就只能是私有的
-		f.Cate = 1
-	}
-
-	f.Id = 0
-	f.CreateBy = me.Id
-
-	ginx.Dangerous(f.Add(rt.Ctx))
-
-	ginx.NewRender(c).Data(f, nil)
-}
-
-// body: ids
-func (rt *Router) metricViewDel(c *gin.Context) {
-	var f idsForm
-	ginx.BindJSON(c, &f)
-	f.Verify()
-
-	me := c.MustGet("user").(*models.User)
-	if me.IsAdmin() {
-		ginx.NewRender(c).Message(models.MetricViewDel(rt.Ctx, f.Ids))
-	} else {
-		ginx.NewRender(c).Message(models.MetricViewDel(rt.Ctx, f.Ids, me.Id))
-	}
-}
-
-// body: id, name, configs, cate
-func (rt *Router) metricViewPut(c *gin.Context) {
-	var f models.MetricView
-	ginx.BindJSON(c, &f)
-
-	view, err := models.MetricViewGet(rt.Ctx, "id = ?", f.Id)
-	ginx.Dangerous(err)
-
-	if view == nil {
-		ginx.NewRender(c).Message("no such item(id: %d)", f.Id)
-		return
-	}
-
-	me := c.MustGet("user").(*models.User)
-	if !me.IsAdmin() {
-		f.Cate = 1
-
-		// 如果是普通用户，只能修改自己的
-		if view.CreateBy != me.Id {
-			ginx.NewRender(c, http.StatusForbidden).Message("forbidden")
-			return
-		}
-	}
-
-	ginx.NewRender(c).Message(view.Update(rt.Ctx, f.Name, f.Configs, f.Cate, me.Id))
-}