Initial commit #1
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
.vscode
|
||||
input.json
|
||||
4
.vscode/settings.json
vendored
Normal file
4
.vscode/settings.json
vendored
Normal file
@@ -0,0 +1,4 @@
|
||||
{
|
||||
"go.gopath": "/home/nedim/go",
|
||||
"go.useLanguageServer": true
|
||||
}
|
||||
20
README.md
20
README.md
@@ -1,2 +1,22 @@
|
||||
# xmpploadtesting
|
||||
A tool for testing xmpp servers load
|
||||
|
||||
## Environment variables
|
||||
|
||||
| Variable | Required | Default Value | Description |
|
||||
| ------------------------------------------------- | -------- | --------------------------------- | ---------------------------------------------- |
|
||||
| **Service:** |
|
||||
| OPENFIRE_PORT | NO | 5222 | Port for openfire server |
|
||||
| OPENFIRE_ADRESS | YES | | Openfire server address |
|
||||
| OPENFIRE_DOMAIN | YES | | Domain |
|
||||
| |
|
||||
| **Credentials:** |
|
||||
| |
|
||||
| CREDENTIALS_FILE_LOCATION | NO | input.json | Database username |
|
||||
| |
|
||||
| **GeneralOptions:** |
|
||||
| |
|
||||
| MASSAGE_DELAY | NO | 120000000000 - 2 mins | Delay for online presence message loop |
|
||||
| STATUS_MASSAGE_DELAY | NO | 10 sec | Delay for online presence message loop |
|
||||
| |
|
||||
|
||||
|
||||
24
config/config.go
Normal file
24
config/config.go
Normal file
@@ -0,0 +1,24 @@
|
||||
package config
|
||||
|
||||
// AppConfig contains application configuration
|
||||
var AppConfig Config
|
||||
|
||||
// Load application configuration
|
||||
func Load() {
|
||||
|
||||
AppConfig = Config{
|
||||
|
||||
Service: Service{
|
||||
Port: getEnv("OPENFIRE_PORT", "5222"),
|
||||
Address: mustGetEnv("OPENFIRE_ADRESS"),
|
||||
Domain: mustGetEnv("OPENFIRE_DOMAIN"),
|
||||
},
|
||||
Credentials: Credentials{
|
||||
CredentialsFileLocation: getEnv("CREDENTIALS_FILE_LOCATION", "input.json"),
|
||||
},
|
||||
GeneralOptions: GeneralOptions{
|
||||
DelayBetweenMassages: int64(getEnvInt("MASSAGE_DELAY", 120000000000)),
|
||||
StatusMessageDelay: getEnvInt("STATUS_MASSAGE_DELAY", 10),
|
||||
},
|
||||
}
|
||||
}
|
||||
56
config/helpers.go
Normal file
56
config/helpers.go
Normal file
@@ -0,0 +1,56 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
// getEnv returns value for give key from environment
|
||||
// if key is not present in environment it returns defaultValue
|
||||
func getEnv(key, defaultValue string) string {
|
||||
v := os.Getenv(key)
|
||||
if len(v) > 0 {
|
||||
return v
|
||||
}
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
// getEnvInt returns integer value for give key from environment
|
||||
// if key is not present in environment it returns defaultValue
|
||||
// if key cannot be parsed to integer function will panic
|
||||
func getEnvInt(key string, defaultValue int) int {
|
||||
v := os.Getenv(key)
|
||||
if len(v) == 0 {
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
valInteger, err := strconv.Atoi(v)
|
||||
if err != nil {
|
||||
log.Fatalf("variable `%s` cannot be parsed to INTEGER", key)
|
||||
}
|
||||
|
||||
return valInteger
|
||||
}
|
||||
|
||||
// getEnv extracts the bool value from environment variable
|
||||
func getEnvBool(env string) bool {
|
||||
envVal := mustGetEnv(env)
|
||||
value, err := strconv.ParseBool(envVal)
|
||||
|
||||
if err != nil {
|
||||
value = false
|
||||
}
|
||||
|
||||
return value
|
||||
}
|
||||
|
||||
// mustGetEnv returns value for give key from environment
|
||||
// if key is not present in environment function will panic
|
||||
func mustGetEnv(key string) string {
|
||||
v := os.Getenv(key)
|
||||
if len(v) == 0 {
|
||||
log.Fatalf(" variable `%s` is not present in ENVIRONMENT", key)
|
||||
}
|
||||
return v
|
||||
}
|
||||
26
config/models.go
Normal file
26
config/models.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package config
|
||||
|
||||
// Config stores application configuration
|
||||
type Config struct {
|
||||
Service Service
|
||||
Credentials Credentials
|
||||
GeneralOptions GeneralOptions
|
||||
}
|
||||
|
||||
// Service contains configuration for service
|
||||
type Service struct {
|
||||
Port string
|
||||
Address string
|
||||
Domain string
|
||||
}
|
||||
|
||||
// Credentials contains information about client credentials
|
||||
type Credentials struct {
|
||||
CredentialsFileLocation string
|
||||
}
|
||||
|
||||
// GeneralOptions contains information for general configuration options
|
||||
type GeneralOptions struct {
|
||||
DelayBetweenMassages int64
|
||||
StatusMessageDelay int
|
||||
}
|
||||
10
go.mod
Normal file
10
go.mod
Normal file
@@ -0,0 +1,10 @@
|
||||
module github.com/xmpploadtesting
|
||||
|
||||
go 1.17
|
||||
|
||||
require (
|
||||
github.com/google/uuid v1.1.1 // indirect
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7 // indirect
|
||||
gosrc.io/xmpp v0.5.1 // indirect
|
||||
nhooyr.io/websocket v1.6.5 // indirect
|
||||
)
|
||||
103
go.sum
Normal file
103
go.sum
Normal file
@@ -0,0 +1,103 @@
|
||||
github.com/agnivade/wasmbrowsertest v0.3.1/go.mod h1:zQt6ZTdl338xxRaMW395qccVE2eQm0SjC/SDz0mPWQI=
|
||||
github.com/chromedp/cdproto v0.0.0-20190614062957-d6d2f92b486d/go.mod h1:S8mB5wY3vV+vRIzf39xDXsw3XKYewW9X6rW2aEmkrSw=
|
||||
github.com/chromedp/cdproto v0.0.0-20190621002710-8cbd498dd7a0/go.mod h1:S8mB5wY3vV+vRIzf39xDXsw3XKYewW9X6rW2aEmkrSw=
|
||||
github.com/chromedp/cdproto v0.0.0-20190812224334-39ef923dcb8d/go.mod h1:0YChpVzuLJC5CPr+x3xkHN6Z8KOSXjNbL7qV8Wc4GW0=
|
||||
github.com/chromedp/cdproto v0.0.0-20190926234355-1b4886c6fad6/go.mod h1:0YChpVzuLJC5CPr+x3xkHN6Z8KOSXjNbL7qV8Wc4GW0=
|
||||
github.com/chromedp/chromedp v0.3.1-0.20190619195644-fd957a4d2901/go.mod h1:mJdvfrVn594N9tfiPecUidF6W5jPRKHymqHfzbobPsM=
|
||||
github.com/chromedp/chromedp v0.4.0/go.mod h1:DC3QUn4mJ24dwjcaGQLoZrhm4X/uPHZ6spDbS2uFhm4=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/edsrzf/mmap-go v1.0.0/go.mod h1:YO35OhQPt3KJa3ryjFM5Bs14WD66h8eGKpfaBNrHW5M=
|
||||
github.com/fatih/color v1.6.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
|
||||
github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
|
||||
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
||||
github.com/go-interpreter/wagon v0.5.1-0.20190713202023-55a163980b6c/go.mod h1:5+b/MBYkclRZngKF5s6qrgWxSLgE9F5dFdO1hAueZLc=
|
||||
github.com/go-interpreter/wagon v0.6.0/go.mod h1:5+b/MBYkclRZngKF5s6qrgWxSLgE9F5dFdO1hAueZLc=
|
||||
github.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee/go.mod h1:L0fX3K22YWvt/FAX9NnzrNzcI4wNYi9Yku4O0LKYflo=
|
||||
github.com/gobwas/pool v0.2.0/go.mod h1:q8bcK0KcYlCgd9e7WYLm9LpyS+YeLd8JVDW6WezmKEw=
|
||||
github.com/gobwas/ws v1.0.2/go.mod h1:szmBTxLgaFppYjEmNtny/v3w89xOydFnnZMcgRRu/EM=
|
||||
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||
github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||
github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
|
||||
github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
|
||||
github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
|
||||
github.com/google/pprof v0.0.0-20190908185732-236ed259b199/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
|
||||
github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
|
||||
github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
|
||||
github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=
|
||||
github.com/knq/sysutil v0.0.0-20181215143952-f05b59f0f307/go.mod h1:BjPj+aVjl9FW/cCGiF3nGh5v+9Gd3VCgBQbod/GlMaQ=
|
||||
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||
github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||
github.com/mailru/easyjson v0.0.0-20190403194419-1ea4449da983/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||
github.com/mailru/easyjson v0.0.0-20190614124828-94de47d64c63/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||
github.com/mailru/easyjson v0.0.0-20190620125010-da37f6c1e481/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||
github.com/mailru/easyjson v0.0.0-20190626092158-b2ccc519800e/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||
github.com/mailru/easyjson v0.7.0/go.mod h1:KAzv3t3aY1NaHWoQz1+4F1ccyAH66Jk7yos7ldAVICs=
|
||||
github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=
|
||||
github.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
||||
github.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4=
|
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||
github.com/mattn/go-isatty v0.0.9/go.mod h1:YNRxwqDuOph6SZLI9vUUz6OYw3QyUt7WiY2yME+cCiQ=
|
||||
github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
||||
github.com/onsi/ginkgo v1.8.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
||||
github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
|
||||
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/sirupsen/logrus v1.0.5/go.mod h1:pMByvHTf9Beacp5x1UXfOR9xyW/9antXMhjMPG0dEzc=
|
||||
github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=
|
||||
github.com/spf13/pflag v1.0.1/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
|
||||
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||
github.com/twitchyliquid64/golang-asm v0.0.0-20190126203739-365674df15fc/go.mod h1:NoCfSFWosfqMqmmD7hApkirIK9ozpHjxRnRxs1l413A=
|
||||
go.coder.com/go-tools v0.0.0-20190317003359-0c6a35b74a16/go.mod h1:iKV5yK9t+J5nG9O3uF6KYdPEz3dyfMyB15MN1rbQ8Qw=
|
||||
go.uber.org/atomic v1.4.0/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=
|
||||
go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=
|
||||
golang.org/x/crypto v0.0.0-20180426230345-b49d69b5da94/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/lint v0.0.0-20190909230951-414d861bb4ac/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
golang.org/x/net v0.0.0-20181102091132-c10e9556a7bc/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190306220234-b354f8bf4d9e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20190618155005-516e3c20635f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20190712062909-fae7ac547cb7/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20190927073244-c990c680b611/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
|
||||
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||
golang.org/x/tools v0.0.0-20190920225731-5eefd052ad72/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7 h1:9zdDQZ7Thm29KFXgAX/+yaf3eVbP7djjWp/dXAppNCc=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
gopkg.in/airbrake/gobrake.v2 v2.0.9/go.mod h1:/h5ZAUhDkGaJfjzjKLSjv6zCL6O0LLBxU4K+aSYdM/U=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
|
||||
gopkg.in/gemnasium/logrus-airbrake-hook.v2 v2.1.2/go.mod h1:Xk6kEKp8OKb+X14hQBKWaSkCsqBpgog8nAV2xsGOxlo=
|
||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
||||
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gosrc.io/xmpp v0.5.1 h1:Rgrm5s2rt+npGggJH3HakQxQXR8ZZz3+QRzakRQqaq4=
|
||||
gosrc.io/xmpp v0.5.1/go.mod h1:L3NFMqYOxyLz3JGmgFyWf7r9htE91zVGiK40oW4RwdY=
|
||||
gotest.tools v2.1.0+incompatible/go.mod h1:DsYFclhRJ6vuDpmuTbkuFWG+y2sxOXAzmJt81HFBacw=
|
||||
gotest.tools/gotestsum v0.3.5/go.mod h1:Mnf3e5FUzXbkCfynWBGOwLssY7gTQgCHObK9tMpAriY=
|
||||
mvdan.cc/sh v2.6.4+incompatible/go.mod h1:IeeQbZq+x2SUGBensq/jge5lLQbS3XT2ktyp3wrt4x8=
|
||||
nhooyr.io/websocket v1.6.5 h1:8TzpkldRfefda5JST+CnOH135bzVPz5uzfn/AF+gVKg=
|
||||
nhooyr.io/websocket v1.6.5/go.mod h1:F259lAzPRAH0htX2y3ehpJe09ih1aSHN7udWki1defY=
|
||||
43
main.go
Normal file
43
main.go
Normal file
@@ -0,0 +1,43 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"time"
|
||||
|
||||
appConfig "github.com/xmpploadtesting/config"
|
||||
xmppService "github.com/xmpploadtesting/services"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
||||
// LOAD APPLICATION CONFIGURATION
|
||||
appConfig.Load()
|
||||
|
||||
// Initialise XMPP service
|
||||
xmppService.Init()
|
||||
|
||||
// Send online presence stanza
|
||||
xmppServiceInstance := xmppService.Instance()
|
||||
|
||||
for {
|
||||
|
||||
for _, xmppClient := range xmppServiceInstance.XMPPClients {
|
||||
|
||||
// Send online presence stanza in go rutines
|
||||
go func(xmppClient xmppService.XMPPClient) {
|
||||
err := xmppService.SendOnlinePresenceStanza(xmppClient.Client, xmppClient.Config.Jid)
|
||||
|
||||
if err != nil {
|
||||
log.Printf("There was an error while sending online presence stanza %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
}(xmppClient)
|
||||
|
||||
}
|
||||
// Delay before sending another message
|
||||
time.Sleep(time.Duration(appConfig.AppConfig.GeneralOptions.DelayBetweenMassages))
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
212
services/xmppsService.go
Normal file
212
services/xmppsService.go
Normal file
@@ -0,0 +1,212 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"math/rand"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
appConfig "github.com/xmpploadtesting/config"
|
||||
"gosrc.io/xmpp"
|
||||
"gosrc.io/xmpp/stanza"
|
||||
)
|
||||
|
||||
const rolingStockPresenceMessage = `ROLLING STOCK PRESENCE MESSAGE
|
||||
{
|
||||
"serial": "79101X02Y0029450001CJ2000",
|
||||
"model": "VEN032FSNWM00",
|
||||
"displayOrientation": "left",
|
||||
"macAddress": "1c:c0:e1:42:5a:c4",
|
||||
"node": "v8.16.2",
|
||||
"nodeJS": "v8.16.2",
|
||||
"fbi": "2.10-2ubuntu1",
|
||||
"libgif7": "5.1.4-2ubuntu0.1",
|
||||
"bsEdgeServer": "13.3.9-1",
|
||||
"initopciVersion": "0.4~24",
|
||||
"bspInstallVersion": "1.0~36",
|
||||
"awsConfigsVersion": "2021.21.26-e7c5738",
|
||||
"outGuard": "2021.36.23-dev-028972a",
|
||||
"val": "1.1.11~129",
|
||||
"broadSignPlayer": "13.3.9-1",
|
||||
"displayUnitExternalID": "L-0141-M7-7281-Right-3SM-G02-02-79101X02Y0020040003EJ2001",
|
||||
"privateIPaddress": "10.1.1.94",
|
||||
"linux": {
|
||||
"os": "Ubuntu 18.04.2 LTS",
|
||||
"kernel": "4.18.20+10+opci",
|
||||
"architecture": "x86-64"
|
||||
},
|
||||
"digiID": "digi-00000000-00000000-0040FFFF-FF801D30"
|
||||
}`
|
||||
|
||||
// XMPPService - struct containing
|
||||
type XMPPService struct {
|
||||
Router *xmpp.Router
|
||||
XMPPClients []XMPPClient
|
||||
}
|
||||
|
||||
// XMPPClient - struct containing client and client configuration
|
||||
type XMPPClient struct {
|
||||
Config xmpp.Config
|
||||
Client *xmpp.Client
|
||||
}
|
||||
|
||||
// ClientCredentials - credentials info for client configuration
|
||||
type ClientCredentials struct {
|
||||
Jid string `json:"username"`
|
||||
Credential string `json:"password"`
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
Ping int `json:"ping"`
|
||||
}
|
||||
|
||||
const format = "2006/01/02 15:04:05"
|
||||
|
||||
// Instance - get instance of xmpp service
|
||||
func Instance() *XMPPService {
|
||||
return &xmppService
|
||||
}
|
||||
|
||||
var xmppService XMPPService
|
||||
|
||||
// Init Initialise XMPP servie, and router
|
||||
func Init() {
|
||||
|
||||
xmppService = XMPPService{
|
||||
Router: xmpp.NewRouter(),
|
||||
}
|
||||
xmppService.Router.HandleFunc("message", handleMessage)
|
||||
|
||||
//Get Client credentials from file
|
||||
clientCredentials, err := getClientConfigsFromFile(appConfig.AppConfig.Credentials.CredentialsFileLocation)
|
||||
|
||||
if err != nil {
|
||||
log.Fatalf("Unable to load credentials from file path %v, ERROR : %v TERMINATING", appConfig.AppConfig.Credentials.CredentialsFileLocation, err)
|
||||
}
|
||||
|
||||
log.Printf("Client credentials length %v", len(clientCredentials))
|
||||
|
||||
xmppService.XMPPClients = make([]XMPPClient, 0)
|
||||
|
||||
for _, credential := range clientCredentials {
|
||||
log.Printf("Host %v", credential.Host+":"+strconv.Itoa(credential.Port))
|
||||
log.Printf("Jid %v", credential.Jid+"@"+credential.Host)
|
||||
log.Printf("Port %v", credential.Port)
|
||||
log.Printf("Credential %v", credential.Credential)
|
||||
|
||||
xmppClient := XMPPClient{
|
||||
Config: xmpp.Config{
|
||||
|
||||
TransportConfiguration: xmpp.TransportConfiguration{
|
||||
Address: credential.Host + ":" + strconv.Itoa(credential.Port),
|
||||
Domain: credential.Host,
|
||||
TLSConfig: &tls.Config{InsecureSkipVerify: true},
|
||||
},
|
||||
Jid: credential.Jid + "@" + credential.Host,
|
||||
Credential: xmpp.Password(credential.Credential),
|
||||
StreamLogger: os.Stdout,
|
||||
Insecure: true,
|
||||
},
|
||||
}
|
||||
|
||||
go func() {
|
||||
|
||||
client, err := xmpp.NewClient(&xmppClient.Config, xmppService.Router, errorHandler)
|
||||
|
||||
// Client connection
|
||||
if err := client.Connect(); err != nil {
|
||||
msg := fmt.Sprintf("XMPP connection failed: %v", err)
|
||||
|
||||
fmt.Printf("Failed to connect to server. Exiting... %v", msg)
|
||||
return
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
log.Printf("Unable to initialise client for %v", xmppClient.Config.Jid)
|
||||
}
|
||||
xmppClient.Client = client
|
||||
xmppService.XMPPClients = append(xmppService.XMPPClients, xmppClient)
|
||||
}()
|
||||
|
||||
}
|
||||
// Delay For two seccond to allow all clients to connect
|
||||
time.Sleep(time.Duration(2000000000))
|
||||
|
||||
}
|
||||
|
||||
// SendOnlinePresenceStanza - send online presence stnza with data, to server
|
||||
func SendOnlinePresenceStanza(client *xmpp.Client, jid string) error {
|
||||
onlinePresencePacket := stanza.NewPresence(stanza.Attrs{From: jid, Type: stanza.StanzaType(stanza.PresenceShowChat)})
|
||||
onlinePresencePacket.Status = rolingStockPresenceMessage
|
||||
|
||||
err := client.Send(onlinePresencePacket)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
log.Printf("<%v> sent online status", jid)
|
||||
return nil
|
||||
}
|
||||
|
||||
func getClientConfigsFromFile(filePath string) ([]ClientCredentials, error) {
|
||||
|
||||
var clientCredentials []ClientCredentials
|
||||
|
||||
f, err := os.Open(filePath)
|
||||
if err != nil {
|
||||
return clientCredentials, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
byteValue, err := ioutil.ReadAll(f)
|
||||
|
||||
if err != nil {
|
||||
return clientCredentials, err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(byteValue, &clientCredentials)
|
||||
|
||||
if err != nil {
|
||||
return clientCredentials, err
|
||||
}
|
||||
|
||||
return clientCredentials, nil
|
||||
|
||||
}
|
||||
|
||||
func handleMessage(s xmpp.Sender, p stanza.Packet) {
|
||||
|
||||
msg, ok := p.(stanza.Message)
|
||||
if !ok {
|
||||
log.Printf(" message not OK")
|
||||
_, _ = fmt.Fprintf(os.Stdout, "Ignoring packet: %T\n", p)
|
||||
return
|
||||
}
|
||||
|
||||
if len(msg.Body) == 0 {
|
||||
log.Printf("<%v> received empty message, and will not respond", msg.To)
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("<%v> received following message: %v", msg.To, msg.Body)
|
||||
statusMesageDelay := int64(rand.Intn(appConfig.AppConfig.GeneralOptions.StatusMessageDelay))
|
||||
log.Printf("DELAYING FOR: %v", statusMesageDelay*1000000000)
|
||||
// Delay For two seccond to allow all clients to connect
|
||||
time.Sleep(time.Duration(statusMesageDelay * 1000000000))
|
||||
|
||||
reply := stanza.Message{Attrs: stanza.Attrs{To: msg.From}, Body: `{"status": "OK"}`}
|
||||
err := s.Send(reply)
|
||||
|
||||
if err != nil {
|
||||
log.Printf("Error sending to %v message %v", msg.From, err)
|
||||
}
|
||||
}
|
||||
|
||||
func errorHandler(err error) {
|
||||
fmt.Println(err.Error())
|
||||
}
|
||||
Reference in New Issue
Block a user