mirror of https://github.com/v2ray/v2ray-core
errors.Combine
parent
d4613f156b
commit
bea521537e
|
@ -7,9 +7,9 @@ import (
|
||||||
"v2ray.com/core/app/proxyman"
|
"v2ray.com/core/app/proxyman"
|
||||||
"v2ray.com/core/common"
|
"v2ray.com/core/common"
|
||||||
"v2ray.com/core/common/dice"
|
"v2ray.com/core/common/dice"
|
||||||
|
"v2ray.com/core/common/errors"
|
||||||
"v2ray.com/core/common/mux"
|
"v2ray.com/core/common/mux"
|
||||||
"v2ray.com/core/common/net"
|
"v2ray.com/core/common/net"
|
||||||
"v2ray.com/core/common/serial"
|
|
||||||
"v2ray.com/core/features/policy"
|
"v2ray.com/core/features/policy"
|
||||||
"v2ray.com/core/features/stats"
|
"v2ray.com/core/features/stats"
|
||||||
"v2ray.com/core/proxy"
|
"v2ray.com/core/proxy"
|
||||||
|
@ -137,17 +137,13 @@ func (h *AlwaysOnInboundHandler) Start() error {
|
||||||
|
|
||||||
// Close implements common.Closable.
|
// Close implements common.Closable.
|
||||||
func (h *AlwaysOnInboundHandler) Close() error {
|
func (h *AlwaysOnInboundHandler) Close() error {
|
||||||
var errors []interface{}
|
var errs []error
|
||||||
for _, worker := range h.workers {
|
for _, worker := range h.workers {
|
||||||
if err := worker.Close(); err != nil {
|
errs = append(errs, worker.Close())
|
||||||
errors = append(errors, err)
|
|
||||||
}
|
}
|
||||||
}
|
errs = append(errs, h.mux.Close())
|
||||||
if err := h.mux.Close(); err != nil {
|
if err := errors.Combine(errs...); err != nil {
|
||||||
errors = append(errors, err)
|
return newError("failed to close all resources").Base(err)
|
||||||
}
|
|
||||||
if len(errors) > 0 {
|
|
||||||
return newError("failed to close all resources").Base(newError(serial.Concat(errors...)))
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,6 +10,7 @@ import (
|
||||||
"v2ray.com/core"
|
"v2ray.com/core"
|
||||||
"v2ray.com/core/app/proxyman"
|
"v2ray.com/core/app/proxyman"
|
||||||
"v2ray.com/core/common"
|
"v2ray.com/core/common"
|
||||||
|
"v2ray.com/core/common/errors"
|
||||||
"v2ray.com/core/features/outbound"
|
"v2ray.com/core/features/outbound"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -64,15 +65,16 @@ func (m *Manager) Close() error {
|
||||||
|
|
||||||
m.running = false
|
m.running = false
|
||||||
|
|
||||||
|
var errs []error
|
||||||
for _, h := range m.taggedHandler {
|
for _, h := range m.taggedHandler {
|
||||||
h.Close()
|
errs = append(errs, h.Close())
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, h := range m.untaggedHandlers {
|
for _, h := range m.untaggedHandlers {
|
||||||
h.Close()
|
errs = append(errs, h.Close())
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return errors.Combine(errs...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetDefaultHandler implements outbound.Manager.
|
// GetDefaultHandler implements outbound.Manager.
|
||||||
|
|
|
@ -7,6 +7,7 @@ import (
|
||||||
|
|
||||||
"v2ray.com/core"
|
"v2ray.com/core"
|
||||||
"v2ray.com/core/common"
|
"v2ray.com/core/common"
|
||||||
|
"v2ray.com/core/common/errors"
|
||||||
"v2ray.com/core/common/net"
|
"v2ray.com/core/common/net"
|
||||||
"v2ray.com/core/features/outbound"
|
"v2ray.com/core/features/outbound"
|
||||||
"v2ray.com/core/features/routing"
|
"v2ray.com/core/features/routing"
|
||||||
|
@ -82,17 +83,14 @@ func (r *Reverse) Start() error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Reverse) Close() error {
|
func (r *Reverse) Close() error {
|
||||||
|
var errs []error
|
||||||
for _, b := range r.bridges {
|
for _, b := range r.bridges {
|
||||||
if err := b.Close(); err != nil {
|
errs = append(errs, b.Close())
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, p := range r.portals {
|
for _, p := range r.portals {
|
||||||
if err := p.Close(); err != nil {
|
errs = append(errs, p.Close())
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return errors.Combine(errs...)
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,30 @@
|
||||||
|
package errors
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type multiError []error
|
||||||
|
|
||||||
|
func (e multiError) Error() string {
|
||||||
|
var r strings.Builder
|
||||||
|
r.WriteString("multierr: ")
|
||||||
|
for _, err := range e {
|
||||||
|
r.WriteString(err.Error())
|
||||||
|
r.WriteString(" | ")
|
||||||
|
}
|
||||||
|
return r.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func Combine(maybeError ...error) error {
|
||||||
|
var errs multiError
|
||||||
|
for _, err := range maybeError {
|
||||||
|
if err != nil {
|
||||||
|
errs = append(errs, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(errs) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return errs
|
||||||
|
}
|
|
@ -2,26 +2,12 @@ package task
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"v2ray.com/core/common"
|
|
||||||
"v2ray.com/core/common/signal/semaphore"
|
"v2ray.com/core/common/signal/semaphore"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Task func() error
|
type Task func() error
|
||||||
|
|
||||||
type MultiError []error
|
|
||||||
|
|
||||||
func (e MultiError) Error() string {
|
|
||||||
var r strings.Builder
|
|
||||||
common.Must2(r.WriteString("multierr: "))
|
|
||||||
for _, err := range e {
|
|
||||||
common.Must2(r.WriteString(err.Error()))
|
|
||||||
common.Must2(r.WriteString(" | "))
|
|
||||||
}
|
|
||||||
return r.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
type executionContext struct {
|
type executionContext struct {
|
||||||
ctx context.Context
|
ctx context.Context
|
||||||
tasks []Task
|
tasks []Task
|
||||||
|
@ -90,30 +76,6 @@ func Sequential(tasks ...Task) ExecutionOption {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func SequentialAll(tasks ...Task) ExecutionOption {
|
|
||||||
return func(c *executionContext) {
|
|
||||||
switch len(tasks) {
|
|
||||||
case 0:
|
|
||||||
return
|
|
||||||
case 1:
|
|
||||||
c.tasks = append(c.tasks, tasks[0])
|
|
||||||
default:
|
|
||||||
c.tasks = append(c.tasks, func() error {
|
|
||||||
var merr MultiError
|
|
||||||
for _, task := range tasks {
|
|
||||||
if err := task(); err != nil {
|
|
||||||
merr = append(merr, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(merr) == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return merr
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func OnSuccess(task Task) ExecutionOption {
|
func OnSuccess(task Task) ExecutionOption {
|
||||||
return func(c *executionContext) {
|
return func(c *executionContext) {
|
||||||
c.onSuccess = task
|
c.onSuccess = task
|
||||||
|
|
|
@ -139,9 +139,10 @@ func New(ctx context.Context, config *Config) (*Handler, error) {
|
||||||
|
|
||||||
// Close implements common.Closable.
|
// Close implements common.Closable.
|
||||||
func (h *Handler) Close() error {
|
func (h *Handler) Close() error {
|
||||||
return task.Run(
|
return errors.Combine(
|
||||||
task.SequentialAll(
|
h.clients.Close(),
|
||||||
task.Close(h.clients), task.Close(h.sessionHistory), task.Close(h.usersByEmail)))()
|
h.sessionHistory.Close(),
|
||||||
|
common.Close(h.usersByEmail))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Network implements proxy.Inbound.Network().
|
// Network implements proxy.Inbound.Network().
|
||||||
|
@ -290,9 +291,10 @@ func (h *Handler) Process(ctx context.Context, network net.Network, connection i
|
||||||
}
|
}
|
||||||
|
|
||||||
responseDone := func() error {
|
responseDone := func() error {
|
||||||
|
defer timer.SetTimeout(sessionPolicy.Timeouts.UplinkOnly)
|
||||||
|
|
||||||
writer := buf.NewBufferedWriter(buf.NewWriter(connection))
|
writer := buf.NewBufferedWriter(buf.NewWriter(connection))
|
||||||
defer writer.Flush()
|
defer writer.Flush()
|
||||||
defer timer.SetTimeout(sessionPolicy.Timeouts.UplinkOnly)
|
|
||||||
|
|
||||||
response := &protocol.ResponseHeader{
|
response := &protocol.ResponseHeader{
|
||||||
Command: h.generateCommand(ctx, request),
|
Command: h.generateCommand(ctx, request),
|
||||||
|
|
Loading…
Reference in New Issue