07fbb8b8f7
- enhance logging throughout the codebase - all packages at pkg/ should never log or perform log.Fatal() instead packages should return errors through functions. - simplified various user, group mapping and removed redundant functions. - deprecate older flags like --tls-certificate --tls-key and --tls-ca as we do not use them anymore, keep them for backward compatibility for some time.
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
// This file is part of MinIO Console Server
|
|
// Copyright (c) 2021 MinIO, Inc.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package restapi
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/gorilla/websocket"
|
|
mc "github.com/minio/mc/cmd"
|
|
)
|
|
|
|
type watchOptions struct {
|
|
BucketName string
|
|
mc.WatchOptions
|
|
}
|
|
|
|
func startWatch(ctx context.Context, conn WSConn, wsc MCClient, options *watchOptions) error {
|
|
wo, pErr := wsc.watch(ctx, options.WatchOptions)
|
|
if pErr != nil {
|
|
LogError("error initializing watch: %v", pErr.Cause)
|
|
return pErr.Cause
|
|
}
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
close(wo.DoneChan)
|
|
return nil
|
|
case events, ok := <-wo.Events():
|
|
// zero value returned because the channel is closed and empty
|
|
if !ok {
|
|
return nil
|
|
}
|
|
for _, event := range events {
|
|
// Serialize message to be sent
|
|
bytes, err := json.Marshal(event)
|
|
if err != nil {
|
|
LogError("error on json.Marshal: %v", err)
|
|
return err
|
|
}
|
|
// Send Message through websocket connection
|
|
err = conn.writeMessage(websocket.TextMessage, bytes)
|
|
if err != nil {
|
|
LogError("error writeMessage: %v", err)
|
|
return err
|
|
}
|
|
}
|
|
case pErr, ok := <-wo.Errors():
|
|
// zero value returned because the channel is closed and empty
|
|
if !ok {
|
|
return nil
|
|
}
|
|
if pErr != nil {
|
|
LogError("error on watch: %v", pErr.Cause)
|
|
return pErr.Cause
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// getWatchOptionsFromReq gets bucket name, events, prefix, suffix from a websocket
|
|
// watch path if defined.
|
|
// path come as : `/watch/bucket1` and query
|
|
// params come on request form
|
|
func getWatchOptionsFromReq(req *http.Request) (*watchOptions, error) {
|
|
wOptions := watchOptions{}
|
|
// Default Events if not defined
|
|
wOptions.Events = []string{"put", "get", "delete"}
|
|
|
|
re := regexp.MustCompile(`(/watch/)(.*?$)`)
|
|
matches := re.FindAllSubmatch([]byte(req.URL.Path), -1)
|
|
// matches comes as e.g.
|
|
// [["...", "/watch/", "bucket1"]]
|
|
// [["/watch/" "/watch/" ""]]
|
|
|
|
if len(matches) == 0 || len(matches[0]) < 3 {
|
|
return nil, fmt.Errorf("invalid url: %s", req.URL.Path)
|
|
}
|
|
|
|
wOptions.BucketName = strings.TrimSpace(string(matches[0][2]))
|
|
|
|
events := req.FormValue("events")
|
|
if strings.TrimSpace(events) != "" {
|
|
wOptions.Events = strings.Split(events, ",")
|
|
}
|
|
wOptions.Prefix = req.FormValue("prefix")
|
|
wOptions.Suffix = req.FormValue("suffix")
|
|
return &wOptions, nil
|
|
}
|