package main
import (
 "encoding/json"
 "errors"
 "flag"
 "fmt"
 "io/ioutil"
 "log"
 "net"
 "os"
 "strings"
 "time"
)
// 鏡像結(jié)構(gòu)
type Image struct {
 Created uint64
 Id string
 ParentId string
 RepoTags []string
 Size uint64
 VirtualSize uint64
}
// 容器結(jié)構(gòu)
type Container struct {
 Id string `json:"Id"`
 Names []string `json:"Names"`
 Image string `json:"Image"`
 ImageID string `json:"ImageID"`
 Command string `json:"Command"`
 Created uint64 `json:"Created"`
 State string `json:"State"`
 Status string `json:"Status"`
 Ports []Port `json:"Ports"`
 Labels map[string]string `json:"Labels"`
 HostConfig map[string]string `json:"HostConfig"`
 NetworkSettings map[string]interface{} `json:"NetworkSettings"`
 Mounts []Mount `json:"Mounts"`
}
// docker 端口映射
type Port struct {
 IP string `json:"IP"`
 PrivatePort int `json:"PrivatePort"`
 PublicPort int `json:"PublicPort"`
 Type string `json:"Type"`
}
// docker 掛載
type Mount struct {
 Type string `json:"Type"`
 Source string `json:"Source"`
 Destination string `json:"Destination"`
 Mode string `json:"Mode"`
 RW bool `json:"RW"`
 Propatation string `json:"Propagation"`
}
// 連接列表
var SockAddr = "/var/run//docker.sock" //這可不是隨便寫(xiě)的，是docker官網(wǎng)文檔的套接字默認(rèn)值，當(dāng)然守護(hù)進(jìn)程通訊方式還有tcp,fd等方式，各自都有適用場(chǎng)景。。。
var imagesSock = "GET /images/json HTTP/1.0\r\n\r\n" //docker對(duì)外的鏡像api操作
var containerSock = "GET /containers/json?all=true HTTP/1.0\r\n\r\n"  //docker對(duì)外的容器查看api
var startContainerSock = "POST /containers/%s/start HTTP/1.0\r\n\r\n" //docker對(duì)外的容器啟動(dòng)api
// 白名單
var whiteList []string
func main() {
 // 讀取命令行參數(shù)
 // 白名單列表
 list := flag.String("list", "", "docker white list to restart, eg: token,explorer")
 // 輪詢(xún)的時(shí)間間隔，單位秒
 times := flag.Int64("time", 10, "time interval to set read docker containers [second], default is 10 second")
 flag.Parse()
 // 解析list => whiteList
 whiteList = strings.Split(*list, ",") //將我們命令行中l(wèi)ist參數(shù)的容器列表解析到代碼中
 log.SetOutput(os.Stdout)
 log.Println("start docker watching...")
 log.Printf("Your whiteList: %v\n", *list)
 log.Printf("Your shedule times: %ds\n", *times)
  //接下來(lái)的這個(gè)for循環(huán)就是每隔一定時(shí)間監(jiān)控docker容器是否正常運(yùn)行，不正常就重新啟動(dòng)它
 for {
  // 輪詢(xún)docker
  err := listenDocker() 
  if err != nil {
   log.Println(err.Error())
  }
  time.Sleep(time.Duration(*times)*time.Second)
 }
}
func listenDocker() error {
 // 獲取容器列表,拿到所有的容器信息
 containers, err := readContainer() 
 if err != nil {
  return errors.New("read container error: " + err.Error())
 }
 // 先遍歷白名單快，次數(shù)少
 for _, name := range whiteList {
 Name:
  for _, container := range containers {
   for _, cname := range container.Names {
    // 如果匹配到白名單
    if cname[1:] == name {
     // 關(guān)心一下容器狀態(tài)
     log.Printf("id=%s, name=%s, state=%s", container.Id[:12], container.Names, container.Status)
     if strings.Contains(container.Status, "Exited") {
      // 如果出現(xiàn)異常退出的容器，啟動(dòng)它
      log.Printf("find container: [%s] has exited, ready to start it. ", name)
      e := startContainer(container.Id)
      if e != nil {
       log.Println("start container error: ", e.Error())
      }
      break Name
     }
    }
   }
  }
 }
 return nil
}
// 獲取 unix sock 連接
func connectDocker() (*net.UnixConn, error) {
 addr := net.UnixAddr{SockAddr, "unix"}  // SockAddr 這個(gè)變量的值被設(shè)定為docker的/var/run/docker 套接字路徑值，也就是說(shuō)此處就是撥通與docker的daemon通訊建立的關(guān)鍵處,其他處的代碼就是些正常的邏輯處理了
 return net.DialUnix("unix", nil, &addr)
}
// 啟動(dòng)容器
func startContainer(id string) error {
 conn, err := connectDocker()
 if err != nil {
  return errors.New("connect error: " + err.Error())
 }
 start := fmt.Sprintf(startContainerSock, id)
 fmt.Println(start)
 cmd := []byte(start)
 code, err := conn.Write(cmd)
 if err != nil {
  return err
 }
 log.Println("start container response code: ", code)
 // 啟動(dòng)容器等待20秒，防止數(shù)據(jù)重發(fā)
 time.Sleep(20*time.Second)
 return nil
}
// 獲取容器列表
func readContainer() ([]Container, error) {
 conn, err := connectDocker() //建立一個(gè)unix連接,這其實(shí)是一個(gè)關(guān)鍵點(diǎn)，需要你了解unix 套接字 建立連接
 if err != nil {
  return nil, errors.New("connect error: " + err.Error())
 }
 _, err = conn.Write([]byte(containerSock))
 if err != nil {
  return nil, err
 }
 result, err := ioutil.ReadAll(conn)
 if err != nil {
  return nil, err
 }
 body := getBody(result)
 var containers []Container
 err = json.Unmarshal(body, &containers)
 if err != nil {
  return nil, err
 }
 log.Println("len of containers: ", len(containers))
 if len(containers) == 0 {
  return nil, errors.New("no containers")
 }
 return containers, nil
}
// 獲取鏡像列表
func readImage(conn *net.UnixConn) ([]Image, error) {
 _, err := conn.Write([]byte(imagesSock))
 if err != nil {
  return nil, err
 }
 result, err := ioutil.ReadAll(conn)
 if err != nil {
  return nil, err
 }
 body := getBody(result[:])
 var images []Image
 err = json.Unmarshal(body, &images)
 if err != nil {
  return nil, err
 }
 return images, nil
}
// 從返回的 http 響應(yīng)中提取 body
func getBody(result []byte) (body []byte) {
 for i:=0; i<=len(result)-4; i++ {
  if result[i] == 13 && result[i+1] == 10 && result[i+2] == 13 && result[i+3] == 10 {
   body = result[i+4:]
   break
  }
 }
 return
}
/*
error log :
 1、write unix @->/var/run/docker.sock: write: broken pipe
  建立的tcp連接不能復(fù)用，每次操作都建立連接
 */