wss-pool/internal/model/pingmessage.go

package model

import (
	"encoding/json"
	"fmt"
	"wss-pool/logging/applogger"
)

type PingMessage struct {
	Ping int64 `json:"ping"`
}

type SymbolMessage struct {
	Type   string `json:"type"`
	Symbol string `json:"symbol"`
}

func ParsePingMessage(message string) *PingMessage {
	result := PingMessage{}
	err := json.Unmarshal([]byte(message), &result)
	if err != nil {
		return nil
	}

	return &result
}

func SubMessage(message string) *SymbolMessage {
	result := SymbolMessage{}
	err := json.Unmarshal([]byte(message), &result)
	applogger.Info("ws param %v", message)
	if err != nil {
		fmt.Println("subMessage", err)
		return nil
	}

	return &result
}

func ReturnValue(str string) string {
	pongMsg := fmt.Sprintf("{\"type\": \"%v\"}", str)

	return pongMsg
}