Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
W
webmagic
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
沈俊林
webmagic
Commits
4e65dac2
Commit
4e65dac2
authored
May 06, 2014
by
Yihua Huang
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #121 from ywooer/master
创建指定编码的Writer
parents
7fbe18b8
259f0a16
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
48 additions
and
2 deletions
+48
-2
FilePipeline.java
...ain/java/us/codecraft/webmagic/pipeline/FilePipeline.java
+4
-2
FilePipelineTest.java
...java/us/codecraft/webmagic/pipeline/FilePipelineTest.java
+44
-0
No files found.
webmagic-core/src/main/java/us/codecraft/webmagic/pipeline/FilePipeline.java
View file @
4e65dac2
...
...
@@ -4,12 +4,14 @@ import org.apache.commons.codec.digest.DigestUtils;
import
org.apache.http.annotation.ThreadSafe
;
import
org.slf4j.Logger
;
import
org.slf4j.LoggerFactory
;
import
us.codecraft.webmagic.ResultItems
;
import
us.codecraft.webmagic.Task
;
import
us.codecraft.webmagic.utils.FilePersistentBase
;
import
java.io.File
Writer
;
import
java.io.File
OutputStream
;
import
java.io.IOException
;
import
java.io.OutputStreamWriter
;
import
java.io.PrintWriter
;
import
java.util.Map
;
...
...
@@ -39,7 +41,7 @@ public class FilePipeline extends FilePersistentBase implements Pipeline {
public
void
process
(
ResultItems
resultItems
,
Task
task
)
{
String
path
=
this
.
path
+
PATH_SEPERATOR
+
task
.
getUUID
()
+
PATH_SEPERATOR
;
try
{
PrintWriter
printWriter
=
new
PrintWriter
(
new
FileWriter
(
getFile
(
path
+
DigestUtils
.
md5Hex
(
resultItems
.
getRequest
().
getUrl
())
+
".html"
)
));
PrintWriter
printWriter
=
new
PrintWriter
(
new
OutputStreamWriter
(
new
FileOutputStream
(
getFile
(
path
+
DigestUtils
.
md5Hex
(
resultItems
.
getRequest
().
getUrl
())
+
".html"
)),
"UTF-8"
));
printWriter
.
println
(
"url:\t"
+
resultItems
.
getRequest
().
getUrl
());
for
(
Map
.
Entry
<
String
,
Object
>
entry
:
resultItems
.
getAll
().
entrySet
())
{
if
(
entry
.
getValue
()
instanceof
Iterable
)
{
...
...
webmagic-core/src/test/java/us/codecraft/webmagic/pipeline/FilePipelineTest.java
0 → 100644
View file @
4e65dac2
package
us
.
codecraft
.
webmagic
.
pipeline
;
import
org.junit.BeforeClass
;
import
org.junit.Test
;
import
us.codecraft.webmagic.Request
;
import
us.codecraft.webmagic.ResultItems
;
import
us.codecraft.webmagic.Site
;
import
us.codecraft.webmagic.Task
;
import
java.util.UUID
;
/**
* Created by ywooer on 2014/5/6 0006.
*/
public
class
FilePipelineTest
{
private
static
ResultItems
resultItems
;
private
static
Task
task
;
@BeforeClass
public
static
void
before
()
{
resultItems
=
new
ResultItems
();
resultItems
.
put
(
"content"
,
"webmagic 爬虫工具"
);
Request
request
=
new
Request
(
"http://www.baidu.com"
);
resultItems
.
setRequest
(
request
);
task
=
new
Task
()
{
@Override
public
String
getUUID
()
{
return
UUID
.
randomUUID
().
toString
();
}
@Override
public
Site
getSite
()
{
return
null
;
}
};
}
@Test
public
void
testProcess
()
{
FilePipeline
filePipeline
=
new
FilePipeline
();
filePipeline
.
process
(
resultItems
,
task
);
}
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment