put_connection_string_operation.go

package ravendb

import (
	"net/http"
)

var (
	_ IMaintenanceOperation = &PutConnectionStringOperation{}
)

type PutConnectionStringOperation struct {
	connectionString interface{}

	Command *PutConnectionStringCommand
}

func NewPutConnectionStringOperation(connectionString interface{}) *PutConnectionStringOperation {
	return &PutConnectionStringOperation{
		connectionString: connectionString,
	}
}

func (o *PutConnectionStringOperation) GetCommand(conventions *DocumentConventions) (RavenCommand, error) {
	o.Command = NewPutConnectionStringCommand(o.connectionString)
	return o.Command, nil
}

var _ RavenCommand = &PutConnectionStringCommand{}

type PutConnectionStringCommand struct {
	RavenCommandBase

	connectionString interface{}

	Result *PutConnectionStringResult
}

// connectionString should be ConnectionString or RavenConnectionString
func NewPutConnectionStringCommand(connectionString interface{}) *PutConnectionStringCommand {
	return &PutConnectionStringCommand{
		RavenCommandBase: NewRavenCommandBase(),

		connectionString: connectionString,
	}
}

func (c *PutConnectionStringCommand) CreateRequest(node *ServerNode) (*http.Request, error) {
	url := node.URL + "/databases/" + node.Database + "/admin/connection-strings"

	d, err := jsonMarshal(c.connectionString)
	if err != nil {
		// TODO: change err into RuntimeError() ?
		return nil, err
	}
	return newHttpPut(url, d)
}

func (c *PutConnectionStringCommand) SetResponse(response []byte, fromCache bool) error {
	if len(response) == 0 {
		return throwInvalidResponse()
	}

	return jsonUnmarshal(response, &c.Result)
}