@ -1,12 +1,26 @@
package rest
import "github.com/eyebluecn/tank/code/core"
import (
"fmt"
"github.com/eyebluecn/tank/code/core"
"github.com/eyebluecn/tank/code/tool/i18n"
"github.com/eyebluecn/tank/code/tool/result"
"github.com/eyebluecn/tank/code/tool/util"
"net/http"
"os"
"regexp"
"strings"
)
//@Service
type PreferenceService struct {
BaseBean
preferenceDao * PreferenceDao
preference * Preference
matterDao * MatterDao
matterService * MatterService
userDao * UserDao
migrating bool
}
func ( this * PreferenceService ) Init ( ) {
@ -17,6 +31,21 @@ func (this *PreferenceService) Init() {
this . preferenceDao = b
}
b = core . CONTEXT . GetBean ( this . matterDao )
if b , ok := b . ( * MatterDao ) ; ok {
this . matterDao = b
}
b = core . CONTEXT . GetBean ( this . matterService )
if b , ok := b . ( * MatterService ) ; ok {
this . matterService = b
}
b = core . CONTEXT . GetBean ( this . userDao )
if b , ok := b . ( * UserDao ) ; ok {
this . userDao = b
}
}
func ( this * PreferenceService ) Fetch ( ) * Preference {
@ -42,3 +71,191 @@ func (this *PreferenceService) Cleanup() {
this . Reset ( )
}
//migrate 2.0's db data and file data to 3.0
func ( this * PreferenceService ) Migrate20to30 ( writer http . ResponseWriter , request * http . Request ) {
matterPath := request . FormValue ( "matterPath" )
if matterPath == "" {
panic ( result . BadRequest ( "matterPath required" ) )
}
this . logger . Info ( "start migrating from 2.0 to 3.0" )
//lock
if this . migrating {
panic ( result . BadRequest ( "migrating work is processing" ) )
} else {
this . migrating = true
}
defer func ( ) {
this . migrating = false
} ( )
//delete all users with _20
this . userDao . DeleteUsers20 ( )
migrateDashboardSql := "INSERT INTO `tank`.`tank30_download_token` ( `uuid`, `sort`, `update_time`, `create_time`, `user_uuid`, `matter_uuid`, `expire_time`, `ip` ) ( SELECT `uuid`, `sort`, `update_time`, `create_time`, `user_uuid`, `matter_uuid`, `expire_time`, `ip` FROM `tank`.`tank20_download_token`)"
this . logger . Info ( migrateDashboardSql )
db := core . CONTEXT . GetDB ( ) . Exec ( migrateDashboardSql )
if db . Error != nil {
this . logger . Error ( "%v" , db . Error )
}
migrateDownloadTokenSql := "INSERT INTO `tank`.`tank30_dashboard` ( `uuid`, `sort`, `update_time`, `create_time`, `invoke_num`, `total_invoke_num`, `uv`, `total_uv`, `matter_num`, `total_matter_num`, `file_size`, `total_file_size`, `avg_cost`, `dt` ) ( SELECT `uuid`, `sort`, `update_time`, `create_time`, `invoke_num`, `total_invoke_num`, `uv`, `total_uv`, `matter_num`, `total_matter_num`, `file_size`, `total_file_size`, `avg_cost`, `dt` FROM `tank`.`tank20_dashboard` )"
this . logger . Info ( migrateDownloadTokenSql )
db = core . CONTEXT . GetDB ( ) . Exec ( migrateDownloadTokenSql )
if db . Error != nil {
this . logger . Error ( "%v" , db . Error )
}
migrateMatterSql := "INSERT INTO `tank`.`tank30_matter` ( `uuid`, `sort`, `update_time`, `create_time`, `puuid`, `user_uuid`, `username`, `dir`, `name`, `md5`, `size`, `privacy`, `path`, `times` ) ( SELECT `uuid`, `sort`, `update_time`, `create_time`, `puuid`, `user_uuid`, '', `dir`, `name`, `md5`, `size`, `privacy`, `path`, `times` FROM `tank`.`tank20_matter` ) "
this . logger . Info ( migrateMatterSql )
db = core . CONTEXT . GetDB ( ) . Exec ( migrateMatterSql )
if db . Error != nil {
this . logger . Error ( "%v" , db . Error )
}
migrateUploadTokenSql := "INSERT INTO `tank`.`tank30_upload_token` ( `uuid`, `sort`, `update_time`, `create_time`, `user_uuid`, `folder_uuid`, `matter_uuid`, `expire_time`, `filename`, `privacy`, `size`, `ip` ) ( SELECT `uuid`, `sort`, `update_time`, `create_time`, `user_uuid`, `folder_uuid`, `matter_uuid`, `expire_time`, `filename`, `privacy`, `size`, `ip` FROM `tank`.`tank20_upload_token` ) "
this . logger . Info ( migrateUploadTokenSql )
db = core . CONTEXT . GetDB ( ) . Exec ( migrateUploadTokenSql )
if db . Error != nil {
this . logger . Error ( "%v" , db . Error )
}
//username in tank2.0 add _20.
migrateUserSql := "INSERT INTO `tank`.`tank30_user` ( `uuid`, `sort`, `update_time`, `create_time`, `role`, `username`, `password`, `avatar_url`, `last_ip`, `last_time`, `size_limit`, `total_size_limit`, `total_size`, `status` ) ( SELECT `uuid`, `sort`, `update_time`, `create_time`, `role`, CONCAT(`username`,'_20') as `username`, `password`, `avatar_url`, `last_ip`, `last_time`, `size_limit`, -1, 0, `status` FROM `tank`.`tank20_user` )"
this . logger . Info ( migrateUserSql )
db = core . CONTEXT . GetDB ( ) . Exec ( migrateUserSql )
if db . Error != nil {
this . logger . Error ( "%v" , db . Error )
}
//find all 2.0 users.
users := this . userDao . FindUsers20 ( )
for _ , user := range users {
this . logger . Info ( "start handling matters for user %s %s" , user . Uuid , user . Username )
rootMatter := NewRootMatter ( user )
firstLevelMatters := this . matterDao . FindByPuuidAndUserUuid ( MATTER_ROOT , user . Uuid , nil )
for _ , firstLevelMatter := range firstLevelMatters {
this . HandleMatter20 ( request , matterPath , rootMatter , firstLevelMatter , user )
}
//adjust all the size.
this . matterService . ComputeAllDirSize ( user )
}
}
//handle matter from 2.0
func ( this * PreferenceService ) HandleMatter20 ( request * http . Request , matterPath string , dirMatter * Matter , matter * Matter , user * User ) {
defer func ( ) {
if err := recover ( ) ; err != nil {
this . logger . Warn ( "HandleMatter20 occur error %v when handle matter %s %s. Ignore the error and continue. \r\n" , err , matter . Uuid , matter . Name )
}
} ( )
this . logger . Info ( "start handling matter %s" , matter . Name )
if matter == nil {
panic ( result . BadRequest ( "matter cannot be nil." ) )
}
if user == nil {
panic ( result . BadRequest ( "user cannot be nil." ) )
}
if dirMatter == nil {
panic ( result . BadRequest ( "dirMatter cannot be nil" ) )
}
if ! dirMatter . Dir {
panic ( result . BadRequest ( "dirMatter must be directory" ) )
}
if dirMatter . UserUuid != user . Uuid {
panic ( result . BadRequest ( "file's user not the same" ) )
}
name := matter . Name
filename := name
if name == "" {
panic ( result . BadRequest ( "name cannot be blank" ) )
}
if len ( name ) > MATTER_NAME_MAX_LENGTH {
panic ( result . BadRequestI18n ( request , i18n . MatterNameLengthExceedLimit , len ( name ) , MATTER_NAME_MAX_LENGTH ) )
}
//if directory. Create it.
if matter . Dir {
if m , _ := regexp . MatchString ( MATTER_NAME_PATTERN , name ) ; m {
panic ( result . BadRequestI18n ( request , i18n . MatterNameContainSpecialChars ) )
}
parts := strings . Split ( dirMatter . Path , "/" )
if len ( parts ) > MATTER_NAME_MAX_DEPTH {
panic ( result . BadRequestI18n ( request , i18n . MatterDepthExceedLimit , len ( parts ) , MATTER_NAME_MAX_DEPTH ) )
}
absolutePath := GetUserMatterRootDir ( user . Username ) + dirMatter . Path + "/" + name
relativePath := dirMatter . Path + "/" + name
//crate directory on disk.
dirPath := util . MakeDirAll ( absolutePath )
this . logger . Info ( "Create Directory: %s" , dirPath )
//change matter info.
matter . Username = user . Username
matter . Path = relativePath
matter = this . matterDao . Save ( matter )
//handle its children.
children := this . matterDao . FindByPuuidAndUserUuid ( matter . Uuid , user . Uuid , nil )
for _ , child := range children {
this . HandleMatter20 ( request , matterPath , matter , child , user )
}
} else {
//if file. copy and adjust it.
dirAbsolutePath := dirMatter . AbsolutePath ( )
dirRelativePath := dirMatter . Path
fileAbsolutePath := dirAbsolutePath + "/" + filename
fileRelativePath := dirRelativePath + "/" + filename
util . MakeDirAll ( dirAbsolutePath )
//if exist, panic it.
exist := util . PathExists ( fileAbsolutePath )
if exist {
this . logger . Error ( "%s exits, overwrite it." , fileAbsolutePath )
removeError := os . Remove ( fileAbsolutePath )
this . PanicError ( removeError )
}
srcAbsolutePath := fmt . Sprintf ( "%s%s" , matterPath , matter . Path )
//find the 2.0 disk file.
fileSize := util . CopyFile ( srcAbsolutePath , fileAbsolutePath )
this . logger . Info ( "copy %s %v " , filename , util . HumanFileSize ( fileSize ) )
//update info.
matter . Path = fileRelativePath
matter . Username = user . Username
matter = this . matterDao . Save ( matter )
}
}
zicla
7 years ago