ㅑ후
This commit is contained in:
commit
766cf34e71
68
.gitignore
vendored
Normal file
68
.gitignore
vendored
Normal file
|
@ -0,0 +1,68 @@
|
||||||
|
# Created by .ignore support plugin (hsz.mobi)
|
||||||
|
### JetBrains template
|
||||||
|
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
|
||||||
|
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
|
||||||
|
|
||||||
|
# User-specific stuff:
|
||||||
|
.idea/**/workspace.xml
|
||||||
|
.idea/**/tasks.xml
|
||||||
|
.idea/dictionaries
|
||||||
|
|
||||||
|
# Sensitive or high-churn files:
|
||||||
|
.idea/**/dataSources/
|
||||||
|
.idea/**/dataSources.ids
|
||||||
|
.idea/**/dataSources.xml
|
||||||
|
.idea/**/dataSources.local.xml
|
||||||
|
.idea/**/sqlDataSources.xml
|
||||||
|
.idea/**/dynamic.xml
|
||||||
|
.idea/**/uiDesigner.xml
|
||||||
|
|
||||||
|
# Gradle:
|
||||||
|
.idea/**/gradle.xml
|
||||||
|
.idea/**/libraries
|
||||||
|
|
||||||
|
# Mongo Explorer plugin:
|
||||||
|
.idea/**/mongoSettings.xml
|
||||||
|
|
||||||
|
## File-based project format:
|
||||||
|
*.iws
|
||||||
|
|
||||||
|
## Plugin-specific files:
|
||||||
|
|
||||||
|
# IntelliJ
|
||||||
|
/out/
|
||||||
|
|
||||||
|
# mpeltonen/sbt-idea plugin
|
||||||
|
.idea_modules/
|
||||||
|
|
||||||
|
# JIRA plugin
|
||||||
|
atlassian-ide-plugin.xml
|
||||||
|
|
||||||
|
# Crashlytics plugin (for Android Studio and IntelliJ)
|
||||||
|
com_crashlytics_export_strings.xml
|
||||||
|
crashlytics.properties
|
||||||
|
crashlytics-build.properties
|
||||||
|
fabric.properties
|
||||||
|
### Go template
|
||||||
|
# Binaries for programs and plugins
|
||||||
|
*.exe
|
||||||
|
*.dll
|
||||||
|
*.so
|
||||||
|
*.dylib
|
||||||
|
|
||||||
|
# Test binary, build with `go test -c`
|
||||||
|
*.test
|
||||||
|
|
||||||
|
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||||
|
*.out
|
||||||
|
|
||||||
|
# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
|
||||||
|
.glide/
|
||||||
|
.idea/
|
||||||
|
*.iml
|
||||||
|
|
||||||
|
vendor/
|
||||||
|
glide.lock
|
||||||
|
.DS_Store
|
||||||
|
dist/
|
||||||
|
debug
|
32
.vscode/launch.json
vendored
Normal file
32
.vscode/launch.json
vendored
Normal file
|
@ -0,0 +1,32 @@
|
||||||
|
{
|
||||||
|
"version": "0.2.0",
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "Debug",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "debug",
|
||||||
|
"remotePath": "",
|
||||||
|
"port": 2345,
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"program": "${workspaceRoot}/main.go",
|
||||||
|
"env": {},
|
||||||
|
"args": [],
|
||||||
|
"showLog": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "File Debug",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "debug",
|
||||||
|
"remotePath": "",
|
||||||
|
"port": 2345,
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"program": "${fileDirname}",
|
||||||
|
"env": {},
|
||||||
|
"args": [],
|
||||||
|
"showLog": true
|
||||||
|
}
|
||||||
|
|
||||||
|
]
|
||||||
|
}
|
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
// Place your settings in this file to overwrite default and user settings.
|
||||||
|
{
|
||||||
|
}
|
5
config/auth.go
Normal file
5
config/auth.go
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
type Auth struct {
|
||||||
|
SigningKey string `json:"signingKey" yaml:"signingKey" toml:"signingKey"`
|
||||||
|
}
|
12
config/config.go
Normal file
12
config/config.go
Normal file
|
@ -0,0 +1,12 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
var Config GatewayConfig
|
||||||
|
|
||||||
|
type GatewayConfig struct {
|
||||||
|
Server Server `json:"server" yaml:"server" toml:"server"`
|
||||||
|
Auth Auth `json:"auth" yaml:"auth" toml:"auth"`
|
||||||
|
Websocket Websocket `json:"websocket" yaml:"websocket" toml:"websocket"`
|
||||||
|
GRPC GRPC `json:"gRPC" yaml:"gRPC" toml:"gRPC"`
|
||||||
|
Redis Redis `json:"redis" yaml:"redis" toml:"redis"`
|
||||||
|
Handlers map[string]Handler `json:"handlers" yaml:"handlers" toml:"handlers"`
|
||||||
|
}
|
6
config/grpc.go
Normal file
6
config/grpc.go
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
type GRPC struct {
|
||||||
|
Server
|
||||||
|
Pool Pool `json:"pool" yaml:"pool" toml:"pool"`
|
||||||
|
}
|
6
config/handler.go
Normal file
6
config/handler.go
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
type Handler struct {
|
||||||
|
Entry string `json:"entry" yaml:"entry" toml:"entry"`
|
||||||
|
Socket Socket `json:"socket" yaml:"socket" toml:"socket"`
|
||||||
|
}
|
10
config/pool.go
Normal file
10
config/pool.go
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type Pool struct {
|
||||||
|
MaxCapacity int `json:"maxCapacity" yaml:"maxCapacity" toml:"maxCapacity"`
|
||||||
|
MaxIdle int `json:"maxIdle" yaml:"maxIdle" toml:"maxIdle"`
|
||||||
|
IdleTimeout time.Duration `json:"idleTimeout" yaml:"idleTimeout" toml:"idleTimeout"`
|
||||||
|
Wait bool `json:"wait" yaml:"wait" toml:"wait"`
|
||||||
|
}
|
6
config/redis.go
Normal file
6
config/redis.go
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
type Redis struct {
|
||||||
|
Server
|
||||||
|
Pool Pool `json:"pool" yaml:"pool" toml:"pool"`
|
||||||
|
}
|
12
config/server.go
Normal file
12
config/server.go
Normal file
|
@ -0,0 +1,12 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type Server struct {
|
||||||
|
Name string `json:"name" yaml:"name" toml:"name"`
|
||||||
|
Network string `json:"network" yaml:"network" toml:"network"`
|
||||||
|
Addr string `json:"addr" yaml:"addr" toml:"addr"`
|
||||||
|
TLS bool `json:"tls" yaml:"tls" toml:"tls"`
|
||||||
|
Concurrency int `json:"concurrency" yaml:"concurrency" toml:"concurrency"`
|
||||||
|
MaxStopWaitTime time.Duration `json:"maxStopWaitTime" yaml:"maxStopWaitTime" toml:"maxStopWaitTime"`
|
||||||
|
}
|
13
config/socket.go
Normal file
13
config/socket.go
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type Socket struct {
|
||||||
|
MaxMessageSize int64 `json:"maxMessageSize" yaml:"maxMessageSize" toml:"maxMessageSize"`
|
||||||
|
WriteTimeout time.Duration `json:"writeTimeout" yaml:"writeTimeout" toml:"writeTimeout"`
|
||||||
|
ReadTimeout time.Duration `json:"readTimeout" yaml:"readTimeout" toml:"readTimeout"`
|
||||||
|
PongTimeout time.Duration `json:"pongTimeout" yaml:"pongTimeout" toml:"pongTimeout"`
|
||||||
|
PingTimeout time.Duration `json:"pingTimeout" yaml:"pingTimeout" toml:"pingTimeout"`
|
||||||
|
PingPeriod time.Duration `json:"pingPeriod" yaml:"pingPeriod" toml:"pingPeriod"`
|
||||||
|
BinaryMessage bool `json:"binaryMessage" yaml:"binaryMessage" toml:"binaryMessage"`
|
||||||
|
}
|
10
config/websocket.go
Normal file
10
config/websocket.go
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type Websocket struct {
|
||||||
|
HandshakeTimeout time.Duration `json:"handshakeTimeout" yaml:"handshakeTimeout" toml:"handshakeTimeout"`
|
||||||
|
ReadBufferSize int `json:"readBufferSize" yaml:"readBufferSize" toml:"readBufferSize"`
|
||||||
|
WriteBufferSize int `json:"writeBufferSize" yaml:"writeBufferSize" toml:"writeBufferSize"`
|
||||||
|
EnableCompression bool `json:"enableCompression" yaml:"enableCompression" toml:"enableCompression"`
|
||||||
|
}
|
15
constants.go
Normal file
15
constants.go
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
package overflow_gateway_websocket
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.loafle.net/commons_go/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ServletSocketKey = util.ContextKey("ServletSocket")
|
||||||
|
ServletContentTypeKey = util.ContextKey("ServletContentType")
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
ConfigPathFlagName = "config-dir"
|
||||||
|
ConfigFileName = "config.json"
|
||||||
|
)
|
16
external/external.go
vendored
Normal file
16
external/external.go
vendored
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
package external
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_app/external/grpc"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_app/external/redis"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ExternalInit() {
|
||||||
|
grpc.ExternalInit()
|
||||||
|
redis.ExternalInit()
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExternalDestroy() {
|
||||||
|
grpc.ExternalDestroy()
|
||||||
|
redis.ExternalDestroy()
|
||||||
|
}
|
35
external/grpc/client.go
vendored
Normal file
35
external/grpc/client.go
vendored
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
package grpc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
ooas "git.loafle.net/overflow/overflow_api_server/golang"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Exec(ctx context.Context, method string, params []string) (string, error) {
|
||||||
|
if nil == grpcPool {
|
||||||
|
return "", fmt.Errorf("App: GRPC Pool is not initialized")
|
||||||
|
}
|
||||||
|
|
||||||
|
var client interface{}
|
||||||
|
var err error
|
||||||
|
if client, err = grpcPool.Get(); nil != err {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer grpcPool.Put(client)
|
||||||
|
|
||||||
|
sm := strings.Split(method, ".")
|
||||||
|
si := &ooas.ServerInput{
|
||||||
|
Target: sm[0],
|
||||||
|
Method: sm[1],
|
||||||
|
Params: params,
|
||||||
|
}
|
||||||
|
var so *ooas.ServerOutput
|
||||||
|
if so, err = client.(ooas.OverflowApiServerClient).Exec(ctx, si); nil != err {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return so.Result, nil
|
||||||
|
}
|
49
external/grpc/pool.go
vendored
Normal file
49
external/grpc/pool.go
vendored
Normal file
|
@ -0,0 +1,49 @@
|
||||||
|
package grpc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
cgp "git.loafle.net/commons_go/grpc_pool"
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
ooas "git.loafle.net/overflow/overflow_api_server/golang"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_app/conf"
|
||||||
|
"google.golang.org/grpc"
|
||||||
|
)
|
||||||
|
|
||||||
|
var grpcPool cgp.Pool
|
||||||
|
|
||||||
|
func ExternalInit() {
|
||||||
|
ph := &grpcPoolHandlers{}
|
||||||
|
ph.MaxCapacity = conf.Config.GRPC.Pool.MaxCapacity
|
||||||
|
ph.MaxIdle = conf.Config.GRPC.Pool.MaxIdle
|
||||||
|
ph.IdleTimeout = conf.Config.GRPC.Pool.IdleTimeout
|
||||||
|
ph.Wait = conf.Config.GRPC.Pool.Wait
|
||||||
|
|
||||||
|
grpcPool = cgp.New(ph)
|
||||||
|
|
||||||
|
if err := grpcPool.Start(); nil != err {
|
||||||
|
logging.Logger().Panic(fmt.Sprintf("App: %v", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExternalDestroy() {
|
||||||
|
if nil != grpcPool {
|
||||||
|
grpcPool.Stop()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type grpcPoolHandlers struct {
|
||||||
|
cgp.PoolHandlers
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *grpcPoolHandlers) Dial() (*grpc.ClientConn, interface{}, error) {
|
||||||
|
var err error
|
||||||
|
conn, err := grpc.Dial(conf.Config.GRPC.Addr, grpc.WithInsecure())
|
||||||
|
if nil != err {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
c := ooas.NewOverflowApiServerClient(conn)
|
||||||
|
|
||||||
|
return conn, c, nil
|
||||||
|
}
|
42
external/redis/pool.go
vendored
Normal file
42
external/redis/pool.go
vendored
Normal file
|
@ -0,0 +1,42 @@
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/config"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
crp "git.loafle.net/commons_go/redis_pool"
|
||||||
|
"github.com/garyburd/redigo/redis"
|
||||||
|
)
|
||||||
|
|
||||||
|
var RedisPool crp.Pool
|
||||||
|
|
||||||
|
func ExternalInit() {
|
||||||
|
ph := &redisPoolHandlers{}
|
||||||
|
ph.MaxCapacity = config.Config.Redis.Pool.MaxCapacity
|
||||||
|
ph.MaxIdle = config.Config.Redis.Pool.MaxIdle
|
||||||
|
ph.IdleTimeout = config.Config.Redis.Pool.IdleTimeout
|
||||||
|
ph.Wait = config.Config.Redis.Pool.Wait
|
||||||
|
|
||||||
|
RedisPool = crp.New(ph)
|
||||||
|
|
||||||
|
if err := RedisPool.Start(); nil != err {
|
||||||
|
logging.Logger().Panic(fmt.Sprintf("App: %v", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExternalDestroy() {
|
||||||
|
if nil != RedisPool {
|
||||||
|
RedisPool.Stop()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type redisPoolHandlers struct {
|
||||||
|
crp.PoolHandlers
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ph *redisPoolHandlers) Dial() (redis.Conn, error) {
|
||||||
|
return redis.Dial(config.Config.Redis.Network, config.Config.Redis.Addr)
|
||||||
|
}
|
19
gateway_websocket.go
Normal file
19
gateway_websocket.go
Normal file
|
@ -0,0 +1,19 @@
|
||||||
|
package overflow_gateway_websocket
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
|
||||||
|
cc "git.loafle.net/commons_go/config"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
configPath := flag.String(ConfigPathFlagName, "./", "The path of config file")
|
||||||
|
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
cc.SetConfigPath(*configPath)
|
||||||
|
if err := cc.Load(&config.Config, ConfigFileName); nil != err {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
19
glide.yaml
Normal file
19
glide.yaml
Normal file
|
@ -0,0 +1,19 @@
|
||||||
|
package: git.loafle.net/overflow/overflow_gateway_websocket
|
||||||
|
import:
|
||||||
|
- package: git.loafle.net/commons_go/grpc_pool
|
||||||
|
- package: git.loafle.net/commons_go/logging
|
||||||
|
- package: git.loafle.net/commons_go/redis_pool
|
||||||
|
- package: git.loafle.net/overflow/overflow_api_server
|
||||||
|
subpackages:
|
||||||
|
- golang
|
||||||
|
- package: git.loafle.net/overflow/overflow_gateway_app
|
||||||
|
subpackages:
|
||||||
|
- conf
|
||||||
|
- external/grpc
|
||||||
|
- external/redis
|
||||||
|
- package: github.com/garyburd/redigo
|
||||||
|
subpackages:
|
||||||
|
- redis
|
||||||
|
- package: google.golang.org/grpc
|
||||||
|
- package: github.com/valyala/fasthttp
|
||||||
|
version: v20160617
|
101
internal/server/rpc_gateway_socket_handlers.go
Normal file
101
internal/server/rpc_gateway_socket_handlers.go
Normal file
|
@ -0,0 +1,101 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
"github.com/valyala/fasthttp"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/rpc/gateway"
|
||||||
|
"git.loafle.net/commons_go/server"
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
oogw "git.loafle.net/overflow/overflow_gateway_websocket"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RPCGatewaySocketHandlers struct {
|
||||||
|
cwf.SocketHandlers
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCGatewaySocketHandlers) Init() error {
|
||||||
|
if err := sh.SocketHandlers.Init(); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCGatewaySocketHandlers) Handshake(ctx *fasthttp.RequestCtx) (id string, extensionsHeader *fasthttp.ResponseHeader) {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnConnect invoked when client is connected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *RPCGatewaySocketHandlers) OnConnect(soc *cwf.Socket) {
|
||||||
|
sh.SocketHandlers.OnConnect(soc)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCGatewaySocketHandlers) Handle(soc *cwf.Socket, stopChan <-chan struct{}, doneChan chan<- struct{}) {
|
||||||
|
codec, err := sh.rpcServerHandler.GetCodec(sh.ContentType)
|
||||||
|
if nil != err {
|
||||||
|
log.Printf("RPC Handle: %v", err)
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var socConn *cwf.SocketConn
|
||||||
|
ctx := context.WithValue(context.Background(), oogw.ServletSocketKey, soc)
|
||||||
|
// conn.SetPongHandler(func(string) error { c.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil })
|
||||||
|
|
||||||
|
for {
|
||||||
|
if socConn, err = soc.WaitRequest(); nil != err {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// // "git.loafle.net/commons_go/websocket_fasthttp/websocket"
|
||||||
|
// switch socConn.MessageType {
|
||||||
|
// case websocket.TextMessage:
|
||||||
|
// case websocket.BinaryMessage:
|
||||||
|
// }
|
||||||
|
|
||||||
|
if err = gateway.Handle(ctx, sh.rpcServerHandler, codec, socConn, socConn); nil != err {
|
||||||
|
if server.IsClientDisconnect(err) {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Printf("RPC: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = socConn.Close(); nil != err {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-stopChan:
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnDisconnect invoked when client is disconnected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *RPCGatewaySocketHandlers) OnDisconnect(soc *cwf.Socket) {
|
||||||
|
|
||||||
|
sh.SocketHandlers.OnDisconnect(soc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Destroy invoked when server is stopped
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *RPCGatewaySocketHandlers) Destroy() {
|
||||||
|
|
||||||
|
sh.SocketHandlers.Destroy()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCGatewaySocketHandlers) Validate() {
|
||||||
|
sh.SocketHandlers.Validate()
|
||||||
|
|
||||||
|
}
|
1
internal/server/socket_handlers.go
Normal file
1
internal/server/socket_handlers.go
Normal file
|
@ -0,0 +1 @@
|
||||||
|
package server
|
1
rpc/rpc_gateway_handlers.go
Normal file
1
rpc/rpc_gateway_handlers.go
Normal file
|
@ -0,0 +1 @@
|
||||||
|
package rpc
|
13
server/server.go
Normal file
13
server/server.go
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
|
||||||
|
cc "git.loafle.net/commons_go/config"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
func New() {
|
||||||
|
|
||||||
|
}
|
14
server/server_handler.go
Normal file
14
server/server_handler.go
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/servlet"
|
||||||
|
oos "git.loafle.net/overflow/overflow_subscriber"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServerHandler interface {
|
||||||
|
cwf.ServerHandler
|
||||||
|
|
||||||
|
RegisterServletHandler(servletHandler servlet.ServletHandler)
|
||||||
|
RegisterSubscriberHandler(subscriberHandler oos.SubscriberHandler)
|
||||||
|
}
|
80
server/server_handlers.go
Normal file
80
server/server_handlers.go
Normal file
|
@ -0,0 +1,80 @@
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/config"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/external"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/external/redis"
|
||||||
|
"git.loafle.net/overflow/overflow_gateway_websocket/servlet"
|
||||||
|
oos "git.loafle.net/overflow/overflow_subscriber"
|
||||||
|
oosr "git.loafle.net/overflow/overflow_subscriber/redis"
|
||||||
|
"github.com/valyala/fasthttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServerHandlers struct {
|
||||||
|
cwf.ServerHandlers
|
||||||
|
|
||||||
|
redisSubscriber oos.Subscriber
|
||||||
|
subscribers []oos.SubscriberHandler
|
||||||
|
}
|
||||||
|
|
||||||
|
// Init invoked before the server is started
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *ServerHandlers) Init() error {
|
||||||
|
if err := sh.ServerHandlers.Init(); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
external.ExternalInit()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) Listen() (net.Listener, error) {
|
||||||
|
return net.Listen(config.Config.Server.Network, config.Config.Server.Addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) OnStart() {
|
||||||
|
sh.ServerHandlers.OnStart()
|
||||||
|
|
||||||
|
sh.redisSubscriber = oosr.New(redis.RedisPool.Get())
|
||||||
|
if err := sh.redisSubscriber.Start(); nil != err {
|
||||||
|
logging.Logger().Error(fmt.Sprintf("App: Redist Subscriber did not start %v", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, subscriber := range sh.subscribers {
|
||||||
|
sh.redisSubscriber.Subscribe(subscriber)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) OnError(ctx *fasthttp.RequestCtx, status int, reason error) {
|
||||||
|
sh.OnError(ctx, status, reason)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) OnStop() {
|
||||||
|
sh.redisSubscriber.Stop()
|
||||||
|
|
||||||
|
external.ExternalDestroy()
|
||||||
|
|
||||||
|
sh.ServerHandlers.OnStop()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) RegisterServletHandler(servletHandler servlet.ServletHandler) {
|
||||||
|
sh.RegisterSocketHandler(servletHandler)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) RegisterSubscriberHandler(subscriberHandler oos.SubscriberHandler) {
|
||||||
|
if nil == sh.subscribers {
|
||||||
|
sh.subscribers = make([]oos.SubscriberHandler, 0)
|
||||||
|
}
|
||||||
|
sh.subscribers = append(sh.subscribers, subscriberHandler)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServerHandlers) Validate() {
|
||||||
|
sh.ServerHandlers.Validate()
|
||||||
|
}
|
5
servlet/rpc_servlet_handler.go
Normal file
5
servlet/rpc_servlet_handler.go
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
package servlet
|
||||||
|
|
||||||
|
type RPCServletHandler interface {
|
||||||
|
ServletHandler
|
||||||
|
}
|
98
servlet/rpc_servlet_handlers.go
Normal file
98
servlet/rpc_servlet_handlers.go
Normal file
|
@ -0,0 +1,98 @@
|
||||||
|
package servlet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
"git.loafle.net/commons_go/rpc/gateway"
|
||||||
|
"git.loafle.net/commons_go/server"
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
|
||||||
|
oogw "git.loafle.net/overflow/overflow_gateway_websocket"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RPCServletHandlers struct {
|
||||||
|
ServletHandlers
|
||||||
|
|
||||||
|
RPCHandler
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCServletHandlers) Init() error {
|
||||||
|
if err := sh.ServletHandlers.Init(); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnConnect invoked when client is connected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *RPCServletHandlers) OnConnect(soc *cwf.Socket) {
|
||||||
|
sh.ServletHandlers.OnConnect(soc)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCServletHandlers) Handle(soc *cwf.Socket, stopChan <-chan struct{}, doneChan chan<- struct{}) {
|
||||||
|
codec, err := sh.rpcServerHandler.GetCodec(sh.ContentType)
|
||||||
|
if nil != err {
|
||||||
|
log.Printf("RPC Handle: %v", err)
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var socConn *cwf.SocketConn
|
||||||
|
ctx := context.WithValue(context.Background(), oogw.ServletSocketKey, soc)
|
||||||
|
// conn.SetPongHandler(func(string) error { c.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil })
|
||||||
|
|
||||||
|
for {
|
||||||
|
if socConn, err = soc.WaitRequest(); nil != err {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// // "git.loafle.net/commons_go/websocket_fasthttp/websocket"
|
||||||
|
// switch socConn.MessageType {
|
||||||
|
// case websocket.TextMessage:
|
||||||
|
// case websocket.BinaryMessage:
|
||||||
|
// }
|
||||||
|
|
||||||
|
if err = gateway.Handle(ctx, sh.rpcServerHandler, codec, socConn, socConn); nil != err {
|
||||||
|
if server.IsClientDisconnect(err) {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Printf("RPC: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = socConn.Close(); nil != err {
|
||||||
|
doneChan <- struct{}{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-stopChan:
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnDisconnect invoked when client is disconnected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *RPCServletHandlers) OnDisconnect(soc *cwf.Socket) {
|
||||||
|
|
||||||
|
sh.ServletHandlers.OnDisconnect(soc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Destroy invoked when server is stopped
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *ServletHandlers) Destroy() {
|
||||||
|
|
||||||
|
sh.SocketHandlers.Destroy()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *RPCServletHandlers) Validate() {
|
||||||
|
sh.ServletHandlers.Validate()
|
||||||
|
|
||||||
|
}
|
9
servlet/servlet_handler.go
Normal file
9
servlet/servlet_handler.go
Normal file
|
@ -0,0 +1,9 @@
|
||||||
|
package servlet
|
||||||
|
|
||||||
|
import (
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServletHandler interface {
|
||||||
|
cwf.SocketHandler
|
||||||
|
}
|
50
servlet/servlet_handlers.go
Normal file
50
servlet/servlet_handlers.go
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
package servlet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.loafle.net/commons_go/logging"
|
||||||
|
cwf "git.loafle.net/commons_go/websocket_fasthttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServletHandlers struct {
|
||||||
|
cwf.SocketHandlers
|
||||||
|
|
||||||
|
// EntryPath is path of url (ex: /web)
|
||||||
|
EntryPath string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServletHandlers) Init() error {
|
||||||
|
if err := sh.SocketHandlers.Init(); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnConnect invoked when client is connected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *ServletHandlers) OnConnect(soc *cwf.Socket) {
|
||||||
|
sh.SocketHandlers.OnConnect(soc)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnDisconnect invoked when client is disconnected
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *ServletHandlers) OnDisconnect(soc *cwf.Socket) {
|
||||||
|
|
||||||
|
sh.SocketHandlers.OnDisconnect(soc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Destroy invoked when server is stopped
|
||||||
|
// If you override ths method, must call
|
||||||
|
func (sh *ServletHandlers) Destroy() {
|
||||||
|
|
||||||
|
sh.SocketHandlers.Destroy()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *ServletHandlers) Validate() {
|
||||||
|
sh.SocketHandlers.Validate()
|
||||||
|
|
||||||
|
if "" == sh.EntryPath {
|
||||||
|
logging.Logger().Panic("Geteway Server: The path of entry must be specified")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user