Comment lire les données jusqu'au délimiteur CRLF dans Golang à l'aide de bufio.Scanner ?

Comment lire des données multilignes jusqu'au délimiteur CRLF dans Golang avec bufio

Dans le package bufio de Golang, les fonctions ReadBytes, ReadSlice et ReadString sont couramment utilisé pour lire des données délimitées par des octets ou des chaînes spécifiques. Cependant, aucune de ces fonctions ne fournit un moyen simple de lire des données jusqu'à un délimiteur CRLF, comme l'exige le protocole de Beanstalkd.

Utiliser bufio.Scanner

Une meilleure approche est pour utiliser le type bufio.Scanner pour créer une fonction de fractionnement personnalisée qui recherche les délimiteurs CRLF. Cela vous permet de lire les données jusqu'à ce qu'un délimiteur CRLF soit rencontré. Voici comment procéder :

<code class="go">import (
    "bufio"
    "bytes"
)

func ScanCRLF(data []byte, atEOF bool) (advance int, token []byte, err error) {
    if atEOF && len(data) == 0 {
        return 0, nil, nil
    }
    if i := bytes.Index(data, []byte{'\r', '\n'}); i >= 0 {
        // We have a full newline-terminated line.
        return i + 2, dropCR(data[0:i]), nil
    }
    // If we're at EOF, we have a final, non-terminated line. Return it.
    if atEOF {
        return len(data), dropCR(data), nil
    }
    // Request more data.
    return 0, nil, nil
}

func dropCR(data []byte) []byte {
    if len(data) > 0 && data[len(data)-1] == '\r' {
        return data[0 : len(data)-1]
    }
    return data
}</code>

Utiliser votre scanner personnalisé

Maintenant, vous pouvez envelopper votre io.Reader avec votre scanner personnalisé et l'utiliser pour lire des données jusqu'aux délimiteurs CRLF :

<code class="go">reader := bufio.NewReader(conn)
scanner := bufio.NewScanner(reader)
scanner.Split(ScanCRLF)

for scanner.Scan() {
    fmt.Println(scanner.Text())
}

if err := scanner.Err(); err != nil {
    fmt.Printf("Invalid input: %s", err)
}</code>

Approche alternative : lecture d'un nombre spécifique d'octets

Vous pouvez également lire un nombre spécifique d'octets si vous connaissez le nombre attendu d'octets octets dans les données. Cependant, cette approche est moins robuste et nécessite une validation minutieuse pour garantir que vous ne lisez pas trop ou pas assez de données. Pour lire un nombre spécifique d'octets à l'aide du type bufio.Reader :

<code class="go">nr_of_bytes := read_number_of_butes_somehow(conn)
buf := make([]byte, nr_of_bytes)
reader.Read(buf)</code>

Gestion des conditions d'erreur

Les deux approches ci-dessus nécessitent une gestion minutieuse des conditions d'erreur, en particulier lorsqu'il s'agit de lectures partielles et d'EOF prématurés. Assurez-vous de valider la saisie et de gérer les erreurs de manière appropriée.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!