database.go

package hive

import (
	"bytes"
	"encoding/json"
	"fmt"
	"log"
)

var (
	ENDPOINT_DATABASE        = `%v:%v/templeton/v1/ddl/database?user.name=%v`
	ENDPOINT_DATABASE_DETAIL = `%v:%v/templeton/v1/ddl/database/%v?user.name=%v`
)

type ListDatabaseResponse struct {
	Databases []string `json:"databases"`
}

type ShowDatabaseResponse struct {
	Location string `json:"location"`
	Params   string `json:"params"`
	Comment  string `json:"comment"`
	Database string `json:"database"`
}

type CreateDatabaseInput struct {
	Comment    string      `json:"comment,omitempty"`
	Location   string      `json:"location,omitempty"`
	Properties interface{} `json:"properties,omitempty"`
}

type CreateDatabaseResponse struct {
	Database string `json:"database"`
}

type DropDatabaseResponse struct {
	Database string `json:"database"`
}

func (this *Client) ListDatabase() (*ListDatabaseResponse, error) {
	endpoint := fmt.Sprintf(ENDPOINT_DATABASE, this.BaseUrl, this.Port, this.User)
	resp, err := this.request(HTTP_GET, endpoint, nil)
	if err != nil {
		log.Println(err)
		return nil, err
	}

	res := &ListDatabaseResponse{}
	if err := json.NewDecoder(resp.Body).Decode(res); err != nil {
		log.Println(err)
		return nil, err
	}

	return res, nil
}

func (this *Client) ShowDatabase(database string) (*ShowDatabaseResponse, error) {
	endpoint := fmt.Sprintf(ENDPOINT_DATABASE_DETAIL, this.BaseUrl, this.Port, database, this.User)
	resp, err := this.request(HTTP_GET, endpoint, nil)
	if err != nil {
		return nil, err
	}

	res := &ShowDatabaseResponse{}
	if err := json.NewDecoder(resp.Body).Decode(res); err != nil {
		log.Println(err)
		return nil, err
	}

	return res, nil
}

func (this *Client) CreateDatabase(database string, in *CreateDatabaseInput) (*CreateDatabaseResponse, error) {
	body := new(bytes.Buffer)
	json.NewEncoder(body).Encode(in)
	endpoint := fmt.Sprintf(ENDPOINT_DATABASE_DETAIL, this.BaseUrl, this.Port, database, this.User)
	resp, err := this.request(HTTP_PUT, endpoint, body)
	if err != nil {
		log.Println(err)
		return nil, err
	}
	res := &CreateDatabaseResponse{}
	if err := json.NewDecoder(resp.Body).Decode(res); err != nil {
		log.Println(err)
		return nil, err
	}

	return res, nil
}

func (this *Client) DropDatabase(database string) (*DropDatabaseResponse, error) {
	endpoint := fmt.Sprintf(ENDPOINT_DATABASE_DETAIL, this.BaseUrl, this.Port, database, this.User)
	resp, err := this.request(HTTP_DELETE, endpoint, nil)
	if err != nil {
		return nil, err
	}

	res := &DropDatabaseResponse{}
	if err := json.NewDecoder(resp.Body).Decode(res); err != nil {
		log.Println(err)
		return nil, err
	}

	return res, nil
}