summaryrefslogtreecommitdiffstats
path: root/model/cluster_info.go
blob: f76a03c0bc2b4a5bc0600980824c7c3401331439 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package model

import (
	"encoding/json"
	"io"
	"strings"
	"sync"
	"sync/atomic"
)

type ClusterInfo struct {
	Id                 string       `json:"id"`
	Version            string       `json:"version"`
	ConfigHash         string       `json:"config_hash"`
	InterNodeUrl       string       `json:"internode_url"`
	Hostname           string       `json:"hostname"`
	LastSuccessfulPing int64        `json:"last_ping"`
	Alive              int32        `json:"is_alive"`
	Mutex              sync.RWMutex `json:"-"`
}

func (me *ClusterInfo) ToJson() string {
	me.Mutex.RLock()
	defer me.Mutex.RUnlock()
	b, err := json.Marshal(me)
	if err != nil {
		return ""
	} else {
		return string(b)
	}
}

func (me *ClusterInfo) Copy() *ClusterInfo {
	json := me.ToJson()
	return ClusterInfoFromJson(strings.NewReader(json))
}

func ClusterInfoFromJson(data io.Reader) *ClusterInfo {
	decoder := json.NewDecoder(data)
	var me ClusterInfo
	me.Mutex = sync.RWMutex{}
	err := decoder.Decode(&me)
	if err == nil {
		return &me
	} else {
		return nil
	}
}

func (me *ClusterInfo) SetAlive(alive bool) {
	if alive {
		atomic.StoreInt32(&me.Alive, 1)
	} else {
		atomic.StoreInt32(&me.Alive, 0)
	}
}

func (me *ClusterInfo) IsAlive() bool {
	return atomic.LoadInt32(&me.Alive) == 1
}

func (me *ClusterInfo) HaveEstablishedInitialContact() bool {
	me.Mutex.RLock()
	defer me.Mutex.RUnlock()
	if me.Id != "" {
		return true
	}

	return false
}

func (me *ClusterInfo) IdEqualTo(in string) bool {
	me.Mutex.RLock()
	defer me.Mutex.RUnlock()
	if me.Id == in {
		return true
	}

	return false
}

func ClusterInfosToJson(objmap []*ClusterInfo) string {
	if b, err := json.Marshal(objmap); err != nil {
		return ""
	} else {
		return string(b)
	}
}

func ClusterInfosFromJson(data io.Reader) []*ClusterInfo {
	decoder := json.NewDecoder(data)

	var objmap []*ClusterInfo
	if err := decoder.Decode(&objmap); err != nil {
		return make([]*ClusterInfo, 0)
	} else {
		return objmap
	}
}