Go 언어 서버 프로그래밍 실습: 간단한 웹 채팅방 만들기

Go 언어는 서버 프로그래밍 분야에서 점점 더 많이 사용되고 있습니다. Go 언어의 빠르고 효율적이며 동시성 기능은 많은 사람들이 Go 언어를 선택하는 이유가 되었습니다. 이 기사에서는 실시간 메시지 전송, 다중 사용자 온라인 상호 작용 및 기타 관련 지식 포인트를 포함하여 Go 언어로 간단한 웹 채팅방을 구축하는 방법을 소개합니다.

1. Go 언어 웹 프레임워크 선택

Go 언어에는 선택할 수 있는 웹 프레임워크가 많이 있으며 그중 가장 인기 있는 것은 Beego, Gin, Echo 등입니다. 이 기사에서는 가볍고 고성능이며 사용하기 쉽고 웹 애플리케이션을 빠르게 구축하는 데 적합하기 때문에 Echo 프레임워크를 사용하기로 선택했습니다. Echo 프레임워크를 사용하기 전에 터미널을 열고 다음 명령을 실행하여 종속성 패키지를 얻으세요.

go get -u github.com/labstack/echo

2. 기능 설계

이 웹 채팅방을 구축하기 전에 필요한 기능 모듈과 프로세스를 설계해야 합니다.

1. 사용자 인증을 받으려면 계정을 등록하고 로그인하세요.
2. 채팅방 홈페이지에는 온라인 사용자 목록, 과거 채팅 기록 등이 표시될 수 있습니다.
3. 실시간 메시지 전송, 다중 사용자 온라인 상호작용 지원.
4. 브라우저가 닫히면 서버에 대한 WebSocket 연결이 끊어지고 사용자가 오프라인으로 표시됩니다.

3. 구현 단계

아래에서는 위의 기능 모듈 및 프로세스를 단계별로 구현합니다.

1. 계정 등록 및 로그인:

Go 언어에서는 타사 패키지 bcrypt를 사용할 수 있습니다. 비밀번호 암호화를 구현합니다. 먼저 사용자 구조 User를 생성한 다음 gorm 라이브러리를 사용하여 데이터베이스 테이블을 생성해야 합니다.

type User struct {
    gorm.Model
    Username string `gorm:"unique;not null"`
    Password string `gorm:"not null"`
}

사용자 등록 및 로그인 처리 기능 코드는 다음과 같습니다.

// 用户注册
func Register(c echo.Context) error {
    username := c.FormValue("username")
    password := c.FormValue("password")
    confirmPwd := c.FormValue("confirm_pwd")
    u := &models.User{
        Username: username,
        Password: util.HashPwd(password),
    }
    if password != confirmPwd {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Passwords do not match",
        })
    }
    err := db.Create(&u).Error
    if err != nil {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Username already exists",
        })
    }
    return c.JSON(http.StatusOK, map[string]interface{}{
        "success": true,
        "message": "Register success",
    })
}

// 用户登录
func Login(c echo.Context) error {
    username := c.FormValue("username")
    password := c.FormValue("password")
    user := &models.User{}
    err := db.Where("username = ?", username).First(&user).Error
    if err != nil {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Username or password is incorrect",
        })
    }
    if util.CheckPwd(password, user.Password) {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": true,
            "message": "Login success",
        })
    }
    return c.JSON(http.StatusOK, map[string]interface{}{
        "success": false,
        "message": "Username or password is incorrect",
    })
}

2. 톡방 홈페이지:

톡방 홈페이지에는 온라인 사용자 목록과 채팅 기록 기록이 표시되어야 합니다. 온라인 사용자 목록은 Map에 저장될 수 있으며, 여기서 키는 사용자 이름이고 값은 연결 ID, 연결 시간 등 WebSocket 연결과 관련된 정보입니다.

// 在线用户列表
var clients = make(map[string]*Client)

과거 채팅 기록은 배열 또는 Redis와 유사한 데이터 구조를 사용하여 저장할 수 있습니다. 다음은 홈 페이지 처리 기능의 코드입니다.

// 聊天室首页
func Room(c echo.Context) error {
    return c.Render(http.StatusOK, "room.html", map[string]interface{}{
        "clients": clients,
        "msgs":    msgs,
    })
}

그 중 room.html이 템플릿 파일이고, 클라이언트와 msgs가 각각 온라인 사용자 목록과 과거 채팅 기록입니다.

3. 실시간 메시지 전송:

Echo 프레임워크에서는 WebSocket을 사용하여 매우 간단하게 실시간 메시지 전송을 구현할 수 있습니다. WebSocket 처리 기능을 정의해야 합니다. 클라이언트가 서버에 성공적으로 연결되면 클라이언트가 메시지를 보내면 해당 정보가 연결된 모든 클라이언트에 전송됩니다.

// WebSocket处理函数
func wsHandler(c echo.Context) error {
    // 建立WebSocket连接
    ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
    if err != nil {
        return err
    }
    defer ws.Close()

    // 获取用户名
    username, ok := c.QueryParams()["username"]
    if !ok {
        return errors.New("username is required")
    }

    // 添加用户
    client := &Client{
        WS:        ws,
        Username: username[0],
        ConnTime:  time.Now(),
    }
    clients[client.Username] = client

    // 遍历用户列表，发送欢迎消息
    for _, c := range clients {
        msg := &Message{
            Type: WelcomeMsg,
            Data: fmt.Sprintf("User '%s' joined the chat room", client.Username),
        }
        err = c.Send(msg)
        if err != nil {
            log.Println(err)
            continue
        }
    }

    // 接收消息，广播给所有用户
    for {
        msg := &Message{}
        err = client.Receive(msg)
        if err != nil {
            if err == io.EOF {
                delete(clients, client.Username)
                for _, c := range clients {
                    msg = &Message{
                        Type: LeaveMsg,
                        Data: fmt.Sprintf("User '%s' left the chat room", client.Username),
                    }
                    err = c.Send(msg)
                    if err != nil {
                        log.Println(err)
                        continue
                    }
                }
                break
            }
            log.Println(err)
            continue
        }
        for _, c := range clients {
            if c.Username == client.Username {
                continue
            }
            err = c.Send(msg)
            if err != nil {
                log.Println(err)
                continue
            }
        }
        msgs = append(msgs, *msg)
    }

    return nil
}

4. WebSocket 연결을 닫을 때 서버와의 연결을 끊고 사용자를 오프라인으로 표시하세요. WebSocket 처리 함수에서 defer 문을 사용하면 함수 마지막에 온라인 사용자 목록의 사용자 정보를 삭제할 수 있습니다.

defer func() {
    delete(clients, client.Username)
}()

IV. 요약

이 글에서는 Echo 프레임워크와 WebSocket을 사용하여 사용자 인증, 온라인 사용자 목록, 채팅 내역, 실시간 메시지 전송 및 기타 기능을 포함하는 간단한 웹 채팅방을 구현합니다. 동시에 우리는 gorm 라이브러리와 bcrypt 패키지를 사용하여 사용자 관리 및 비밀번호 암호화 기능을 구현합니다.

실제로는 메시지 필터링, 민감한 단어 필터링 등 보안 문제도 고려해야 합니다. 독자들이 이 기사를 통해 Go 언어를 사용하여 웹 애플리케이션을 구축하는 방법은 물론 일반적으로 사용되는 라이브러리와 프레임워크를 배울 수 있기를 바랍니다.

위 내용은 Go 언어 서버 프로그래밍 실습: 간단한 웹 채팅방 만들기의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!