1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-16 09:26:21 -05:00
v2fly/app/space.go

83 lines
1.5 KiB
Go
Raw Normal View History

2015-12-05 16:55:45 -05:00
package app
2016-05-18 11:12:04 -04:00
import (
"errors"
2016-08-20 14:55:45 -04:00
"v2ray.com/core/common"
2016-05-18 11:12:04 -04:00
)
var (
2016-06-27 02:53:35 -04:00
ErrMissingApplication = errors.New("App: Failed to found one or more applications.")
2016-05-18 11:12:04 -04:00
)
2016-01-31 11:01:28 -05:00
type ID int
// Context of a function call from proxy to app.
2015-12-10 17:55:39 -05:00
type Context interface {
CallerTag() string
}
2016-05-18 02:05:52 -04:00
type Caller interface {
Tag() string
}
2016-05-18 11:12:04 -04:00
type Application interface {
common.Releasable
}
type ApplicationInitializer func() error
// A Space contains all apps that may be available in a V2Ray runtime.
// Caller must check the availability of an app by calling HasXXX before getting its instance.
type Space interface {
2016-05-18 11:12:04 -04:00
Initialize() error
InitializeApplication(ApplicationInitializer)
2016-01-31 11:01:28 -05:00
HasApp(ID) bool
2016-05-18 11:12:04 -04:00
GetApp(ID) Application
BindApp(ID, Application)
2016-01-31 11:01:28 -05:00
}
type spaceImpl struct {
cache map[ID]Application
appInit []ApplicationInitializer
2016-01-31 11:01:28 -05:00
}
2015-12-11 09:56:10 -05:00
2016-05-18 02:05:52 -04:00
func NewSpace() Space {
return &spaceImpl{
cache: make(map[ID]Application),
appInit: make([]ApplicationInitializer, 0, 32),
2016-05-18 11:12:04 -04:00
}
}
func (this *spaceImpl) InitializeApplication(f ApplicationInitializer) {
this.appInit = append(this.appInit, f)
2016-05-18 11:12:04 -04:00
}
func (this *spaceImpl) Initialize() error {
for _, f := range this.appInit {
err := f()
if err != nil {
return err
}
2016-01-31 11:01:28 -05:00
}
2016-05-18 11:12:04 -04:00
return nil
2016-01-31 11:01:28 -05:00
}
func (this *spaceImpl) HasApp(id ID) bool {
_, found := this.cache[id]
return found
}
2016-05-18 11:12:04 -04:00
func (this *spaceImpl) GetApp(id ID) Application {
2016-01-31 11:01:28 -05:00
obj, found := this.cache[id]
if !found {
return nil
}
return obj
}
2016-05-18 11:12:04 -04:00
func (this *spaceImpl) BindApp(id ID, application Application) {
this.cache[id] = application
2015-12-05 16:55:45 -05:00
}