Go language nail send notifications

package main

import (
   "bufio"
   "io"
   "fmt"
   "io/ioutil"
   "os/exec"
   "net/http"
   "os"
   "strings"
)

func GetKey() (timestamp string ,sign string, err error){
   cmd := exec.Command("/bin/python", "/application/scripts/hezhong_host_status/aaa.py")
   stdout , err := cmd.StdoutPipe()
   if err != nil {
      fmt.Println("Error:can not obtain stdout pipe for command:%s \n", err)
      return timestamp , sign,  err
   }


   if err := cmd.Start() ; err != nil {
      fmt.Println("Error:The command is err ",err)
      return  timestamp,  sign,err
   }

   var Key []string
   //读取所有输出
   bytes := bufio.NewReader(stdout)
   for {
      line ,err:= bytes.ReadString('\n')
      if err == io.EOF{
         break
      }
      if err != nil {
         fmt.Println("Read Err:",err)
      }
      Key = append(Key,line)
   }


   if err := cmd.Wait();err!= nil{
      fmt.Println("Wait",err.Error())
      return  timestamp, sign ,err
   }

   timestamp = Key[0]
   sign = Key[1]
   return timestamp ,sign ,nil
}

func SendDingMsg(msg string) {
   //请求地址模板

   timestamp , sign , err := GetKey()
   if err != nil {
      fmt.Println("Get Key Err:",err)
   }
   timestamp = strings.TrimRight(timestamp, "\n")
   sign = strings.TrimRight(sign, "\n")
   //timestamp := time.Now().UnixNano() / 1e6
   //timestamp := int64(1572870805748)
   //sign := "oJ5SHD3IwCWVIcL78k3pgX0tfQnjDfcDZMPXAI%2BvrFE%3D"
   webHook := "https://oapi.dingtalk.com/robot/send?access_token=628d11124aef5f9efe2a8c8a6b5afa2b67ab01dxxxxxxxxxxxxxxxxxxxxx&"+ "timestamp=" + timestamp + "&sign=" + sign
   content := `{"msgtype": "text",
      "text": {"content": "`+ msg + `"},
                "at": {
                     "atMobiles": [
                         "18301371817"
                     ], 
                     "isAtAll": true
                } 
   } ` 
   // Create a request 
   REQ, ERR: = http.NewRequest (" the POST ", WebHook, strings.NewReader (Content)) 
   IF ERR = nil {! 
      Fmt.Println (ERR) 
   } 
   Client: = & HTTP. {} Client 
   // setting request header 
   req.Header.Set ( "the Content-the Type", "file application / JSON") 
   req.Header.Set ( "the User-Agent", "Firefox") 
   // send request 
   resp, err: client.Do = (REQ) 
   // close request 
   the defer resp.Body.Close () 
   fmt.Println (resp.StatusCode) 
   body, _: = ioutil.ReadAll (resp.Body) 
   fmt.Println (String (body)) 
   IF ERR! = nil { 
      fmt.Println ( "handle error")
   }
}


func main(){
   SendDingMsg (os.Args [1]) 
}

Auxiliary Python script:

#!python 2.7
import time
import hmac
import hashlib
import base64
import urllib
timestamp = long(round(time.time() * 1000))
secret = 'this is secret'
secret_enc = bytes(secret).encode('utf-8')
string_to_sign = '{}\n{}'.format(timestamp, secret)
string_to_sign_enc = bytes(string_to_sign).encode('utf-8')
hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
sign = urllib.quote_plus(base64.b64encode(hmac_code))
print(timestamp)
print(sign)




Guess you like

Origin blog.51cto.com/dellinger/2447790