ing
This commit is contained in:
parent
1430aa098c
commit
3f11e81d80
54
const.go
54
const.go
|
@ -1,9 +1,6 @@
|
||||||
package gateway
|
package gateway
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
|
|
||||||
cuc "git.loafle.net/commons/util-go/context"
|
cuc "git.loafle.net/commons/util-go/context"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -12,55 +9,4 @@ const (
|
||||||
SessionIDKey = cuc.ContextKey("SessionID")
|
SessionIDKey = cuc.ContextKey("SessionID")
|
||||||
SessionTargetIDKey = cuc.ContextKey("SessionTargetID")
|
SessionTargetIDKey = cuc.ContextKey("SessionTargetID")
|
||||||
SessionWriteChanKey = cuc.ContextKey("SessionWriteChan")
|
SessionWriteChanKey = cuc.ContextKey("SessionWriteChan")
|
||||||
|
|
||||||
GRPCClientTypeKey = cuc.ContextKey("OVERFLOW_GRPC_CLIENT_TYPE")
|
|
||||||
GRPCSessionIDKey = cuc.ContextKey("OVERFLOW_GRPC_SESSION_ID")
|
|
||||||
GRPCTargetIDKey = cuc.ContextKey("OVERFLOW_GRPC_TARGET_ID")
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type ClientType int
|
|
||||||
|
|
||||||
const (
|
|
||||||
MEMBER ClientType = iota
|
|
||||||
PROBE
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
clientTypeID = map[ClientType]string{
|
|
||||||
MEMBER: "MEMBER",
|
|
||||||
PROBE: "PROBE",
|
|
||||||
}
|
|
||||||
|
|
||||||
clientTypeName = map[string]ClientType{
|
|
||||||
"MEMBER": MEMBER,
|
|
||||||
"PROBE": PROBE,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
func (ct ClientType) String() string {
|
|
||||||
return clientTypeID[ct]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ct ClientType) MarshalJSON() ([]byte, error) {
|
|
||||||
value, ok := clientTypeID[ct]
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("Invalid EnumType[%s] value", ct)
|
|
||||||
}
|
|
||||||
return json.Marshal(value)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ct ClientType) UnmarshalJSON(b []byte) error {
|
|
||||||
// unmarshal as string
|
|
||||||
var s string
|
|
||||||
err := json.Unmarshal(b, &s)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
value, ok := clientTypeName[s]
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Invalid EnumType[%s] value", s)
|
|
||||||
}
|
|
||||||
ct = value
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
44
external/external.go
vendored
44
external/external.go
vendored
|
@ -1,44 +0,0 @@
|
||||||
package external
|
|
||||||
|
|
||||||
import (
|
|
||||||
occe "git.loafle.net/overflow/commons-go/config/external"
|
|
||||||
"git.loafle.net/overflow/gateway/external/grpc"
|
|
||||||
"git.loafle.net/overflow/gateway/external/kafka"
|
|
||||||
"git.loafle.net/overflow/gateway/external/redis"
|
|
||||||
)
|
|
||||||
|
|
||||||
func InitPackage(config *occe.External) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
grpc.InitPackage(config.GRPC)
|
|
||||||
redis.InitPackage(config.Redis)
|
|
||||||
kafka.InitPackage(config.Kafka)
|
|
||||||
}
|
|
||||||
|
|
||||||
func StartPackage(config *occe.External) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
grpc.StartPackage(config.GRPC)
|
|
||||||
redis.StartPackage(config.Redis)
|
|
||||||
kafka.StartPackage(config.Kafka)
|
|
||||||
}
|
|
||||||
|
|
||||||
func StopPackage(config *occe.External) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
grpc.StopPackage(config.GRPC)
|
|
||||||
redis.StopPackage(config.Redis)
|
|
||||||
kafka.StopPackage(config.Kafka)
|
|
||||||
}
|
|
||||||
|
|
||||||
func DestroyPackage(config *occe.External) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
grpc.DestroyPackage(config.GRPC)
|
|
||||||
redis.DestroyPackage(config.Redis)
|
|
||||||
kafka.DestroyPackage(config.Kafka)
|
|
||||||
}
|
|
74
external/grpc/grpc.go
vendored
74
external/grpc/grpc.go
vendored
|
@ -1,74 +0,0 @@
|
||||||
package grpc
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
"git.loafle.net/commons/logging-go"
|
|
||||||
oci "git.loafle.net/overflow/central-api/golang"
|
|
||||||
occe "git.loafle.net/overflow/commons-go/config/external"
|
|
||||||
"google.golang.org/grpc"
|
|
||||||
)
|
|
||||||
|
|
||||||
var grpcClient oci.CentralAPIClient
|
|
||||||
|
|
||||||
func InitPackage(config *occe.GRPC) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
conn, err := grpc.Dial(config.Address, grpc.WithInsecure())
|
|
||||||
if nil != err {
|
|
||||||
logging.Logger().Panic(err)
|
|
||||||
}
|
|
||||||
grpcClient = oci.NewCentralAPIClient(conn)
|
|
||||||
logging.Logger().Infof("GRPC: connected to %s", config.Address)
|
|
||||||
}
|
|
||||||
|
|
||||||
func StartPackage(config *occe.GRPC) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func StopPackage(config *occe.GRPC) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func DestroyPackage(config *occe.GRPC) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
var execMtx sync.RWMutex
|
|
||||||
|
|
||||||
func Exec(ctx context.Context, method string, params ...string) (string, error) {
|
|
||||||
if nil == grpcClient {
|
|
||||||
return "", fmt.Errorf("GRPC Client is not initialized")
|
|
||||||
}
|
|
||||||
|
|
||||||
var err error
|
|
||||||
|
|
||||||
sm := strings.Split(method, ".")
|
|
||||||
si := &oci.ServerInput{
|
|
||||||
Target: sm[0],
|
|
||||||
Method: sm[1],
|
|
||||||
Params: params,
|
|
||||||
}
|
|
||||||
|
|
||||||
execMtx.Lock()
|
|
||||||
defer execMtx.Unlock()
|
|
||||||
so, err := grpcClient.(oci.CentralAPIClient).Exec(ctx, si)
|
|
||||||
if nil != err {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
return so.Result, nil
|
|
||||||
}
|
|
89
external/kafka/kafka.go
vendored
89
external/kafka/kafka.go
vendored
|
@ -1,89 +0,0 @@
|
||||||
package kafka
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"git.loafle.net/commons/logging-go"
|
|
||||||
occe "git.loafle.net/overflow/commons-go/config/external"
|
|
||||||
"github.com/segmentio/kafka-go"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
kafkaWriters map[string]*kafka.Writer
|
|
||||||
)
|
|
||||||
|
|
||||||
func InitPackage(config *occe.Kafka) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if nil == config.Producers || 0 == len(config.Producers) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
kafkaWriters = make(map[string]*kafka.Writer)
|
|
||||||
|
|
||||||
for n, c := range config.Producers {
|
|
||||||
wc := kafka.WriterConfig{
|
|
||||||
Brokers: c.Brokers,
|
|
||||||
Topic: c.Topic,
|
|
||||||
Balancer: &kafka.LeastBytes{},
|
|
||||||
MaxAttempts: c.MaxAttempts,
|
|
||||||
QueueCapacity: c.QueueCapacity,
|
|
||||||
BatchSize: c.BatchSize,
|
|
||||||
BatchTimeout: c.BatchTimeout,
|
|
||||||
ReadTimeout: c.ReadTimeout,
|
|
||||||
WriteTimeout: c.WriteTimeout,
|
|
||||||
RebalanceInterval: c.RebalanceInterval,
|
|
||||||
RequiredAcks: c.RequiredAcks,
|
|
||||||
Async: c.Async,
|
|
||||||
}
|
|
||||||
kafkaWriters[n] = kafka.NewWriter(wc)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func StartPackage(config *occe.Kafka) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func StopPackage(config *occe.Kafka) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func DestroyPackage(config *occe.Kafka) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, w := range kafkaWriters {
|
|
||||||
if err := w.Close(); nil != err {
|
|
||||||
logging.Logger().Errorf("%v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func Write(name string, key []byte, value []byte) error {
|
|
||||||
if nil == kafkaWriters {
|
|
||||||
return fmt.Errorf("Kafka client is not valid")
|
|
||||||
}
|
|
||||||
|
|
||||||
w, ok := kafkaWriters[name]
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Kafka client is not valid")
|
|
||||||
}
|
|
||||||
|
|
||||||
err := w.WriteMessages(context.Background(),
|
|
||||||
kafka.Message{
|
|
||||||
Key: key,
|
|
||||||
Value: value,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
48
external/redis/redis.go
vendored
48
external/redis/redis.go
vendored
|
@ -1,48 +0,0 @@
|
||||||
package redis
|
|
||||||
|
|
||||||
import (
|
|
||||||
"git.loafle.net/commons/logging-go"
|
|
||||||
occe "git.loafle.net/overflow/commons-go/config/external"
|
|
||||||
"github.com/gomodule/redigo/redis"
|
|
||||||
)
|
|
||||||
|
|
||||||
var Pool *redis.Pool
|
|
||||||
|
|
||||||
func InitPackage(config *occe.Redis) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
Pool = &redis.Pool{
|
|
||||||
MaxIdle: 1,
|
|
||||||
MaxActive: 3,
|
|
||||||
IdleTimeout: 240,
|
|
||||||
Wait: true,
|
|
||||||
MaxConnLifetime: 1,
|
|
||||||
Dial: func() (redis.Conn, error) {
|
|
||||||
return redis.Dial(config.Network, config.Address)
|
|
||||||
},
|
|
||||||
}
|
|
||||||
logging.Logger().Infof("Redis: initialized [%s:%s]", config.Network, config.Address)
|
|
||||||
}
|
|
||||||
|
|
||||||
func StartPackage(config *occe.Redis) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func StopPackage(config *occe.Redis) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func DestroyPackage(config *occe.Redis) {
|
|
||||||
if nil == config {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
Pool.Close()
|
|
||||||
}
|
|
10
glide.yaml
10
glide.yaml
|
@ -1,10 +1,10 @@
|
||||||
package: git.loafle.net/overflow/gateway
|
package: git.loafle.net/overflow/gateway
|
||||||
import:
|
import:
|
||||||
- package: git.loafle.net/commons/logging-go
|
- package: git.loafle.net/commons/logging-go
|
||||||
- package: google.golang.org/grpc
|
- package: git.loafle.net/commons/util-go
|
||||||
version: ^1.11.2
|
subpackages:
|
||||||
|
- context
|
||||||
- package: github.com/gomodule/redigo
|
- package: github.com/gomodule/redigo
|
||||||
version: ^2.0.0
|
version: ^2.0.0
|
||||||
- package: github.com/segmentio/kafka-go
|
subpackages:
|
||||||
- package: git.loafle.net/commons/util-go
|
- redis
|
||||||
- package: git.loafle.net/overflow/central-api
|
|
||||||
|
|
|
@ -5,7 +5,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
logging "git.loafle.net/commons/logging-go"
|
"git.loafle.net/commons/logging-go"
|
||||||
"git.loafle.net/overflow/gateway/subscribe"
|
"git.loafle.net/overflow/gateway/subscribe"
|
||||||
"github.com/gomodule/redigo/redis"
|
"github.com/gomodule/redigo/redis"
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user