package models
|
|
import (
|
"errors"
|
"fmt"
|
"strconv"
|
"time"
|
"vamicro/config"
|
"vamicro/extend/util"
|
|
"basic.com/syncdb.git"
|
)
|
|
type Cluster struct {
|
ClusterId string `gorm:"primary_key;column:cluster_id" json:"cluster_id"`
|
ClusterName string `gorm:"column:cluster_name" json:"cluster_name"`
|
Password string `gorm:"column:password" json:"password"`
|
VirtualIp string `gorm:"column:virtual_ip" json:"virtual_ip"`
|
}
|
|
func (Cluster) TableName() string {
|
return "cluster"
|
}
|
|
func (c *Cluster) FindAll() (arr []Cluster, err error) {
|
if err = db.Table("cluster").Find(&arr).Error; err != nil {
|
return nil, err
|
}
|
return arr, nil
|
}
|
|
func (c *Cluster) FindByClusterId() (result Cluster, err error) {
|
if c.ClusterId == "" {
|
return Cluster{}, fmt.Errorf("param clusterId is nil error")
|
}
|
if err = db.Table("cluster").Where("cluster_id=?", c.ClusterId).First(&result).Error; err != nil {
|
return Cluster{}, err
|
}
|
return result, nil
|
}
|
|
func (c *Cluster) FindByVirtualIp() (result Cluster, err error) {
|
if c.VirtualIp == "" {
|
return Cluster{}, fmt.Errorf("param virtual_ip is nil error")
|
}
|
if err = db.Table("cluster").Where("virtual_ip=?", c.VirtualIp).First(&result).Error; err != nil {
|
return Cluster{}, err
|
}
|
return result, nil
|
}
|
|
func (c *Cluster) Create() error {
|
var localConfig LocalConfig
|
err := localConfig.Select()
|
if err != nil {
|
return err
|
}
|
|
serverId := config.Server.AnalyServerId
|
if serverId == "" {
|
return errors.New("serverId 为空")
|
}
|
|
serverIp, _, e := util.GetLocalIP(config.Server.NetworkAdapter)
|
if e != nil {
|
return e
|
}
|
|
tx := db.Begin()
|
defer func() {
|
if err != nil && tx != nil {
|
tx.Rollback()
|
}
|
}()
|
|
sql := "insert into cluster (cluster_id,cluster_name,password,virtual_ip) values ('" + c.ClusterId + "','" + c.ClusterName + "','" + c.Password + "','" + c.VirtualIp + "')"
|
if err = tx.Exec(sql).Error; err != nil {
|
return err
|
}
|
timeUnix := time.Now().Unix()
|
fmtTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")
|
//添加本身节点
|
sql = "insert into cluster_node (id,cluster_id,node_name,node_id,node_ip,create_time,isDelete,device_type,drift_state) values ('" + serverId + "','" + c.ClusterId + "','" + localConfig.ServerName + "','" + serverId + "','" + (serverIp + ":" + strconv.Itoa(syncdb.DefaultBindPort)) + "','" + fmtTimeStr + "',0,'" + config.Server.DeviceType + "','master')"
|
if err = tx.Exec(sql).Error; err != nil {
|
return err
|
}
|
|
tx.Commit()
|
|
return nil
|
}
|
|
func (c *Cluster) UpdateClusterName(clusterName string, virtualIp string) bool {
|
arr, e := c.FindAll()
|
if e == nil && arr != nil && len(arr) > 0 {
|
result := db.Exec("update cluster set cluster_name='" + clusterName + "',virtual_ip='" + virtualIp + "' where cluster_id='" + arr[0].ClusterId + "'")
|
if result.Error != nil {
|
return false
|
}
|
if result.RowsAffected > 0 {
|
return true
|
}
|
}
|
return false
|
}
|
|
func (c *Cluster) UpdateNodeName(nodeName string, nodeId string) bool {
|
arr, _ := c.FindAll()
|
if arr != nil && len(arr) > 0 {
|
c := arr[0]
|
result := db.Exec("update cluster_node set node_name='" + nodeName + "' where cluster_id='" + c.ClusterId + "' and id='" + nodeId + "'")
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
return false
|
}
|
|
func (c *Cluster) FindAllNodeMap() map[string]Node {
|
m := make(map[string]Node, 0)
|
arr, err := c.FindAll()
|
if err == nil && arr != nil {
|
var nodeE Node
|
for _, ct := range arr {
|
nodes, e := nodeE.FindNodesByClusterId(ct.ClusterId)
|
if e == nil && nodes != nil {
|
for _, nd := range nodes {
|
m[nd.NodeId] = nd
|
}
|
}
|
}
|
}
|
|
return m
|
}
|
