mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
7677aad450
*) I/O protocol parse issues should shutdown the container as the container goes to inconsistent state between calls. (eg. next call may receive previous calls left overs.) *) Move ghost read/write code into io_utils in common. *) Clean unused error from docker Wait() *) We can catch one case in JSON, if there's remaining unparsed data in decoder buffer, we can shut the container *) stdout/stderr when container is not handling a request are now blocked if freezer is also enabled. *) if a fatal err is set for slot, we do not requeue it and proceed to shutdown *) added a test function for a few cases with freezer strict behavior
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/fnproject/fn/api/agent/drivers"
|
|
)
|
|
|
|
func New() drivers.Driver {
|
|
return &Mocker{}
|
|
}
|
|
|
|
type Mocker struct {
|
|
count int
|
|
}
|
|
|
|
func (m *Mocker) Prepare(context.Context, drivers.ContainerTask) (drivers.Cookie, error) {
|
|
return &cookie{m}, nil
|
|
}
|
|
|
|
type cookie struct {
|
|
m *Mocker
|
|
}
|
|
|
|
func (c *cookie) Freeze(context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *cookie) Unfreeze(context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *cookie) Close(context.Context) error { return nil }
|
|
|
|
func (c *cookie) Run(ctx context.Context) (drivers.WaitResult, error) {
|
|
c.m.count++
|
|
if c.m.count%100 == 0 {
|
|
return nil, fmt.Errorf("Mocker error! Bad.")
|
|
}
|
|
return &runResult{
|
|
err: nil,
|
|
status: "success",
|
|
start: time.Now(),
|
|
}, nil
|
|
}
|
|
|
|
type runResult struct {
|
|
err error
|
|
status string
|
|
start time.Time
|
|
}
|
|
|
|
func (r *runResult) Wait(context.Context) drivers.RunResult { return r }
|
|
func (r *runResult) Status() string { return r.status }
|
|
func (r *runResult) Error() error { return r.err }
|