runtime/shim: check the namespace flag first

Signed-off-by: Iceber Gu <wei.cai-nat@daocloud.io>
This commit is contained in:
Iceber Gu 2021-04-16 15:44:57 +08:00
parent f0890f9b3a
commit 34780d67ad

View File

@ -167,12 +167,17 @@ func run(id string, initFunc Init, config Config) error {
return nil return nil
} }
if namespaceFlag == "" {
return fmt.Errorf("shim namespace cannot be empty")
}
setRuntime() setRuntime()
signals, err := setupSignals(config) signals, err := setupSignals(config)
if err != nil { if err != nil {
return err return err
} }
if !config.NoSubreaper { if !config.NoSubreaper {
if err := subreaper(); err != nil { if err := subreaper(); err != nil {
return err return err
@ -180,17 +185,12 @@ func run(id string, initFunc Init, config Config) error {
} }
ttrpcAddress := os.Getenv(ttrpcAddressEnv) ttrpcAddress := os.Getenv(ttrpcAddressEnv)
publisher, err := NewPublisher(ttrpcAddress) publisher, err := NewPublisher(ttrpcAddress)
if err != nil { if err != nil {
return err return err
} }
defer publisher.Close() defer publisher.Close()
if namespaceFlag == "" {
return fmt.Errorf("shim namespace cannot be empty")
}
ctx := namespaces.WithNamespace(context.Background(), namespaceFlag) ctx := namespaces.WithNamespace(context.Background(), namespaceFlag)
ctx = context.WithValue(ctx, OptsKey{}, Opts{BundlePath: bundlePath, Debug: debugFlag}) ctx = context.WithValue(ctx, OptsKey{}, Opts{BundlePath: bundlePath, Debug: debugFlag})
ctx = log.WithLogger(ctx, log.G(ctx).WithField("runtime", id)) ctx = log.WithLogger(ctx, log.G(ctx).WithField("runtime", id))
@ -199,6 +199,7 @@ func run(id string, initFunc Init, config Config) error {
if err != nil { if err != nil {
return err return err
} }
switch action { switch action {
case "delete": case "delete":
logger := logrus.WithFields(logrus.Fields{ logger := logrus.WithFields(logrus.Fields{