-
Notifications
You must be signed in to change notification settings - Fork 2.2k
/
gray.go
125 lines (110 loc) · 3.74 KB
/
gray.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.
*/
package traffic
import (
"net/http"
"github.com/apache/dubbo-admin/pkg/admin/model"
"github.com/apache/dubbo-admin/pkg/admin/services/traffic"
"github.com/apache/dubbo-admin/pkg/logger"
"github.com/gin-gonic/gin"
)
var graySVC = &traffic.GrayService{}
// CreateGray create rule
// @Summary create rule
// @Description create rule
// @Tags TrafficGray
// @Accept json
// @Produce json
// @Param gray body model.Gray true "rule"
// @Success 200 {bool} true
// @Failure 400 {object} model.HTTPError
// @Failure 500 {object} model.HTTPError
// @Router /api/{env}/traffic/gray [post]
func CreateGray(c *gin.Context) {
doGrayUpdate(c, func(g *model.Gray) error {
return graySVC.CreateOrUpdate(g)
})
}
// UpdateGray update rule
// @Summary update rule
// @Description update rule
// @Tags TrafficGray
// @Accept json
// @Produce json
// @Param gray body model.Gray true "rule"
// @Success 200 {bool} true
// @Failure 400 {object} model.HTTPError
// @Failure 500 {object} model.HTTPError
// @Router /api/{env}/traffic/gray [put]
func UpdateGray(c *gin.Context) {
doGrayUpdate(c, func(g *model.Gray) error {
return graySVC.CreateOrUpdate(g)
})
}
// DeleteGray delete rule
// @Summary delete rule
// @Description delete rule
// @Tags TrafficGray
// @Accept json
// @Produce json
// @Param application query string true "application name"
// @Success 200 {bool} true
// @Failure 500 {object} model.HTTPError
// @Router /api/{env}/traffic/gray [delete]
func DeleteGray(c *gin.Context) {
g := &model.Gray{
Application: c.Query("application"),
}
err := graySVC.Delete(g)
if err != nil {
c.JSON(http.StatusInternalServerError, model.HTTPError{Error: err.Error()})
}
c.JSON(http.StatusOK, true)
}
// SearchGray get rule list
// @Summary get rule list
// @Description get rule list
// @Tags TrafficGray
// @Accept json
// @Produce json
// @Param application query string true "application name"
// @Success 200 {object} []model.Gray
// @Failure 500 {object} model.HTTPError
// @Router /api/{env}/traffic/gray [get]
func SearchGray(c *gin.Context) {
g := &model.Gray{
Application: c.Query("application"),
}
result, err := graySVC.Search(g)
if err != nil {
c.JSON(http.StatusInternalServerError, model.HTTPError{Error: err.Error()})
}
c.JSON(http.StatusOK, result)
}
func doGrayUpdate(c *gin.Context, handle func(g *model.Gray) error) {
var g *model.Gray
if err := c.ShouldBindJSON(&g); err != nil {
logger.Errorf("Error parsing rule input when trying to create override rule, err msg is %s.", err.Error())
c.JSON(http.StatusBadRequest, model.HTTPError{Error: err.Error()})
return
}
err := handle(g)
if err != nil {
c.JSON(http.StatusInternalServerError, model.HTTPError{Error: err.Error()})
}
c.JSON(http.StatusOK, true)
}