@@ -37,17 +37,27 @@ import (
3737 "time"
3838)
3939
40+ var (
41+ ErrUnsupportedTRuntimeType = errors .New ("unsupported runtime type" )
42+ ErrUnsupportedTubeType = errors .New ("unsupported tube type" )
43+ )
44+
4045type Server struct {
4146 options * serverOptions
4247 httpSvr atomic.Pointer [http.Server ]
4348 log * slog.Logger
4449 Manager * fs.FunctionManager
4550}
4651
52+ type TubeLoaderType func (c * FactoryConfig ) (contube.TubeFactory , error )
53+ type RuntimeLoaderType func (c * FactoryConfig ) (api.FunctionRuntimeFactory , error )
54+
4755type serverOptions struct {
48- httpListener net.Listener
49- managerOpts []fs.ManagerOption
50- httpTubeFact * contube.HttpTubeFactory
56+ httpListener net.Listener
57+ managerOpts []fs.ManagerOption
58+ httpTubeFact * contube.HttpTubeFactory
59+ tubeLoader TubeLoaderType
60+ runtimeLoader RuntimeLoaderType
5161}
5262
5363type ServerOption interface {
@@ -86,6 +96,24 @@ func WithHttpTubeFactory(factory *contube.HttpTubeFactory) ServerOption {
8696 })
8797}
8898
99+ // WithTubeLoader sets the loader for the tube factory.
100+ // This must be called before WithConfig.
101+ func WithTubeLoader (loader TubeLoaderType ) ServerOption {
102+ return serverOptionFunc (func (o * serverOptions ) (* serverOptions , error ) {
103+ o .tubeLoader = loader
104+ return o , nil
105+ })
106+ }
107+
108+ // WithRuntimeLoader sets the loader for the runtime factory.
109+ // This must be called before WithConfig.
110+ func WithRuntimeLoader (loader RuntimeLoaderType ) ServerOption {
111+ return serverOptionFunc (func (o * serverOptions ) (* serverOptions , error ) {
112+ o .runtimeLoader = loader
113+ return o , nil
114+ })
115+ }
116+
89117func getRefFactory (m map [string ]* FactoryConfig , name string , visited set.Set [string ]) (string , error ) {
90118 if visited .Has (name ) {
91119 return "" , errors .Errorf ("circular reference of factory %s" , name )
@@ -101,7 +129,7 @@ func getRefFactory(m map[string]*FactoryConfig, name string, visited set.Set[str
101129 return name , nil
102130}
103131
104- func initFactories [T any ](m map [string ]* FactoryConfig , newFactory func (n string , c * FactoryConfig ) (T , error ), setup func (n string , f T )) error {
132+ func initFactories [T any ](m map [string ]* FactoryConfig , newFactory func (c * FactoryConfig ) (T , error ), setup func (n string , f T )) error {
105133 factoryMap := make (map [string ]T )
106134
107135 for name := range m {
@@ -114,7 +142,10 @@ func initFactories[T any](m map[string]*FactoryConfig, newFactory func(n string,
114142 if ! exist {
115143 return errors .Errorf ("factory %s not found, which the factory %s is pointed to" , refName , name )
116144 }
117- f , err := newFactory (refName , fc )
145+ if fc .Type == nil {
146+ return errors .Errorf ("factory %s type is not set" , refName )
147+ }
148+ f , err := newFactory (fc )
118149 if err != nil {
119150 return err
120151 }
@@ -126,43 +157,40 @@ func initFactories[T any](m map[string]*FactoryConfig, newFactory func(n string,
126157 return nil
127158}
128159
160+ func DefaultTubeLoader (c * FactoryConfig ) (contube.TubeFactory , error ) {
161+ switch strings .ToLower (* c .Type ) {
162+ case common .PulsarTubeType :
163+ return contube .NewPulsarEventQueueFactory (context .Background (), contube .ConfigMap (* c .Config ))
164+ case common .MemoryTubeType :
165+ return contube .NewMemoryQueueFactory (context .Background ()), nil
166+ }
167+ return nil , errors .WithMessagef (ErrUnsupportedTubeType , "unsupported tube type :%s" , * c .Type )
168+ }
169+
170+ func DefaultRuntimeLoader (c * FactoryConfig ) (api.FunctionRuntimeFactory , error ) {
171+ switch strings .ToLower (* c .Type ) {
172+ case WASMRuntime :
173+ return wazero .NewWazeroFunctionRuntimeFactory (), nil
174+ }
175+ return nil , errors .WithMessagef (ErrUnsupportedTRuntimeType , "unsupported runtime type: %s" , * c .Type )
176+ }
177+
129178func WithConfig (config * Config ) ServerOption {
130179 return serverOptionFunc (func (o * serverOptions ) (* serverOptions , error ) {
131180 ln , err := net .Listen ("tcp" , config .ListenAddr )
132181 if err != nil {
133182 return nil , err
134183 }
135184 o .httpListener = ln
136- err = initFactories [contube.TubeFactory ](config .TubeFactory , func (n string , c * FactoryConfig ) (contube.TubeFactory , error ) {
137- if c .Type == nil {
138- return nil , errors .Errorf ("tube factory %s type is not set" , n )
139- }
140- switch strings .ToLower (* c .Type ) {
141- case common .PulsarTubeType :
142- return contube .NewPulsarEventQueueFactory (context .Background (), contube .ConfigMap (* c .Config ))
143- case common .MemoryTubeType :
144- return contube .NewMemoryQueueFactory (context .Background ()), nil
145- }
146- return nil , errors .Errorf ("unsupported tube type %s" , * c .Type )
147- }, func (n string , f contube.TubeFactory ) {
185+ err = initFactories [contube.TubeFactory ](config .TubeFactory , o .tubeLoader , func (n string , f contube.TubeFactory ) {
148186 o .managerOpts = append (o .managerOpts , fs .WithTubeFactory (n , f ))
149187 })
150188 if err != nil {
151189 return nil , err
152190 }
153- err = initFactories [api.FunctionRuntimeFactory ](config .RuntimeFactory ,
154- func (n string , c * FactoryConfig ) (api.FunctionRuntimeFactory , error ) {
155- if c .Type == nil {
156- return nil , errors .Errorf ("runtime factory %s type is not set" , n )
157- }
158- switch strings .ToLower (* c .Type ) {
159- case WASMRuntime :
160- return wazero .NewWazeroFunctionRuntimeFactory (), nil
161- }
162- return nil , errors .Errorf ("unsupported runtime type %s" , * c .Type )
163- }, func (n string , f api.FunctionRuntimeFactory ) {
164- o .managerOpts = append (o .managerOpts , fs .WithRuntimeFactory (n , f ))
165- })
191+ err = initFactories [api.FunctionRuntimeFactory ](config .RuntimeFactory , o .runtimeLoader , func (n string , f api.FunctionRuntimeFactory ) {
192+ o .managerOpts = append (o .managerOpts , fs .WithRuntimeFactory (n , f ))
193+ })
166194 if err != nil {
167195 return nil , err
168196 }
@@ -178,6 +206,8 @@ func NewServer(opts ...ServerOption) (*Server, error) {
178206 fs .WithTubeFactory ("http" , httpTubeFact ),
179207 }
180208 options .httpTubeFact = httpTubeFact
209+ options .tubeLoader = DefaultTubeLoader
210+ options .runtimeLoader = DefaultRuntimeLoader
181211 for _ , o := range opts {
182212 if o == nil {
183213 continue
0 commit comments