Function EncodeAgentIDChunks

coderd/agentapi/metadatabatcher/agentid_chunks.go:35–59

EncodeAgentIDChunks encodes agent IDs into chunks that fit within the PostgreSQL NOTIFY 8KB payload size limit. Each UUID is base64-encoded (without padding) and concatenated into a single byte slice per chunk.

(agentIDs []uuid.UUID)

// PostgreSQL NOTIFY 8KB payload size limit. Each UUID is base64-encoded
// (without padding) and concatenated into a single byte slice per chunk.
func EncodeAgentIDChunks(agentIDs []uuid.UUID) ([][]byte, error) {
	chunks := make([][]byte, 0, (len(agentIDs)+maxAgentIDsPerChunk-1)/maxAgentIDsPerChunk)

	for i := 0; i < len(agentIDs); i += maxAgentIDsPerChunk {
		end := i + maxAgentIDsPerChunk
		if end > len(agentIDs) {
			end = len(agentIDs)
		}

		chunk := agentIDs[i:end]

		// Build payload by base64-encoding each UUID (without padding) and
		// concatenating them. This is UTF-8 safe for PostgreSQL NOTIFY.
		payload := make([]byte, len(chunk)*UUIDBase64Size)
		for i, agentID := range chunk {
			err := EncodeAgentID(agentID, payload[i*UUIDBase64Size:(i+1)*UUIDBase64Size])
			if err != nil {
				return nil, err
			}
		}
		chunks = append(chunks, payload)
	}

	return chunks, nil
}