Class: LgPodPlugin::LSqliteDb

Inherits:
Object
  • Object
show all
Includes:
Singleton
Defined in:
lib/lg_pod_plugin/db/database.rb

Constant Summary collapse

K_USER_TABLE = 
"user_tab"
K_USER_PROJECTS = 
"user_projects"
K_POD_LATEST_REFS = 
"user_pod_latest_refs"
K_POD_SHA_BRANCH = 
"user_pod_sha_with_branch"

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ LSqliteDb

初始化 db



88
89
 
# File 'lib/lg_pod_plugin/db/database.rb', line 88

def initialize
end

Class Method Details

.shared ⇒ Object 



83
84
85
 
# File 'lib/lg_pod_plugin/db/database.rb', line 83

def self.shared
  return LSqliteDb.instance
end

Instance Method Details

#delete_project_by_id(project_id) ⇒ Object 



218
219
220
221
 
# File 'lib/lg_pod_plugin/db/database.rb', line 218

def delete_project_by_id(project_id)
  ps = @db.prepare("DELETE FROM #{K_USER_PROJECTS} WHERE id = :n")
  ps.execute('n' => project_id)
end

#delete_user_info(id) ⇒ Object

删除用户信息



181
182
183
184
 
# File 'lib/lg_pod_plugin/db/database.rb', line 181

def (id)
  ps = @db.prepare("DELETE FROM #{K_USER_TABLE} WHERE id = :n")
  ps.execute('n' => id)
end

#init_database ⇒ Object

初始化database



92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
 
# File 'lib/lg_pod_plugin/db/database.rb', line 92

def init_database
  root_path = LFileManager.download_director.join("database")
  db_file_path = root_path.join("my.db")
  unless root_path.exist? && db_file_path.exist?
    FileUtils.mkdir(root_path)
    FileUtils.chdir(root_path)
    FileUtils.touch("my.db")
  end
  # FileUtils.chdir(root_path)
  self.db = SQLite3::Database.new(db_file_path.to_path)
  #添加用户表
  sql1 = "create table if not exists #{K_USER_TABLE}(
    id varchar(100) primary key not null,
    username varchar(100),
   password varchar(100),
   host varchar(100),
    access_token varchar(100),
    refresh_token varchar(100),
    expires_in TimeStamp NOT NULL DEFAULT CURRENT_TIMESTAMP);"
  self.db.execute(sql1)

  #添加项目表
  sql2 = "create table if not exists #{K_USER_PROJECTS}(
    id varchar(100) primary key not null,
    name varchar(100),
   desc varchar(100),
   path varchar(100),
    ssh_url_to_repo varchar(100),
    http_url_to_repo varchar(100),
    web_url varchar(100),
    name_with_namespace varchar(100),
    path_with_namespace varchar(100)
   );"
  self.db.execute(sql2)

  #添加项目表
  sql3 = "create table if not exists #{K_POD_LATEST_REFS}(
    id varchar(100) primary key not null,
    name varchar(100),
   git varchar(100),
   branch varchar(100),
    tag varchar(100),
    sha varchar(100)
    );"
  self.db.execute(sql3)

  #添加项目表
  sql4 = "create table if not exists #{K_POD_SHA_BRANCH}(
    id varchar(100) primary key not null,
    name varchar(100),
    git varchar(100),
   branch varchar(100),
    sha varchar(100)
    );"
  self.db.execute(sql4)
end

#insert_pod_refs(name, git, branch, tag, commit) ⇒ Object 



223
224
225
226
227
228
229
230
231
232
233
234
235
 
# File 'lib/lg_pod_plugin/db/database.rb', line 223

def insert_pod_refs(name, git, branch, tag, commit)
  id = LPodLatestRefs.get_pod_id(name, git)
  pod = LPodLatestRefs.new(id, name, git, branch, tag, commit)
  if self.query_pod_refs(id) != nil
    self.db.execute_batch(
      "UPDATE #{K_POD_LATEST_REFS} SET name = (:name), git = (:git), branch = (:branch), tag = (:tag), sha = (:sha) where (id = :id)", { "name" => pod.name, "git" => pod.git, "sha" => pod.commit, "branch" => pod.branch, :tag => pod.tag, :id => pod.id }
    )
  else
    self.db.execute("INSERT INTO #{K_POD_LATEST_REFS} (id, name, git, branch, tag, sha)
        VALUES (?, ?, ?, ?, ?, ?)", [pod.id, pod.name, pod.git, pod.branch, pod.tag, pod.commit])
  end
  self.insert_pod_sha_with_branch(name, git, commit, branch)
end

#insert_pod_sha_with_branch(name, git, sha, branch) ⇒ Object

保存 sha, branch 到数据库



252
253
254
255
256
257
258
259
260
261
262
263
 
# File 'lib/lg_pod_plugin/db/database.rb', line 252

def insert_pod_sha_with_branch(name, git, sha, branch)
  return unless name && git && sha
  id = Digest::MD5.hexdigest(name + git + sha)
  if self.query_branch_with_sha(name, git, sha)[:sha] != nil
    self.db.execute_batch(
      "UPDATE #{K_POD_SHA_BRANCH} SET name = (:name), git = (:git), branch = (:branch), sha = (:sha) where (id = :id)", { "name" => name, "git" => git, "branch" => branch, :id => id, :sha => sha }
    )
  else
    self.db.execute("INSERT INTO #{K_POD_SHA_BRANCH} (id, name, git, branch, sha)
        VALUES (?, ?, ?, ?, ?)", [id, name, git, branch, sha])
  end
end

#insert_project(project) ⇒ Object

保存项目信息到数据库



187
188
189
190
191
192
193
194
195
196
 
# File 'lib/lg_pod_plugin/db/database.rb', line 187

def insert_project(project)
  if self.query_project_info(project.name, project.http_url_to_repo) != nil
    self.db.execute_batch(
      "UPDATE #{K_USER_PROJECTS} SET name = (:name), desc = (:desc), path = (:path), ssh_url_to_repo = (:ssh_url_to_repo), http_url_to_repo = (:http_url_to_repo), web_url = (:web_url), name_with_namespace = (:name_with_namespace), path_with_namespace = (:path_with_namespace) where (id = :id)", { "name" => project.name, "desc" => project.description, "path" => project.path, "ssh_url_to_repo" => project.ssh_url_to_repo, :http_url_to_repo => project.http_url_to_repo, :web_url => project.web_url, :id => project.id, :path_with_namespace => project.path_with_namespace, :name_with_namespace => project.name_with_namespace }
    )
  else
    self.db.execute("INSERT INTO #{K_USER_PROJECTS} (id, name, desc, path, ssh_url_to_repo, http_url_to_repo, web_url,name_with_namespace, path_with_namespace)
        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", [project.id, project.name, project.description, project.path, project.ssh_url_to_repo, project.http_url_to_repo, project.web_url, project.name_with_namespace, project.path_with_namespace])
  end
end

#insert_user_info(user) ⇒ Object 



150
151
152
153
154
155
156
157
158
159
160
161
 
# File 'lib/lg_pod_plugin/db/database.rb', line 150

def (user)
  # pp "user.id => #{user.id}"
  if self.(user.id) != nil
    self.db.execute_batch(
      "UPDATE #{K_USER_TABLE} SET username = (:username), password = (:password), host = (:host), access_token = (:access_token), expires_in = (:expires_in), refresh_token = (:refresh_token) where (id = :id)", { "username" => user.username, "password" => user.password, "host" => user.host, "access_token" => user.access_token, :expires_in => user.expires_in, :id => user.id, :refresh_token => user.refresh_token }
    )
  else
    self.db.execute("INSERT INTO #{K_USER_TABLE} (id, username, password, host, access_token,refresh_token, expires_in)
        VALUES (?, ?, ?, ?,?,?,?)", [user.id, user.username, user.password, user.host, user.access_token, user.refresh_token, user.expires_in])
  end

end

#query_branch_with_sha(id = nil, name, git, sha) ⇒ Object

通过 sha 查询对应 branch



266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
 
# File 'lib/lg_pod_plugin/db/database.rb', line 266

def query_branch_with_sha(id = nil, name, git, sha)
  hash_map = Hash.new
  id = Digest::MD5.hexdigest(name + git + sha) unless id
  self.db.execute("select * from #{K_POD_SHA_BRANCH} where id = '#{id}' and name = '#{name}' and git = '#{git}' ;") do |row|
    new_id = row[0]
    new_sha = row[4]
    new_git = row[2]
    new_name = row[1]
    new_branch = row[3]
    hash_map[:id] = new_id
    hash_map[:git] = new_git
    hash_map[:sha] = new_sha
    hash_map[:name] = new_name
    hash_map[:branch] = new_branch
  end
  hash_map
end

#query_pod_refs(id) ⇒ Object 



237
238
239
240
241
242
243
244
245
246
247
248
249
 
# File 'lib/lg_pod_plugin/db/database.rb', line 237

def query_pod_refs(id)
  pod_info = nil
  self.db.execute("select * from #{K_POD_LATEST_REFS} where id = '#{id}';") do |row|
    id = row[0]
    name = row[1]
    git = row[2]
    branch = row[3]
    tag = row[4]
    commit = row[5]
    pod_info = LPodLatestRefs.new(id, name, git, branch, tag, commit)
  end
  pod_info
end

#query_project_info(name, git) ⇒ Object

通过名称查询项目信息



199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
 
# File 'lib/lg_pod_plugin/db/database.rb', line 199

def query_project_info(name, git)
  project_info = nil
  self.db.execute("select * from #{K_USER_PROJECTS} where name = '#{name}' or path = '#{name}' ;") do |row|
    name_with_namespace = row[7]
    path_with_namespace = row[8]
    next unless git.include?(name_with_namespace) || git.include?(path_with_namespace)
    id = row[0]
    path = row[3]
    name = row[1]
    web_url = row[6]
    description = row[2]
    ssh_url_to_repo = row[4]
    http_url_to_repo = row[5]
    project_info = ProjectModel.new(id, name, description, path, ssh_url_to_repo, http_url_to_repo, web_url, name_with_namespace, path_with_namespace)
    return project_info
  end
  project_info
end

#query_user_info(user_id) ⇒ Object 



165
166
167
168
169
170
171
172
173
174
175
176
177
178
 
# File 'lib/lg_pod_plugin/db/database.rb', line 165

def (user_id)
   = nil
  self.db.execute("select * from #{K_USER_TABLE} where id = '#{user_id}';") do |row|
     = LUserAuthInfo.new
    .id = row[0]
    .username = row[1]
    .password = row[2]
    .host = row[3]
    .access_token = row[4]
    .refresh_token = row[5]
    .expires_in = row[6]
  end
  
end