-
Notifications
You must be signed in to change notification settings - Fork 362
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
2. Support choose log color mode. 3. Support silence usage, because usage info is noisy in some cases; 4. Silence error, because sealer has already print error itself. 5. Use logrus to print the final error. Signed-off-by: huaiyou <[email protected]>
- Loading branch information
Showing
6 changed files
with
147 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,111 @@ | ||
| // alibaba-inc.com Inc. | ||
| // Copyright (c) 2004-2022 All Rights Reserved. | ||
| // | ||
| // @Author : huaiyou.cyz | ||
| // @Time : 2022/6/7 11:39 PM | ||
| // @File : remote-logger | ||
| // | ||
|
|
||
| package remotelogger | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "fmt" | ||
| "net/http" | ||
| "net/url" | ||
| "os" | ||
| "sync" | ||
|
|
||
| "github.com/google/uuid" | ||
| "github.com/sirupsen/logrus" | ||
| corev1 "k8s.io/api/core/v1" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| ) | ||
|
|
||
| // RemoteLogHook to send logs via remote URL. | ||
| type RemoteLogHook struct { | ||
| sync.RWMutex | ||
|
|
||
| TaskName string | ||
| URL string | ||
| } | ||
|
|
||
| func NewRemoteLogHook(remoteURL, taskName string) (*RemoteLogHook, error) { | ||
| reqURL, err := url.Parse(remoteURL) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return &RemoteLogHook{ | ||
| TaskName: taskName, | ||
| URL: reqURL.String(), | ||
| }, err | ||
| } | ||
|
|
||
| // #nosec | ||
| func httpSend(url string, method string, body []byte) error { | ||
| var resp *http.Response | ||
|
|
||
| var err error | ||
| switch method { | ||
| case http.MethodGet: | ||
| resp, err = http.Get(url) | ||
| case http.MethodPost: | ||
| resp, err = http.Post(url, "application/json", bytes.NewBuffer(body)) | ||
| } | ||
|
|
||
| if err != nil { | ||
| return fmt.Errorf("bad %s request to server : %w", method, err) | ||
| } | ||
| defer func() { | ||
| _ = resp.Body.Close() | ||
| }() | ||
|
|
||
| if resp.StatusCode != http.StatusOK { | ||
| return fmt.Errorf("bad status code from server: [%d] %s ", resp.StatusCode, resp.Status) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (hook *RemoteLogHook) Fire(entry *logrus.Entry) error { | ||
| line, err := entry.String() | ||
| if err != nil { | ||
| _, _ = fmt.Fprintf(os.Stderr, "Unable to read entry, %v", err) | ||
| return err | ||
| } | ||
|
|
||
| t := "Info" | ||
| if entry.Level <= logrus.ErrorLevel { | ||
| t = "Error" | ||
| } | ||
|
|
||
| event := &corev1.Event{ | ||
| ObjectMeta: metav1.ObjectMeta{ | ||
| Name: uuid.New().String(), | ||
| }, | ||
| InvolvedObject: corev1.ObjectReference{ | ||
| Name: hook.TaskName, | ||
| }, | ||
| Message: line, | ||
| Type: t, | ||
| } | ||
|
|
||
| bytesData, _ := json.Marshal(event) | ||
|
|
||
| hook.Lock() | ||
| defer hook.Unlock() | ||
|
|
||
| return httpSend(hook.URL, http.MethodPost, bytesData) | ||
| } | ||
|
|
||
| func (hook *RemoteLogHook) Levels() []logrus.Level { | ||
| return []logrus.Level{ | ||
| logrus.PanicLevel, | ||
| logrus.FatalLevel, | ||
| logrus.ErrorLevel, | ||
| logrus.WarnLevel, | ||
| logrus.InfoLevel, | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters