Go 言語 Websocket アプリケーションの同時実行セキュリティ問題を解決する-Golang-php.cn

Go 言語 Websocket アプリケーションの同時実行セキュリティ問題を解決する

王林

リリース： 2023-12-14 13:47:39

オリジナル

690 人が閲覧しました

Go 言語 Websocket アプリケーションの同時実行セキュリティ問題を解決する

WebSocket は、強力なリアルタイムパフォーマンスを備えた双方向通信を実現できる最新のネットワーク通信プロトコルです。 Go 言語は本質的に同時実行性をサポートしているため、Websocket アプリケーションで非常に優れたパフォーマンスを発揮します。ただし、同時実行にはいくつかの問題も伴い、Websocket アプリケーションでは、これは主に同時実行のセキュリティに反映されます。この記事では、Go Websocket アプリケーションの同時実行セキュリティの問題を解決する方法を説明し、デモンストレーションします。

問題の背景

Websocket アプリケーションでは、クライアントはいつでもサーバーにメッセージを送信でき、サーバーもいつでもクライアントにメッセージを送信できます。。したがって、Websocket メッセージを処理するときは、同時実行の問題を考慮する必要があります。 Go 言語では、Goroutine を使用して WebSocket メッセージを同時に処理できます。

ただし、同時実行により、競合状態やデッドロックなど、同時実行のセキュリティ上の問題が発生することがあります。競合状態によりデータの不整合が発生したり、デッドロックによりプログラムがフリーズしたりすることがあります。したがって、Websocket アプリケーションでは、これらの同時実行セキュリティの問題を解決する必要があります。

解決策

2.1 ミューテックスロック

ミューテックスロックは、Go 言語で最も一般的な同時実行制御メカニズムの 1 つです。共有リソースを保護し、複数のゴルーチンが同時に共有リソースにアクセスすることを防ぎ、データの正確性と一貫性を保証します。

Websocket アプリケーションでは、ミューテックスロックを通じて共有リソースの同時実行セキュリティを確保できます。たとえば、次のコードは、ミューテックスロックを使用して、共有マップに同時に書き込む複数のゴルーチンの安全性を確保する方法を示しています。

type safeMap struct {
    m map[string]int
    sync.Mutex
}

func (sm *safeMap) Inc(key string) {
    sm.Lock()
    sm.m[key]++
    sm.Unlock()
}

ログイン後にコピー

この例では、構造体に sync.Mutex を埋め込みます。共有リソースを保護するための、safeMap タイプのミューテックス。この構造では、複数のゴルーチンによって共有されるリソースを表すマップ型変数 m を定義します。次に、マップ内のデータに対して自動インクリメント操作を実行するために、safeMap のメソッド Inc を定義しました。メソッド Inc では、最初にロックし、次にインクリメント操作を実行し、最後にロックを解除します。

2.2 ロックフリーの同時実行性

同時実行のセキュリティ問題を解決するもう 1 つの方法は、ロックフリーの同時実行性を使用することです。ロックフリーの同時実行では、非ブロッキングアルゴリズムを使用することで、ミューテックスロックによって引き起こされるパフォーマンスの損失を回避します。システムの同時実行性とスループットを向上させることができ、高性能、低遅延、高スループットのシステムでよく使用されます。

Go 言語では、sync/atomic パッケージのアトミック操作関数を使用して、ロックフリーの同時実行性を実現できます。たとえば、次のコードは、アトミック操作を使用して共有変数に対する同時操作を実装する方法を示しています。

type Counter struct {
    count int32
}

func (c *Counter) Inc() {
    atomic.AddInt32(&c.count, 1)
}

func (c *Counter) Dec() {
    atomic.AddInt32(&c.count, -1)
}

func (c *Counter) Get() int32 {
    return atomic.LoadInt32(&c.count)
}

ログイン後にコピー

この例では、アトミックパッケージの AddInt32 関数と LoadInt32 関数を使用してカウンターを実装します。 int32 型の count 変数を含む構造体 Counter を定義します。 Counter 構造体は、Inc、Dec、Get という 3 つのメソッドも実装します。 Inc メソッドと Dec メソッドでは、アトミック操作 AddInt32 を使用して、共有変数のカウントをインクリメントおよびデクリメントします。 Get メソッドでは、アトミック操作 LoadInt32 を使用して共有変数 count の値を取得します。

サンプルコード

次に、同時実行の安全性を確保するためにミューテックスロックを使用する Websocket アプリケーションのサンプルコードを示します。

package main

import (
    "fmt"
    "net/http"
    "sync"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

type Connection struct {
    ws *websocket.Conn
    mu sync.Mutex
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

func handler(w http.ResponseWriter, r *http.Request) {
    c, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println(err)
        return
    }

    conn := &Connection{ws: c}

    go conn.WriteLoop()
    conn.ReadLoop()
}

func (conn *Connection) ReadLoop() {
    defer conn.ws.Close()
    for {
        _, message, err := conn.ws.ReadMessage()
        if err != nil {
            fmt.Println(err)
            break
        }

        fmt.Printf("Received message: %s
", message)
    }
}

func (conn *Connection) WriteLoop() {
    defer conn.ws.Close()
    for {
        conn.mu.Lock()
        err := conn.ws.WriteMessage(websocket.TextMessage, []byte("Hello, world!"))
        conn.mu.Unlock()
        if err != nil {
            fmt.Println(err)
            break
        }
    }
}

ログイン後にコピー

この例では、私たちは単純な Websocket アプリケーションを実装しました。これには、クライアントメッセージを読み取る ReadLoop と、クライアントにメッセージを送信する WriteLoop が含まれています。このアプリケーションでは、各クライアントの接続を Connection 構造にカプセル化し、sync.Mutex タイプの mutex mu を埋め込みます。 WriteLoop でこのミューテックスロックを使用して、共有リソース conn.ws の同時実行の安全性を確保します。ミューテックスロックを使用すると、複数のゴルーチンが同じ Websocket 接続に同時にデータを書き込む問題を回避できます。

以下は、アトミック操作を使用してロックフリーの同時実行性を実現する Websocket アプリケーションのサンプルコードです:

package main

import (
    "fmt"
    "net/http"
    "sync/atomic"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

type Connection struct {
    ws    *websocket.Conn
    count int32
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

func handler(w http.ResponseWriter, r *http.Request) {
    c, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println(err)
        return
    }

    conn := &Connection{ws: c}

    go conn.WriteLoop()
    conn.ReadLoop()
}

func (conn *Connection) ReadLoop() {
    defer conn.ws.Close()
    for {
        _, message, err := conn.ws.ReadMessage()
        if err != nil {
            fmt.Println(err)
            break
        }

        fmt.Printf("Received message: %s
", message)
    }
}

func (conn *Connection) WriteLoop() {
    defer conn.ws.Close()
    for {
        n := atomic.AddInt32(&conn.count, 1)
        if n > 10 {
            break
        }

        err := conn.ws.WriteMessage(websocket.TextMessage, []byte("Hello, world!"))
        if err != nil {
            fmt.Println(err)
            break
        }
    }
}

ログイン後にコピー

この例では、アトミックパッケージの AddInt32 関数と LoadInt32 関数を使用して、カウンターを実装します。 int32 型の count 変数を含む構造体 Connection を定義します。 Connection 構造体は、ReadLoop と WriteLoop という 2 つのメソッドも実装します。 WriteLoop メソッドでは、アトミック操作 AddInt32 を使用して、共有変数の数をインクリメントします。次に、カウンター値が 10 を超えているかどうかを判断し、超えている場合はループを終了します。この例では、ミューテックスを使用する代わりに、アトミック操作を使用してロックフリーの同時実行性を実現します。

この記事では、Go 言語の Websocket アプリケーションにおける同時実行セキュリティの問題を解決する方法を紹介します。ミューテックスロックとロックフリーの同時実行という 2 つの解決策を提供します。ミューテックスロックであっても、ロックなしの同時実行であっても、同時実行のセキュリティは保証されます。どの方法を選択するかは、特定のアプリケーションのシナリオと要件によって異なります。これらのテクノロジの使用方法を特定のサンプルコードを通じて示し、読者がこれらのテクノロジをよりよく理解して適用できるようにしたいと考えています。

以上がGo 言語 Websocket アプリケーションの同時実行セキュリティ問題を解決するの詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。