29 lines
623 B
Go
29 lines
623 B
Go
package deployment
|
|
|
|
import "sync"
|
|
|
|
const commandOutputLimit = 64 << 10
|
|
|
|
// commandOutput 限制远程标准输出和错误输出占用，防止异常命令耗尽管理端内存。
|
|
type commandOutput struct {
|
|
mu sync.Mutex
|
|
data []byte
|
|
}
|
|
|
|
func (b *commandOutput) Write(data []byte) (int, error) {
|
|
b.mu.Lock()
|
|
defer b.mu.Unlock()
|
|
remaining := commandOutputLimit - len(b.data)
|
|
if remaining > len(data) {
|
|
remaining = len(data)
|
|
}
|
|
b.data = append(b.data, data[:remaining]...)
|
|
return len(data), nil
|
|
}
|
|
|
|
func (b *commandOutput) Bytes() []byte {
|
|
b.mu.Lock()
|
|
defer b.mu.Unlock()
|
|
return append([]byte(nil), b.data...)
|
|
}
|