Start9/registry
2
0
Fork 0
mirror of https://github.com/Start9Labs/registry.git synced 2026-03-26 02:11:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

migrations

Browse Source
This commit is contained in:
Keagan McClelland
2022-05-24 12:21:06 -06:00
parent fe423f1ed2
commit 1cd0b78fa6
4 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/Application.hs
View File

@@ -83,6 +83,8 @@ import Control.Lens
import Data.List ( lookup )
import Data.String.Interpolate.IsString
( i )
import qualified Database.Persist.Migration
import qualified Database.Persist.Migration.Postgres
import Database.Persist.Sql ( SqlBackend )
import Foundation
import Handler.Admin
@@ -95,6 +97,7 @@ import Lib.PkgRepository ( watchEosRepoRoot
, watchPkgRepoRoot
)
import Lib.Ssl
import Migration ( manualMigration )
import Model
import Network.HTTP.Types.Header ( hOrigin )
import Network.Wai.Middleware.Gzip ( GzipFiles(GzipCompress)
@@ -107,6 +110,7 @@ import Settings
import System.Directory ( createDirectoryIfMissing )
import System.Posix.Process
import System.Time.Extra
import qualified UnliftIO
import Yesod
-- This line actually creates our YesodDispatch instance. It is the second half
@@ -152,6 +156,16 @@ makeFoundation appSettings = do
-- Preform database migration using application logging settings
runLoggingT (runSqlPool (runMigration migrateAll) pool) logFunc
`UnliftIO.catch` (\(e :: SomeException) -> do
print e
runSqlPool
(Database.Persist.Migration.Postgres.runMigration
Database.Persist.Migration.defaultSettings
manualMigration
)
pool
runLoggingT (runSqlPool (runMigration migrateAll) pool) logFunc
)
-- Return the foundation
return $ mkFoundation pool stopPkgWatch stopEosWatch

34
src/Migration.hs Normal file
View File

@@ -0,0 +1,34 @@
module Migration where
import Database.Persist.Migration
import Database.Persist.Sql ( Single(..) )
import Startlude ( ($)
, (<<$>>)
, Maybe(Just)
)
manualMigration :: Migration
manualMigration = [(0, 1) := migration_0_2_0]
migration_0_2_0 :: [Operation]
migration_0_2_0 =
[ AddColumn "version" (Column "title" SqlString [NotNull]) (Just $ PersistText "")
, AddColumn "version" (Column "desc_short" SqlString [NotNull]) (Just $ PersistText "")
, AddColumn "version" (Column "desc_long" SqlString [NotNull]) (Just $ PersistText "")
, AddColumn "version" (Column "icon_type" SqlString [NotNull]) (Just $ PersistText "")
, populateMetadata
, DropColumn ("pkg_record", "title")
, DropColumn ("pkg_record", "desc_short")
, DropColumn ("pkg_record", "desc_long")
, DropColumn ("pkg_record", "icon_type")
]
populateMetadata :: Operation
populateMetadata =
RawOperation "Populating Metadata"
$ migrateTitles
<<$>> rawSql "SELECT pkg_id, title, desc_short, desc_long, icon_type FROM pkg_record" []
where
migrateTitles (Single id', Single title', Single descShort', Single descLong', Single iconType') = MigrateSql
"UPDATE version SET title = ?, desc_short = ?, desc_long = ?, icon_type = ? where pkg_id = ?"
[PersistText title', PersistText descShort', PersistText descLong', PersistText iconType', PersistText id']