firestore.go

package firebase

import (
	"context"
	"fmt"

	firebase "firebase.google.com/go"
	"github.com/ziphttpd/ziphttpd.com/util"
	"google.golang.org/api/iterator"
)

// Collection はfirestoreのコレクションを表します。
type Collection struct {
	collection string
}

// NewCollection は Collecction のインスタンスを生成します。
func NewCollection(collection string) *Collection {
	return &Collection{collection: collection}
}

// SetMap は Collection にマップを保存します。
func (f *Collection) SetMap(key string, data map[string]string) error {
	util.LogOutput(fmt.Sprintf("key:%s, value:%+v", key, data))
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	app, err := firebase.NewApp(ctx, nil, GetOption())
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	client, err := app.Firestore(ctx)
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	defer client.Close()
	// 書き出し
	_, err = client.Collection(f.collection).Doc(key).Set(ctx, data)
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	util.LogOutput("done")
	return nil
}

// Set は Collection に値を保存します。
func (f *Collection) Set(key, value string) error {
	data := map[string]string{}
	data["data"] = value
	return f.SetMap(key, data)
}

// GetMap は生のドキュメントを取得します。
func (f *Collection) GetMap(key string) (map[string]string, error) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	app, err := firebase.NewApp(ctx, nil, GetOption())
	if err != nil {
		return nil, err
	}
	client, err := app.Firestore(ctx)
	if err != nil {
		return nil, err
	}
	defer client.Close()
	// 読み出し
	dsnap, err := client.Collection(f.collection).Doc(key).Get(ctx)
	if err != nil {
		return nil, err
	}
	if dsnap.Exists() == false {
		return nil, err
	}
	data := dsnap.Data()
	// map[string]string マップへの変換
	ret := map[string]string{}
	for subkey, intr := range data {
		if value, ok := intr.(string); ok {
			ret[subkey] = value
		}
	}
	return ret, nil
}

// Get は Collection でキーから値を検索します。
func (f *Collection) Get(key string) (string, error) {
	util.LogOutput(fmt.Sprintf("key:%s", key))
	data, err := f.GetMap(key)
	if err != nil {
		util.LogOutput(err.Error())
		return "", err
	}
	util.LogOutput(fmt.Sprintf("%+v", data))
	if txt, ok := data["data"]; ok {
		util.LogOutput(fmt.Sprintf("return \"%s\"", txt))
		return txt, nil
	}
	util.LogOutput("return empty")
	return "", nil
}

// Delete は Collection からキーを削除します。
func (f *Collection) Delete(key string) error {
	util.LogOutput(fmt.Sprintf("key:%s", key))
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	app, err := firebase.NewApp(ctx, nil, GetOption())
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	client, err := app.Firestore(ctx)
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	defer client.Close()
	// 削除
	_, err = client.Collection(f.collection).Doc(key).Delete(ctx)
	if err != nil {
		util.LogOutput(err.Error())
		return err
	}
	return err
}

// Keys はキーの一覧を返します。
func (f *Collection) Keys() []string {
	ret := []string{}
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	app, err := firebase.NewApp(ctx, nil, GetOption())
	if err != nil {
		util.LogOutput(err.Error())
		return ret
	}
	client, err := app.Firestore(ctx)
	if err != nil {
		util.LogOutput(err.Error())
		return ret
	}
	defer client.Close()
	// キーの取得
	di := client.Collection(f.collection).Documents(ctx)
	for {
		doc, err := di.Next()
		if err == iterator.Done {
			break
		}
		if err != nil {
			break
		}
		ret = append(ret, doc.Ref.ID)
	}
	return ret
}