Integrating a Go Web Server for Image Generation Services

Welcome to the final lesson of our course on building an image generation service with Go! In our previous lessons, we've built several key components of our application: the PromptManager for formatting user inputs, the ImageManager for storing and processing images, the ImageGeneratorService for connecting to Google's Gemini API, and most recently, the ImageGeneratorHandler, which handles input validation and response formatting.

Now it's time to bring everything together by creating a Go web server that will expose our functionality through HTTP. This is the final piece of our backend architecture that will allow users to interact with our image generation service through a web interface.

In the previous lesson, we built the ImageGeneratorHandler, which acts as an intermediary between our service layer and the API routes we'll create today. The handler handles the business logic of validating inputs, calling the appropriate service methods, and formatting responses. Now, we'll create the HTTP routes that will receive requests from clients and pass them to our handler.

Our Go API server has two main routes:

1. A route to handle image generation requests -- /generate-image?prompt=
2. A route to retrieve all previously generated images -- /images

By the end of this lesson, you'll have a complete Go web server that integrates with the controller we built previously, providing a clean interface for clients to generate and retrieve images.

Let's start by creating a function to run our Go web server. We'll build upon the InitializeApp function we wrote in the previous lesson and add a new RunServer function:

server/server.go

package server

import (
    "github.com/gofiber/fiber/v2"
    "imagegenerator/server/handlers"
)

func InitializeApp() *fiber.App {
    app := fiber.New()
    
    // Instantiate the ImageGeneratorHandler
    imageGeneratorHandler := handlers.NewImageGeneratorHandler()
    app.Get("/generate-image", imageGeneratorHandler.GenerateImage)
    app.Get("/images", imageGeneratorHandler.GetImages)
    
    return app
}

func RunServer() {
    app := InitializeApp()
    app.Listen(":3000")
}

In this code, we're calling the InitializeApp() function and listening on port 3000. This is the port we want our server to run on to work properly in the CodeSignal environment.