| 382 | } |
| 383 | |
| 384 | func (b *DBBatcher) buildParams() database.BatchUpsertConnectionLogsParams { |
| 385 | count := b.batchLen() |
| 386 | var ( |
| 387 | ids = make([]uuid.UUID, 0, count) |
| 388 | connectTime = make([]time.Time, 0, count) |
| 389 | organizationID = make([]uuid.UUID, 0, count) |
| 390 | workspaceOwnerID = make([]uuid.UUID, 0, count) |
| 391 | workspaceID = make([]uuid.UUID, 0, count) |
| 392 | workspaceName = make([]string, 0, count) |
| 393 | agentName = make([]string, 0, count) |
| 394 | connType = make([]database.ConnectionType, 0, count) |
| 395 | code = make([]int32, 0, count) |
| 396 | codeValid = make([]bool, 0, count) |
| 397 | ip = make([]pqtype.Inet, 0, count) |
| 398 | userAgent = make([]string, 0, count) |
| 399 | userID = make([]uuid.UUID, 0, count) |
| 400 | slugOrPort = make([]string, 0, count) |
| 401 | connectionID = make([]uuid.UUID, 0, count) |
| 402 | disconnectReason = make([]string, 0, count) |
| 403 | disconnectTime = make([]time.Time, 0, count) |
| 404 | ) |
| 405 | |
| 406 | appendEntry := func(e batchEntry) { |
| 407 | ids = append(ids, e.ID) |
| 408 | connectTime = append(connectTime, e.connectTime) |
| 409 | organizationID = append(organizationID, e.OrganizationID) |
| 410 | workspaceOwnerID = append(workspaceOwnerID, e.WorkspaceOwnerID) |
| 411 | workspaceID = append(workspaceID, e.WorkspaceID) |
| 412 | workspaceName = append(workspaceName, e.WorkspaceName) |
| 413 | agentName = append(agentName, e.AgentName) |
| 414 | connType = append(connType, e.Type) |
| 415 | code = append(code, e.Code.Int32) |
| 416 | codeValid = append(codeValid, e.Code.Valid) |
| 417 | ip = append(ip, e.IP) |
| 418 | userAgent = append(userAgent, e.UserAgent.String) |
| 419 | userID = append(userID, e.UserID.UUID) |
| 420 | slugOrPort = append(slugOrPort, e.SlugOrPort.String) |
| 421 | connectionID = append(connectionID, e.ConnectionID.UUID) |
| 422 | disconnectReason = append(disconnectReason, e.DisconnectReason.String) |
| 423 | disconnectTime = append(disconnectTime, e.disconnectTime) |
| 424 | } |
| 425 | |
| 426 | for _, entry := range b.dedupedBatch { |
| 427 | appendEntry(entry) |
| 428 | } |
| 429 | for _, entry := range b.nullConnIDBatch { |
| 430 | appendEntry(entry) |
| 431 | } |
| 432 | |
| 433 | return database.BatchUpsertConnectionLogsParams{ |
| 434 | ID: ids, |
| 435 | ConnectTime: connectTime, |
| 436 | OrganizationID: organizationID, |
| 437 | WorkspaceOwnerID: workspaceOwnerID, |
| 438 | WorkspaceID: workspaceID, |
| 439 | WorkspaceName: workspaceName, |
| 440 | AgentName: agentName, |
| 441 | Type: connType, |