(body []byte, stream flight.FlightService_DoActionServer)
| 291 | return status.Errorf(codes.Internal, "start activation drain tracking: %v", err) |
| 292 | } |
| 293 | defer finishDrain() |
| 294 | |
| 295 | if err := h.pool.activateTenantFunc(payload); err != nil { |
| 296 | return status.Errorf(codes.FailedPrecondition, "activate tenant: %v", err) |
| 297 | } |
| 298 | |
| 299 | resp, _ := json.Marshal(map[string]bool{"ok": true}) |
| 300 | return sendActionResult(stream, &flight.Result{Body: resp}) |
| 301 | } |
| 302 | |
| 303 | func (h *FlightSQLHandler) doDestroySession(body []byte, stream flight.FlightService_DoActionServer) error { |
| 304 | var req server.WorkerDestroySessionPayload |
| 305 | if err := json.Unmarshal(body, &req); err != nil { |
| 306 | return status.Errorf(codes.InvalidArgument, "invalid DestroySession request: %v", err) |
| 307 | } |
| 308 | if req.SessionToken == "" { |
| 309 | return status.Error(codes.InvalidArgument, "session_token is required") |
| 310 | } |
| 311 | if err := h.pool.validateControlMetadata(req.WorkerControlMetadata); err != nil { |
| 312 | return status.Errorf(codes.FailedPrecondition, "stale worker owner: %v", err) |
| 313 | } |
| 314 |
no test coverage detected