diff --git a/plugins/drivers/utils.go b/plugins/drivers/utils.go index 5b780faf1..c146e96ef 100644 --- a/plugins/drivers/utils.go +++ b/plugins/drivers/utils.go @@ -49,21 +49,22 @@ func taskConfigFromProto(pb *proto.TaskConfig) *TaskConfig { return &TaskConfig{} } return &TaskConfig{ - ID: pb.Id, - JobName: pb.JobName, - TaskGroupName: pb.TaskGroupName, - Name: pb.Name, - Env: pb.Env, - DeviceEnv: pb.DeviceEnv, - rawDriverConfig: pb.MsgpackDriverConfig, - Resources: ResourcesFromProto(pb.Resources), - Devices: DevicesFromProto(pb.Devices), - Mounts: MountsFromProto(pb.Mounts), - User: pb.User, - AllocDir: pb.AllocDir, - StdoutPath: pb.StdoutPath, - StderrPath: pb.StderrPath, - AllocID: pb.AllocId, + ID: pb.Id, + JobName: pb.JobName, + TaskGroupName: pb.TaskGroupName, + Name: pb.Name, + Env: pb.Env, + DeviceEnv: pb.DeviceEnv, + rawDriverConfig: pb.MsgpackDriverConfig, + Resources: ResourcesFromProto(pb.Resources), + Devices: DevicesFromProto(pb.Devices), + Mounts: MountsFromProto(pb.Mounts), + User: pb.User, + AllocDir: pb.AllocDir, + StdoutPath: pb.StdoutPath, + StderrPath: pb.StderrPath, + AllocID: pb.AllocId, + NetworkIsolation: NetworkIsolationSpecFromProto(pb.NetworkIsolationSpec), } } @@ -72,21 +73,22 @@ func taskConfigToProto(cfg *TaskConfig) *proto.TaskConfig { return &proto.TaskConfig{} } pb := &proto.TaskConfig{ - Id: cfg.ID, - JobName: cfg.JobName, - TaskGroupName: cfg.TaskGroupName, - Name: cfg.Name, - Env: cfg.Env, - DeviceEnv: cfg.DeviceEnv, - Resources: ResourcesToProto(cfg.Resources), - Devices: DevicesToProto(cfg.Devices), - Mounts: MountsToProto(cfg.Mounts), - User: cfg.User, - AllocDir: cfg.AllocDir, - MsgpackDriverConfig: cfg.rawDriverConfig, - StdoutPath: cfg.StdoutPath, - StderrPath: cfg.StderrPath, - AllocId: cfg.AllocID, + Id: cfg.ID, + JobName: cfg.JobName, + TaskGroupName: cfg.TaskGroupName, + Name: cfg.Name, + Env: cfg.Env, + DeviceEnv: cfg.DeviceEnv, + Resources: ResourcesToProto(cfg.Resources), + Devices: DevicesToProto(cfg.Devices), + Mounts: MountsToProto(cfg.Mounts), + User: cfg.User, + AllocDir: cfg.AllocDir, + MsgpackDriverConfig: cfg.rawDriverConfig, + StdoutPath: cfg.StdoutPath, + StderrPath: cfg.StderrPath, + AllocId: cfg.AllocID, + NetworkIsolationSpec: NetworkIsolationSpecToProto(cfg.NetworkIsolation), } return pb }