Skip to content

Commit

Permalink
修复ws bug
Browse files Browse the repository at this point in the history
  • Loading branch information
leon.chen committed Jan 22, 2021
1 parent 6ce9867 commit 601f4ea
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 38 deletions.
38 changes: 4 additions & 34 deletions backend/controllers/query.go
Original file line number Diff line number Diff line change
Expand Up @@ -455,42 +455,12 @@ func LokiContext(c *gin.Context) {
// @Router /ws/tail/ [get]
func LokiTail(c *gin.Context) {
level := c.DefaultQuery("level", "")
pod := c.DefaultQuery("pod", "")

filtersStr := c.DefaultQuery("filters", "")
filters := strings.Split(filtersStr, ",")
queryExpr := c.DefaultQuery("logql", "")
filterStr := c.DefaultQuery("filters", "")
filters := strings.Split(filterStr, ",")
start := c.DefaultQuery("start", "")

queryExprArray := []string{}

t := time.Now().Unix()
end := fmt.Sprintf("%d000000000", t)
labels := utils.Labels(start, end)
for _, label := range labels {
if c.DefaultQuery(label.(string), "") != "" {
queryExprArray = append(queryExprArray, utils.GetExpr(label.(string), c.DefaultQuery(label.(string), "")))
}
}

if pod != "" {
queryExprArray = append(queryExprArray, utils.GetPodExpr(pod))
}

if len(queryExprArray) == 0 {
c.AbortWithStatusJSON(400, gin.H{"success": false, "message": "缺少查询条件"})
return
}

queryExpr := fmt.Sprintf("{%s}", strings.Join(queryExprArray, ","))
for _, filter := range filters {
_, err := regexp.Compile(filter)
if err != nil {
utils.Log4Zap(zap.WarnLevel).Warn(fmt.Sprintf("regex compile error, %s", err))
c.AbortWithStatusJSON(500, gin.H{"success": false, "message": "请查看服务器日志"})
return
}
queryExpr = fmt.Sprintf("%s |~ `%s`", queryExpr, strings.Trim(filter, ""))
}
queryExpr, _ = url.QueryUnescape(queryExpr)
if level != "" {
levelExpr := utils.GenerateLevelRegex(level)
if levelExpr != "" {
Expand Down
3 changes: 0 additions & 3 deletions backend/utils/lokiwebsocket.go
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,6 @@ func LokiWebsocketMessageConstruct(data []byte, filters []string) []byte {

results := message["streams"]
if results != nil {
for _, filter := range filters {
filter = strings.ReplaceAll(filter, "\\\\", "\\")
}
for _, result := range results.([]interface{}) {
resultEle := result.(map[string]interface{})
stream := resultEle["stream"].(map[string]interface{})
Expand Down
1 change: 0 additions & 1 deletion ui/src/views/logs/viewer/viewer.vue
Original file line number Diff line number Diff line change
Expand Up @@ -696,7 +696,6 @@ export default {
const data = {
start: Date.parse(new Date()).toString() + '000000',
level: this.level.join(','),
pod: this.pod,
filters: this.filters,
logql: encodeURIComponent(this.logQL),
}
Expand Down

0 comments on commit 601f4ea

Please sign in to comment.