func node_lost_handler(event models.AppEvent, ctxt string) (models.AppEvent, error) {
//worry about the node only if the node in active state
state, err := skyringutils.GetNodeStateById(ctxt, event.NodeId)
if state == models.NODE_STATE_ACTIVE && err == nil {
if err := skyringutils.Update_node_status_byId(ctxt, event.NodeId, models.NODE_STATUS_ERROR); err != nil {
return event, err
}
}
event.Name = skyring.EventTypes["NODE_STATE_CHANGED"]
event.Message = fmt.Sprintf("Host: %s lost contact", event.NodeName)
event.EntityId = event.NodeId
event.Severity = models.ALARM_STATUS_MAJOR
event.NotificationEntity = models.NOTIFICATION_ENTITY_HOST
event.Notify = true
util.AppendServiceToNode(bson.M{"nodeid": event.NodeId}, models.SkyringServices[1], models.STATUS_DOWN, ctxt)
if err := update_alarm_count(event, ctxt); err != nil {
logger.Get().Error("%s-could not update alarm"+
" count for event: %s", ctxt, event.EventId.String())
return event, err
}
app := skyring.GetApp()
if err := app.RouteProviderEvents(ctxt, event); err != nil {
logger.Get().Error("%s-Event:%s could not be handled for node: %s. error: %v", ctxt, event.Name, event.NodeName, err)
}
return event, nil
}
func handle_node_start_event(node string) error {
sessionCopy := db.GetDatastore().Copy()
defer sessionCopy.Close()
coll := sessionCopy.DB(conf.SystemConfig.DBConfig.Database).C(models.COLL_NAME_STORAGE_NODES)
var storage_node models.Node
_ = coll.Find(bson.M{"hostname": node}).One(&storage_node)
if storage_node.State != models.NODE_STATE_INITIALIZING {
logger.Get().Warning(fmt.Sprintf("Node with name: %s not in activating state to update other details", node))
return nil
}
asyncTask := func(t *task.Task) {
t.UpdateStatus("started the task for InitializeNode: %s", t.ID)
// Process the request
if err := initializeStorageNode(node, t); err != nil {
t.UpdateStatus("Failed")
t.Done(models.TASK_STATUS_FAILURE)
} else {
t.UpdateStatus("Success")
t.Done(models.TASK_STATUS_SUCCESS)
}
}
if taskId, err := skyring.GetApp().GetTaskManager().Run(fmt.Sprintf("Initialize Node: %s", node), asyncTask, nil, nil, nil); err != nil {
logger.Get().Error("Unable to create the task for Initialize Node: %s. error: %v", node, err)
} else {
logger.Get().Debug("Task created for initialize node. Task id: %s", taskId.String())
}
return nil
}
func RouteEvent(event models.NodeEvent) {
var e models.Event
e.Timestamp = event.Timestamp
e.Tag = event.Tag
e.Tags = event.Tags
e.Message = event.Message
e.Severity = event.Severity
eventId, err := uuid.New()
if err != nil {
logger.Get().Error("Uuid generation for the event failed for node: %s. error: %v", event.Node, err)
return
}
e.EventId = *eventId
// querying DB to get node ID and Cluster ID for the event
sessionCopy := db.GetDatastore().Copy()
defer sessionCopy.Close()
var node models.Node
coll := sessionCopy.DB(conf.SystemConfig.DBConfig.Database).C(models.COLL_NAME_STORAGE_NODES)
if err := coll.Find(bson.M{"hostname": event.Node}).One(&node); err != nil {
logger.Get().Error("Node information read from DB failed for node: %s. error: %v", event.Node, err)
return
}
// Push the event to DB only if the node is managed
if node.Hostname == "" || !node.Enabled {
return
}
e.ClusterId = node.ClusterId
e.NodeId = node.NodeId
// Invoking the event handler
for tag, handler := range handlermap {
if match, err := filepath.Match(tag, e.Tag); err == nil {
if match {
if err := handler.(func(models.Event) error)(e); err != nil {
logger.Get().Error("Event Handling Failed for event for node: %s. error: %v", node.Hostname, err)
return
}
if err := Persist_event(e); err != nil {
logger.Get().Error("Could not persist the event to DB for node: %s. error: %v", node.Hostname, err)
return
} else {
// For upcoming any new event , broadcasting to all connected clients
eventObj, err := json.Marshal(e)
if err != nil {
logger.Get().Error("Error marshalling the event data for node: %s. error: %v", node.Hostname, err)
}
GetBroadcaster().chBroadcast <- string(eventObj)
return
}
}
} else {
logger.Get().Error("Error while maping handler for event for node: %s. error: %v", node.Hostname, err)
return
}
}
// Handle Provider specific events
app := skyring.GetApp()
if err := app.RouteProviderEvents(e); err != nil {
logger.Get().Error("Event:%s could not be handled for node: %s. error: %v", e.Tag, node.Hostname, err)
}
return
}
func drive_add_handler(event models.AppEvent, ctxt string) (models.AppEvent, error) {
sessionCopy := db.GetDatastore().Copy()
defer sessionCopy.Close()
var node models.Node
coll := sessionCopy.DB(conf.SystemConfig.DBConfig.Database).C(models.COLL_NAME_STORAGE_NODES)
if err := coll.Find(bson.M{"nodeid": event.NodeId}).One(&node); err != nil {
logger.Get().Error("%s-Node information read from DB failed for node: %s. error: %v", ctxt, event.NodeId, err)
return event, err
}
if node.State != models.NODE_STATE_ACTIVE {
return event, nil
}
existing_disks := node.StorageDisks
sProfiles, err := skyring.GetDbProvider().StorageProfileInterface().StorageProfiles(ctxt, nil, models.QueryOps{})
if err != nil {
logger.Get().Error("%s-Unable to get the storage profiles. err:%v", ctxt, err)
return event, err
}
// sync the nodes to get new disks
if ok, err := skyring.GetCoreNodeManager().SyncStorageDisks(node.Hostname, sProfiles, ctxt); err != nil || !ok {
logger.Get().Error("%s-Failed to sync disk for host: %s Error: %v", ctxt, node.Hostname, err)
return event, err
}
// get the list of disks after syncing.
if err := coll.Find(bson.M{"nodeid": event.NodeId}).One(&node); err != nil {
logger.Get().Error("%s-Node information read from DB failed for node: %s. error: %v", ctxt, event.NodeId, err)
return event, err
}
var cluster_nodes []models.ClusterNode
var cluster_node models.ClusterNode
cluster_node.NodeId = event.NodeId.String()
cluster_node.NodeType = []string{models.NODE_TYPE_OSD}
var exists bool
for _, disk := range node.StorageDisks {
exists = false
for _, existing_disk := range existing_disks {
if disk.DevName == existing_disk.DevName {
exists = true
break
}
}
if !exists && !disk.Used && disk.Type == "disk" {
cluster_node.Devices = append(cluster_node.Devices, models.ClusterNodeDevice{Name: disk.DevName, FSType: "xfs"})
event.EntityId = disk.DiskId
event.Tags["DevName"] = disk.DevName
event.Tags["size"] = strconv.FormatFloat(disk.Size, 'E', -1, 64)
event.Tags["Type"] = disk.Type
}
}
// adding the details for event
event.Name = skyring.EventTypes["DRIVE_ADD"]
event.NotificationEntity = models.NOTIFICATION_ENTITY_HOST
event.Message = fmt.Sprintf("New Storage Drive: %s added to Host:%s", event.Tags["DevName"], node.Hostname)
event.Severity = models.ALARM_STATUS_CLEARED
event.Notify = false
c := sessionCopy.DB(conf.SystemConfig.DBConfig.Database).C(models.COLL_NAME_STORAGE_CLUSTERS)
var cluster models.Cluster
if err := c.Find(bson.M{"clusterid": event.ClusterId}).One(&cluster); err != nil {
logger.Get().Error("%s-Cluster information read from DB failed for Cluster: %s. error: %v", ctxt, event.ClusterId, err)
return event, nil
}
// check if cluster is in managed/un-managed state
ok, err := skyring.ClusterUnmanaged(event.ClusterId)
if err != nil {
logger.Get().Warning("%s-Error checking managed state of cluster: %v. error: %v", ctxt, event.ClusterId, err)
return event, err
}
if ok {
logger.Get().Error("%s-Cluster: %v is in un-managed state", ctxt, event.ClusterId)
return event, err
}
vars := map[string]string{"cluster-id": event.ClusterId.String()}
cluster_nodes = append(cluster_nodes, cluster_node)
body, err := json.Marshal(cluster_nodes)
if err != nil {
logger.Get().Error(fmt.Sprintf("%s-Error forming request body. error: %v", ctxt, err))
return event, nil
}
// lock the node for expanding
var nodes models.Nodes
if err := coll.Find(bson.M{"clusterid": event.ClusterId}).All(&nodes); err != nil {
logger.Get().Error("%s-Node information read from DB failed . error: %v", ctxt, err)
return event, nil
}
// check if autoExpand is enabled or not
if !cluster.AutoExpand {
return event, nil
}
// Expand cluster
var result models.RpcResponse
var providerTaskId *uuid.UUID
asyncTask := func(t *task.Task) {
sessionCopy := db.GetDatastore().Copy()
defer sessionCopy.Close()
for {
select {
case <-t.StopCh:
return
default:
t.UpdateStatus("Started task for cluster expansion: %v", t.ID)
appLock, err := skyring.LockNodes(ctxt, nodes, "Expand_Cluster")
if err != nil {
util.FailTask("Failed to acquire lock", fmt.Errorf("%s-%v", ctxt, err), t)
return
}
defer skyring.GetApp().GetLockManager().ReleaseLock(ctxt, *appLock)
provider := skyring.GetApp().GetProviderFromClusterId(ctxt, node.ClusterId)
if provider == nil {
util.FailTask("", errors.New(fmt.Sprintf("%s-Error etting provider for cluster: %v", ctxt, event.ClusterId)), t)
return
}
err = provider.Client.Call(fmt.Sprintf("%s.%s",
provider.Name, "ExpandCluster"),
models.RpcRequest{RpcRequestVars: vars, RpcRequestData: body, RpcRequestContext: ctxt},
&result)
if err != nil || (result.Status.StatusCode != http.StatusOK && result.Status.StatusCode != http.StatusAccepted) {
util.FailTask(fmt.Sprintf("Error expanding cluster: %v", event.ClusterId), fmt.Errorf("%s-%v", ctxt, err), t)
return
}
// Update the master task id
providerTaskId, err = uuid.Parse(result.Data.RequestId)
if err != nil {
util.FailTask(fmt.Sprintf("Error parsing provider task id while expand cluster: %v", event.ClusterId), fmt.Errorf("%s-%v", ctxt, err), t)
return
}
t.UpdateStatus(fmt.Sprintf("Started provider task: %v", *providerTaskId))
if ok, err := t.AddSubTask(*providerTaskId); !ok || err != nil {
util.FailTask(fmt.Sprintf("Error adding sub task while expand cluster: %v", event.ClusterId), fmt.Errorf("%s-%v", ctxt, err), t)
return
}
// Check for provider task to complete and update the disk info
for {
time.Sleep(2 * time.Second)
coll := sessionCopy.DB(conf.SystemConfig.DBConfig.Database).C(models.COLL_NAME_TASKS)
var providerTask models.AppTask
if err := coll.Find(bson.M{"id": *providerTaskId}).One(&providerTask); err != nil {
util.FailTask(fmt.Sprintf("Error getting sub task status while expand cluster: %v", event.ClusterId), fmt.Errorf("%s-%v", ctxt, err), t)
return
}
if providerTask.Completed {
if providerTask.Status == models.TASK_STATUS_SUCCESS {
t.UpdateStatus("Starting disk sync")
if ok, err := skyring.GetCoreNodeManager().SyncStorageDisks(node.Hostname, sProfiles, ctxt); err != nil || !ok {
logger.Get().Error("%s-Failed to sync disk for host: %s Error: %v", ctxt, node.Hostname, err)
return
}
go skyring.ComputeSystemSummary(make(map[string]interface{}))
t.UpdateStatus("Success")
t.Done(models.TASK_STATUS_SUCCESS)
} else {
logger.Get().Error("%s-Failed to expand the cluster %s", ctxt, event.ClusterId)
t.UpdateStatus("Failed")
t.Done(models.TASK_STATUS_FAILURE)
}
break
}
}
return
}
}
}
if taskId, err := skyring.GetApp().GetTaskManager().Run(
models.ENGINE_NAME,
fmt.Sprintf("Expand Cluster: %s", event.ClusterId.String()),
asyncTask,
nil,
nil,
nil); err != nil {
logger.Get().Error("%s-Unable to create task to expand cluster: %v. error: %v", ctxt, event.ClusterId, err)
return event, nil
} else {
logger.Get().Debug("%s-Task Created: %v to expand cluster: %v", ctxt, taskId, event.ClusterId)
return event, nil
}
}