Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5e1e4b9daf | ||
|
|
0f06128ce8 | ||
|
|
be07dc8749 | ||
|
|
e75d7c8b03 | ||
|
|
59c7705ca1 |
22
args/args.go
22
args/args.go
@@ -2,7 +2,6 @@ package args
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"flag"
|
"flag"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/tecamino/tecamino-dbm/cert"
|
"github.com/tecamino/tecamino-dbm/cert"
|
||||||
"github.com/tecamino/tecamino-dbm/models"
|
"github.com/tecamino/tecamino-dbm/models"
|
||||||
@@ -19,21 +18,10 @@ type Args struct {
|
|||||||
Debug bool
|
Debug bool
|
||||||
}
|
}
|
||||||
|
|
||||||
type StringSlice []string
|
|
||||||
|
|
||||||
func (s *StringSlice) String() string {
|
|
||||||
return strings.Join(*s, ",")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *StringSlice) Set(value string) error {
|
|
||||||
*s = append(*s, value)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// initialte cli arguments
|
// initialte cli arguments
|
||||||
func Init() *Args {
|
func Init() *Args {
|
||||||
|
|
||||||
var allowOrigins StringSlice
|
var allowOrigins models.StringSlice
|
||||||
|
|
||||||
flag.Var(&allowOrigins, "allowOrigin", "Allowed origin (can repeat this flag)")
|
flag.Var(&allowOrigins, "allowOrigin", "Allowed origin (can repeat this flag)")
|
||||||
|
|
||||||
@@ -43,6 +31,7 @@ func Init() *Args {
|
|||||||
keyFile := flag.String("keyFile", "./cert/key.pem", "path of keyfile")
|
keyFile := flag.String("keyFile", "./cert/key.pem", "path of keyfile")
|
||||||
portHttp := flag.Uint("http-port", 8100, "json server communication for http/ws")
|
portHttp := flag.Uint("http-port", 8100, "json server communication for http/ws")
|
||||||
portHttps := flag.Uint("https-port", 8101, "json server communication for http/wss")
|
portHttps := flag.Uint("https-port", 8101, "json server communication for http/wss")
|
||||||
|
remotePort := flag.Uint("remotePort", 9500, "remote Port of gui user interface")
|
||||||
rootDir := flag.String("workingDir", "./", "working directory")
|
rootDir := flag.String("workingDir", "./", "working directory")
|
||||||
dbmFile := flag.String("dbm", "/test/test", "dbm file name")
|
dbmFile := flag.String("dbm", "/test/test", "dbm file name")
|
||||||
debug := flag.Bool("debug", false, "debug flag")
|
debug := flag.Bool("debug", false, "debug flag")
|
||||||
@@ -56,8 +45,9 @@ func Init() *Args {
|
|||||||
KeyFile: *keyFile,
|
KeyFile: *keyFile,
|
||||||
},
|
},
|
||||||
Port: models.Port{
|
Port: models.Port{
|
||||||
Http: *portHttp,
|
Http: *portHttp,
|
||||||
Https: *portHttps,
|
Https: *portHttps,
|
||||||
|
Remote: *remotePort,
|
||||||
},
|
},
|
||||||
RootDir: *rootDir,
|
RootDir: *rootDir,
|
||||||
DBMFile: *dbmFile,
|
DBMFile: *dbmFile,
|
||||||
@@ -65,7 +55,7 @@ func Init() *Args {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(allowOrigins) == 0 {
|
if len(allowOrigins) == 0 {
|
||||||
allowOrigins = StringSlice{"http://localhost:9500"}
|
allowOrigins = models.StringSlice{"http://localhost:9500"}
|
||||||
}
|
}
|
||||||
|
|
||||||
a.AllowOrigins = allowOrigins
|
a.AllowOrigins = allowOrigins
|
||||||
|
|||||||
@@ -35,6 +35,7 @@ func (d *DBMHandler) Json_Data(c *gin.Context) {
|
|||||||
Path: res.Path,
|
Path: res.Path,
|
||||||
Type: res.Type,
|
Type: res.Type,
|
||||||
Value: res.Value,
|
Value: res.Value,
|
||||||
|
Drivers: &res.Drivers,
|
||||||
HasChild: res.Datapoints != nil,
|
HasChild: res.Datapoints != nil,
|
||||||
Rights: res.ReadWrite,
|
Rights: res.ReadWrite,
|
||||||
})
|
})
|
||||||
|
|||||||
2
go.mod
2
go.mod
@@ -7,7 +7,7 @@ require (
|
|||||||
github.com/gin-gonic/gin v1.10.0
|
github.com/gin-gonic/gin v1.10.0
|
||||||
github.com/google/uuid v1.6.0
|
github.com/google/uuid v1.6.0
|
||||||
github.com/gorilla/websocket v1.5.3
|
github.com/gorilla/websocket v1.5.3
|
||||||
github.com/tecamino/tecamino-json_data v0.0.16
|
github.com/tecamino/tecamino-json_data v0.0.17
|
||||||
github.com/tecamino/tecamino-logger v0.2.0
|
github.com/tecamino/tecamino-logger v0.2.0
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
4
go.sum
4
go.sum
@@ -71,8 +71,8 @@ github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o
|
|||||||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
||||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/tecamino/tecamino-json_data v0.0.16 h1:aZFxnhm4g6WMDPoqy4HosUk7vl0DB0iIcVs8bbT4MzU=
|
github.com/tecamino/tecamino-json_data v0.0.17 h1:M12UzKbfIgu/q3ERsEhGNDV3DYOvc0TioU288kQjDCA=
|
||||||
github.com/tecamino/tecamino-json_data v0.0.16/go.mod h1:LLlyD7Wwqplb2BP4PeO86EokEcTRidlW5MwgPd1T2JY=
|
github.com/tecamino/tecamino-json_data v0.0.17/go.mod h1:LLlyD7Wwqplb2BP4PeO86EokEcTRidlW5MwgPd1T2JY=
|
||||||
github.com/tecamino/tecamino-logger v0.2.0 h1:NPH/Gg9qRhmVoW8b39i1eXu/LEftHc74nyISpcRG+XU=
|
github.com/tecamino/tecamino-logger v0.2.0 h1:NPH/Gg9qRhmVoW8b39i1eXu/LEftHc74nyISpcRG+XU=
|
||||||
github.com/tecamino/tecamino-logger v0.2.0/go.mod h1:0M1E9Uei/qw3e3WA1x3lBo1eP3H5oeYE7GjYrMahnj8=
|
github.com/tecamino/tecamino-logger v0.2.0/go.mod h1:0M1E9Uei/qw3e3WA1x3lBo1eP3H5oeYE7GjYrMahnj8=
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||||
|
|||||||
2
main.go
2
main.go
@@ -23,7 +23,7 @@ func main() {
|
|||||||
|
|
||||||
//initialize new server
|
//initialize new server
|
||||||
dbmHandler.Log.Debug("main", "initialize new server instance")
|
dbmHandler.Log.Debug("main", "initialize new server instance")
|
||||||
s := ws.NewServer(a.AllowOrigins)
|
s := ws.NewServer(a.AllowOrigins, a.Port.Remote)
|
||||||
|
|
||||||
//set routes
|
//set routes
|
||||||
dbmHandler.Log.Debug("main", "setting routes")
|
dbmHandler.Log.Debug("main", "setting routes")
|
||||||
|
|||||||
@@ -385,6 +385,7 @@ func (d *Datapoint) Publish(eventType string) error {
|
|||||||
Event: eventType,
|
Event: eventType,
|
||||||
Uuid: d.Uuid,
|
Uuid: d.Uuid,
|
||||||
Path: d.Path,
|
Path: d.Path,
|
||||||
|
Type: d.Type,
|
||||||
Value: d.Value,
|
Value: d.Value,
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
type Port struct {
|
type Port struct {
|
||||||
Http uint
|
Http uint
|
||||||
Https uint
|
Https uint
|
||||||
|
Remote uint
|
||||||
}
|
}
|
||||||
|
|||||||
14
models/stringSlices.go
Normal file
14
models/stringSlices.go
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
type StringSlice []string
|
||||||
|
|
||||||
|
func (s *StringSlice) String() string {
|
||||||
|
return strings.Join(*s, ",")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StringSlice) Set(value string) error {
|
||||||
|
*s = append(*s, value)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -142,7 +142,7 @@ func TestUpdateDps(t *testing.T) {
|
|||||||
|
|
||||||
func TestServer(t *testing.T) {
|
func TestServer(t *testing.T) {
|
||||||
fmt.Println("start")
|
fmt.Println("start")
|
||||||
server := ws.NewServer([]string{".*"})
|
server := ws.NewServer([]string{".*"}, 9500)
|
||||||
|
|
||||||
t.Fatal(server.ServeHttp("http://localhost", 8100))
|
t.Fatal(server.ServeHttp("http://localhost", 8100))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,10 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
)
|
||||||
|
|
||||||
func ListofA2ZZ() (list []string) {
|
func ListofA2ZZ() (list []string) {
|
||||||
for i := 'A'; i <= 'Z'; i++ {
|
for i := 'A'; i <= 'Z'; i++ {
|
||||||
list = append(list, string(i))
|
list = append(list, string(i))
|
||||||
@@ -9,12 +14,21 @@ func ListofA2ZZ() (list []string) {
|
|||||||
list = append(list, string(i)+string(j))
|
list = append(list, string(i)+string(j))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// for i := 'A'; i <= 'Z'; i++ {
|
|
||||||
// for j := 'A'; j <= 'Z'; j++ {
|
|
||||||
// for k := 'A'; k <= 'Z'; k++ {
|
|
||||||
// list = append(list, string(i)+string(j)+string(k))
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetLocalIP() (string, error) {
|
||||||
|
addrs, err := net.InterfaceAddrs()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, addr := range addrs {
|
||||||
|
if ipNet, ok := addr.(*net.IPNet); ok && !ipNet.IP.IsLoopback() {
|
||||||
|
if ipNet.IP.To4() != nil {
|
||||||
|
return ipNet.IP.String(), nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("no local IP address found")
|
||||||
|
}
|
||||||
|
|||||||
@@ -27,13 +27,14 @@ const (
|
|||||||
type Client struct {
|
type Client struct {
|
||||||
Id string
|
Id string
|
||||||
Connected bool `json:"connected"`
|
Connected bool `json:"connected"`
|
||||||
Conn *websocket.Conn `json:"-"`
|
conn *websocket.Conn `json:"-"`
|
||||||
OnOpen func()
|
OnOpen func()
|
||||||
OnMessage func(data []byte)
|
OnMessage func(data []byte)
|
||||||
OnClose func(code int, reason string)
|
OnClose func(code int, reason string)
|
||||||
OnError func(err error)
|
OnError func(err error)
|
||||||
OnPing func()
|
OnPing func()
|
||||||
OnPong func()
|
OnPong func()
|
||||||
|
sendPong chan string
|
||||||
send chan []byte
|
send chan []byte
|
||||||
unregister chan []byte
|
unregister chan []byte
|
||||||
}
|
}
|
||||||
@@ -66,7 +67,8 @@ func ConnectNewClient(id string, c *gin.Context) (*Client, error) {
|
|||||||
client := &Client{
|
client := &Client{
|
||||||
Id: id,
|
Id: id,
|
||||||
Connected: true,
|
Connected: true,
|
||||||
Conn: conn,
|
conn: conn,
|
||||||
|
sendPong: make(chan string),
|
||||||
send: make(chan []byte),
|
send: make(chan []byte),
|
||||||
unregister: make(chan []byte),
|
unregister: make(chan []byte),
|
||||||
}
|
}
|
||||||
@@ -77,7 +79,8 @@ func ConnectNewClient(id string, c *gin.Context) (*Client, error) {
|
|||||||
}
|
}
|
||||||
conn.SetWriteDeadline(time.Now().Add(writeWait))
|
conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||||
conn.SetReadDeadline(time.Now().Add(writeWait))
|
conn.SetReadDeadline(time.Now().Add(writeWait))
|
||||||
return conn.WriteMessage(websocket.PongMessage, []byte(appData))
|
client.sendPong <- appData
|
||||||
|
return nil
|
||||||
})
|
})
|
||||||
|
|
||||||
conn.SetPongHandler(func(string) error {
|
conn.SetPongHandler(func(string) error {
|
||||||
@@ -99,9 +102,9 @@ func (c *Client) Read() {
|
|||||||
c.OnOpen()
|
c.OnOpen()
|
||||||
}
|
}
|
||||||
|
|
||||||
c.Conn.SetReadDeadline(time.Now().Add(writeWait))
|
c.conn.SetReadDeadline(time.Now().Add(writeWait))
|
||||||
for c.Connected {
|
for c.Connected {
|
||||||
msgType, msg, err := c.Conn.ReadMessage()
|
msgType, msg, err := c.conn.ReadMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.handleError(fmt.Errorf("read error (id:%s): %w", c.Id, err))
|
c.handleError(fmt.Errorf("read error (id:%s): %w", c.Id, err))
|
||||||
return
|
return
|
||||||
@@ -128,30 +131,30 @@ func (c *Client) Write() {
|
|||||||
ticker := time.NewTicker(pingPeriod)
|
ticker := time.NewTicker(pingPeriod)
|
||||||
defer func() {
|
defer func() {
|
||||||
ticker.Stop()
|
ticker.Stop()
|
||||||
c.Conn.Close()
|
c.conn.Close()
|
||||||
}()
|
}()
|
||||||
for {
|
for {
|
||||||
|
|
||||||
select {
|
select {
|
||||||
case message, ok := <-c.send:
|
case message, ok := <-c.send:
|
||||||
c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
|
c.conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||||
if !ok {
|
if !ok {
|
||||||
// The hub closed the channel.
|
// The hub closed the channel.
|
||||||
if err := c.Conn.WriteMessage(websocket.CloseMessage, []byte{}); err != nil {
|
if err := c.conn.WriteMessage(websocket.CloseMessage, []byte{}); err != nil {
|
||||||
c.handleError(err)
|
c.handleError(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
c.handleError(fmt.Errorf("server %s closed channel", c.Id))
|
c.handleError(fmt.Errorf("server %s closed channel", c.Id))
|
||||||
return
|
return
|
||||||
} else {
|
} else {
|
||||||
if err := c.Conn.WriteMessage(websocket.TextMessage, message); err != nil {
|
if err := c.conn.WriteMessage(websocket.TextMessage, message); err != nil {
|
||||||
c.handleError(err)
|
c.handleError(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
case <-ticker.C:
|
case <-ticker.C:
|
||||||
c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
|
c.conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||||
if err := c.Conn.WriteMessage(websocket.PingMessage, nil); err != nil {
|
if err := c.conn.WriteMessage(websocket.PingMessage, nil); err != nil {
|
||||||
c.handleError(err)
|
c.handleError(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -159,9 +162,15 @@ func (c *Client) Write() {
|
|||||||
if c.OnPing != nil {
|
if c.OnPing != nil {
|
||||||
c.OnPing()
|
c.OnPing()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
case message, ok := <-c.sendPong:
|
||||||
|
if ok {
|
||||||
|
c.conn.WriteMessage(websocket.PongMessage, []byte(message))
|
||||||
|
}
|
||||||
case message := <-c.unregister:
|
case message := <-c.unregister:
|
||||||
c.Conn.WriteMessage(websocket.CloseMessage, message)
|
c.conn.WriteMessage(websocket.CloseMessage, message)
|
||||||
c.Connected = false
|
c.Connected = false
|
||||||
|
close(c.sendPong)
|
||||||
close(c.send)
|
close(c.send)
|
||||||
close(c.unregister)
|
close(c.unregister)
|
||||||
return
|
return
|
||||||
@@ -175,7 +184,7 @@ func (c *Client) handleJsonPing(msg []byte) (isPing bool) {
|
|||||||
err := json.Unmarshal(msg, &wsMsg)
|
err := json.Unmarshal(msg, &wsMsg)
|
||||||
|
|
||||||
if err == nil && wsMsg.IsPing() {
|
if err == nil && wsMsg.IsPing() {
|
||||||
c.Conn.SetReadDeadline(time.Now().Add(writeWait))
|
c.conn.SetReadDeadline(time.Now().Add(writeWait))
|
||||||
// Respond with pong JSON
|
// Respond with pong JSON
|
||||||
select {
|
select {
|
||||||
case c.send <- GetPongByteSlice():
|
case c.send <- GetPongByteSlice():
|
||||||
|
|||||||
@@ -2,12 +2,14 @@ package websocket
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/gin-contrib/cors"
|
"github.com/gin-contrib/cors"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/tecamino/tecamino-dbm/cert"
|
"github.com/tecamino/tecamino-dbm/cert"
|
||||||
|
"github.com/tecamino/tecamino-dbm/utils"
|
||||||
"github.com/tecamino/tecamino-logger/logging"
|
"github.com/tecamino/tecamino-logger/logging"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -19,8 +21,17 @@ type Server struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// initalizes new dbm server
|
// initalizes new dbm server
|
||||||
func NewServer(allowOrigins []string) *Server {
|
func NewServer(allowOrigins []string, port uint) *Server {
|
||||||
r := gin.Default()
|
r := gin.Default()
|
||||||
|
|
||||||
|
allowOrigins = append(allowOrigins, fmt.Sprintf("http://localhost:%d", port))
|
||||||
|
|
||||||
|
localIP, err := utils.GetLocalIP()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("get local ip : %s", err.Error())
|
||||||
|
} else {
|
||||||
|
allowOrigins = append(allowOrigins, fmt.Sprintf("http://%s:%d", localIP, port))
|
||||||
|
}
|
||||||
r.Use(cors.New(cors.Config{
|
r.Use(cors.New(cors.Config{
|
||||||
AllowOrigins: allowOrigins,
|
AllowOrigins: allowOrigins,
|
||||||
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
||||||
|
|||||||
Reference in New Issue
Block a user