820 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			Go
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			820 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			Go
		
	
	
		
			Executable File
		
	
	
	
	
| /*
 | |
| Package alils implements the SDK(v0.5.0) of Simple Log Service(abbr. SLS).
 | |
| 
 | |
| For more description about SLS, please read this article:
 | |
| http://gitlab.alibaba-inc.com/sls/doc.
 | |
| */
 | |
| package alils
 | |
| 
 | |
| import (
 | |
| 	"encoding/json"
 | |
| 	"fmt"
 | |
| 	"io/ioutil"
 | |
| 	"net/http"
 | |
| 	"net/http/httputil"
 | |
| )
 | |
| 
 | |
| // Error message in SLS HTTP response.
 | |
| type errorMessage struct {
 | |
| 	Code    string `json:"errorCode"`
 | |
| 	Message string `json:"errorMessage"`
 | |
| }
 | |
| 
 | |
| // LogProject Define the Ali Project detail
 | |
| type LogProject struct {
 | |
| 	Name            string // Project name
 | |
| 	Endpoint        string // IP or hostname of SLS endpoint
 | |
| 	AccessKeyID     string
 | |
| 	AccessKeySecret string
 | |
| }
 | |
| 
 | |
| // NewLogProject creates a new SLS project.
 | |
| func NewLogProject(name, endpoint, AccessKeyID, accessKeySecret string) (p *LogProject, err error) {
 | |
| 	p = &LogProject{
 | |
| 		Name:            name,
 | |
| 		Endpoint:        endpoint,
 | |
| 		AccessKeyID:     AccessKeyID,
 | |
| 		AccessKeySecret: accessKeySecret,
 | |
| 	}
 | |
| 	return p, nil
 | |
| }
 | |
| 
 | |
| // ListLogStore returns all logstore names of project p.
 | |
| func (p *LogProject) ListLogStore() (storeNames []string, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/logstores")
 | |
| 	r, err := request(p, "GET", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to list logstore")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	type Body struct {
 | |
| 		Count     int
 | |
| 		LogStores []string
 | |
| 	}
 | |
| 	body := &Body{}
 | |
| 
 | |
| 	err = json.Unmarshal(buf, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	storeNames = body.LogStores
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // GetLogStore returns logstore according by logstore name.
 | |
| func (p *LogProject) GetLogStore(name string) (s *LogStore, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "GET", "/logstores/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to get logstore")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	s = &LogStore{}
 | |
| 	err = json.Unmarshal(buf, s)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	s.project = p
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // CreateLogStore creates a new logstore in SLS,
 | |
| // where name is logstore name,
 | |
| // and ttl is time-to-live(in day) of logs,
 | |
| // and shardCnt is the number of shards.
 | |
| func (p *LogProject) CreateLogStore(name string, ttl, shardCnt int) (err error) {
 | |
| 
 | |
| 	type Body struct {
 | |
| 		Name       string `json:"logstoreName"`
 | |
| 		TTL        int    `json:"ttl"`
 | |
| 		ShardCount int    `json:"shardCount"`
 | |
| 	}
 | |
| 
 | |
| 	store := &Body{
 | |
| 		Name:       name,
 | |
| 		TTL:        ttl,
 | |
| 		ShardCount: shardCnt,
 | |
| 	}
 | |
| 
 | |
| 	body, err := json.Marshal(store)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "POST", "/logstores", h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to create logstore")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // DeleteLogStore deletes a logstore according by logstore name.
 | |
| func (p *LogProject) DeleteLogStore(name string) (err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "DELETE", "/logstores/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to delete logstore")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // UpdateLogStore updates a logstore according by logstore name,
 | |
| // obviously we can't modify the logstore name itself.
 | |
| func (p *LogProject) UpdateLogStore(name string, ttl, shardCnt int) (err error) {
 | |
| 
 | |
| 	type Body struct {
 | |
| 		Name       string `json:"logstoreName"`
 | |
| 		TTL        int    `json:"ttl"`
 | |
| 		ShardCount int    `json:"shardCount"`
 | |
| 	}
 | |
| 
 | |
| 	store := &Body{
 | |
| 		Name:       name,
 | |
| 		TTL:        ttl,
 | |
| 		ShardCount: shardCnt,
 | |
| 	}
 | |
| 
 | |
| 	body, err := json.Marshal(store)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "PUT", "/logstores", h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to update logstore")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // ListMachineGroup returns machine group name list and the total number of machine groups.
 | |
| // The offset starts from 0 and the size is the max number of machine groups could be returned.
 | |
| func (p *LogProject) ListMachineGroup(offset, size int) (m []string, total int, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	if size <= 0 {
 | |
| 		size = 500
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/machinegroups?offset=%v&size=%v", offset, size)
 | |
| 	r, err := request(p, "GET", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to list machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	type Body struct {
 | |
| 		MachineGroups []string
 | |
| 		Count         int
 | |
| 		Total         int
 | |
| 	}
 | |
| 	body := &Body{}
 | |
| 
 | |
| 	err = json.Unmarshal(buf, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	m = body.MachineGroups
 | |
| 	total = body.Total
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // GetMachineGroup retruns machine group according by machine group name.
 | |
| func (p *LogProject) GetMachineGroup(name string) (m *MachineGroup, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "GET", "/machinegroups/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to get machine group:%v", name)
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	m = &MachineGroup{}
 | |
| 	err = json.Unmarshal(buf, m)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	m.project = p
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // CreateMachineGroup creates a new machine group in SLS.
 | |
| func (p *LogProject) CreateMachineGroup(m *MachineGroup) (err error) {
 | |
| 
 | |
| 	body, err := json.Marshal(m)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "POST", "/machinegroups", h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to create machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // UpdateMachineGroup updates a machine group.
 | |
| func (p *LogProject) UpdateMachineGroup(m *MachineGroup) (err error) {
 | |
| 
 | |
| 	body, err := json.Marshal(m)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "PUT", "/machinegroups/"+m.Name, h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to update machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // DeleteMachineGroup deletes machine group according machine group name.
 | |
| func (p *LogProject) DeleteMachineGroup(name string) (err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "DELETE", "/machinegroups/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to delete machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // ListConfig returns config names list and the total number of configs.
 | |
| // The offset starts from 0 and the size is the max number of configs could be returned.
 | |
| func (p *LogProject) ListConfig(offset, size int) (cfgNames []string, total int, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	if size <= 0 {
 | |
| 		size = 100
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/configs?offset=%v&size=%v", offset, size)
 | |
| 	r, err := request(p, "GET", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to delete machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	type Body struct {
 | |
| 		Total   int
 | |
| 		Configs []string
 | |
| 	}
 | |
| 	body := &Body{}
 | |
| 
 | |
| 	err = json.Unmarshal(buf, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	cfgNames = body.Configs
 | |
| 	total = body.Total
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // GetConfig returns config according by config name.
 | |
| func (p *LogProject) GetConfig(name string) (c *LogConfig, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "GET", "/configs/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to delete config")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	c = &LogConfig{}
 | |
| 	err = json.Unmarshal(buf, c)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	c.project = p
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // UpdateConfig updates a config.
 | |
| func (p *LogProject) UpdateConfig(c *LogConfig) (err error) {
 | |
| 
 | |
| 	body, err := json.Marshal(c)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "PUT", "/configs/"+c.Name, h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to update config")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // CreateConfig creates a new config in SLS.
 | |
| func (p *LogProject) CreateConfig(c *LogConfig) (err error) {
 | |
| 
 | |
| 	body, err := json.Marshal(c)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": fmt.Sprintf("%v", len(body)),
 | |
| 		"Content-Type":      "application/json",
 | |
| 		"Accept-Encoding":   "deflate", // TODO: support lz4
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "POST", "/configs", h, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err = ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to update config")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // DeleteConfig deletes a config according by config name.
 | |
| func (p *LogProject) DeleteConfig(name string) (err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	r, err := request(p, "DELETE", "/configs/"+name, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	body, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(body, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to delete config")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // GetAppliedMachineGroups returns applied machine group names list according config name.
 | |
| func (p *LogProject) GetAppliedMachineGroups(confName string) (groupNames []string, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/configs/%v/machinegroups", confName)
 | |
| 	r, err := request(p, "GET", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to get applied machine groups")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	type Body struct {
 | |
| 		Count         int
 | |
| 		Machinegroups []string
 | |
| 	}
 | |
| 
 | |
| 	body := &Body{}
 | |
| 	err = json.Unmarshal(buf, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	groupNames = body.Machinegroups
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // GetAppliedConfigs returns applied config names list according machine group name groupName.
 | |
| func (p *LogProject) GetAppliedConfigs(groupName string) (confNames []string, err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/machinegroups/%v/configs", groupName)
 | |
| 	r, err := request(p, "GET", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to applied configs")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	type Cfg struct {
 | |
| 		Count   int      `json:"count"`
 | |
| 		Configs []string `json:"configs"`
 | |
| 	}
 | |
| 
 | |
| 	body := &Cfg{}
 | |
| 	err = json.Unmarshal(buf, body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	confNames = body.Configs
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // ApplyConfigToMachineGroup applies config to machine group.
 | |
| func (p *LogProject) ApplyConfigToMachineGroup(confName, groupName string) (err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/machinegroups/%v/configs/%v", groupName, confName)
 | |
| 	r, err := request(p, "PUT", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to apply config to machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // RemoveConfigFromMachineGroup removes config from machine group.
 | |
| func (p *LogProject) RemoveConfigFromMachineGroup(confName, groupName string) (err error) {
 | |
| 	h := map[string]string{
 | |
| 		"x-sls-bodyrawsize": "0",
 | |
| 	}
 | |
| 
 | |
| 	uri := fmt.Sprintf("/machinegroups/%v/configs/%v", groupName, confName)
 | |
| 	r, err := request(p, "DELETE", uri, h, nil)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	buf, err := ioutil.ReadAll(r.Body)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if r.StatusCode != http.StatusOK {
 | |
| 		errMsg := &errorMessage{}
 | |
| 		err = json.Unmarshal(buf, errMsg)
 | |
| 		if err != nil {
 | |
| 			err = fmt.Errorf("failed to remove config from machine group")
 | |
| 			dump, _ := httputil.DumpResponse(r, true)
 | |
| 			fmt.Printf("%s\n", dump)
 | |
| 			return
 | |
| 		}
 | |
| 		err = fmt.Errorf("%v:%v", errMsg.Code, errMsg.Message)
 | |
| 		return
 | |
| 	}
 | |
| 	return
 | |
| }
 |