drop pkg closer from pj, now import it from common
This commit is contained in:
parent
8f7f0db142
commit
228edf6a1e
@ -4,6 +4,7 @@ import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"gitea.pena/PenaSide/common/closer"
|
||||
"gitea.pena/PenaSide/common/mongo"
|
||||
"gitea.pena/PenaSide/treasurer/internal/server/grpc"
|
||||
"gitea.pena/PenaSide/treasurer/internal/server/http"
|
||||
@ -11,7 +12,6 @@ import (
|
||||
|
||||
"gitea.pena/PenaSide/treasurer/internal/initialize"
|
||||
"gitea.pena/PenaSide/treasurer/internal/worker"
|
||||
"gitea.pena/PenaSide/treasurer/pkg/closer"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
|
@ -1,37 +0,0 @@
|
||||
package closer
|
||||
|
||||
import (
|
||||
"context"
|
||||
)
|
||||
|
||||
type Closer interface {
|
||||
Close(ctx context.Context) error
|
||||
}
|
||||
|
||||
type CloserFunc func(ctx context.Context) error
|
||||
|
||||
func (cf CloserFunc) Close(ctx context.Context) error {
|
||||
return cf(ctx)
|
||||
}
|
||||
|
||||
type CloserGroup struct {
|
||||
closers []Closer
|
||||
}
|
||||
|
||||
func NewCloserGroup() *CloserGroup {
|
||||
return &CloserGroup{}
|
||||
}
|
||||
|
||||
func (cg *CloserGroup) Add(c Closer) {
|
||||
cg.closers = append(cg.closers, c)
|
||||
}
|
||||
|
||||
func (cg *CloserGroup) Call(ctx context.Context) error {
|
||||
var closeErr error
|
||||
for i := len(cg.closers) - 1; i >= 0; i-- {
|
||||
if err := cg.closers[i].Close(ctx); err != nil && closeErr == nil {
|
||||
closeErr = err
|
||||
}
|
||||
}
|
||||
return closeErr
|
||||
}
|
Loading…
Reference in New Issue
Block a user