From 0383c58d01c3f67552e2a3f018e4667206a0363c Mon Sep 17 00:00:00 2001
From: Corey Hulen <corey@hulen.com>
Date: Mon, 28 Nov 2016 07:40:02 -0800
Subject: PLT-4357 Adding stats to HA (#4638)

* PLT-4357 adding stats to HA

* PLT-4357 adding stats to HA

* Fixing getting stats from the current server
---
 model/cluster_stats.go      | 36 ++++++++++++++++++++++++++++++++++++
 model/cluster_stats_test.go | 19 +++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 model/cluster_stats.go
 create mode 100644 model/cluster_stats_test.go

(limited to 'model')

diff --git a/model/cluster_stats.go b/model/cluster_stats.go
new file mode 100644
index 000000000..f2efa323e
--- /dev/null
+++ b/model/cluster_stats.go
@@ -0,0 +1,36 @@
+// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package model
+
+import (
+	"encoding/json"
+	"io"
+)
+
+type ClusterStats struct {
+	Id                        string `json:"id"`
+	TotalWebsocketConnections int    `json:"total_websocket_connections"`
+	TotalReadDbConnections    int    `json:"total_read_db_connections"`
+	TotalMasterDbConnections  int    `json:"total_master_db_connections"`
+}
+
+func (me *ClusterStats) ToJson() string {
+	b, err := json.Marshal(me)
+	if err != nil {
+		return ""
+	} else {
+		return string(b)
+	}
+}
+
+func ClusterStatsFromJson(data io.Reader) *ClusterStats {
+	decoder := json.NewDecoder(data)
+	var me ClusterStats
+	err := decoder.Decode(&me)
+	if err == nil {
+		return &me
+	} else {
+		return nil
+	}
+}
diff --git a/model/cluster_stats_test.go b/model/cluster_stats_test.go
new file mode 100644
index 000000000..13920b5d1
--- /dev/null
+++ b/model/cluster_stats_test.go
@@ -0,0 +1,19 @@
+// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package model
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestClusterStatsJson(t *testing.T) {
+	cluster := ClusterStats{Id: NewId(), TotalWebsocketConnections: 1, TotalReadDbConnections: 1}
+	json := cluster.ToJson()
+	result := ClusterStatsFromJson(strings.NewReader(json))
+
+	if cluster.Id != result.Id {
+		t.Fatal("Ids do not match")
+	}
+}
-- 
cgit v1.2.3-1-g7c22