api/main.go

package main

import (
	"context"
	"log"
	"net/http"

	"api/ent"
	"api/ent/ogent"
	"entgo.io/ent/dialect"
	_ "github.com/mattn/go-sqlite3"
	"entgo.io/ent/dialect/sql/schema"
	"time"
)

type User struct {
	username string `json:"username"`
	created_at time.Time `json:"created_at"`
}

type handler struct {
	*ogent.OgentHandler
	client *ent.Client
}

func (h handler) DrawStart(ctx context.Context, params ogent.DrawStartParams) error {
	error := h.client.Card.UpdateOneID(params.ID).Exec(ctx)
	return (error)
}

//func (h handler) DrawDone(ctx context.Context, params ogent.DrawDoneParams) (ogent.DrawDoneNoContent, error) {
func (h handler) DrawDone(ctx context.Context, params ogent.DrawDoneParams) error {
	body := h.client.Card.GetX(ctx, params.ID)
	u_body := h.client.User.GetX(ctx, params.ID)
	jst, err := time.LoadLocation("Asia/Tokyo")
	if err != nil {
		panic(err)
	}
	t := time.Now().In(jst)
	tt := t.Format("20060102")
	f := body.CreatedAt.Add(time.Hour * 24 * 1).In(jst)
	ff := f.Format("20060102")
	fff := u_body.Next
	if tt < fff {
		error := h.client.User.UpdateOneID(params.ID).SetNext(ff).Exec(ctx)
		return	(error)
	} 
	error := h.client.User.UpdateOneID(params.ID).SetUpdatedAt(t).Exec(ctx)
	return	(error)
}

func main() {
	// Create ent client.
	//// sqlite
	//url := "file:/app/data/new.sqlite?_fk=1"
	//client, err := ent.Open(dialect.SQLite, url)
	////client, err := ent.Open(dialect.SQLite, "file:data?mode=memory&cache=shared&_fk=1")

	// postgres
	//url := os.Getenv("DATABASE_URL") + "?sslmode=require"
	url := "postgres://api:api@database/api"
	client, err := ent.Open("postgres", url)

	if err != nil {
		log.Fatal(err)
	}
	// Run the migrations.
	if err := client.Schema.Create(context.Background()); err != nil {
		log.Fatal(err)
	}
	ctx := context.Background()
	err = client.Schema.Create(ctx, schema.WithAtlas(true))
	if err != nil {
		log.Fatalf("failed creating schema resources: %v", err)
	}
	// Create the handler.
	h := handler{
		OgentHandler: ogent.NewOgentHandler(client),
		client:       client,
	}
	// Start listening.
	srv, err := ogent.NewServer(h)
	//srv,err := ogent.NewServer(ogent.NewOgentHandler(client))
	if err != nil {
		log.Fatal(err)
	}
	if err := http.ListenAndServe(":8080", srv); err != nil {
		log.Fatal(err)
	}
}