save and refresh EBS volume state

agent/ebs/watcher.go

@@ -247,7 +247,6 @@ func (w *EBSWatcher) notifyAttachedEBS(volumeId string) error {
 	if err := w.sendEBSStateChange(ebs); err != nil {
 		return fmt.Errorf("Unable to send state EBS change, %s", err)
 	}
-	ebs.SetSentStatus()
 	ebs.StopAckTimer()
 	log.Infof("We've set sent status for %v", ebs.EBSToString())
 	return nil

agent/engine/data.go

@@ -40,6 +40,11 @@ func (engine *DockerTaskEngine) LoadState() error {
 		return err
 	}
 
+	// for now this will only load EBS Attachments -- see TODO below
+	if err := engine.loadResourceAttachments(); err != nil {
+		return err
+	}
+
 	return engine.loadENIAttachments()
 }
 
@@ -119,6 +124,18 @@ func (engine *DockerTaskEngine) loadImageStates() error {
 	return nil
 }
 
+func (engine *DockerTaskEngine) loadResourceAttachments() error {
+        resAttachments, err := engine.dataClient.GetResourceAttachments()
+        if err != nil {
+                return err
+        }
+        // TODO update docker state to handle more than just EBS Resource Type
+        for _, resAttachment := range resAttachments {
+                engine.state.AddEBSAttachment(Attachment)
+        }
+        return nil
+}
+
 func (engine *DockerTaskEngine) loadENIAttachments() error {
 	eniAttachments, err := engine.dataClient.GetENIAttachments()
 	if err != nil {