Пример #1
0
func createBackend(vs, rs string, b gdc.APIPort) error {
	log.Infof("creating [%s] with %s:%d -> %d", path.Join(vs, rs), b.IP,
		b.PublicPort, b.PrivatePort)

	if _, exists := exposed[vs]; !exists {
		r, err := http.Get(
			fmt.Sprintf("http://%s", path.Join(*remote, "service", vs)))
		if err != nil {
			return err
		}

		if r.StatusCode == 200 {
			// Service was pre-exposed earlier.
			exposed[vs] = struct{}{}
		} else if err := createService(vs, b.PrivatePort, b.Type); err != nil {
			return err
		}
	}

	opts := core.BackendOptions{Host: b.IP, Port: uint16(b.PublicPort)}
	data := bytes.NewBuffer(util.MustMarshal(opts, util.JSONOptions{}))

	rqst, _ := http.NewRequest(
		"PUT",
		fmt.Sprintf("http://%s", path.Join(*remote, "service", vs, rs)),
		data)

	return roundtrip(rqst, map[int]func() error{
		http.StatusConflict: func() error {
			return fmt.Errorf("backend [%s] was already created", path.Join(vs, rs))
		},
		http.StatusNotFound: func() error {
			return fmt.Errorf("service parent [%s] cannot be found", vs)
		}})
}
Пример #2
0
func (h backendStatusHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)

	if opts, err := h.ctx.GetBackend(vars["vsID"], vars["rsID"]); err != nil {
		writeError(w, err)
	} else {
		w.Write(util.MustMarshal(opts, util.JSONOptions{Indent: true}))
	}
}
Пример #3
0
func createService(vs string, n int64, proto string) error {
	log.Infof("creating service [%s] on port %d/%s", vs, n, proto)

	opts := core.ServiceOptions{Port: uint16(n), Protocol: proto}
	data := bytes.NewBuffer(util.MustMarshal(opts, util.JSONOptions{}))

	rqst, _ := http.NewRequest(
		"PUT",
		fmt.Sprintf("http://%s", path.Join(*remote, "service", vs)),
		data)

	return roundtrip(rqst, map[int]func() error{
		http.StatusConflict: func() error {
			exposed[vs] = struct{}{}
			return nil // not actually an error.
		}})
}
Пример #4
0
func writeError(w http.ResponseWriter, err error) {
	code := http.StatusOK

	switch err {
	case core.ErrIpvsSyscallFailed:
		code = http.StatusInternalServerError
	case core.ErrObjectExists:
		code = http.StatusConflict
	case core.ErrObjectNotFound:
		code = http.StatusNotFound
	default:
		code = http.StatusBadRequest
	}

	w.WriteHeader(code)
	w.Write(util.MustMarshal(&errorResponse{err.Error()}, util.JSONOptions{Indent: true}))
}
Пример #5
0
func (c *consulDisco) Expose(name, host string, port uint16) error {
	u := *c.consul
	u.Path = "v1/agent/service/register"

	r, err := c.client.Post(
		u.String(),
		"application/json",
		bytes.NewBuffer(util.MustMarshal(exposeRequest{
			Name: name,
			Host: host,
			Port: port,
		}, util.JSONOptions{})))
	if err != nil {
		return err
	}

	if r.StatusCode != http.StatusOK {
		return errConsulError
	}

	return nil
}
Пример #6
0
func writeJSON(w http.ResponseWriter, obj interface{}) {
	w.Header().Add("Content-Type", "application/json")
	w.Write(util.MustMarshal(obj, util.JSONOptions{Indent: true}))
}