parent
b6c08f3056
commit
4eca54b6ed
6 changed files with 16 additions and 12 deletions
|
|
@ -16,10 +16,10 @@ type Server struct {
|
|||
v2 *v2.Server
|
||||
}
|
||||
|
||||
func NewServer(workers *worker.Server, rpmMetadata rpmmd.RPMMD, distros *distroregistry.Registry) *Server {
|
||||
func NewServer(workers *worker.Server, rpmMetadata rpmmd.RPMMD, distros *distroregistry.Registry, awsBucket string) *Server {
|
||||
server := &Server{
|
||||
v1: v1.NewServer(workers, rpmMetadata, distros),
|
||||
v2: v2.NewServer(workers, rpmMetadata, distros),
|
||||
v2: v2.NewServer(workers, rpmMetadata, distros, awsBucket),
|
||||
}
|
||||
return server
|
||||
}
|
||||
|
|
|
|||
|
|
@ -28,15 +28,12 @@ import (
|
|||
"github.com/osbuild/osbuild-composer/internal/worker"
|
||||
)
|
||||
|
||||
const (
|
||||
Bucket = "image-builder.service"
|
||||
)
|
||||
|
||||
// Server represents the state of the cloud Server
|
||||
type Server struct {
|
||||
workers *worker.Server
|
||||
rpmMetadata rpmmd.RPMMD
|
||||
distros *distroregistry.Registry
|
||||
awsBucket string
|
||||
}
|
||||
|
||||
type apiHandlers struct {
|
||||
|
|
@ -45,11 +42,12 @@ type apiHandlers struct {
|
|||
|
||||
type binder struct{}
|
||||
|
||||
func NewServer(workers *worker.Server, rpmMetadata rpmmd.RPMMD, distros *distroregistry.Registry) *Server {
|
||||
func NewServer(workers *worker.Server, rpmMetadata rpmmd.RPMMD, distros *distroregistry.Registry, bucket string) *Server {
|
||||
server := &Server{
|
||||
workers: workers,
|
||||
rpmMetadata: rpmMetadata,
|
||||
distros: distros,
|
||||
awsBucket: bucket,
|
||||
}
|
||||
return server
|
||||
}
|
||||
|
|
@ -294,7 +292,7 @@ func (h *apiHandlers) PostCompose(ctx echo.Context) error {
|
|||
t := target.NewAWSTarget(&target.AWSTargetOptions{
|
||||
Filename: imageType.Filename(),
|
||||
Region: awsUploadOptions.Region,
|
||||
Bucket: Bucket,
|
||||
Bucket: h.server.awsBucket,
|
||||
Key: key,
|
||||
ShareWithAccounts: awsUploadOptions.ShareWithAccounts,
|
||||
})
|
||||
|
|
@ -322,7 +320,7 @@ func (h *apiHandlers) PostCompose(ctx echo.Context) error {
|
|||
t := target.NewAWSS3Target(&target.AWSS3TargetOptions{
|
||||
Filename: imageType.Filename(),
|
||||
Region: awsS3UploadOptions.Region,
|
||||
Bucket: Bucket,
|
||||
Bucket: h.server.awsBucket,
|
||||
Key: key,
|
||||
})
|
||||
t.ImageName = key
|
||||
|
|
|
|||
|
|
@ -28,7 +28,7 @@ func newV2Server(t *testing.T, dir string) (*v2.Server, *worker.Server) {
|
|||
require.NoError(t, err)
|
||||
require.NotNil(t, distros)
|
||||
|
||||
v2Server := v2.NewServer(rpmFixture.Workers, rpm, distros)
|
||||
v2Server := v2.NewServer(rpmFixture.Workers, rpm, distros, "image-builder.service")
|
||||
require.NotNil(t, v2Server)
|
||||
|
||||
return v2Server, rpmFixture.Workers
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue