DeveloperDurp/DurpAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add dad jokes

Browse Source
This commit is contained in:
DeveloperDurp
2023-06-24 10:34:49 -04:00
parent b01b26b12d
commit 31a71fac7f
11 changed files with 193 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
controller/controller.go
View File

@@ -11,11 +11,13 @@ import (
"github.com/sashabaranov/go-openai"
"gitlab.com/DeveloperDurp/DurpAPI/model"
"gitlab.com/DeveloperDurp/DurpAPI/storage"
)
type Controller struct {
Cfg model.Config
dbcfg model.DBConfig
Dbcfg model.DBConfig
Db model.Repository
}
func NewController() *Controller {
@@ -23,19 +25,30 @@ func NewController() *Controller {
if err != nil {
log.Fatalf("unable to load file: %e", err)
}
controller := &Controller{}
controller.Cfg = model.Config{}
controller.dbcfg = model.DBConfig{}
controller := &Controller{
Cfg: model.Config{},
Dbcfg: model.DBConfig{},
}
err = env.Parse(&controller.Cfg)
if err != nil {
log.Fatalf("unable to parse environment variables: %e", err)
}
err = env.Parse(&controller.dbcfg)
err = env.Parse(&controller.Dbcfg)
if err != nil {
log.Fatalf("unable to parse database variables: %e", err)
}
controller.Cfg.OpenaiClient = *openai.NewClient(controller.Cfg.OpenaiApiKey)
Db, err := storage.Connect(controller.Dbcfg)
if err != nil {
panic("Failed to connect to database")
}
controller.Db = *Db
return controller
}

38
controller/dadjoke.go
View File

@@ -6,6 +6,7 @@ import (
"github.com/gin-gonic/gin"
"gitlab.com/DeveloperDurp/DurpAPI/model"
"gitlab.com/DeveloperDurp/DurpAPI/service"
)
// GetDadJoke godoc
@@ -19,15 +20,40 @@ import (
// @failure 400 {object} model.Message "error"
// @Router /jokes/dadjoke [get]
func (c *Controller) GetDadJoke(ctx *gin.Context) {
joke, err := service.GetRandomDadJoke(c.Db.DB)
if err != nil {
ctx.JSON(http.StatusInternalServerError, gin.H{"message": err})
return
}
ctx.JSON(http.StatusOK, gin.H{"message": joke})
}
// PostDadJoke godoc
//
// @Summary Generate dadjoke
// @Description create a dad joke
// @Tags DadJoke
// @Accept json
// @Produce application/json
// @Success 200 {object} model.Message "response"
// @failure 400 {object} model.Message "error"
// @Router /jokes/dadjoke [post]
func (c *Controller) PostDadJoke(ctx *gin.Context) {
var req model.DadJoke
if err := ctx.ShouldBindJSON(&req); err != nil {
ctx.JSON(http.StatusInternalServerError, gin.H{"message": err})
return
}
message := "test"
// message, err := service.GetDadJoke(req)
// if err != nil {
// ctx.JSON(http.StatusInternalServerError, gin.H{"message": err})
// }
ctx.JSON(http.StatusOK, gin.H{"message": message})
entry := model.DadJoke{
JOKE: req.JOKE,
}
err := c.Db.DB.Create(&entry).Error
if err != nil {
ctx.JSON(http.StatusInternalServerError, gin.H{"message": err})
return
}
ctx.JSON(http.StatusOK, gin.H{"message": "OK"})
}

27
docs/docs.go
View File

@@ -81,6 +81,33 @@ const docTemplate = `{
}
}
}
},
"post": {
"description": "create a dad joke",
"consumes": [
"application/json"
],
"produces": [
"application/json"
],
"tags": [
"DadJoke"
],
"summary": "Generate dadjoke",
"responses": {
"200": {
"description": "response",
"schema": {
"$ref": "#/definitions/model.Message"
}
},
"400": {
"description": "error",
"schema": {
"$ref": "#/definitions/model.Message"
}
}
}
}
},
"/openai/general": {

27
docs/swagger.json
View File

@@ -72,6 +72,33 @@
}
}
}
},
"post": {
"description": "create a dad joke",
"consumes": [
"application/json"
],
"produces": [
"application/json"
],
"tags": [
"DadJoke"
],
"summary": "Generate dadjoke",
"responses": {
"200": {
"description": "response",
"schema": {
"$ref": "#/definitions/model.Message"
}
},
"400": {
"description": "error",
"schema": {
"$ref": "#/definitions/model.Message"
}
}
}
}
},
"/openai/general": {

18
docs/swagger.yaml
View File

@@ -97,6 +97,24 @@ paths:
summary: Generate dadjoke
tags:
- DadJoke
post:
consumes:
- application/json
description: create a dad joke
produces:
- application/json
responses:
"200":
description: response
schema:
$ref: '#/definitions/model.Message'
"400":
description: error
schema:
$ref: '#/definitions/model.Message'
summary: Generate dadjoke
tags:
- DadJoke
/openai/general:
get:
consumes:

3
main.go
View File

@@ -40,6 +40,9 @@ func main() {
jokes := v1.Group("/jokes")
{
jokes.GET("dadjoke", c.GetDadJoke)
jokes.Use(c.AuthMiddleware([]string{"rw-jokes"}, c.Cfg.Groupsenv))
jokes.POST("dadjoke", c.PostDadJoke)
}
openai := v1.Group("/openai")
{

13
model/dadjoke.go
View File

@@ -1,13 +0,0 @@
package model
type DadJokes struct {
ID uint `gorm: "primary key;autoIncrement" json:"id"`
Author *string `json:"author"`
Title *string `json:"title"`
Publisher *string `json:"publisher"`
}
type DadJoke struct {
Author string `json:"author"`
Title string `json:"title"`
Publisher string `json:"publisher"`
}

5
model/jokes.go Normal file
View File

@@ -0,0 +1,5 @@
package model
type DadJoke struct {
JOKE string `json:"joke"`
}

30
service/dadjoke.go Normal file
View File

@@ -0,0 +1,30 @@
package service
import (
"math/rand"
"gorm.io/gorm"
"gitlab.com/DeveloperDurp/DurpAPI/model"
)
func GetRandomDadJoke(db *gorm.DB) (string, error) {
jokes, err := getDadJokes(db)
if err != nil {
return "", err
}
randomIndex := rand.Intn(len(jokes))
randomElement := jokes[randomIndex]
return randomElement.JOKE, err
}
func getDadJokes(db *gorm.DB) ([]model.DadJoke, error) {
req := []model.DadJoke{}
err := db.Find(&req).Error
return req, err
}

22
service/shared.go
View File

@@ -1,22 +0,0 @@
package service
import (
"fmt"
"time"
)
func RetryOperation(maxRetries int, delay time.Duration, operation func() error) error {
var err error
for i := 0; i <= maxRetries; i++ {
err = operation()
if err == nil {
return nil
}
fmt.Printf("Error encountered: %v\n", err)
if i < maxRetries {
fmt.Printf("Retrying after %v...\n", delay)
time.Sleep(delay)
}
}
return err
}

33
storage/postgres.go Normal file
View File

@@ -0,0 +1,33 @@
package storage
import (
"fmt"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"gitlab.com/DeveloperDurp/DurpAPI/model"
)
func Connect(config model.DBConfig) (*model.Repository, error) {
dsn := fmt.Sprintf(
"host=%s port=%s user=%s password=%s dbname=%s sslmode=%s",
config.Host, config.Port, config.User, config.Password, config.DBName, config.SSLMode,
)
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
return nil, err
}
runMigrations(db)
return &model.Repository{
DB: db,
}, nil
}
func runMigrations(db *gorm.DB) error {
err := db.AutoMigrate(&model.DadJoke{})
if err != nil {
return err
}
return nil
}